[DeprecationWarning] Replacing soon-to-be-deprecated lazyInitCUDA with lazyInitDevice(at::kCUDA) (dmlc#7855)

Author: drivanov

graphbolt/src/cuda/common.h

@@ -58,7 +58,9 @@ struct CUDAWorkspaceAllocator {
   // Required by thrust to satisfy allocator requirements.
   using value_type = value_t;
 
-  explicit CUDAWorkspaceAllocator() { at::globalContext().lazyInitCUDA(); }
+  explicit CUDAWorkspaceAllocator() {
+    at::globalContext().lazyInitDevice(at::kCUDA);
+  }
 
   template <class U>
   CUDAWorkspaceAllocator(CUDAWorkspaceAllocator<U> const&) noexcept {}

tensoradapter/pytorch/torch.cpp

@@ -28,7 +28,7 @@ TA_EXPORTS void CPURawDelete(void* ptr) {
 
 #ifdef DGL_USE_CUDA
 TA_EXPORTS void* CUDARawAlloc(size_t nbytes, cudaStream_t stream) {
-  at::globalContext().lazyInitCUDA();
+  at::globalContext().lazyInitDevice(at::kCUDA);
   return c10::cuda::CUDACachingAllocator::raw_alloc_with_stream(nbytes, stream);
 }