diff --git a/Project.toml b/Project.toml index 3d0abc9b1..9fa5bd6ca 100644 --- a/Project.toml +++ b/Project.toml @@ -25,11 +25,10 @@ CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba" ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4" FiniteDifferences = "26cc04aa-876d-5657-8c51-4c34ba976000" Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6" -cuTENSOR = "011b41b2-24ef-40a8-b3eb-fa098493e9e1" [extensions] TensorKitAMDGPUExt = "AMDGPU" -TensorKitCUDAExt = ["CUDA", "cuTENSOR"] +TensorKitCUDAExt = "CUDA" TensorKitChainRulesCoreExt = "ChainRulesCore" TensorKitFiniteDifferencesExt = "FiniteDifferences" TensorKitMooncakeExt = "Mooncake" @@ -57,5 +56,4 @@ TensorKitSectors = "0.3.7" TensorOperations = "5.5" TupleTools = "1.5" VectorInterface = "0.4.8, 0.5, 0.6" -cuTENSOR = "6" julia = "1.10" diff --git a/ext/TensorKitCUDAExt/TensorKitCUDAExt.jl b/ext/TensorKitCUDAExt/TensorKitCUDAExt.jl index aa7320f32..dc48042a4 100644 --- a/ext/TensorKitCUDAExt/TensorKitCUDAExt.jl +++ b/ext/TensorKitCUDAExt/TensorKitCUDAExt.jl @@ -2,7 +2,6 @@ module TensorKitCUDAExt using CUDA, CUDA.cuBLAS, CUDA.cuSOLVER, CUDA.cuRAND, LinearAlgebra using CUDA: @allowscalar -using cuTENSOR: cuTENSOR import CUDA.cuRAND: rand as curand, rand! as curand!, randn as curandn, randn! as curandn! using Strided: StridedViews using CUDA.CUDACore.KernelAbstractions: @kernel, @index, get_backend diff --git a/test/Project.toml b/test/Project.toml index 18af8af80..f8602fad0 100644 --- a/test/Project.toml +++ b/test/Project.toml @@ -35,6 +35,7 @@ Aqua = "0.6, 0.7, 0.8" AllocCheck = "0.2" ChainRulesTestUtils = "1" Combinatorics = "1" +cuTENSOR = "6" GPUArrays = "11.3.1" JET = "0.9, 0.10, 0.11" ParallelTestRunner = "2"