From 9dacab41539dfe06358d5fa07c0d840a7dc51fc6 Mon Sep 17 00:00:00 2001 From: dshi7 Date: Mon, 24 Nov 2025 15:57:53 -0800 Subject: [PATCH] bug fix --- third_party/tlx/language/tlx/types.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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):