Fix rccl test suite to use hip_bf16.h instead of hip_bfloat16.h for the __bf16 intrinsic (#2082)
[ROCm/rccl commit: cc6e259a02]
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
e4aef19511
Коммит
0ced7aede8
@@ -225,7 +225,7 @@ namespace RcclUnitTesting
|
||||
case ncclFloat32: ss << scalarsPerRank.F4[this->globalRank]; break;
|
||||
case ncclFloat64: ss << scalarsPerRank.F8[this->globalRank]; break;
|
||||
case ncclFloat8e5m2: ss << (float)scalarsPerRank.B1[this->globalRank]; break;
|
||||
case ncclBfloat16: ss << scalarsPerRank.B2[this->globalRank]; break;
|
||||
case ncclBfloat16: ss << (float)scalarsPerRank.B2[this->globalRank]; break;
|
||||
default: ss << "(UNKNOWN)";
|
||||
}
|
||||
ss << " ";
|
||||
|
||||
Ссылка в новой задаче
Block a user