Skip to content
This repository was archived by the owner on Mar 12, 2021. It is now read-only.

Commit 84b908c

Browse files
authored
Merge pull request #565 from JuliaGPU/tb/cutensor_coverage
CUTENSOR coverage fixes.
2 parents 6d19577 + 8196bc9 commit 84b908c

File tree

2 files changed

+2
-8
lines changed

2 files changed

+2
-8
lines changed

src/CuArrays.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ function __init__()
105105
end
106106

107107
cuda = CUDAnative.version()
108-
cutensor_cuda = CUDNN.cuda_version()
108+
cutensor_cuda = CUTENSOR.cuda_version()
109109
if cutensor_cuda.major != cuda.major || cutensor_cuda.minor != cuda.minor
110110
silent || @warn("You are using CUTENSOR $cutensor for CUDA $cutensor_cuda with CUDA toolkit $cuda; these might be incompatible.")
111111
end

src/tensor/wrappers.jl

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -12,19 +12,13 @@ function version()
1212
end
1313

1414
function cuda_version()
15-
ver = cudnnGetCudartVersion()
15+
ver = cutensorGetCudartVersion()
1616
major, ver = divrem(ver, 1000)
1717
minor, patch = divrem(ver, 10)
1818

1919
VersionNumber(major, minor, patch)
2020
end
2121

22-
function cutensorCreate()
23-
handle = Ref{cutensorHandle_t}()
24-
cutensorCreate(handle)
25-
handle[]
26-
end
27-
2822
const ModeType = AbstractVector{<:Union{Char, Integer}}
2923

3024
is_unary(op::cutensorOperator_t) =

0 commit comments

Comments
 (0)