Skip to content

GitLab

Unverified Commit ac3a453a authored by Boyuan Yao's avatar Boyuan Yao Committed by GitHub
Browse files

[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
parent 31fffd3f
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
colossalai/fx/passes/algorithms/ckpt_solver_rotor.py
View file @ ac3a453a
...@@ -142,8 +142,6 @@ def _construct_chain(node_dict: Dict[int, Node], data: torch.Tensor, mem_unit: i ...@@ -142,8 +142,6 @@ def _construct_chain(node_dict: Dict[int, Node], data: torch.Tensor, mem_unit: i
bwd_time.append(0) 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) xbar_sizes = _discretize(mem_unit, xbar_sizes)
x_sizes = _discretize(mem_unit, x_sizes) x_sizes = _discretize(mem_unit, x_sizes)
tmp_fwd = _discretize(mem_unit, tmp_fwd) tmp_fwd = _discretize(mem_unit, tmp_fwd)
......
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