github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-22 09:59:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

reorgnize colotensor directory (#1062)

Browse Source 
* reorgnize colotensor directory

* polish code
This commit is contained in:
Jiarui Fang
2022-06-03 18:04:22 +08:00
committed by GitHub
parent 3d10be33bd
commit a00644079e
25 changed files with 130 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
colossalai/tensor/utils.py
View File

@@ -1,10 +1,8 @@
import torch
from colossalai.tensor.colo_tensor import ColoTensor
from typing import Iterator, Tuple, Union
import torch.nn as nn
from colossalai.tensor import ColoTensor
from colossalai.tensor.colo_tensor import ColoTensor
# The function is credited to PyTorch Team