fix: guard tensormap with cuda version check (#18107)

spectrometerHBH · web-flow · commit 956b65910c47 · 2025-07-02T10:30:37.000-04:00
diff --git a/src/runtime/cuda/cuda_device_api.cc b/src/runtime/cuda/cuda_device_api.cc
@@ -357,6 +357,7 @@ TVM_DLL int GetCudaDeviceCount() {
 
 TVM_FFI_REGISTER_GLOBAL("runtime.GetCudaDeviceCount").set_body_typed(GetCudaDeviceCount);
 
+#if (CUDA_VERSION >= 12000)
 /**
  * \brief FFI wrapper for cuTensorMapEncodeTiled.
  *
@@ -562,6 +563,7 @@ TVM_FFI_REGISTER_GLOBAL("runtime.cuTensorMapEncodeTiled")
         CHECK_EQ(res, CUDA_SUCCESS) << "Error in cuTensorMapEncodeTiled: " << errstr;
       }
     });
+#endif  // CUDA_VERSION >= 12000
 
 }  // namespace runtime
 }  // namespace tvm

Original file line number	Diff line number	Diff line change
`@@ -357,6 +357,7 @@ TVM_DLL int GetCudaDeviceCount() {`
`357`	`357`
`358`	`358`	`TVM_FFI_REGISTER_GLOBAL("runtime.GetCudaDeviceCount").set_body_typed(GetCudaDeviceCount);`
`359`	`359`
	`360`	`+#if (CUDA_VERSION >= 12000)`
`360`	`361`	`/**`
`361`	`362`	`* \brief FFI wrapper for cuTensorMapEncodeTiled.`
`362`	`363`	`*`
`@@ -562,6 +563,7 @@ TVM_FFI_REGISTER_GLOBAL("runtime.cuTensorMapEncodeTiled")`
`562`	`563`	`CHECK_EQ(res, CUDA_SUCCESS) << "Error in cuTensorMapEncodeTiled: " << errstr;`
`563`	`564`	`}`
`564`	`565`	`});`
	`566`	`+#endif // CUDA_VERSION >= 12000`
`565`	`567`
`566`	`568`	`} // namespace runtime`
`567`	`569`	`} // namespace tvm`