xoscar 0.7.10__cp310-cp310-win_amd64.whl → 0.7.11__cp310-cp310-win_amd64.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

@@ -1,5 +1,5 @@
 xoscar/__init__.py,sha256=lzCXUmkuIjcjkiNQFekysdJR_ZhlbjcfR5ka1bZZNQg,1683
-xoscar/_utils.cp310-win_amd64.pyd,sha256=chmDOXoAJrQihkdZAFU5RPDfyRyIYE76hMK63hM5zgE,110080
+xoscar/_utils.cp310-win_amd64.pyd,sha256=tTTKPlYCyj1NtIU5Op6HZfTfMyP6A5oNNWvfQ-BOTpY,110080
 xoscar/_utils.pxd,sha256=rlNbTg5lhXA-jCOLksqF4jhUlNn0xw2jx1HxdLa34pc,1193
 xoscar/_utils.pyx,sha256=TWScgqmJGYzjbWBOShBLkq07ldfYEQ5fw6V4OytX_IA,7626
 xoscar/_version.py,sha256=bsfCVAo_o9LkiP3AjPsP4SRRqhjuS0t4D1WGJPzbdls,24412
@@ -7,10 +7,10 @@ xoscar/api.py,sha256=QQDHn-_FiDExmOxEk8BUnq4Qrx13VX3shSlEEqPQJo0,15175
 xoscar/backend.py,sha256=8G5JwjoOT6Q2slb11eXNApxgcmvNQUCdQzkoIMDwLcQ,1917
 xoscar/batch.py,sha256=Jk5BSpvMFAV9DrRy0a9tgPvIo_dt8cbJReZBL0cnOPc,8128
 xoscar/constants.py,sha256=GJ1KEOxwnqksc9K_GH42TSWpQECeC6ti3KJmE3PUcTw,810
-xoscar/context.cp310-win_amd64.pyd,sha256=i4N6DdBMW85RoJdRA-1X2B0q4bAviDtXlhBbovem53Q,154112
+xoscar/context.cp310-win_amd64.pyd,sha256=lGx9Q0xShrJIPIEHZ5o9zLnO97teAIB2uGGvkZvwVj8,154112
 xoscar/context.pxd,sha256=6n6IAbmArSRq8EjcsbS6npW8xP1jI0qOoS1fF0oyj-o,746
 xoscar/context.pyx,sha256=FOJVerGOvxe2USryXEQA0rpaFX_ScxISH6QWKUcahY8,11310
-xoscar/core.cp310-win_amd64.pyd,sha256=AQVAD4bvzNXyp5THcJ48o3hynSeAMQqNc_dZuUpfKMQ,304640
+xoscar/core.cp310-win_amd64.pyd,sha256=tcmhwlpOyvSF9EBp0bZGhl8vIdUgbGc0HtUTGuQMLyo,304640
 xoscar/core.pxd,sha256=9IZP7dYGfnF1I-obIls8R8b6forxDOPbiM3v5nVslLk,1368
 xoscar/core.pyx,sha256=U6jCZN74MQHi7HkQRaVGm_w5q-FMsw0nnE3aU6533_Q,22756
 xoscar/debug.py,sha256=hrmxIH6zvTKasQo6PUUgXu5mgEsR0g87Fvpw7CoHipg,5257
@@ -30,7 +30,7 @@ xoscar/backends/allocate_strategy.py,sha256=DzvTlixwzTANURI2mDLHm3vcaugSPDxU6UQZ
 xoscar/backends/config.py,sha256=86j0g_Xrl8ENPzBWi296yWg9QEcljvdKK-yJbfYTvQ0,5532
 xoscar/backends/context.py,sha256=qWwksx8JxYcKR-LQA3PvXh4ReuuTTEyDaLbjpxGXcTA,16766
 xoscar/backends/core.py,sha256=fbekAxys_t1Dv7if-1R6uVvC_yAxNkXLeQ1V1ZSAfC0,11161
-xoscar/backends/message.cp310-win_amd64.pyd,sha256=Anfg42EvIZ7R3_qOVkGEaIKZwzZuh7xXdAoLVRpZmgI,248320
+xoscar/backends/message.cp310-win_amd64.pyd,sha256=BIaZgosVipZEZQAGsgfcTSvkkKzOwp_QKtR4u5laJSY,248320
 xoscar/backends/message.pyx,sha256=lBEjMJv4VyxmeO9lHXJJazOajbFnTLak4PSBcLoPZvU,20331
 xoscar/backends/pool.py,sha256=bS_m8XBkfQQsCOaLEzM6HkV5e78dPPp1bCH6yjvPEss,62153
 xoscar/backends/router.py,sha256=EjfNpQUrhFU15eYe1kRPueziHgI2gfDViUzm7ruvXDE,10817
@@ -57,8 +57,8 @@ xoscar/collective/common.py,sha256=9c7xq3IOUvfA0I9GnpalUqXZOzmF6IEILv4zL64BYVE,3
 xoscar/collective/core.py,sha256=191aPxbUgWpjzrqyozndImDAQhZFmqoQdBkHFLDfXN0,24239
 xoscar/collective/process_group.py,sha256=kTPbrLMJSGhqbiWvTIiz-X3W0rZWd_CFn_zUIlXbOlM,23286
 xoscar/collective/utils.py,sha256=p3WEVtXvnVhkuO5mRgQBhBRFr1dKHcDKMjrbMyuiyfg,1219
-xoscar/collective/uv.dll,sha256=ur4aO_Tc0snpYesZ62C2b74obez48GIoM9SFDqn0NJU,620544
-xoscar/collective/xoscar_pygloo.cp310-win_amd64.pyd,sha256=D89ml1Na0yx5cTa_jZqSWpUxRe5QWcwbLTeJVYXRr2E,829952
+xoscar/collective/uv.dll,sha256=tK0iWk3p16AdNy5fs83yyG1EN4vUkf6vIXd13UM_p3w,620544
+xoscar/collective/xoscar_pygloo.cp310-win_amd64.pyd,sha256=jCX0rMY-QP-zequ-d6jREbUfTiJwXFpRDi5V8UAKrjk,829952
 xoscar/metrics/__init__.py,sha256=RjXuuYw4I2YYgD8UY2Z5yCZk0Z56xMJ1n40O80Dtxf8,726
 xoscar/metrics/api.py,sha256=dtJ4QrIqQNXhJedeqOPs4TXKgrRGZFFN50xAd9SCfec,9144
 xoscar/metrics/backends/__init__.py,sha256=ZHepfhCDRuK9yz4pAM7bjpWDvS3Ijp1YgyynoUFLeuU,594
@@ -69,12 +69,12 @@ xoscar/metrics/backends/prometheus/__init__.py,sha256=ZHepfhCDRuK9yz4pAM7bjpWDvS
 xoscar/metrics/backends/prometheus/prometheus_metric.py,sha256=65hb8O3tmsEJ7jgOrIwl_suj9SE5Tmqcfjuk0urkLvE,2120
 xoscar/serialization/__init__.py,sha256=tS8C49yrW_geWNEsbgW3phK1q4YN1ojI6CN-vroIFYM,876
 xoscar/serialization/aio.py,sha256=7YLXgkWpQ3ANy-TZ1qO8Mt4_J3cZFhFh2FEgUgxMT60,4873
-xoscar/serialization/core.cp310-win_amd64.pyd,sha256=wLCqZpSYPl021j8kDFUtfYCBbFiyFM_4U8oqiQN0HAg,268800
+xoscar/serialization/core.cp310-win_amd64.pyd,sha256=D6PGQl-vcXwDvfotN4aC-ew5kiwsUglfB9HdjPLd2PE,268800
 xoscar/serialization/core.pxd,sha256=X-47bqBM2Kzw5SkLqICdKD0gU6CpmLsBxC3kfW--wVk,1013
 xoscar/serialization/core.pyx,sha256=ZKexLRnRwZXXn2045kR7xfM_szcoPNrDuouQCWtpFp8,30570
 xoscar/serialization/cuda.py,sha256=Fj4Cpr_YmkGceUCo0mQn8fRvmHP_5WcLdRx6epZ3RC0,3869
 xoscar/serialization/exception.py,sha256=t6yZn_Ate04UE1RbabNh7mu739sdtwarjuPXWhASx7c,1682
-xoscar/serialization/mlx.py,sha256=wQ1Nv524VNiGRca-bdg0QGBqD3jdUpSBhNuwdZ5cnDk,2115
+xoscar/serialization/mlx.py,sha256=8QOWlYmS0eFZ1bYweDQzUkSRw5MI0dt118wOGhyGk1s,2185
 xoscar/serialization/numpy.py,sha256=C6WVx-Sdl2OHBAvVY34DFjAKXlekMbpc2ni6bR8wxYo,3001
 xoscar/serialization/pyfury.py,sha256=3ucal29Hr7PX9_1SfB2x43FE2xw_C0rLkVv3foL7qwM,1200
 xoscar/serialization/scipy.py,sha256=9ph-yoRoNiwUZTwQrn35U60VPirWlncXNAg6EXvqMR4,2554
@@ -82,7 +82,7 @@ xoscar/virtualenv/__init__.py,sha256=rhJ7I6x7aXjKOCzSqsKLwqFJMh4YC2sqchEIJNEfI58
 xoscar/virtualenv/core.py,sha256=6u7zcMiAZVOKSBUXBlJOC5pvwbmycn6svJApYMSJDIY,2943
 xoscar/virtualenv/utils.py,sha256=tZYB1kAEym5e8WK9SxhW7VgZzXk6fJMm6MoBA7pY4gM,2523
 xoscar/virtualenv/uv.py,sha256=CbOr9556htAvN3Jy-nO2MNIVs53tcEM6ZuYFFyFFbVw,5025
-xoscar-0.7.10.dist-info/METADATA,sha256=nywxpoQ_jE9MtilnDnylJ3tKP_aNt0idYXfPy7Glap0,9361
-xoscar-0.7.10.dist-info/WHEEL,sha256=NVXpD7b4Gxps0cd2ds5rr5TG8W4ApEwx_i5J99qMZ5E,102
-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=iUy9nGB1H_iSHbwMFbzaWqcyBbHZhRpLCcAyQFLUv54,9361
+xoscar-0.7.11.dist-info/WHEEL,sha256=NVXpD7b4Gxps0cd2ds5rr5TG8W4ApEwx_i5J99qMZ5E,102
+xoscar-0.7.11.dist-info/top_level.txt,sha256=vYlqqY4Nys8Thm1hePIuUv8eQePdULVWMmt7lXtX_ZA,21
+xoscar-0.7.11.dist-info/RECORD,,