python - 如何将 Numba 中另一个库中的函数用于 Cuda GPU？

Question

我是 numba 的新手，我想在 CUDA GPU 上的 python 中使用 hashlib 库中的 md5 函数。

我已经搜索了如何在我的函数中使用外部函数@cuda.jit，但我没有找到任何有用的东西或任何教程。我在https://numba.pydata.org/numba-doc/dev/cuda-reference/host.html中发现，我可以在上下文中将我的 python 函数包装到一个线程中，但我真的不明白如何去做吧。

我想在result = hashlib.md5(str.encode()) 将在 GPU 上运行的代码中使用此函数。

标签： pythonnumba