replacing rccl_bfloat16 with hip_bfloat16 (#1126)

Co-authored-by: mberenjk <mberenjk@amd.com>

[ROCm/rccl commit: 428837ffe4]
This commit is contained in:
mberenjk
2024-04-11 11:30:37 -05:00
zatwierdzone przez GitHub
rodzic 165d51b255
commit da835cff9c
12 zmienionych plików z 30 dodań i 306 usunięć
+2 -2
Wyświetl plik
@@ -1558,7 +1558,7 @@ static ncclResult_t hostToDevRedOp(
float f32;
double f64;
#if defined(RCCL_BFLOAT16)
rccl_bfloat16 bf16;
hip_bfloat16 bf16;
#endif
#if defined(RCCL_FLOAT8)
rccl_float8 fp8_e4m3;
@@ -1602,7 +1602,7 @@ static ncclResult_t hostToDevRedOp(
#if defined(RCCL_BFLOAT16)
case ncclBfloat16:
opFull->op = ncclDevPreMulSum;
bf16 = (rccl_bfloat16)(float(1.0/comm->nRanks));
bf16 = (hip_bfloat16)(float(1.0/comm->nRanks));
break;
#endif
#if defined(RCCL_FLOAT8)