Conditionalize code.

JuliaGPU · May 2, 2019 · dcc92d3 · dcc92d3
1 parent 429f02b
commit dcc92d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/CuArrays.jl b/src/CuArrays.jl
@@ -91,9 +91,9 @@ function __init__()
         CUSOLVER._dense_handle[] = C_NULL
         CUSOLVER._sparse_handle[] = C_NULL
         CUSPARSE._handle[] = C_NULL
-        CUTENSOR._handle[] = C_NULL
         CURAND._generator[] = nothing
-        isdefined(CuArrays, :CUDNN) && (CUDNN._handle[] = C_NULL)
+        isdefined(CuArrays, :CUDNN)    && (CUDNN._handle[] = C_NULL)
+        isdefined(CuArrays, :CUTENSOR) && (CUTENSOR._handle[] = C_NULL)
     end
     push!(CUDAnative.device!_listeners, callback)