Make swizzle in pycute work (#2553)

2025-08-20 10:21:00 +08:00
parent 5b76420d6a
commit ec18e8043b
1 changed files with 1 additions and 1 deletions
--- a/python/pycute/swizzle.py
+++ b/python/pycute/swizzle.py
@ -75,7 +75,7 @@ class Swizzle:

  # Size of the domain
  def size(self):
-    return 1 << (bits + base + abs(shift))
+    return 1 << (self.bits + self.base + abs(self.shift))

  # Size of the codomain
  def cosize(self):