cupyx.jit.range

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

支持循环展开的范围。

参数:

• start (int) – Same as that of built-in range.

• stop (int) – Same as that of built-in range.

• step (int) – Same as that of built-in range.

• unroll (int or bool or None) –

  • 如果为 True，则在循环前添加 #pragma unroll 指令。

  • 如果为 False，在循环前添加 #pragma unroll(1) 指令以禁用展开。

  • 如果是一个 int，则在循环前添加 #pragma unroll(n) 指令，其中整数 n 表示要展开的迭代次数。

  • 如果为 None`（默认），则将循环展开的控制权交给编译器（不使用 ``#pragma`）。

参见

#pragma unroll