fix cudaDeviceProp

4ab3e3db · zhuwenwen · a4b438d0 · 4ab3e3db
Commit 4ab3e3db authored Aug 19, 2025 by zhuwenwen
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 3 deletions

csrc/cuda_compat.h csrc/cuda_compat.h +6 -3

No files found.
--- a/csrc/cuda_compat.h
+++ b/csrc/cuda_compat.h
@@ -12,9 +12,12 @@ struct Utils {
    if (!is_cached) {
      int device_id;
-      cudaDeviceProp deviceProp;
+      // cudaDeviceProp deviceProp;
-      cudaGetDevice(&device_id);
+      // cudaGetDevice(&device_id);
-      cudaGetDeviceProperties(&deviceProp, device_id);
+      // cudaGetDeviceProperties(&deviceProp, device_id);
+      hipDeviceProp_t deviceProp;
+      hipGetDevice(&device_id);
+      hipGetDeviceProperties(&deviceProp, device_id);
      result = deviceProp.warpSize;
      is_cached = true;