0
代码
代码
Issues
Pull Requests
讨论
Wiki
分析
0
ascend-robotascend-robot[feature]分布式支持context parallel
8720af51创建于 12 天前历史提交
import torch_npu.distributed.tensor._attention
import torch_npu.distributed.tensor._dtensor_patch  # patch before register strategy
import torch_npu.distributed.tensor._math_ops
import torch_npu.distributed.tensor._matrix_ops
import torch_npu.distributed.tensor._moe_ops
import torch_npu.distributed.tensor._pointwise_ops
import torch_npu.distributed.tensor._sharded_tensor_patch
import torch_npu.distributed.tensor._view_ops
import torch_npu.distributed.tensor.experimental