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

[autoparallel] fix bugs caused by negative dim key (#1808)

Browse Source 
* [autoparallel] fix bugs caused by negative dim key

* fix import error

* fix matmul test issue

* fix unit test issue
This commit is contained in:
YuliangLiu0306
2022-11-08 17:03:50 +08:00
committed by GitHub
parent 4268ae017b
commit 49216d7ab1
12 changed files with 108 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
colossalai/tensor/colo_parameter.py
View File

@@ -1,11 +1,11 @@
import torch
from typing import Optional
import torch
from colossalai.tensor.colo_tensor import ColoTensor
from colossalai.tensor.const import TensorType
from colossalai.tensor import ColoTensorSpec
from colossalai.tensor.param_op_hook import ParamOpHookManager
from colossalai.tensor.tensor_spec import ColoTensorSpec
def filter_args(func, *args):