Just support all memory alloc modes

JuliaGPU · Jan 23, 2025 · cc7e01f · cc7e01f
1 parent a2dedad
commit cc7e01f
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 57 deletions.
diff --git a/lib/cublas/CUBLAS.jl b/lib/cublas/CUBLAS.jl
@@ -4,7 +4,7 @@ using ..APIUtils
 
 using ..CUDA
 using ..CUDA: CUstream, cuComplex, cuDoubleComplex, libraryPropertyType, cudaDataType, i32
-using ..CUDA: unsafe_free!, retry_reclaim, isdebug, @sync, initialize_context, CuRefArray, DeviceMemory, UnifiedMemory
+using ..CUDA: unsafe_free!, retry_reclaim, isdebug, @sync, initialize_context, CuRefArray, AbstractMemory
 
 using ..CUDA: CUDA_Runtime
 using ..CUDA_Runtime