[inductor] require shape in TritonCSEVariable (#162275)

Pull Request resolved: #162275
Approved by: https://github.com/mlazos
ghstack dependencies: #164158

Author: isuruf

torch/_inductor/codegen/triton.py

@@ -969,8 +969,7 @@ def __init__(
         # 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: