Fixing the AR_Bias issue for FP8 (#155)

Authored-by: Marzieh Berenjkoub <146776561+mberenjk@users.noreply.github.com>
Co-authored-by: corey-derochie-amd <161367113+corey-derochie-amd@users.noreply.github.com>
Co-authored-by: Nilesh M Negi <Nilesh.Negi@amd.com>

[ROCm/rccl-tests commit: 33cc4df1e4]
Tento commit je obsažen v:
mberenjk
2025-10-18 12:46:31 -07:00
odevzdal GitHub
rodič 75a69211a0
revize abf0605823
+8 -3
Zobrazit soubor
@@ -1083,9 +1083,14 @@ void applyBias1(
#if HAVE_ncclBfloat16
case ncclBfloat16: CASE_TY(hip_bfloat16)
#endif
#if HAVE_ncclfp8
case ncclFp8E4M3: CASE_TY(rccl_float8)
case ncclFp8E5M2: CASE_TY(rccl_bfloat8)
#if HAVE_ncclfp8_DEVICE || HIP_VERSION < 60300000
case ncclFloat8e4m3: CASE_TY(rccl_float8)
case ncclFloat8e5m2: CASE_TY(rccl_bfloat8)
#elif HAVE_ncclfp8_HOST
case ncclFloat8e4m3: if (rccl_float8_useFnuz) { CASE_TY(__hip_fp8_e4m3_fnuz) }
else { CASE_TY(__hip_fp8_e4m3) }
case ncclFloat8e5m2: if (rccl_float8_useFnuz) { CASE_TY(__hip_fp8_e5m2_fnuz) }
else { CASE_TY(__hip_fp8_e5m2) }
#endif
case ncclFloat32: CASE_TY(float)
case ncclFloat64: CASE_TY(double)