o i@sddlmZddlZeddddZedddd Zeddd d Zeddd d ZeddddZeddddZeddddZ eddddZ dS))jitNT)ZdevicecCtj|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. rrnumbacudaZvote_sync_intrinsicmask predicater k/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/numba/cuda/intrinsic_wrapper.pyall_syncr cCstj|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. rrrr r r any_syncrrcCr)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. rrrr r r eq_syncrrcCr)zo Returns a mask of all threads in the warp whose predicate is true, and are within the given mask. rrrr r r ballot_sync rrcCstj|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. rrrZshfl_sync_intrinsic)r valueZsrc_laner r r shfl_sync)rcCstj|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 rdeltar r r shfl_up_sync3rrcCtj|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. rrrrrr r r shfl_down_sync=rrcCr)zd Shuffles value across the masked warp and returns the value from (laneid ^ lane_mask). rrrr)r rZ lane_maskr r r shfl_xor_syncGsr) Z decoratorsrrr rrrrrrrr r r r s$