Searched refs:WrapperCodeGen (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_torchinductor_dynamic_shapes.py | 18 from torch._inductor.codegen.wrapper import WrapperCodeGen 877 WrapperCodeGen.statically_known_int_or_none(batch_dim) is None 882 WrapperCodeGen.statically_known_int_or_none(batch_dim) == 5 887 WrapperCodeGen.statically_known_int_or_none(batch_dim) == 5 890 class TestWrapperCodegen(WrapperCodeGen): 899 register_backend_for_device("cpu", CppScheduling, WrapperCodeGen)
|
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
H A D | wrapper.py | 161 wrapper: Optional[WrapperCodeGen] = None, argument 275 wrapper: WrapperCodeGen 288 wrapper: WrapperCodeGen 353 wrapper: WrapperCodeGen 458 class WrapperCodeGen(CodeGen): class 2041 num = WrapperCodeGen.statically_known_int_or_none(x) 2049 return WrapperCodeGen.statically_known_list_of_ints_or_none(lst) is not None 2053 return WrapperCodeGen.statically_known_list_of_ints_or_none(buffer.get_size()) 2057 return WrapperCodeGen.static_shape_for_buffer_or_none(buffer) is not None
|
H A D | common.py | 203 from .wrapper import WrapperCodeGen 210 WrapperCodeGen, 220 WrapperCodeGen, 225 register_backend_for_device("xpu", TritonScheduling, WrapperCodeGen)
|
H A D | cpp_wrapper_cpu.py | 30 from .wrapper import EnterSubgraphLine, ExitSubgraphLine, WrapperCodeGen 33 class CppWrapperCpu(WrapperCodeGen):
|
/aosp_15_r20/external/pytorch/torch/_inductor/kernel/ |
H A D | mm.py | 23 from ..codegen.wrapper import WrapperCodeGen 262 static_size = WrapperCodeGen.statically_known_list_of_ints_or_none(layout.size) 266 sz = WrapperCodeGen.statically_known_int_or_none(s) 393 WrapperCodeGen.statically_known_int_or_none(inp_expanded.layout.stride[-1])
|
/aosp_15_r20/external/pytorch/torch/_inductor/ |
H A D | graph.py | 58 from .codegen.wrapper import WrapperCodeGen 281 "cpu", CppScheduling, WrapperCodeGen, CppWrapperCpu 289 "cuda", CUDACombinedScheduling, WrapperCodeGen, CppWrapperCuda 295 register_backend_for_device("xpu", TritonScheduling, WrapperCodeGen) 371 self.wrapper_code: WrapperCodeGen = None # type: ignore[assignment]
|
H A D | utils.py | 1782 from .codegen.wrapper import WrapperCodeGen 1784 orig_define_kernel = WrapperCodeGen.define_kernel 1791 with unittest.mock.patch.object(WrapperCodeGen, "define_kernel", new_define_kernel):
|
/aosp_15_r20/external/pytorch/test/inductor/extension_backends/triton/ |
H A D | extension_codegen_backend.py | 6 class ExtensionWrapperCodegen(wrapper.WrapperCodeGen):
|
/aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/ |
H A D | extension_codegen_backend.py | 6 class ExtensionWrapperCodegen(wrapper.WrapperCodeGen):
|