Fix cuTensor, cuTensorNet and cuStateVec when using local Toolkit (#2274)

Author: bjoe2k4

lib/custatevec/src/cuStateVec.jl

@@ -119,8 +119,8 @@ function __init__()
     # find the library
     global libcustatevec
     if CUDA.local_toolkit
-        dirs = CUDA_Runtime.find_toolkit()
-        path = CUDA_Runtime.get_library(dirs, "custatevec"; optional=true)
+        dirs = CUDA_Runtime_Discovery.find_toolkit()
+        path = CUDA_Runtime_Discovery.get_library(dirs, "custatevec"; optional=true)
         if path === nothing
             precompiling || @error "cuQuantum is not available on your system (looked for custatevec in $(join(dirs, ", ")))"
             return

lib/cutensor/src/cuTENSOR.jl

@@ -95,8 +95,8 @@ function __init__()
     # find the library
     global libcutensor
     if CUDA.local_toolkit
-        dirs = CUDA_Runtime.find_toolkit()
-        path = CUDA_Runtime.get_library(dirs, "cutensor"; optional=true)
+        dirs = CUDA_Runtime_Discovery.find_toolkit()
+        path = CUDA_Runtime_Discovery.get_library(dirs, "cutensor"; optional=true)
         if path === nothing
             precompiling || @error "cuTENSOR is not available on your system (looked in $(join(dirs, ", ")))"
             return

lib/cutensornet/src/cuTensorNet.jl

@@ -120,8 +120,8 @@ function __init__()
     # find the library
     global libcutensornet
     if CUDA.local_toolkit
-        dirs = CUDA_Runtime.find_toolkit()
-        path = CUDA_Runtime.get_library(dirs, "cutensornet"; optional=true)
+        dirs = CUDA_Runtime_Discovery.find_toolkit()
+        path = CUDA_Runtime_Discovery.get_library(dirs, "cutensornet"; optional=true)
         if path === nothing
             precompiling || @error "cuQuantum is not available on your system (looked for cutensornet in $(join(dirs, ", ")))"
             return