fix 8bit int8 param device

Signed-off-by: jiqing-feng <[email protected]>

Author: jiqing-feng

bitsandbytes/nn/modules.py

@@ -657,7 +657,7 @@ def cpu(self):
 
     def xpu(self):
         # we store the 8-bit rows-major weight
-        B = self.data.contiguous().bfloat16().xpu()
+        B = self.data.contiguous().float16().xpu()
         CB, CBt, SCB, SCBt, coo_tensorB = bnb.functional.double_quant(B)
         if CBt is not None:
             del CBt
@@ -695,8 +695,9 @@ def to(self, *args, **kwargs):
                 return self.cpu()
         elif device.type == "xpu":
             if self.data.dtype == torch.int8:
+                self.data = self.data.contiguous().xpu()
                 self.CB = self.data
-                return super().xpu(device)
+                return self
             else:
                 return self.xpu()
         else: