tvm_tirx.tirx.op.gpu_warp_shuffle_xor

tvm_tirx.tirx.op.gpu_warp_shuffle_xor(mask, value, lane_mask, width, warp_size, *, ty=None, loc=UNKNOWN_LOC)

Copy value from a lane with index computed by src_lane_idx ^ lane_mask.

Parameters:

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

• value (Expr) – The value to exchange.

• lane_mask (Expr) – The mask to compute source lane index:

• 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.