tvm_tirx.tirx.script.gpu_warp_shuffle

tvm_tirx.tirx.script.gpu_warp_shuffle#

tvm_tirx.tirx.script.gpu_warp_shuffle(mask, value, warp_id, width, warp_size, *, ty=None, loc=UNKNOWN_LOC)#

Exchange value between threads inside a warp.

Parameters:
  • mask (Expr) – The warp mask indicates active threads inside warp.

  • value (Expr) – The value to exchange.

  • warp_id (Expr) – The source lane index to fetch value.

  • width (Expr) – The width of sub-sections to perform warp shuffle.

  • warp_size (Expr) – The warp size.

  • loc (Location)

Returns:

call (Expr) – The call expression.