mirror of https://github.com/hpcaitech/ColossalAI
code style
parent
fa5e6fbf96
commit
e0ae68e736
|
@ -1497,8 +1497,8 @@ class ChunkRegionSearch(object):
|
||||||
chunk_info = self._step_search(mem_peak, active_node, chunk_infos)
|
chunk_info = self._step_search(mem_peak, active_node, chunk_infos)
|
||||||
if chunk_info is None:
|
if chunk_info is None:
|
||||||
break
|
break
|
||||||
|
|
||||||
chunk_infos.append(chunk_info)
|
chunk_infos.append(chunk_info)
|
||||||
|
|
||||||
(
|
(
|
||||||
mem_peak,
|
mem_peak,
|
||||||
_,
|
_,
|
||||||
|
|
Loading…
Reference in New Issue