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

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

Browse Source 
This reverts commit c257570e6cd25753f9f0a640b965148ead2cf918.

Reverted https://github.com/pytorch/pytorch/pull/162275 on behalf of https://github.com/jeffdaily due to sorry this broke rocm CI; inductor/test_select_algorithm.py::TestTemplateRender::test_finalized_subclass_hooks [GH job link](https://github.com/pytorch/pytorch/actions/runs/18048893250/job/51366715091) [HUD commit link](c257570e6c) ([comment](https://github.com/pytorch/pytorch/pull/162275#issuecomment-3348159095))
This commit is contained in:
PyTorch MergeBot
2025-09-29 17:26:54 +00:00
parent 6bcc6bbc85
commit 6b473c90cf
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -875,7 +875,8 @@ 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"
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):
for arg in args: