github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-13 05:35:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NFC] polish colossalai/auto_parallel/tensor_shard/deprecated/cost_graph.py code style (#2720)

Browse Source 
Co-authored-by: Fuzhao Xue <fuzhao@login2.ls6.tacc.utexas.edu>
This commit is contained in:
Xue Fuzhao 2023-02-15 16:12:45 +08:00 committed by GitHub
parent 51c45c2460
commit e81caeb4bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
colossalai/auto_parallel/tensor_shard/deprecated/cost_graph.py
View File

@ -1,6 +1,8 @@
from typing import List
import math
from typing import List
from torch.fx.node import Node
from .constants import INFINITY_COST