xoscar 0.7.10__cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl → 0.7.11__cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

xoscar/serialization/mlx.py

@@ -40,25 +40,27 @@ dtype_map = {
 class MLXSerislizer(Serializer):
     @buffered
     def serial(self, obj: "mx.array", context: dict):  # type: ignore
-        mv = memoryview(obj)
-        header = dict(shape=obj.shape, format=mv.format)
-        # If the memoryview is a multi-dimension view, then there could
-        # trigger a bug of asyncio write: https://github.com/python/cpython/issues/135862
-        if mv.ndim > 1 or not mv.c_contiguous:
+        ravel_obj = obj.reshape(-1).view(mx.uint8)
+        mv = memoryview(ravel_obj)
+        header = dict(
+            shape=obj.shape, format=mv.format, dtype=str(obj.dtype).rsplit(".", 1)[-1]
+        )
+        if not mv.c_contiguous:
+            # NOTE: we only consider c contiguous here,
+            # MLX has no way to create f contiguous arrays.
             mv = memoryview(bytes(mv))
         return (header,), [mv], True
 
     def deserial(self, serialized: tuple, context: dict, subs: List[Any]):
         header = serialized[0]
-        shape, format = header["shape"], header["format"]
+        shape, format, dtype = header["shape"], header["format"], header["dtype"]
         mv = memoryview(subs[0])
         if mv.format != format:
             dtype = dtype_map.get(format, np.uint8)
             np_arr = np.frombuffer(mv, dtype=dtype).reshape(shape)  # parse
             mv = memoryview(np_arr)  # recreate memoryview
-        elif mv.shape != shape:
-            mv = mv.cast(format, shape)  # cast directly
-        return mx.array(mv)
+        ravel_array = mx.array(mv)
+        return ravel_array.view(getattr(mx, dtype)).reshape(shape)
 
 
 if mx is not None:

{xoscar-0.7.10.dist-info → xoscar-0.7.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xoscar
-Version: 0.7.10
+Version: 0.7.11
 Summary: Python actor framework for heterogeneous computing.
 Home-page: http://github.com/xorbitsai/xoscar
 Author: Qin Xuye

{xoscar-0.7.10.dist-info → xoscar-0.7.11.dist-info}/RECORD

@@ -73,7 +73,7 @@ xoscar/serialization/core.pxd,sha256=k4RoJgX5E5LGs4jdCQ7vvcn26MabXbrWoWhkO49X6YI
 xoscar/serialization/core.pyx,sha256=bjR-zXGm9qersk7kYPzpjpMIxDl_Auur4BCubRfKmfA,29626
 xoscar/serialization/cuda.py,sha256=iFUEnN4SiquBIhyieyOrfw3TnKnW-tU_vYgqOxO_DrA,3758
 xoscar/serialization/exception.py,sha256=Jy8Lsk0z-VJyEUaWeuZIwkmxqaoB-nLKMa1D15Cl4js,1634
-xoscar/serialization/mlx.py,sha256=o7m6L5fqi7brIgNe0-qi8zhXtiJMzGgAJXl2Zvh1q10,2050
+xoscar/serialization/mlx.py,sha256=tRu_7o6RizdRhbr88EasHrZtShimAsLy3pIEO-by29o,2118
 xoscar/serialization/numpy.py,sha256=5Kem87CvpJmzUMp3QHk4WeHU30FoQWTJJP2SwIcaQG0,2919
 xoscar/serialization/pyfury.py,sha256=sifOnVMYoS82PzZEkzkfxesmMHei23k5UAUUKUyoOYQ,1163
 xoscar/serialization/scipy.py,sha256=yOEi0NB8cqQ6e2UnCZ1w006RsB7T725tIL-DM_hNcsU,2482
@@ -81,7 +81,7 @@ xoscar/virtualenv/__init__.py,sha256=65t9_X1DvbanNjFy366SiiWZrRTpa9SXWMXPmqayE-4
 xoscar/virtualenv/core.py,sha256=qHKqI6R92SN0_OWVAX1xy9wJiZr-r0VfLHTskZtZE4U,2851
 xoscar/virtualenv/utils.py,sha256=mL_uATHhj82xec0-0IZ6N8yI-laPAB4t8G3alPUGtPA,2439
 xoscar/virtualenv/uv.py,sha256=ef7InUP0VwPJe-ssZpWCMDN69lEp4aIRFtEgVff82eI,4884
-xoscar-0.7.10.dist-info/METADATA,sha256=g6djQpUz02xfI-DyU0u_Rejcxt8nnZVOXeTs2wKYCPA,9135
-xoscar-0.7.10.dist-info/WHEEL,sha256=Zqf17R98VQuI9JZEwsfDDfsqkexN9tbZpiCgdvPM64U,154
-xoscar-0.7.10.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
-xoscar-0.7.10.dist-info/RECORD,,
+xoscar-0.7.11.dist-info/METADATA,sha256=PDSZhxwM8g2RzmkuPLu6aynwy5WrFKV1jyv8XVkUY_4,9135
+xoscar-0.7.11.dist-info/WHEEL,sha256=Zqf17R98VQuI9JZEwsfDDfsqkexN9tbZpiCgdvPM64U,154
+xoscar-0.7.11.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
+xoscar-0.7.11.dist-info/RECORD,,