您好，请教一个关于代码中reduce相关的问题

[ss]

1. sum = warp_reduce_sum<NUM_WARPS>(sum);
2. if(warp==0) sum = warp_reduce_sum<NUM_WARPS>(sum);

0x03 warp/block reduce sum/max 、0x09 softmax, softmax + vec4
做final sum的时候，用的是第一种形式
0x04 block all reduce + vec4
而用的是第二种形式
我的理解是，最后final sum的时候是不是应该用第二种形式？最后都集中在第一个warp束中。
感谢！

[github-actions]

This issue is stale because it has been open for 30 days with no activity.

[github-actions]

This issue was closed because it has been inactive for 7 days since being marked as stale.