small things from endrange (#11339)

* small things from endrange * store
2025-07-22 19:45:37 -07:00
parent 53339e62f7
commit c65b5aab62
4 changed files with 11 additions and 11 deletions
@@ -56,7 +56,6 @@ class PythonProgram:
        assert dtype is not None, f"{uop} is missing a dtype"
        dl[i] = dtype
        if uop is Ops.STORE:
-          assert len(inp) == 2, "expected store is ([(memory, offset, gate)], [value])"
          for j,val in enumerate(inp[1] if dtp[1].count > 1 else [inp[1]]):
            for (m,o,g),v in zip(inp[0], val):
              if g: _store(m, o+j, v)