[Flang][MLIR][OpenMP] Simplify data layot setting using helper setMLI…

…RDataLayout
agozillon · Jan 17, 2024 · 62d6b6b · 62d6b6b
1 parent b20b7ee
commit 62d6b6b
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/flang/lib/Optimizer/CodeGen/CodeGen.cpp b/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