从 C 代码创建 cupy.array() 的 C 头文件

Posted 2023-02-16

【中文标题】从 C 代码创建 cupy.array() 的 C 头文件

【英文标题】：C headers to create cupy.array() from C code

【发布时间】：2019-09-27 00:12:54

【问题描述】：

我想在 C 代码中创建一个 cupy.array()，类似于我通过 PyArray_SimpleNew() 使用 numpy 的方式。到目前为止，我无法找到我应该用来执行此操作的 C 标头。这目前可能吗？ 谢谢

【参考方案1】：

您可以使用 Python C API 从 C 构造任何 Python 对象。https://docs.python.org/3/c-api/index.html

【参考方案2】：

据我所知，没有 C 标头可以创建 cupy 对象。我最终使用的解决方案是创建自己的 C 定义对象，该对象包含 CUDA 指针，并使用 cupy 的 memory.UnownedMemory 特性将其包装到 Python 代码中的 cupy 对象中。