Move things to GPUToolbox. (#2826)

maleadt · web-flow · commit 832eba990add · 2025-07-26T19:09:10.000-04:00
diff --git a/Project.toml b/Project.toml
@@ -66,7 +66,7 @@ EnzymeCore = "0.8.2"
 ExprTools = "0.1"
 GPUArrays = "11.2.1"
 GPUCompiler = "0.24, 0.25, 0.26, 0.27, 1"
-GPUToolbox = "0.2"
+GPUToolbox = "0.3"
 KernelAbstractions = "0.9.32"
 LLVM = "9.1"
 LLVMLoopInfo = "1"
diff --git a/lib/cublas/CUBLAS.jl b/lib/cublas/CUBLAS.jl
@@ -1,6 +1,7 @@
 module CUBLAS
 
 using ..APIUtils
+using ..GPUToolbox
 
 using ..CUDA
 using ..CUDA: CUstream, cuComplex, cuDoubleComplex, libraryPropertyType, cudaDataType, i32
diff --git a/lib/cufft/CUFFT.jl b/lib/cufft/CUFFT.jl
@@ -1,6 +1,7 @@
 module CUFFT
 
 using ..APIUtils
+using ..GPUToolbox
 
 using ..CUDA_Runtime
 
diff --git a/lib/cupti/CUPTI.jl b/lib/cupti/CUPTI.jl
@@ -1,6 +1,7 @@
 module CUPTI
 
 using ..APIUtils
+using ..GPUToolbox
 
 using ..CUDA_Runtime
 
diff --git a/lib/curand/CURAND.jl b/lib/curand/CURAND.jl
@@ -1,6 +1,7 @@
 module CURAND
 
 using ..APIUtils
+using ..GPUToolbox
 
 using ..CUDA_Runtime
 
diff --git a/lib/cusolver/CUSOLVER.jl b/lib/cusolver/CUSOLVER.jl
@@ -1,6 +1,7 @@
 module CUSOLVER
 
 using ..APIUtils
+using ..GPUToolbox
 
 using ..CUDA_Runtime
 
diff --git a/lib/cusparse/CUSPARSE.jl b/lib/cusparse/CUSPARSE.jl
@@ -1,6 +1,7 @@
 module CUSPARSE
 
 using ..APIUtils
+using ..GPUToolbox
 
 using ..CUDA_Runtime
 
diff --git a/lib/nvml/NVML.jl b/lib/nvml/NVML.jl
@@ -1,6 +1,7 @@
 module NVML
 
-using ..APIUtils
+
+using ..GPUToolbox
 
 using ..CUDA
 
diff --git a/lib/utils/APIUtils.jl b/lib/utils/APIUtils.jl
@@ -5,15 +5,8 @@ using ..CUDA
 using LLVM
 using LLVM.Interop
 
-# helpers that facilitate working with CUDA APIs
-using GPUToolbox: @checked, @debug_ccall, @gcsafe_ccall
-export @checked, @debug_ccall, @gcsafe_ccall
-
 include("call.jl")
-include("enum.jl")
-include("threading.jl")
 include("cache.jl")
-include("memoization.jl")
 include("struct_size.jl")
 
 end
diff --git a/lib/utils/enum.jl b/lib/utils/enum.jl
diff --git a/lib/utils/memoization.jl b/lib/utils/memoization.jl
diff --git a/lib/utils/threading.jl b/lib/utils/threading.jl
diff --git a/test/core/apiutils.jl b/test/core/apiutils.jl
