Home
last modified time | relevance | path

Searched refs:WrapperCodeGen (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor_dynamic_shapes.py18 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 Dwrapper.py161 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 Dcommon.py203 from .wrapper import WrapperCodeGen
210 WrapperCodeGen,
220 WrapperCodeGen,
225 register_backend_for_device("xpu", TritonScheduling, WrapperCodeGen)
H A Dcpp_wrapper_cpu.py30 from .wrapper import EnterSubgraphLine, ExitSubgraphLine, WrapperCodeGen
33 class CppWrapperCpu(WrapperCodeGen):
/aosp_15_r20/external/pytorch/torch/_inductor/kernel/
H A Dmm.py23 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 Dgraph.py58 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 Dutils.py1782 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 Dextension_codegen_backend.py6 class ExtensionWrapperCodegen(wrapper.WrapperCodeGen):
/aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/
H A Dextension_codegen_backend.py6 class ExtensionWrapperCodegen(wrapper.WrapperCodeGen):