From ec18e8043bb53382d7639d4b6753c3a4dfaffcb4 Mon Sep 17 00:00:00 2001
From: melonedo <44501064+melonedo@users.noreply.github.com>
Date: Wed, 20 Aug 2025 10:21:00 +0800
Subject: [PATCH] Make swizzle in pycute work (#2553)

---
 python/pycute/swizzle.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/pycute/swizzle.py b/python/pycute/swizzle.py
index 3f73d1bc..308aee0c 100644
--- 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):