+ KV-jk^RIHtHt^RIt^RIHt^RIHtH t RRlt RRlt R R lt R R lt RR RlltRRltRRltRRltR#))TupleListN)NDArray) LLMatcher LLExecutorc^V^8dQhR\R\R\\\3,/# batch_size vocab_sizereturn)intr)formats"a/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/llguidance/numpy.py __annotate__rs)11#1315c?1c$W^,^ ,3#)r r s&&rget_bitmask_shapers bR/ 00rcfV^8dQhR\R\R\\P,/#r )rrnpint32)rs"rrr s,s'*/6rxx/@rcb\P!\W4R\PR7#)dtype)rfullrrrs&&rallocate_token_bitmaskr! s% 77*1 hh rcV^8dQhR\\P,R\\P,RR/#r logitsmaskr Nrrfloat32r)rs"rrrs5%%wrzz/B%-4RXX->%CG%rcV\P!V^ ^R7p\P!\P!^ \PR7VP ^,4pW#, ^,pVRRVP ^,13,p\P )W^8H&R#) axisrNNNN)rrepeattilearangershapeinf)r$r% mask_expanded bit_indices bit_maskss&& r"apply_token_bitmask_inplace_kernelr5swIIdBQ/M''"))Bbhh7AGK-2I!-fll1o--.I ffWF>rcV^8dQhR\\P,R\\P,RR/#r#r&)rs"rrrs555 (;5&-bhh&75<@5rcVP^8Xd\P!V^R7pVP^8Xd\P!V^R7pVP\P8XgQR4hVP^8XgQR4hVP wr#VP wrEW$8XgQR4h^ V,pW68d&\P )VRVR13&VRRV13,p\W4R#)rr*Mask must be int32zLogits must be 2DzBatch size mismatchr,N)ndimr expand_dimsrrr0r1r5)r$r%batchvocabm_batchm_vocabcutoffs&& rapply_token_bitmask_inplacer@s {{aQ/ yyA~~~d+ :: !7#77 ! ;;! 000 < " "=$== " ""188==!&&1::2EFrcV^8dQhR\R\\\\3,,R\ \ P,RR/#r executormatchersrCr Nrrrrrrrr)rs"rrr9sH V V* V*.uY^/D*E V)0): V?C Vrc:VP\P8XgQR4hVP^8XgQR4hVPwr4VP R,'gQR4hVP WPPV^,V4R#zs Compute the token mask directly into the specified array. For each matcher, provide the index of the target mask. r8rFrGrHN) rrrr9r0rIrJrKrLrSrTrCr;r<s&&& rfill_next_token_bitmask_parrY9s~ ==BHH $:&:: $ <<1 /// ==LE == ( (C*CC ( $$X~~/B/BEAIuUrc V^8dQhR\R\\\\\\,3,,R\ \ P,RR/#rRrU)rs"rrrGsO g gJ g*.uYT#Y5N/O*P g)0): g?C grc:VP\P8XgQR4hVP^8XgQR4hVPwr4VP R,'gQR4hVP WPPV^,V4R#rW) rrrr9r0rI(unsafe_compute_mask_ptr_with_draft_tokenrKrLrXs&&& r-fill_next_token_bitmask_par_with_draft_tokensr]Gs ==BHH $:&:: $ <<1 /// ==LE == ( (C*CC ( 55h@S@SUZ]^U^`efrcV^8dQhR\R\\\\3,,R\\ ,/#)r rSrTr )rrrrrbool)rs"rrrUs500 0 $U9c>%: ;0@DT 0rc$VPV4#)a Consume a single token for each matcher in parallel. Args: executor: The LLExecutor to use for parallel execution. matchers: List of tuples containing (LLMatcher, token_id). Returns: List[bool]: Success/failure for each matcher (in order). Note: Matchers that fail (return False) are left in an error state, consistent with the behavior of consume_token on a single matcher. )consume_token_par)rSrTs&&rraraUs  % %h //r)r)typingrrnumpyr numpy.typingr_librrrr!r5r@rPrYr]rarrrrfs< '1%5$G V g0r