CheckTensorNANOrInf support checking SelectedRows

7 years ago · 3033841b4a
parent 02c31458bb
commit 3033841b4a
1 changed files with 2 additions and 0 deletions
--- a/paddle/fluid/framework/operator.cc
+++ b/paddle/fluid/framework/operator.cc
@ -679,6 +679,8 @@ void OperatorWithKernel::RunImpl(const Scope& scope,
      if (var == nullptr) continue;
      if (var->IsType<framework::LoDTensor>()) {
        CheckTensorNANOrInf(vname, var->Get<framework::LoDTensor>());
+      } else if (var->IsType<framework::SelectedRows>()) {
+        CheckTensorNANOrInf(vname, var->Get<framework::SelectedRows>().value());
      }
    }
  }