NVIDIA-NeMo / NVIDIA-NeMo/RL

gpt-oss sft megatron does not support sequence packing

Open
#1,685 7 comments 1 reaction 1 assignee Claimed by @seonjinn View on GitHub
bug community-request waiting-on-maintainers x-mini x-sf
Dominant language
Python
Stars
2k
Forks
561
Avg merge
4d 5h
Merged PRs (30d)
145

Description

**Describe the bug**

When set `sequence_packing=true` and use Megatron backend for fine-tuning gpt-oss. The debug info says no attention backend available.
If `sequence_packing=False`, only `UnfusedAttention` is available, `FlashAttention` disabled for softmax_type = learnable, `FusedAttention` disabled as no backend supports the provided input.

This is different from the fine-tuning in megatron-bridge, which support `FusedAttention` with learnable softmax.

Update: The different backends enabled are due to different versions of cuDNN in docker image. Latest nemo-rl nano image has cuDNN version = 91002, but in nemo image cuDNN version is 91310.

**Steps/Code to reproduce bug**

Please list *minimal* steps or code snippet for us to be able to reproduce the bug.

A helpful guide on on how to craft a minimal bug report http://matthewrocklin.com/blog/work/2018/02/28/minimal-bug-reports.

**Expected behavior**

A clear and concise description of what you expected to happen.

**Additional context**

Add any other context about the problem here.

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.