Update

[ghstack-poisoned]
Update (base update)
2025-11-05 08:24:57 +08:00 · 2025-09-30 14:15:23 +00:00 · 2025-09-30 14:15:23 +00:00 · 2025-09-26 16:15:09 +00:00 · 2025-09-26 16:15:09 +00:00 · 2025-09-16 14:09:45 +00:00
2 changed files with 2 additions and 3 deletions
--- a/test/inductor/test_select_algorithm.py
+++ b/test/inductor/test_select_algorithm.py
@ -513,7 +513,7 @@ class TestTemplateRender(TestCase):
    tmp0 = tl.load(A + xindex)
    tmp1 = tl.load(B + xindex)
    tmp2 = tmp0 + tmp1
-    {{store_output(("xindex",), "tmp2", mask="xmask")}}
+    {{store_output(("xindex",), "tmp2", mask="xmask", val_shape=("XBLOCK",))}}
    """
            ),
        )
--- a/torch/_inductor/codegen/triton.py
+++ b/torch/_inductor/codegen/triton.py
@ -875,8 +875,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:
Author	SHA1	Message	Date
Isuru Fernando	ef387d3709	Update [ghstack-poisoned]	2025-09-30 14:15:23 +00:00
Isuru Fernando	3817eb2ba2	Update (base update) [ghstack-poisoned]	2025-09-30 14:15:23 +00:00
Isuru Fernando	904a1960eb	Update [ghstack-poisoned]	2025-09-26 16:15:09 +00:00
Isuru Fernando	2ad1314ab3	Update (base update) [ghstack-poisoned]	2025-09-26 16:15:09 +00:00
Isuru Fernando	70eb95e408	Update [ghstack-poisoned]	2025-09-16 14:09:45 +00:00
Isuru Fernando	c08e751dde	Update (base update) [ghstack-poisoned]	2025-09-16 14:09:45 +00:00
Isuru Fernando	2c6584fb92	Update [ghstack-poisoned]	2025-09-15 23:44:57 +00:00
Isuru Fernando	480bddd81b	Update (base update) [ghstack-poisoned]	2025-09-15 23:44:57 +00:00
Isuru Fernando	3da8dc6a9c	Update [ghstack-poisoned]	2025-09-12 16:10:26 +00:00
Isuru Fernando	1869b45507	Update (base update) [ghstack-poisoned]	2025-09-12 16:10:26 +00:00
Isuru Fernando	d030cb74a9	Update [ghstack-poisoned]	2025-09-11 18:26:29 +00:00
Isuru Fernando	bafcba93b5	Update (base update) [ghstack-poisoned]	2025-09-11 18:26:29 +00:00
Isuru Fernando	a96ff8e52b	Update [ghstack-poisoned]	2025-09-10 20:06:51 +00:00
Isuru Fernando	2216524bf7	Update (base update) [ghstack-poisoned]	2025-09-10 20:06:51 +00:00
Isuru Fernando	5a481eae37	Update [ghstack-poisoned]	2025-09-09 18:20:10 +00:00
Isuru Fernando	836336e8c6	Update (base update) [ghstack-poisoned]	2025-09-09 18:20:10 +00:00
Isuru Fernando	f0386d1017	Update [ghstack-poisoned]	2025-09-08 22:13:21 +00:00
Isuru Fernando	01f7f44aff	Update (base update) [ghstack-poisoned]	2025-09-08 22:13:21 +00:00
Isuru Fernando	cf25e88371	Update [ghstack-poisoned]	2025-09-05 16:41:18 +00:00
Isuru Fernando	8e4a5b0e3f	Update (base update) [ghstack-poisoned]	2025-09-05 16:41:18 +00:00