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

Commit c78efca

Browse files
committed
Add back lost function.
1 parent 68edbfc commit c78efca

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

src/tensor/wrappers.jl

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -341,3 +341,29 @@ function reduction!(
341341

342342
return C
343343
end
344+
345+
function cutensorComputeType(T::DataType)
346+
if T == Float32
347+
return CUTENSOR_R_MIN_32F
348+
elseif T == ComplexF32
349+
return CUTENSOR_C_MIN_32F
350+
elseif T == Float16
351+
return CUTENSOR_R_MIN_16F
352+
elseif T == ComplexF16
353+
return CUTENSOR_C_MIN_16F
354+
elseif T == Float64
355+
return CUTENSOR_R_MIN_64F
356+
elseif T == ComplexF64
357+
return CUTENSOR_C_MIN_64F
358+
elseif T == Int8
359+
return CUTENSOR_R_MIN_8I
360+
elseif T == Int32
361+
return CUTENSOR_R_MIN_32I
362+
elseif T == UInt8
363+
return CUTENSOR_R_MIN_8U
364+
elseif T == UInt32
365+
return CUTENSOR_R_MIN_32U
366+
else
367+
throw(ArgumentError("cutensorComputeType equivalent for input type $T does not exist!"))
368+
end
369+
end

0 commit comments

Comments
 (0)