calculate timestamp mask

Dewei-Wang-sh · Jun 22, 2022 · 601cec3 · 601cec3
1 parent 40f1236
commit 601cec3
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-0.45.1
+0.46.0
diff --git a/utils/ze_utils.h b/utils/ze_utils.h
@@ -366,7 +366,13 @@ inline uint64_t GetMetricTimestampMask(ze_device_handle_t device) {
   ze_device_properties_t props{ZE_STRUCTURE_TYPE_DEVICE_PROPERTIES_1_2, };
   ze_result_t status = zeDeviceGetProperties(device, &props);
   PTI_ASSERT(status == ZE_RESULT_SUCCESS);
-  return (1ull << props.kernelTimestampValidBits) - 1ull;
+  uint32_t devicemask = (props.deviceId & 0xFF00);
+  if ((devicemask == 0x5600) || (devicemask == 0x4F00) || (devicemask == 0x0B00)) {
+      return (1ull << (props.kernelTimestampValidBits - 1)) - 1ull;
+  }
+  else {
+      return (1ull << props.kernelTimestampValidBits) - 1ull;
+  }
 #endif
 }
 
@@ -391,4 +397,4 @@ inline ze_api_version_t GetVersion() {
 } // namespace ze
 } // namespace utils
 
-#endif // PTI_UTILS_ZE_UTILS_H_
+#endif // PTI_UTILS_ZE_UTILS_H_