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

-