Merge pull request #12439 from jacquesqiao/CheckTensorNANOrInf-support-selectedrows

jacquesqiao · web-flow · commit 297cbeb1c615 · 2018-07-30T13:41:18.000+08:00
CheckTensorNANOrInf support checking SelectedRows
diff --git 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());
       }
     }
   }

Original file line number	Diff line number	Diff line change
`@@ -679,6 +679,8 @@ void OperatorWithKernel::RunImpl(const Scope& scope,`
`679`	`679`	`if (var == nullptr) continue;`
`680`	`680`	`if (var->IsType<framework::LoDTensor>()) {`
`681`	`681`	`CheckTensorNANOrInf(vname, var->Get<framework::LoDTensor>());`
	`682`	`+ } else if (var->IsType<framework::SelectedRows>()) {`
	`683`	`+ CheckTensorNANOrInf(vname, var->Get<framework::SelectedRows>().value());`
`682`	`684`	`}`
`683`	`685`	`}`
`684`	`686`	`}`