fix bug that some op has no op_role attr (#41040)

paddle/fluid/framework/details/nan_inf_utils_detail.cc

@@ -422,8 +422,11 @@ void CheckVarHasNanOrInf(const std::string& op_type,
 bool IsSkipOp(const framework::OperatorBase& op) {
   if (op_type_nan_inf_white_list().count(op.Type()) != 0) return true;
 
-  int op_role = op.template Attr<int>(
-      framework::OpProtoAndCheckerMaker::OpRoleAttrName());
+  int op_role = 0;
+  if (op.HasAttr(framework::OpProtoAndCheckerMaker::OpRoleAttrName())) {
+    op_role = op.template Attr<int>(
+        framework::OpProtoAndCheckerMaker::OpRoleAttrName());
+  }
 
   // kForward=0, can't filter
   if (op_role == static_cast<int>(framework::OpRole::kForward)) {