From 8ea50d999eaad11c614c103835bf42e99f9a3fbd Mon Sep 17 00:00:00 2001
From: Super Daniel <78588128+super-dainiu@users.noreply.github.com>
Date: Tue, 3 Jan 2023 18:05:06 +0800
Subject: [PATCH] [hotfix] pass a parameter. (#2288)

* [autockpt] make it work.

* [autockpt] linearize / merge shape-consistency nodes.

* [autockpt] considering parameter and optimizer weights.

* [hotfix] pass a parameter.
---
 colossalai/auto_parallel/checkpoint/ckpt_solver_rotor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colossalai/auto_parallel/checkpoint/ckpt_solver_rotor.py b/colossalai/auto_parallel/checkpoint/ckpt_solver_rotor.py
index 5cc57fca0..41d23be5c 100644
--- a/colossalai/auto_parallel/checkpoint/ckpt_solver_rotor.py
+++ b/colossalai/auto_parallel/checkpoint/ckpt_solver_rotor.py
@@ -50,7 +50,7 @@ class CheckpointSolverRotor(CheckpointSolverBase):
             optim_multiplier (float, optional): The multiplier of extra weight storage for the
             ``torch.optim.Optimizer``. Default to 1.0.
         """
-        super().__init__(graph, free_memory, True, cnode)
+        super().__init__(graph, free_memory, True, cnode, optim_multiplier)
         self.memory_slots = memory_slots
 
         # construct chain