Searched refs:get_scheduling_for_device (Results 1 – 5 of 5) sorted by relevance
31 get_scheduling_for_device,87 self.assertTrue(get_scheduling_for_device("cpu") == ExtensionScheduling)
29 get_scheduling_for_device,117 get_scheduling_for_device("extension_device") == ExtensionScheduling
169 scheduling = get_scheduling_for_device(device_type)179 def get_scheduling_for_device(device: str): function205 if get_scheduling_for_device("cpu") is None:214 if get_scheduling_for_device("cuda") is None:224 if get_scheduling_for_device("xpu") is None:230 and get_scheduling_for_device(private_backend) is None
52 get_scheduling_for_device,277 if get_scheduling_for_device("cpu") is None:284 if get_scheduling_for_device("cuda") is None:292 if get_scheduling_for_device("xpu") is None:
45 from .codegen.common import BackendFeature, get_scheduling_for_device, Kernel3339 device_scheduling = get_scheduling_for_device(device.type)