github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-07 12:01:39 +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

10
colossalai/tensor/param_op_hook.py
View File

@@ -1,9 +1,11 @@
import torch
from contextlib import contextmanager
from abc import ABC, abstractmethod
from typing import List, Tuple, Any
from contextlib import contextmanager
from typing import Any, List, Tuple
import torch
from colossalai.tensor.colo_tensor import ColoTensor
from colossalai.tensor import ColoTensorSpec
from colossalai.tensor.tensor_spec import ColoTensorSpec
class ParamOpHook(ABC):