From 3de0f612e8b5340b502045a4fd0d2b41b4465ff4 Mon Sep 17 00:00:00 2001
From: minqiyang <minqiyang@baidu.com>
Date: Fri, 21 Dec 2018 13:09:31 +0800
Subject: [PATCH] Polish code
test=develop
---
paddle/fluid/pybind/imperative.cc | 5 ++---
python/paddle/fluid/imperative/base.py | 3 +--
2 files changed, 3 insertions(+), 5 deletions(-)
diff --git a/paddle/fluid/pybind/imperative.cc b/paddle/fluid/pybind/imperative.cc
index be63fb8778..34e9c897d9 100644
--- a/paddle/fluid/pybind/imperative.cc
+++ b/paddle/fluid/pybind/imperative.cc
@@ -24,9 +24,8 @@ namespace pybind {
void BindTracer(pybind11::module *m) {
pybind11::class_<imperative::Tracer>(*m, "Tracer", "")
.def("__init__",
- [](imperative::Tracer &self, framework::BlockDesc *root_block,
- framework::BlockDesc *startup_block) {
- new (&self) imperative::Tracer(root_block, startup_block);
+ [](imperative::Tracer &self, framework::BlockDesc *root_block) {
+ new (&self) imperative::Tracer(root_block);
})
.def("trace", &imperative::Tracer::Trace)
.def("get_scope", &imperative::Tracer::GetScope,
diff --git a/python/paddle/fluid/imperative/base.py b/python/paddle/fluid/imperative/base.py
index c9e7c86483..a33ad4b72c 100644
--- a/python/paddle/fluid/imperative/base.py
+++ b/python/paddle/fluid/imperative/base.py
@@ -28,8 +28,7 @@ def enabled():
def guard():
train = framework.Program()
startup = framework.Program()
- tracer = core.Tracer(train.current_block().desc,
- startup.current_block().desc)
+ tracer = core.Tracer(train.current_block().desc)
with framework.program_guard(train, startup):
with framework.unique_name.guard():
with framework._imperative_guard(tracer):