Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "6b1a39b17bdbb4dd9d1e15bf96672895f305818f"
Unverified Commit e81caeb4 authored by Xue Fuzhao's avatar Xue Fuzhao Committed by GitHub
Browse files

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


Co-authored-by: default avatarFuzhao Xue <fuzhao@login2.ls6.tacc.utexas.edu>
parent 51c45c24
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 6 deletions
colossalai/auto_parallel/tensor_shard/deprecated/cost_graph.py
View file @ e81caeb4
from typing import List
import math import math
from typing import List
from torch.fx.node import Node from torch.fx.node import Node
from .constants import INFINITY_COST from .constants import INFINITY_COST
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment