diff --git a/python/bitblas/ops/general_matmul.py b/python/bitblas/ops/general_matmul.py
index ef6b3dfcb..4c518f55d 100644
--- a/python/bitblas/ops/general_matmul.py
+++ b/python/bitblas/ops/general_matmul.py
@@ -417,6 +417,7 @@ def transform_weight(self, weight, scale=None, zeros=None, bias=None):
         Returns:
         A list containing the transformed weight tensor and optionally the scale, zeros, and bias.
         """
+        weight = weight.contiguous()
         if self.W_dtype == self.A_dtype:
             if self.weight_transform is not None:
                 return self.weight_transform(weight.cpu()).cuda().contiguous()