我有两个问题想请教一下:
1、 ti_import_vulkan_memory 读取到的VKbuffer是存储在CPU端还是存储在GPU端的?
2、除了ti_import_vulkan_memory这个函数之外,是否还有别的接口函数可以将ndarray转化为在GPU端其它图形API(比如cuda)可以操作的内存?
您好,还有个问题想请教一下, ti_map_memory这个函数是将TiMemory map回cpu端吗?如果是的话那么我想在GPU端得到TiMemory的内存指针有什么办法呢?
import_vulkan_memory并不会读取到VkBuffer,import是将你创建的VkBuffer导入到Taichi变成一个TiMemory;其次VkBuffer不是一个指针而是一个句柄(handle);如果你是想把Taichi的vulkan内存内容给cuda使用,请参见 taichi-aot-demo/cross_device_copy.cpp at master · taichi-dev/taichi-aot-demo · GitHub