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

2024-03-28 19:26:40 发布

您现在位置:Python中文网/ 问答频道 /正文

如果我有一个简单的三维采样点的形式

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

在cupy中实现的体素函数中,我可以从c++代码调用该函数吗?我们有c++代码调用tensorflow graph并在车上执行。我能把我的cupy代码编译成c++二进制代码并从c++代码调用它吗?在


Tags: 函数代码形式x1x2cupyz3y1
1条回答
网友
1楼 · 发布于 2024-03-28 19:26:40

我对CuPy不熟悉,但是看一下pythoncapi(https://docs.python.org/3/c-api/)。 这允许您在C和C++中运行Python函数。在

下面是一个示例,在中运行一个名为functionname的函数Python.pythonfile.py但这不应该被检测过。在

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)。在

相关问题 更多 >