Skip to content

GitLab

You need to sign in or sign up before continuing.
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) {
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 {
......
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