search

mars-project / mars

Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.
https://mars-project.readthedocs.io
Apache License 2.0
2.68k stars 325 forks source link

[ENH] Enrich `nvutils.py` for GPU/CUDA initialization #3336

Closed dlee992 closed 1 year ago

dlee992 commented 1 year ago

CP one commit from xprobe-inc/mars, for enhancing GPU-related utils.

After merging this, I would open another PR, coming from xprobe-inc/xorbits, to fix a series of similar issues.

All credits in this PR belong to xprobe-inc.

qinxuye commented 1 year ago

Oh, we should do this, anyway, thanks.

qinxuye commented 1 year ago

https://github.com/xprobe-inc/xorbits/pull/242

This PR is helpful to address #3325 .

dlee992 commented 1 year ago

xprobe-inc/xorbits#242

This PR is helpful to address #3325 .

Thanks, xuye! I have added related code in #3337.