Skip to content

GitLab

Commit dd14b36d authored by Chenggang Zhao's avatar Chenggang Zhao
Browse files

Fix compilation

parent c5facf5c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
csrc/kernels/internode_ll.cu
View file @ dd14b36d
...@@ -501,7 +501,7 @@ __forceinline__ __device__ void logfmt_check_amaxmin(uint8_t* meta_buffer, float ...@@ -501,7 +501,7 @@ __forceinline__ __device__ void logfmt_check_amaxmin(uint8_t* meta_buffer, float
} }
const auto& casted = warp_reduce_and<kNumSendUnrolls>(enable_cast) ? 1u << (lane_id / kNumRecvUnrolls): 0u; const auto& casted = warp_reduce_and<kNumSendUnrolls>(enable_cast) ? 1u << (lane_id / kNumRecvUnrolls): 0u;
const auto& num_casted_prefix = std::__popcount(warp_reduce_or<kNumRecvUnrolls, true>(casted) & ((1u << (lane_id / kNumRecvUnrolls)) - 1)); const auto& num_casted_prefix = __popc(warp_reduce_or<kNumRecvUnrolls, true>(casted) & ((1u << (lane_id / kNumRecvUnrolls)) - 1));
if (lane_id < kNumLanes and lane_id % kNumRecvUnrolls == 0) if (lane_id < kNumLanes and lane_id % kNumRecvUnrolls == 0)
shared_cast_info[lane_id / kNumRecvUnrolls] = (num_casted_prefix << 1) | (casted ? 1u : 0u); shared_cast_info[lane_id / kNumRecvUnrolls] = (num_casted_prefix << 1) | (casted ? 1u : 0u);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment