Change MAX_MEMORY_BANDWIDTH device query to uint64

oneapi-src · Feb 3, 2025 · cde90bd · cde90bd
1 parent af4f331
commit cde90bd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/source/adapters/cuda/device.cpp b/source/adapters/cuda/device.cpp
@@ -1008,8 +1008,10 @@ UR_APIEXPORT ur_result_t UR_APICALL urDeviceGetInfo(ur_device_handle_t hDevice,
           &MemoryBusWidth, CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH,
           hDevice->get()));
     }
+    uint64_t MemoryBandwidthConstant = 250;
 
-    uint32_t MemoryBandwidth = MemoryClockKHz * MemoryBusWidth * 250;
+    uint64_t MemoryBandwidth = static_cast<uint64_t>(
+        MemoryClockKHz * MemoryBusWidth * MemoryBandwidthConstant);
 
     return ReturnValue(MemoryBandwidth);
   }