OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Revert "[inductor] require shape in TritonCSEVariable (#162275)"

Browse Source 
This reverts commit 3af2f0c12a.

Reverted https://github.com/pytorch/pytorch/pull/162275 on behalf of https://github.com/clee2000 due to still failing due to the above D84932446 ([comment](https://github.com/pytorch/pytorch/pull/162275#issuecomment-3423153819))
This commit is contained in:
PyTorch MergeBot 2025-10-20 17:55:54 +00:00
parent 150682ba7f
commit 240c13394e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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