FIX

lightx2v/common/ops/conv/conv2d.py

@@ -58,7 +58,7 @@ class Conv2dWeight(Conv2dWeightTemplate):
         return destination
 
     def clear(self):
-        attrs = ["weight", "bias"]
+        attrs = ["weight", "bias", "pinned_weight", "pinned_bias"]
         for attr in attrs:
             if hasattr(self, attr):
                 delattr(self, attr)

lightx2v/common/ops/conv/conv3d.py

@@ -68,7 +68,7 @@ class Conv3dWeight(Conv3dWeightTemplate):
         return destination
 
     def clear(self):
-        attrs = ["weight", "bias"]
+        attrs = ["weight", "bias", "pinned_weight", "pinned_bias"]
         for attr in attrs:
             if hasattr(self, attr):
                 delattr(self, attr)

lightx2v/common/ops/mm/mm_weight.py

@@ -145,7 +145,7 @@ class MMWeightQuantTemplate(MMWeightTemplate):
                 self.pinned_weight = self.pinned_weight.t()
 
     def clear(self):
-        attrs = ["weight", "weight_scale", "bias"]
+        attrs = ["weight", "weight_scale", "bias", "pinned_weight", "pinned_weight_scale", "pinned_bias"]
         for attr in attrs:
             if hasattr(self, attr):
                 delattr(self, attr)

lightx2v/common/ops/norm/layer_norm_weight.py

@@ -34,7 +34,7 @@ class LNWeightTemplate(metaclass=ABCMeta):
         return self.weight.numel() * self.weight.element_size()
 
     def clear(self):
-        attrs = ["weight", "bias"]
+        attrs = ["weight", "bias", "pinned_weight", "pinned_bias"]
         for attr in attrs:
             if hasattr(self, attr):
                 delattr(self, attr)

lightx2v/common/ops/norm/rms_norm_weight.py

@@ -23,7 +23,7 @@ class RMSWeightTemplate(metaclass=ABCMeta):
             self.pinned_weight = torch.empty(self.weight.shape, pin_memory=True, dtype=self.weight.dtype)
 
     def clear(self):
-        attrs = ["weight"]
+        attrs = ["weight", "pinned_weight"]
         for attr in attrs:
             if hasattr(self, attr):
                 delattr(self, attr)

lightx2v/common/ops/tensor/tensor.py

@@ -22,7 +22,7 @@ class DefaultTensor:
             self.pinned_tensor = torch.empty(self.tensor.shape, pin_memory=True, dtype=self.tensor.dtype)
 
     def clear(self):
-        attrs = ["tensor"]
+        attrs = ["tensor", "pinned_tensor"]
         for attr in attrs:
             if hasattr(self, attr):
                 delattr(self, attr)