hgemm ctype is fp32 in default

400cda78 · yuguo · d95c28c4 · 400cda78
Commit 400cda78 authored Aug 24, 2023 by yuguo
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

oneflow/core/ep/cuda/primitive/broadcast_matmul.cpp oneflow/core/ep/cuda/primitive/broadcast_matmul.cpp +1 -1

No files found.
--- a/oneflow/core/ep/cuda/primitive/broadcast_matmul.cpp
+++ b/oneflow/core/ep/cuda/primitive/broadcast_matmul.cpp
@@ -286,7 +286,7 @@ hipblasDatatype_t GetComputeType(DataType data_type) {
    case kDouble: return HIPBLAS_R_64F;
    case kFloat16: {
      const bool allow_half_accumulation =
-          ParseBooleanFromEnv("ONEFLOW_MATMUL_ALLOW_HALF_PRECISION_ACCUMULATION", true);
+          ParseBooleanFromEnv("ONEFLOW_MATMUL_ALLOW_HALF_PRECISION_ACCUMULATION", false);
      if (allow_half_accumulation) {
        return HIPBLAS_R_16F;
      } else {