10000 set_mat_mul_precision is flakey · Issue #9129 · pytorch/xla · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

set_mat_mul_precision is flakey #9129

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.

Sign up for GitHub

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

Jump to bottom
Open
yaoshiang opened this issue May 9, 2025 · 1 comment
Open

set_mat_mul_precision is flakey #9129

yaoshiang opened this issue May 9, 2025 · 1 comment

Comments

@yaoshiang
Copy link
Collaborator

🐛 Bug

set_mat_mul_precision seems to allow switching the precision within a single process... sometimes, like in the precision_tutorial.py/ipynb. But in the unit test test_mat_mul_precision, there's an example of a test that switches the precision unsuccessfully.

To Reproduce

One unit test in test_mat_mul_precision.py is decorated "@expectedfailure". Once this issue is resolved, we should be able to remove that decorator and see that these tests work as intended, within a loop.

PYTHONPATH="$TEST_CDIR${PYTHONPATH:+:$PYTHONPATH}" python3 -m unittest test_mat_mul_precision.TestMatMulPrecision.test_all

Expected behavior

Program can switch mat_mul_precision between default, high, and highest dynamically in a single program.
@yaoshiang
Copy link
Collaborator Author

See this discussion for some analysis from @tengyifei. #9103 (comment)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@yaoshiang
0