关于如何使用Continuum加速包和numba包的Python库,我有一个问题。是在target = gpu
中使用与@cuda.jit
相同的装饰符@cuda.jit
发布于 2016-03-09 21:01:43
不,它们是不一样的,虽然最终编译到PTX到汇编程序的路径是。@jit
装饰器是通用的编译器路径,可以选择引导到CUDA设备上。@cuda.jit
装饰器实际上是连续分析所发展的底层Python核心方言。因此,您可以获得对CUDA内置变量(如threadIdx
)和内存空间说明符(如__shared__
in @cuda.jit
)的支持。
如果您想用Python编写一个CUDA内核并编译并运行它,请使用@cuda.jit
。否则,如果您想要加速现有的Python部分,请使用@jit
和一个CUDA目标。
https://stackoverflow.com/questions/35890045
复制相似问题