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

Question

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

Answer 1

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

Answer 2

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