diff --git a/python/dllib/src/bigdl/utils/common.py b/python/dllib/src/bigdl/utils/common.py
index 6bb1fb6d873..a58d61629dd 100644
--- a/python/dllib/src/bigdl/utils/common.py
+++ b/python/dllib/src/bigdl/utils/common.py
@@ -425,6 +425,12 @@ def init_engine(bigdl_type="float"):
 def get_bigdl_engine_type(bigdl_type="float"):
     return callBigDlFunc(bigdl_type, "getEngineType")
 
+def set_optimizer_version(optimizerVersion, bigdl_type="float"):
+    return callBigDlFunc(bigdl_type, "setOptimizerVersion", optimizerVersion)
+
+def get_optimizer_version(bigdl_type="float"):
+    return callBigDlFunc(bigdl_type, "getOptimizerVersion")
+
 def init_executor_gateway(sc, bigdl_type="float"):
     callBigDlFunc(bigdl_type, "initExecutorGateway", sc, sc._gateway._gateway_client.port)