From eb60c55ebef75387021e6103c11da4ad2ea979d9 Mon Sep 17 00:00:00 2001 From: Varun Thumbe Date: Wed, 1 Jul 2026 22:42:32 +0000 Subject: [PATCH 1/2] skip tests on hopper Signed-off-by: Varun Thumbe --- qa/L1_pytorch_mcore_fsdp_integration/test.sh | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/qa/L1_pytorch_mcore_fsdp_integration/test.sh b/qa/L1_pytorch_mcore_fsdp_integration/test.sh index d63c66f2ea..918ac7fecf 100644 --- a/qa/L1_pytorch_mcore_fsdp_integration/test.sh +++ b/qa/L1_pytorch_mcore_fsdp_integration/test.sh @@ -4,6 +4,14 @@ set -e +# This test uses the MXFP8 recipe (--fp8-recipe mxfp8), which is only supported +# on Blackwell (compute capability 10.0) and newer. +DEVICE_ARCH=$(nvidia-smi --query-gpu=compute_cap --format=csv,noheader | head -n 1 | sed 's/[^0-9]//g') +if [[ -z "${DEVICE_ARCH}" || ${DEVICE_ARCH} -lt 100 ]]; then + echo "Skipping L1_pytorch_mcore_fsdp_integration: MXFP8 requires compute capability 10.0+ (Blackwell), detected compute_cap=${DEVICE_ARCH:-unknown}." + exit 0 +fi + # Megatron-LM / Megatron-FSDP commit for main branch on Apr. 10, 2026. # Necessary to support wgrad accumulate fusion and Megatron-FSDP NCCL UBR, # and fixes decoupled_grad <> DistOpt usage in Megatron-LM. From 6058b7c8021fb57553722d771019659c3a256789 Mon Sep 17 00:00:00 2001 From: vthumbe1503 Date: Wed, 1 Jul 2026 15:54:19 -0700 Subject: [PATCH 2/2] Update qa/L1_pytorch_mcore_fsdp_integration/test.sh Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com> Signed-off-by: vthumbe1503 --- qa/L1_pytorch_mcore_fsdp_integration/test.sh | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/qa/L1_pytorch_mcore_fsdp_integration/test.sh b/qa/L1_pytorch_mcore_fsdp_integration/test.sh index 918ac7fecf..e08cb8bb98 100644 --- a/qa/L1_pytorch_mcore_fsdp_integration/test.sh +++ b/qa/L1_pytorch_mcore_fsdp_integration/test.sh @@ -6,9 +6,10 @@ set -e # This test uses the MXFP8 recipe (--fp8-recipe mxfp8), which is only supported # on Blackwell (compute capability 10.0) and newer. -DEVICE_ARCH=$(nvidia-smi --query-gpu=compute_cap --format=csv,noheader | head -n 1 | sed 's/[^0-9]//g') +DEVICE_ARCH_RAW=$(nvidia-smi --query-gpu=compute_cap --format=csv,noheader | head -n 1) +DEVICE_ARCH=$(echo "${DEVICE_ARCH_RAW}" | sed 's/[^0-9]//g') if [[ -z "${DEVICE_ARCH}" || ${DEVICE_ARCH} -lt 100 ]]; then - echo "Skipping L1_pytorch_mcore_fsdp_integration: MXFP8 requires compute capability 10.0+ (Blackwell), detected compute_cap=${DEVICE_ARCH:-unknown}." + echo "Skipping L1_pytorch_mcore_fsdp_integration: MXFP8 requires compute capability 10.0+ (Blackwell), detected compute_cap=${DEVICE_ARCH_RAW:-unknown}." exit 0 fi