Function tvm::tirx::gpu_warp_shuffle_down_op

Function tvm::tirx::gpu_warp_shuffle_down_op#

Function Documentation#

const Op &tvm::tirx::gpu_warp_shuffle_down_op()#

Read the value from a higher lane in the warp.

Arguments, in order:

  • args[0]: mask, The mask.

  • args[1]: value, The value to use.

  • args[2]: offset, The offset.

  • args[3]: width, The width.

  • args[4]: warp_size, The number of threads per warp.