add USE_ROCM ifdef

takeshi-yoshimura · takeshi-yoshimura · commit 3efccfeefc3b · 2025-12-05T17:25:12.000+09:00
Signed-off-by: Takeshi Yoshimura &lt;tyos@jp.ibm.com&gt;
diff --git a/fastsafetensors/cpp/ext.cpp b/fastsafetensors/cpp/ext.cpp
@@ -301,9 +301,10 @@ void init_gil_release_from_env() {
 
 int is_gds_supported(int deviceId)
 {
-    cudaError_t err;
-    int driverVersion = 0;
+#ifndef USE_ROCM
     int gdr_support = 1;
+    int driverVersion = 0;
+    cudaError_t err;
 
     err = cuda_fns.cudaDriverGetVersion(&driverVersion);
     if (err != cudaSuccess) {
@@ -319,6 +320,9 @@ int is_gds_supported(int deviceId)
         }
     }
     return gdr_support;
+#endif
+    // ROCm does not have GDS
+    return 0;
 }
 
 int init_gds()

Original file line number	Diff line number	Diff line change
`@@ -301,9 +301,10 @@ void init_gil_release_from_env() {`
`301`	`301`
`302`	`302`	`int is_gds_supported(int deviceId)`
`303`	`303`	`{`
`304`		`- cudaError_t err;`
`305`		`- int driverVersion = 0;`
	`304`	`+#ifndef USE_ROCM`
`306`	`305`	`int gdr_support = 1;`
	`306`	`+ int driverVersion = 0;`
	`307`	`+ cudaError_t err;`
`307`	`308`
`308`	`309`	`err = cuda_fns.cudaDriverGetVersion(&driverVersion);`
`309`	`310`	`if (err != cudaSuccess) {`
`@@ -319,6 +320,9 @@ int is_gds_supported(int deviceId)`
`319`	`320`	`}`
`320`	`321`	`}`
`321`	`322`	`return gdr_support;`
	`323`	`+#endif`
	`324`	`+ // ROCm does not have GDS`
	`325`	`+ return 0;`
`322`	`326`	`}`
`323`	`327`
`324`	`328`	`int init_gds()`