Skip to content

Commit dc89057

Browse files
committed
Add complex types for hipBLAS GEAM
1 parent 22ebb3b commit dc89057

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

hip/base/hipblas_bindings.hip.hpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,9 @@ GKO_BIND_HIPBLAS_GEMM(ValueType, detail::not_implemented);
119119

120120
GKO_BIND_HIPBLAS_GEAM(float, hipblasSgeam);
121121
GKO_BIND_HIPBLAS_GEAM(double, hipblasDgeam);
122-
// Hipblas does not provide geam complex version yet.
122+
GKO_BIND_HIPBLAS_GEAM(std::complex<float>, hipblasCgeam);
123+
GKO_BIND_HIPBLAS_GEAM(std::complex<double>, hipblasZgeam);
124+
123125
template <typename ValueType>
124126
GKO_BIND_HIPBLAS_GEAM(ValueType, detail::not_implemented);
125127

0 commit comments

Comments
 (0)