[GraphBolt][CUDA] Don't link against `libcuda.so`. (#7313)

graphbolt/src/cuda/common.h

@@ -11,7 +11,6 @@
 #include <c10/cuda/CUDACachingAllocator.h>
 #include <c10/cuda/CUDAException.h>
 #include <c10/cuda/CUDAStream.h>
-#include <cuda.h>
 #include <cuda_runtime.h>
 #include <torch/script.h>
 
@@ -89,18 +88,12 @@ inline bool is_zero<dim3>(dim3 size) {
   return size.x == 0 || size.y == 0 || size.z == 0;
 }
 
-#define CUDA_DRIVER_CHECK(EXPR)                       \
+#define CUDA_RUNTIME_CHECK(EXPR)                           \
   do {                                                     \
-    CUresult __err = EXPR;                            \
-    if (__err != CUDA_SUCCESS) {                      \
-      const char* err_str;                            \
-      CUresult get_error_str_err C10_UNUSED =         \
-          cuGetErrorString(__err, &err_str);          \
-      if (get_error_str_err != CUDA_SUCCESS) {        \
-        AT_ERROR("CUDA driver error: unknown error"); \
-      } else {                                        \
-        AT_ERROR("CUDA driver error: ", err_str);     \
-      }                                               \
+    cudaError_t __err = EXPR;                              \
+    if (__err != cudaSuccess) {                            \
+      auto get_error_str_err = cudaGetErrorString(__err);  \
+      AT_ERROR("CUDA runtime error: ", get_error_str_err); \
     }                                                      \
   } while (0)
 

graphbolt/src/cuda/unique_and_compact_impl.cu

@@ -274,8 +274,8 @@ UniqueAndCompactBatched(
       return it->second;
     } else {
       int major;
-      CUDA_DRIVER_CHECK(cuDeviceGetAttribute(
-          &major, CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR, dev_id));
+      CUDA_RUNTIME_CHECK(cudaDeviceGetAttribute(
+          &major, cudaDevAttrComputeCapabilityMajor, dev_id));
       return compute_capability_cache[dev_id] = major;
     }
   }();