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] require shape in TritonCSEVariable (#162275)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/162275
Approved by: https://github.com/mlazos
ghstack dependencies: #164158
This commit is contained in:
Isuru Fernando
2025-09-30 14:15:23 +00:00
committed by PyTorch MergeBot
parent 6ece527fc5
commit 3af2f0c12a
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/_inductor/codegen/triton.py
View File

@ -951,8 +951,7 @@ class TritonCSEVariable(CSEVariable):
# We'll use this to track which masks the variable needs when used for indirect indexing
self.mask_vars: OrderedSet[str] = OrderedSet()
assert dtype is not None, "TritonCSEVariable must have dtype"
# TODO: uncomment this and fix the few failures left
# assert shape is not None, "TritonCSEVariable must have shape"
assert shape is not None, "TritonCSEVariable must have shape"
def update_on_args(self, name, args, kwargs):
for arg in args: