Function tvm::tirx::gpu_warp_shuffle_op

Function tvm::tirx::gpu_warp_shuffle_op#

Function Documentation#

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

Read the value from a selected lane in the warp.

Arguments, in order:

  • args[0]: mask, The mask.

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

  • args[2]: warp_id, The warp identifier.

  • args[3]: width, The width.

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