cupyx.jit.range#

cupyx.jit.range(*args, unroll=None) = <cupyx.jit function>#

支持循环展开的范围。

参数:

start (int) – 与内置 range 相同。
stop (int) – 与内置 range 相同。
step (int) – 与内置 range 相同。
unroll (int 或 bool 或 None) –
- 如果为 True，在循环前添加 #pragma unroll 指令。
- 如果为 False，在循环前添加 #pragma unroll(1) 指令以禁用展开。
- 如果为 int，在循环前添加 #pragma unroll(n) 指令，其中整数 n 表示要展开的迭代次数。
- 如果为 None（默认），将循环展开的控制权留给编译器（无 #pragma）。

另请参阅