Change the returned device interface (#3960)

intel · Mar 26, 2024 · e842872 · e842872
1 parent 17c0d71
commit e842872
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/intel_extension_for_pytorch/_dynamo/device_interface.py b/intel_extension_for_pytorch/_dynamo/device_interface.py
@@ -65,9 +65,4 @@ def is_available() -> bool:
 
     @staticmethod
     def get_compute_capability(device: _device_t = None) -> int:
-        # TODO :Return 0x80860001 for ATSM
-        # Currently return 0x80860002 for PVC
-        # currently, torch.xpu.get_device_capability returns a dict,
-        # but we want int for now
-        return 86
-        # return torch.xpu.get_device_capability(device)
+        return torch.xpu.get_device_capability(device)