hgemm fp32 ctype

ab239e0a · yuguo960516yuguo · d2d32668 · ab239e0a
Commit ab239e0a authored May 12, 2023 by yuguo960516yuguo
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

paddle/phi/kernels/funcs/blas/blas_impl.hip.h paddle/phi/kernels/funcs/blas/blas_impl.hip.h +2 -2

No files found.
--- a/paddle/phi/kernels/funcs/blas/blas_impl.hip.h
+++ b/paddle/phi/kernels/funcs/blas/blas_impl.hip.h
@@ -922,7 +922,7 @@ inline void Blas<paddle::platform::CUDADeviceContext>::GEMM(
                                       C,
                                       rocblas_datatype_f16_r,
                                       N,
-                                       rocblas_datatype_f16_r);
+                                       rocblas_datatype_f32_r);
 }
 template <>
 template <>
@@ -977,7 +977,7 @@ inline void Blas<phi::GPUContext>::GEMM(CBLAS_TRANSPOSE transA,
                                       C,
                                       rocblas_datatype_f16_r,
                                       N,
-                                       rocblas_datatype_f16_r);
+                                       rocblas_datatype_f32_r);
 }

 template <>