Skip to content

GitLab

Commit 400cda78 authored by yuguo's avatar yuguo
Browse files

hgemm ctype is fp32 in default

parent d95c28c4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
oneflow/core/ep/cuda/primitive/broadcast_matmul.cpp
View file @ 400cda78
...@@ -286,7 +286,7 @@ hipblasDatatype_t GetComputeType(DataType data_type) { ...@@ -286,7 +286,7 @@ hipblasDatatype_t GetComputeType(DataType data_type) {
case kDouble: return HIPBLAS_R_64F; case kDouble: return HIPBLAS_R_64F;
case kFloat16: { case kFloat16: {
const bool allow_half_accumulation = 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) { if (allow_half_accumulation) {
return HIPBLAS_R_16F; return HIPBLAS_R_16F;
} else { } else {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment