upadte backbone cuda

jindavid · jindavid · commit f643a0d234e2 · 2025-05-14T15:20:30.000-04:00
diff --git a/src/backbone_cuda.jl b/src/backbone_cuda.jl
@@ -1,4 +1,5 @@
 using CUDA
+using CUDA.CUSOLVER, CUDA.CUBLAS
 
 for (Xpotrf_buffer, Xpotrf, Xtrsm, Xgemm, T) in (
     (:cusolverDnSpotrf_bufferSize, :cusolverDnSpotrf, :cublasStrsm_v2, :cublasSgemm_v2, :Float32),

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`using CUDA`
	`2`	`+using CUDA.CUSOLVER, CUDA.CUBLAS`
`2`	`3`
`3`	`4`	`for (Xpotrf_buffer, Xpotrf, Xtrsm, Xgemm, T) in (`
`4`	`5`	`(:cusolverDnSpotrf_bufferSize, :cusolverDnSpotrf, :cublasStrsm_v2, :cublasSgemm_v2, :Float32),`