frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[inductor] don't try to reorder loops for template (#165601)

Browse Source 
fix https://github.com/pytorch/pytorch/issues/165579

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165601
Approved by: https://github.com/yushangdi
This commit is contained in:
Shunting Zhang
2025-10-15 17:52:57 -07:00
committed by PyTorch MergeBot
parent 7669ac9402
commit a303d6dda9
2 changed files with 31 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/_inductor/scheduler.py
View File

@ -3994,6 +3994,12 @@ class Scheduler:
):
return -1
# in some rare case, a template can be passed in.
# Check test_interaction_with_multi_template in test_loop_ordering.py
# and https://github.com/pytorch/pytorch/issues/165579
if node1.is_template() or node2.is_template():
return -1
node1_buffer_names = node1.read_writes.buffer_names()
node2_buffer_names = node2.read_writes.buffer_names()
# Fast path: no common buffers.