o DÉli¿ã@s¨ddlmZddlZedddd„ƒZedddd „ƒZeddd d „ƒZeddd d „ƒZedddd„ƒZedddd„ƒZedddd„ƒZ edddd„ƒZ dS)é)ÚjitéNT)ÚdevicecCótj |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. rr©ÚnumbaÚcudaÚvote_sync_intrinsic©ÚmaskÚ predicate©r úl/var/www/html/tatsat2dev/dashboard-backend/venv/lib/python3.10/site-packages/numba/cuda/intrinsic_wrapper.pyÚall_syncórcCstj |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 r 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. érrr r r rÚeq_syncrrcCr)zo Returns a mask of all threads in the warp whose predicate is true, and are within the given mask. érrr r r rÚ ballot_sync rrcCstj |d||d¡dS)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. ré©rrÚshfl_sync_intrinsic)r ÚvalueÚsrc_laner r rÚ shfl_sync)órcCstj |d||d¡dS)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. rrr©r rÚdeltar r rÚ shfl_up_sync3rrcCótj |d||d¡dS)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=rr!cCr )zd Shuffles value across the masked warp and returns the value from (laneid ^ lane_mask). rrrr)r rÚ lane_maskr r rÚ shfl_xor_syncGsr#) Ú decoratorsrrrrrrrrr!r#r r r rÚs$