若API“是否支持”为“是”,“限制与说明”为空,说明此API和原生API支持度保持一致。
API名称 |
是否支持 |
限制与说明 |
---|---|---|
torch.distributed.tensor.parallel.parallelize_module |
是 |
|
torch.distributed.tensor.parallel.style.RowwiseParallel |
是 |
|
torch.distributed.tensor.parallel.style.ColwiseParallel |
是 |
|
torch.distributed.tensor.parallel.fsdp.enable_2d_with_fsdp |
否 |