提交 9774c8ff 编写于 作者: H hjchen2

Fix crash when DEBUG is enabled

上级 2bbf01d1
...@@ -64,8 +64,7 @@ void OperatorBase<Dtype>::Run() { ...@@ -64,8 +64,7 @@ void OperatorBase<Dtype>::Run() {
for (const auto key : input_keys) { for (const auto key : input_keys) {
auto var_vec_in = inputs_.at(key); auto var_vec_in = inputs_.at(key);
for (int i = 0; i < var_vec_in.size(); ++i) { for (int i = 0; i < var_vec_in.size(); ++i) {
DLOG << var_vec_in[i]; auto vari = this->scope_->FindVar(var_vec_in[i]);
auto vari = this->scope_->FindVar("input");
if (vari->IsInitialized()) { if (vari->IsInitialized()) {
const Tensor *tensor = vari->template Get<framework::LoDTensor>(); const Tensor *tensor = vari->template Get<framework::LoDTensor>();
if (tensor) DLOG << type_ << " input- " << key << "=" << *tensor; if (tensor) DLOG << type_ << " input- " << key << "=" << *tensor;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册