C headers 从 C 代码创建 cupy.array()

Question

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

Answer 1

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

Answer 2

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