Home
last modified time | relevance | path

Searched refs:src_code (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/
H A Drocm_cpp_scheduling.py44 def define_kernel(self, src_code: str, node_schedule) -> str:
46 if src_code in wrapper.src_to_kernel:
47 kernel_name = wrapper.src_to_kernel[src_code]
56 wrapper.src_to_kernel[src_code] = kernel_name
57 src_code = src_code.replace("KERNEL_NAME", kernel_name)
59 _, _, kernel_path = get_path(code_hash(src_code), "py")
63 compile_wrapper.splice(src_code, strip=True)
92 src_code = render()
96 kernel_name = self.define_kernel(src_code, node_schedule)
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/
H A Dcuda_cpp_scheduling.py49 def define_kernel(self, src_code: str, node_schedule) -> str:
51 if src_code in wrapper.src_to_kernel:
52 kernel_name = wrapper.src_to_kernel[src_code]
61 wrapper.src_to_kernel[src_code] = kernel_name
62 src_code = src_code.replace("KERNEL_NAME", kernel_name)
64 _, _, kernel_path = get_path(code_hash(src_code), "py")
68 compile_wrapper.splice(src_code, strip=True)
98 src_code = render()
102 kernel_name = self.define_kernel(src_code, node_schedule)
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/
H A DShader.cpp15 : src_code{ in ShaderInfo()
25 : src_code{ in ShaderInfo()
40 : src_code{ in ShaderInfo()
56 _1.src_code.bin == _2.src_code.bin && in operator ==()
57 _1.src_code.size == _2.src_code.size); in operator ==()
123 const uint32_t* code = source.src_code.bin;
124 uint32_t size = source.src_code.size;
H A DShader.h53 } src_code; member
171 seed, std::hash<const uint32_t*>()(source.src_code.bin)); in operator()
173 seed, std::hash<uint32_t>()(source.src_code.size)); in operator()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/vk_api/
H A DShader.cpp21 : src_code{ in ShaderInfo()
32 : src_code{ in ShaderInfo()
43 _1.src_code.bin == _2.src_code.bin && in operator ==()
44 _1.src_code.size == _2.src_code.size); in operator ==()
110 const uint32_t* code = source.src_code.bin;
111 uint32_t size = source.src_code.size;
H A DShader.h58 } src_code; member
76 return src_code.bin != nullptr;
169 seed, std::hash<const uint32_t*>()(source.src_code.bin)); in operator()
171 seed, std::hash<uint32_t>()(source.src_code.size)); in operator()
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dsimd.py1364 src_code = kernel.codegen_kernel()
1366 kernel_name = self.define_kernel(src_code, node_schedule, kernel)
1369 kernel.code_hash = code_hash(src_code)
1479 src_code = partial_code
1482 src_code = partial_code.code
1490 src_code = (
1497 return src_code
1499 kernel_name = self.define_kernel(src_code, node_schedule, kernel)
1581 src_code = kernel.codegen_kernel()
1582 kernel_code_list.append((src_code, kernel, node_group))
[all …]
H A Dtriton.py3011 def define_kernel(self, src_code, node_schedule, kernel): argument
3013 if src_code in wrapper.src_to_kernel:
3014 kernel_name = wrapper.src_to_kernel[src_code]
3021 kernel_category = get_kernel_category_by_source_code(src_code)[:3]
3026 wrapper.src_to_kernel[src_code] = kernel_name
3032 src_code = src_code.replace(str(Placeholder.DESCRIPTIVE_NAME), kernel_name)
3033 src_code = src_code.replace(str(Placeholder.KERNEL_NAME), subs_name)
3037 src_code = src_code.replace("#pragma CMT", "#")
3039 basename, _, kernel_path = get_path(code_hash(src_code.strip()), "py")
3043 compile_wrapper.splice(src_code, strip=True)
[all …]
H A Dhalide.py1672 def define_kernel(self, src_code, node_schedule, kernel): argument
1675 if src_code in wrapper.src_to_kernel:
1676 kernel_name = wrapper.src_to_kernel[src_code]
1679 wrapper.src_to_kernel[src_code] = kernel_name
1688 compile_wrapper.splice(src_code, strip=True)
1697 log_kernel_metadata(kernel_name, "", src_code)
H A Dcpp.py4526 src_code = render()
4530 kernel_name = self.define_kernel(src_code, node_schedule, kernel.args)
4544 def define_kernel(self, src_code, nodes, kernel_args=None): argument
4553 src_code = src_code.replace(str(Placeholder.KERNEL_NAME), kernel_decl_name)
4554 src_code = src_code.replace(str(Placeholder.DESCRIPTIVE_NAME), kernel_name)
4557 src_code = src_code.replace("#pragma CMT", "//")
4564 compile_wrapper.splice(src_code, strip=True)
4571 src_code = self.kernel_group.codegen_group()
4572 if src_code:
4574 src_code, self.kernel_group.scheduled_nodes
/aosp_15_r20/external/sdv/vsomeip/third_party/boost/unordered/doc/
Dhash_equality.qbk23 [import src_code/dictionary.cpp]
58 [import src_code/point1.cpp]
65 [import src_code/point2.cpp]
Dintro.qbk74 [import src_code/intro.cpp]
/aosp_15_r20/external/zucchini/aosp/include/components/zucchini/
H A Drel32_utils.h170 CODE_T src_code = ADDR_TRAITS::Fetch(src_view, src_idx); in ArmCopyDisp() local
172 if (ADDR_TRAITS::Decode(src_code, &disp)) { in ArmCopyDisp()
/aosp_15_r20/external/zucchini/
H A Drel32_utils.h170 CODE_T src_code = ADDR_TRAITS::Fetch(src_view, src_idx); in ArmCopyDisp() local
172 if (ADDR_TRAITS::Decode(src_code, &disp)) { in ArmCopyDisp()
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dwrapper_benchmark.py23 def get_kernel_category_by_source_code(src_code): argument
29 ch for ch in _kernel_category_choices if f"@triton_heuristics.{ch}" in src_code
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_aot_inductor.py166 src_code = cpp.read()
171 ).run(src_code)
2928 src_code = cpp.read()
2933 ).run(src_code)
2938 ).run(src_code)
2943 ).run(src_code)
2948 ).run(src_code)
/aosp_15_r20/external/executorch/backends/vulkan/test/
H A Dvulkan_compute_api_test.cpp180 EXPECT_TRUE(empty_shader_info.src_code.bin == nullptr); in TEST_F()
181 EXPECT_TRUE(empty_shader_info.src_code.size == 0u); in TEST_F()
/aosp_15_r20/out/.module_paths/
Dfiles.db[all...]