Searched defs:CommBlock (Results 1 – 1 of 1) sorted by relevance
74 class CommBlock: class168 def always_true(comm_block: CommBlock) -> bool:187 last_comm_block: CommBlock,245 last_comm_block: CommBlock,305 fused_comm_block: CommBlock,306 orig_comm_blocks: List[CommBlock],391 comm_blocks: List[CommBlock],437 graph: fx.Graph, comm_blocks: List[CommBlock], bucket_size_mb: int482 def ddp_reducer_filter(block: CommBlock) -> bool: