github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-06 11:32:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Tensor] remove ParallelAction, use ComputeSpec instread (#1166)

Browse Source
This commit is contained in:
Jiarui Fang
2022-06-23 17:34:59 +08:00
committed by GitHub
parent 177c374401
commit f4ef224358
20 changed files with 87 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
colossalai/tensor/colo_parameter.py
View File

@@ -1,10 +1,12 @@
import torch
from typing import Optional
from copy import copy
from colossalai.tensor.colo_tensor import ColoTensor
from colossalai.tensor.const import TensorType
import torch
from colossalai.tensor import TensorSpec, distspec
from copy import copy
from colossalai.tensor.param_op_hook import ParamOpHookManager
from typing import Optional
def filter_args(func, *args):