Home
last modified time | relevance | path

Searched refs:get_wrapper_codegen_for_device (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_extension_backend.py30 get_wrapper_codegen_for_device,
120 get_wrapper_codegen_for_device("extension_device")
124 get_wrapper_codegen_for_device("extension_device", True)
H A Dtest_triton_extension_backend.py32 get_wrapper_codegen_for_device,
89 get_wrapper_codegen_for_device("cpu") == ExtensionWrapperCodegen
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dgraph.py53 get_wrapper_codegen_for_device,
1530 wrapper_code_gen_cls = get_wrapper_codegen_for_device(
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcommon.py183 def get_wrapper_codegen_for_device(device: str, cpp_wrapper: bool = False): function