[Flang][MLIR][OpenMP] Simplify data layot setting using helper setMLIRDataLayout

Author: agozillon

flang/lib/Optimizer/CodeGen/CodeGen.cpp

@@ -3869,13 +3869,7 @@ class FIRToLLVMLowering
 
     if (!forcedDataLayout.empty()) {
       llvm::DataLayout dl(forcedDataLayout);
-      mlir::MLIRContext *context = mod.getContext();
-      mod->setAttr(
-          mlir::LLVM::LLVMDialect::getDataLayoutAttrName(),
-          mlir::StringAttr::get(context, dl.getStringRepresentation()));
-      mlir::DataLayoutSpecInterface dlSpec =
-          mlir::translateDataLayout(dl, context);
-      mod->setAttr(mlir::DLTIDialect::kDataLayoutAttrName, dlSpec);
+      fir::support::setMLIRDataLayout(mod, dl);
     }
 
     // Run dynamic pass pipeline for converting Math dialect