Searched refs:device_codegens (Results 1 – 2 of 2) sorted by relevance
96 device_codegens: Dict[str, DeviceCodegen] = {} variable143 device_codegens[device] = DeviceCodegen(180 return device_codegens[device].scheduling if device in device_codegens else None184 if device in device_codegens:185 wrapper_codegen_obj: DeviceCodegen = device_codegens[device]
16 from torch._inductor.codegen.common import device_codegens, register_backend_for_device898 if "cpu" not in device_codegens:900 orig_cpu_codegens = device_codegens["cpu"]