Home
last modified time | relevance | path

Searched refs:CppScheduling (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/extension_backends/cpp/
H A Dextension_codegen_backend.py19 self._scheduling = cpp.CppScheduling(scheduler)
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_torchinductor_dynamic_shapes.py17 from torch._inductor.codegen.cpp import CppScheduling
899 register_backend_for_device("cpu", CppScheduling, WrapperCodeGen)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dgraph.py278 from .codegen.cpp import CppScheduling
281 "cpu", CppScheduling, WrapperCodeGen, CppWrapperCpu
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcommon.py197 from .cpp import CppScheduling
206 cpu_backends = {"cpp": CppScheduling, "halide": HalideScheduling}
H A Dcpp.py3889 class CppScheduling(BaseScheduling): class
4473 if args_num > CppScheduling.MAX_FUSED_KERNEL_ARGS_NUM: