CheckTensorNANOrInf support checking SelectedRows

3033841b · qiaolongfei · 02c31458 · 3033841b
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/framework/operator.cc paddle/fluid/framework/operator.cc +2 -0

未找到文件。
--- 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());
      }
    }
  }