C头文件，用于从C代码创建cupy.array（）

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对象中。