NVIDIA / NVIDIA/Megatron-LM

ci: test_fused_indexer_loss_gradient_tp_consistency is flaky in dev

Open
#4,300 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
17.9k
Forks
4.5k
Avg merge
4d 6h
Merged PRs (30d)
271

Description

## Description

`test_fused_indexer_loss_gradient_tp_consistency` in `tests/unit_tests/transformer/experimental_attention_variant/test_attention_variant_dsa.py` is flaky in the dev container CI.

## Failure

- CI run: https://github.com/NVIDIA/Megatron-LM/actions/runs/24403010080/job/71283603051#step:3:998

## Fix

The test has been marked with `@pytest.mark.flaky_in_dev` in:
- PR: #4299

## Owner

Test introduced by @kunlunl.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.