Skip to content

Commit 41b1fc9

Browse files
authored
Remove explicit dependence on cuTENSOR for CUDA ext (#455)
1 parent 848a3b4 commit 41b1fc9

3 files changed

Lines changed: 2 additions & 4 deletions

File tree

Project.toml

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,10 @@ CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
2525
ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
2626
FiniteDifferences = "26cc04aa-876d-5657-8c51-4c34ba976000"
2727
Mooncake = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
28-
cuTENSOR = "011b41b2-24ef-40a8-b3eb-fa098493e9e1"
2928

3029
[extensions]
3130
TensorKitAMDGPUExt = "AMDGPU"
32-
TensorKitCUDAExt = ["CUDA", "cuTENSOR"]
31+
TensorKitCUDAExt = "CUDA"
3332
TensorKitChainRulesCoreExt = "ChainRulesCore"
3433
TensorKitFiniteDifferencesExt = "FiniteDifferences"
3534
TensorKitMooncakeExt = "Mooncake"
@@ -57,5 +56,4 @@ TensorKitSectors = "0.3.7"
5756
TensorOperations = "5.5"
5857
TupleTools = "1.5"
5958
VectorInterface = "0.4.8, 0.5, 0.6"
60-
cuTENSOR = "6"
6159
julia = "1.10"

ext/TensorKitCUDAExt/TensorKitCUDAExt.jl

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ module TensorKitCUDAExt
22

33
using CUDA, CUDA.cuBLAS, CUDA.cuSOLVER, CUDA.cuRAND, LinearAlgebra
44
using CUDA: @allowscalar
5-
using cuTENSOR: cuTENSOR
65
import CUDA.cuRAND: rand as curand, rand! as curand!, randn as curandn, randn! as curandn!
76
using Strided: StridedViews
87
using CUDA.CUDACore.KernelAbstractions: @kernel, @index, get_backend

test/Project.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ Aqua = "0.6, 0.7, 0.8"
3535
AllocCheck = "0.2"
3636
ChainRulesTestUtils = "1"
3737
Combinatorics = "1"
38+
cuTENSOR = "6"
3839
GPUArrays = "11.3.1"
3940
JET = "0.9, 0.10, 0.11"
4041
ParallelTestRunner = "2"

0 commit comments

Comments
 (0)