Fix overflow issue flagged by asan.

Apparently `memoryClockRate * 2` overflowing int. PiperOrigin-RevId: 670143739
tumelo-mapheto · Sep 2, 2024 · 733dc8d · 733dc8d
1 parent c0d155d
commit 733dc8d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensorflow/core/grappler/clusters/utils.cc b/tensorflow/core/grappler/clusters/utils.cc
@@ -100,7 +100,7 @@ DeviceProperties GetLocalGPUInfo(PlatformDeviceId platform_device_id) {
   // 8 is the number of bits per byte. 2 is accounted for
   // double data rate (DDR).
   device.set_bandwidth(properties.memoryBusWidth / 8 *
-                       properties.memoryClockRate * 2);
+                       properties.memoryClockRate * 2ULL);
 
   (*device.mutable_environment())["architecture"] =
       strings::StrCat(properties.major, ".", properties.minor);
@@ -132,7 +132,7 @@ DeviceProperties GetLocalGPUInfo(PlatformDeviceId platform_device_id) {
   // 8 is the number of bits per byte. 2 is accounted for
   // double data rate (DDR).
   device.set_bandwidth(properties.memoryBusWidth / 8 *
-                       properties.memoryClockRate * 2);
+                       properties.memoryClockRate * 2ULL);
 
   (*device.mutable_environment())["architecture"] =
       strings::StrCat("gfx", properties.gcnArchName);