@cuda.jit与@jit之间的差异(target=‘gpu’)

Question

关于如何使用Continuum加速包和numba包的Python库，我有一个问题。是在target = gpu中使用与@cuda.jit相同的装饰符@cuda.jit

Answer 1

不，它们是不一样的，虽然最终编译到PTX到汇编程序的路径是。@jit装饰器是通用的编译器路径，可以选择引导到CUDA设备上。@cuda.jit装饰器实际上是连续分析所发展的底层Python核心方言。因此，您可以获得对CUDA内置变量(如threadIdx )和内存空间说明符(如__shared__ in @cuda.jit )的支持。

如果您想用Python编写一个CUDA内核并编译并运行它，请使用@cuda.jit。否则，如果您想要加速现有的Python部分，请使用@jit和一个CUDA目标。