diff --git a/third_party/tlx/language/tlx/types.py b/third_party/tlx/language/tlx/types.py
index 6f3132ff41..d8d5967dd5 100644
--- a/third_party/tlx/language/tlx/types.py
+++ b/third_party/tlx/language/tlx/types.py
@@ -180,6 +180,7 @@ def to_ir(self, builder: ir.builder) -> None:
             self.numCTASplit,
             self.numCTAOrder,
             self.fp4Padded,
+            self.swizzled,
         )
 
     def __str__(self) -> str:
@@ -189,7 +190,7 @@ def __eq__(self, other) -> bool:
         return (type(self) is type(other) and self.shape == other.shape and self.order == other.order
                 and self.elemType == other.elemType and self.numCTAsPerCGA == other.numCTAsPerCGA
                 and self.numCTASplit == other.numCTASplit and self.numCTAOrder == other.numCTAOrder
-                and self.fp4Padded == other.fp4Padded)
+                and self.fp4Padded == other.fp4Padded and self.swizzled == other.swizzled)
 
 
 class storage_kind(enum.Enum):