o ۾i@s(ddlZGdddZGdddZdS)Ncs@eZdZd deddffdd ZdefddZd d d ZZS) CounterrstartreturnNcst||_dSN)super__init__counter)selfr __class__F/home/ubuntu/.local/lib/python3.10/site-packages/vllm/utils/counter.pyrs  zCounter.__init__cCs|j}|jd7_|S)Nr)r ir r r __next__ szCounter.__next__cCs d|_dS)Nrrr r r r resets z Counter.resetr)rN)__name__ __module__ __qualname__intrrr __classcell__r r r r rsrcsfeZdZdZddeddffdd Zedefdd Zdd edefd d Zdd edefddZ Z S) AtomicCounterzAn atomic, thread-safe counterrinitialrNcst||_t|_dS)z6Initialize a new atomic counter to given initial valueN)rr_value threadingLock_lock)r rr r r rs zAtomicCounter.__init__cCs|jSr)rrr r r valueszAtomicCounter.valuernumcCs>|j|j|7_|jWdS1swYdS)z@Atomically increment the counter by num and return the new valueNrrr r!r r r inc#$zAtomicCounter.inccCs>|j|j|8_|jWdS1swYdS)z@Atomically decrement the counter by num and return the new valueNr"r#r r r dec)r%zAtomicCounter.decr)r) rrr__doc__rrpropertyr r$r&rr r r r rsr)rrrr r r r s