From abf0605823aa047798c885b6c1b22f63be3dcc19 Mon Sep 17 00:00:00 2001 From: mberenjk <146776561+mberenjk@users.noreply.github.com> Date: Sat, 18 Oct 2025 12:46:31 -0700 Subject: [PATCH] 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 [ROCm/rccl-tests commit: 33cc4df1e4631d98a7a9ff1b1e0221f77ec81470] --- projects/rccl-tests/verifiable/verifiable.cu | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/projects/rccl-tests/verifiable/verifiable.cu b/projects/rccl-tests/verifiable/verifiable.cu index 0896c724d0..b845bd99e9 100644 --- a/projects/rccl-tests/verifiable/verifiable.cu +++ b/projects/rccl-tests/verifiable/verifiable.cu @@ -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)