iddlmZddlZeddZeddZeddZedd Zedd Zedd Zedd Z edd Z y))jitNT)devicecJtjj|d|dS)z If for all threads in the masked warp the predicate is true, then a non-zero value is returned, otherwise 0 is returned. rrnumbacudavote_sync_intrinsicmask predicates q/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/cuda/intrinsic_wrapper.pyall_syncr" :: ) )$9 =a @@cJtjj|d|dS)z If for any thread in the masked warp the predicate is true, then a non-zero value is returned, otherwise 0 is returned. rrr s rany_syncrrrcJtjj|d|dS)z If for all threads in the masked warp the boolean predicate is the same, then a non-zero value is returned, otherwise 0 is returned. rrr s req_syncrrrcJtjj|d|dS)zo Returns a mask of all threads in the warp whose predicate is true, and are within the given mask. rrr s r ballot_syncr rrcNtjj|d||ddS)z Shuffles value across the masked warp and returns the value from src_lane. If this is outside the warp, then the given value is returned. rrr shfl_sync_intrinsic)r valuesrc_lanes r shfl_syncr )s& :: ) )$5(D I! LLrcNtjj|d||ddS)z Shuffles value across the masked warp and returns the value from (laneid - delta). If this is outside the warp, then the given value is returned. rrrr rdeltas r shfl_up_syncr$3s& :: ) )$5% CA FFrcNtjj|d||ddS)z Shuffles value across the masked warp and returns the value from (laneid + delta). If this is outside the warp, then the given value is returned. rrrrr"s rshfl_down_syncr&=s& :: ) )$5% Fq IIrcNtjj|d||ddS)zd Shuffles value across the masked warp and returns the value from (laneid ^ lane_mask). rrrr)r r lane_masks r shfl_xor_syncr)Gs& :: ) )$5)T J1 MMr) decoratorsrrrrrrr r$r&r)rrr,s DAADAADAADAADMMDGGDJJDNNr