fix cuda ext

QuantumBFS · Oct 7, 2023 · c8db4f4 · c8db4f4
1 parent ae6c465
commit c8db4f4
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/ext/YaoToEinsumCUDAExt.jl b/ext/YaoToEinsumCUDAExt.jl
@@ -1,9 +1,7 @@
 module YaoToEinsumCUDAExt
 using CUDA, YaoToEinsum
 
-export cu
-
-function cu(tnet::TensorNetwork)
+function CUDA.cu(tnet::TensorNetwork)
     return TensorNetwork(tnet.code, tnet.tensors .|> CuArray)
 end
-end
+end