Update DeviceProperties.h

intel · Aug 28, 2024 · 242ab85 · 242ab85
1 parent 71d212d
commit 242ab85
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/comm/DeviceProperties.h b/src/comm/DeviceProperties.h
@@ -190,5 +190,11 @@ uint32_t syclNativeVectorWidth(
       "Invalid data type to fetch native vector width!");
 }
 
+static inline bool syclHasFloat64(
+    at::DeviceIndex dev_id = at::xpu::getDeviceIndexOfCurrentQueue()) {
+  auto* dev_prop = at::xpu::getDeviceProperties(dev_id);
+  return dev_prop->has_fp64;
+}
+
 } // namespace sycl
 } // namespace xpu