AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

replacing rccl_bfloat16 with hip_bfloat16 (#1126)

Browse Source 
Co-authored-by: mberenjk <mberenjk@amd.com>
This commit is contained in:
mberenjk
2024-04-11 11:30:37 -05:00
committed by GitHub
parent d3899c0581
commit 428837ffe4
12 changed files with 30 additions and 306 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/common/PtrUnion.cpp
+2 -2
View File
@@ -167,7 +167,7 @@ namespace RcclUnitTesting
case ncclFloat32: F4[idx] = valueF; break;
case ncclFloat64: F8[idx] = valueF; break;
case ncclFp8E5M2: B1[idx] = rccl_bfloat8(valueF); break;
case ncclBfloat16: B2[idx] = rccl_bfloat16(static_cast<float>(valueF)); break;
case ncclBfloat16: B2[idx] = hip_bfloat16(static_cast<float>(valueF)); break;
default:
ERROR("Unsupported datatype\n");
return TEST_FAIL;
@@ -286,7 +286,7 @@ namespace RcclUnitTesting
case ncclFloat32: F4[idx] /= divisor; break;
case ncclFloat64: F8[idx] /= divisor; break;
case ncclFp8E5M2: B1[idx] = (rccl_bfloat8((float)(B1[idx]) / divisor)); break;
case ncclBfloat16: B2[idx] = (rccl_bfloat16((float)(B2[idx]) / divisor)); break;
case ncclBfloat16: B2[idx] = (hip_bfloat16((float)(B2[idx]) / divisor)); break;
default:
ERROR("Unsupported datatype\n");
return TEST_FAIL;