Closed dlee992 closed 1 year ago
If this is a Mars bug, any idea about how to fix this? I'm unfamiliar with Mars source code, but would like to give it a try.
I guess this PR https://github.com/mars-project/mars/pull/3330 can probably fix this issue. Since this PR changes the way of reading/writing buffer for CUDA, see https://github.com/mars-project/mars/pull/3330/files#diff-e288ecb73e105160acd53edb9a8c00be733cc5587fc1b322c65db73aa1ba8c1aR43-R57
Use Mars==
0.10.0
, and execute code below in Google Colab. PS, I use the Colab template from the answer of https://stackoverflow.com/questions/69407234/not-able-to-install-cudf-cupy-and-cuml-into-colab-with-rapids-ai-version-21-08, which can help me install and configure the RAPIDS runtime without any difficulty.Traceback:
As contrast, Dask can execute the same computation without any error.
outputs: