CheckTensorNANOrInf support checking SelectedRows

jacquesqiao · jacquesqiao · commit 3033841b4ae9 · 2018-07-30T12:47:53.000+08:00
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`	`}`