github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[fx] fix the discretize bug (#1506) 

* [fx] fix wrong variable name in solver rotor

* [fx] fix wrong variable name in solver rotor

* code modification

* [fx] fix the discretize bug
pull/1507/head^2
Boyuan Yao 2022-08-26 17:15:52 +08:00 committed by GitHub
parent 31fffd3fc5
commit ac3a453a50
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/fx/passes/algorithms/ckpt_solver_rotor.py
View File

@ -142,8 +142,6 @@ def _construct_chain(node_dict: Dict[int, Node], data: torch.Tensor, mem_unit: i
bwd_time.append(0)
fwd_time = _discretize(mem_unit, fwd_time)
bwd_time = _discretize(mem_unit, bwd_time)
xbar_sizes = _discretize(mem_unit, xbar_sizes)
x_sizes = _discretize(mem_unit, x_sizes)
tmp_fwd = _discretize(mem_unit, tmp_fwd)