diff options
author | Hieu Hoang <hieuhoang@gmail.com> | 2017-12-01 18:28:01 +0300 |
---|---|---|
committer | Hieu Hoang <hieuhoang@gmail.com> | 2017-12-01 18:28:01 +0300 |
commit | 22079a4aed6483d158075ea65b424fa488790ed8 (patch) | |
tree | 5dfb01cc1ebcdc1c03f2e42e8111ca8b46c0a8df | |
parent | 668aa6331cb9a3dc3ebe630788c2968bf38ffe90 (diff) |
custom Transpose. cublasHgeam does not exist
-rw-r--r-- | src/amun/half/mblas/matrix_functions.cu | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/src/amun/half/mblas/matrix_functions.cu b/src/amun/half/mblas/matrix_functions.cu index ee196bbd..0483fa27 100644 --- a/src/amun/half/mblas/matrix_functions.cu +++ b/src/amun/half/mblas/matrix_functions.cu @@ -407,11 +407,8 @@ Matrix& Prod(cublasHandle_t handle, Matrix& C, const Matrix& A, const Matrix& B, { assert((A.dim(2) == A.dim(3) == 1) || (B.dim(2) == B.dim(3) == 1)); - //HH - //Matrix::value_type alpha = 1.0; - //Matrix::value_type beta = 0.0; - Matrix::value_type alpha; - Matrix::value_type beta; + Matrix::value_type alpha = float2half_rn(1.0); + Matrix::value_type beta = float2half_rn(0.0); size_t m = A.dim(0) * A.dim(2) * A.dim(3); size_t k = A.dim(1); |