-
Notifications
You must be signed in to change notification settings - Fork 321
Fix NVFP4 to_copy #2812
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix NVFP4 to_copy #2812
Conversation
**Summary:** Fixes #2811 **Test Plan:** ``` pytest test/prototype/mx_formats/test_nvfp4_tensor.py -k to_copy ```
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/2812
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 New Failure, 1 Cancelled JobAs of commit fb5e41c with merge base 083361b ( NEW FAILURE - The following job has failed:
CANCELLED JOB - The following job was cancelled. Please retry:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
Great catch! |
@pytest.mark.skipif( | ||
not TORCH_VERSION_AT_LEAST_2_8, reason="NVFP4 requires PyTorch 2.8+" | ||
) | ||
def test_nvfp4_to_copy(): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
should we verify the effect for to_copy as well?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
added
1ddb363
to
6b794bb
Compare
6b794bb
to
fb5e41c
Compare
* Fix NVFP4 to_copy **Summary:** Fixes #2811 **Test Plan:** ``` pytest test/prototype/mx_formats/test_nvfp4_tensor.py -k to_copy ``` * Update test_nvfp4_tensor.py
Summary: Fixes #2811
Test Plan: