python - 我可以从c ++调用用cupy代码编写的cuda代码吗？

Question

如果我有一个简单的采样 3D 点的形式

s = [[x1, y1, z1], [x2, y2, z2],[x3, y3, z3],[x4, y4, z4],.....]

在cupy中实现的体素函数中，我可以从c ++代码中调用该函数吗？我们有 c++ 代码调用 tensorflow 图并在汽车中执行。我可以将我的 Cupy 代码编译成 C++ 二进制文件并从 C++ 代码中调用吗？

Answer 1

我不熟悉 CuPy，但可以看看 Python C API ( https://docs.python.org/3/c-api/ )。这允许您在 C 和 C++ 中运行 Python 函数。

这是一个示例，在pythonfile.py中运行一个名为functionname的函数（尚未对此进行测试，但应该没问题）。

PyObject* pyModuleStr = PyString_FromString((char*)"pythonfile");
PyObject* pyModule = PyImport_Import(pyModuleStr);
PyObject* pyFunction = PyObject_GetAttrString(pyModule,(char*)"functionname");
PyObject_CallObject(pyFunction);

也可以提取函数的返回值，并添加参数。

希望这可以帮助！

8)

编辑：这是一个可能有帮助的 StackOverflow 帖子（从 C/C++ 调用 python 方法，并提取其返回值）。