Skip to content
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

New Pytorch Triton breaks custom cast kernel MX #1799

Open
drisspg opened this issue Feb 28, 2025 · 0 comments
Open

New Pytorch Triton breaks custom cast kernel MX #1799

drisspg opened this issue Feb 28, 2025 · 0 comments
Labels

Comments

@drisspg
Copy link
Contributor

drisspg commented Feb 28, 2025

Summary

To run: TRITON_ALWAYS_COMPILE=1 TRITON_DUMP_DIR=my_directory_2 TRITON_KERNEL_DUMP=1 pytest -s -v test/prototype/mx_formats/test_custom_cast.py -k "test_fp4_triton_unscaled_cast"

Bad ttir on left good on right. No real differences
https://www.diffchecker.com/ueX5YZw4

TTGIR:
https://www.diffchecker.com/M5PS6QJg/

Differences in PTX
https://www.diffchecker.com/8mseNnKA/

@drisspg drisspg changed the title New Pytorch Triton breaks custom cast kernel New Pytorch Triton breaks custom cast kernel MX Feb 28, 2025
@drisspg drisspg added the mx label Feb 28, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

1 participant