各位同学, 这边项目有个CUDA的加速代码, 是一个.cu文件, 包含CPU部分和GPU部分, 请问如果要将GPU部分接入taichi, 大概有什么方法?
感谢!
您好，非常高兴您能使用Taichi语言。
.cu 的cuda代码，CPU部分可以使用普通的python来实现，GPU部分可以使用Taichi的ti.kernel或者ti.func修饰的函数来实现。在kernel函数当中，Taichi会自动实现一些语句的并行化，具体可以参考Taichi的一些小例子。
各位同学, 这边项目有个CUDA的加速代码, 是一个.cu文件, 包含CPU部分和GPU部分, 请问如果要将GPU部分接入taichi, 大概有什么方法?
感谢!
您好，非常高兴您能使用Taichi语言。
.cu 的cuda代码，CPU部分可以使用普通的python来实现，GPU部分可以使用Taichi的ti.kernel或者ti.func修饰的函数来实现。在kernel函数当中，Taichi会自动实现一些语句的并行化，具体可以参考Taichi的一些小例子。