o }oi@s.ddlmZddlZdeedefddZdS))OptionalNrankreturncCs@t}|dur dSt|tr||kStd|dt|d)z Check if the current MPI rank matches the specified rank. Args: rank (Optional[int]): The rank to check against. Returns: bool: True if the current rank matches the specified rank or if rank is None. NTzInvalid rank argument z of type .) tensorrt_llmmpi_rank isinstanceint ValueErrortype)r current_rankr M/home/ubuntu/.local/lib/python3.10/site-packages/nemo/export/trt_llm/utils.pyis_ranks   r)typingrrr boolrr r r rs