torch.distributed.tensor.parallel
若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.distributed.tensor.parallel.parallelize_module |
否 |
- |
torch.distributed.tensor.parallel.ColwiseParallel |
否 |
- |
torch.distributed.tensor.parallel.RowwiseParallel |
否 |
- |
torch.distributed.tensor.parallel.PrepareModuleInput |
否 |
- |
torch.distributed.tensor.parallel.PrepareModuleOutput |
是 |
- |
父主题： PyTorch原生API支持度