Commit feced2f1 authored by 王敏's avatar 王敏
Browse files

[BUGFIX]修复cplb调度处理的笔误

parent 35c61e91
...@@ -1528,7 +1528,7 @@ class GPUModelRunner( ...@@ -1528,7 +1528,7 @@ class GPUModelRunner(
local_scatter_indexes_tensor = None local_scatter_indexes_tensor = None
gather_indexes_tensor = None gather_indexes_tensor = None
if self.enable_lightly_cp: if self.enable_lightly_cplb:
rank_tokens = 0 rank_tokens = 0
rank_pad_tokens = 0 rank_pad_tokens = 0
accu_q_start = 0 accu_q_start = 0
......
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