Skip to content

Use nvfuser_direct for version checking#2812

Merged
KaelanDt merged 1 commit intomainfrom
wjy/direct
Jan 29, 2026
Merged

Use nvfuser_direct for version checking#2812
KaelanDt merged 1 commit intomainfrom
wjy/direct

Conversation

@wujingyue
Copy link
Collaborator

@wujingyue wujingyue commented Jan 28, 2026

PyTorch 2.8 CIs have been using nvFuser 0.2.34, the version that supports direct bindings: https://github.com/Lightning-AI/lightning-thunder/blob/main/thunder/executors/nvfuserex_impl.py#L74

image

@wujingyue wujingyue marked this pull request as ready for review January 28, 2026 18:38
@wujingyue
Copy link
Collaborator Author

@KaelanDt could you merge this? The CI errors are pre-existing.

FAILED thunder/tests/test_jit_general.py::test_autograd_function_apply - KeyError: 'saved_for_backward_idx'

That has existed for about a month.

@KaelanDt KaelanDt merged commit 1e22aac into main Jan 29, 2026
49 of 54 checks passed
@KaelanDt KaelanDt deleted the wjy/direct branch January 29, 2026 18:44
wujingyue added a commit that referenced this pull request Jan 29, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants