Pipeline Parallelism

若API“是否支持”“是”“限制与说明”为空,说明此API和原生API支持度保持一致。

API名称

是否支持

限制与说明

torch.distributed.pipeline.sync.Pipe

  

torch.distributed.pipeline.sync.Pipe.forward

  

torch.distributed.pipeline.sync.skip.skippable.skippable

  

torch.distributed.pipeline.sync.skip.skippable.stash

  

torch.distributed.pipeline.sync.skip.skippable.pop

  

torch.distributed.pipeline.sync.skip.skippable.verify_skippables