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

Question

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

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

在 cupy 中实现的体素函数，我可以从 C++ 代码调用该函数吗？我们有调用 tensorflow graph 并在车上执行的 C++ 代码。我可以将我的 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 帖子 ( Calling a python method from C/C++, and extracting its return value )。