[mlir] Let ModuleTranslate propagate LLVM triple

[nicolasvasilache]

Summary

This adds LLVM triple propagation and updates the test that did not check it properly.

[nicolasvasilache]

Don't break another test :p

mlir/lib/Target/LLVMIR/ModuleTranslation.cpp

 
 std::unique_ptr<llvm::Module> ModuleTranslation::prepareLLVMModule(
     Operation *m, llvm::LLVMContext &llvmContext, StringRef name) {
   m->getContext()->getOrLoadDialect<LLVM::LLVMDialect>();
   auto llvmModule = std::make_unique<llvm::Module>(name, llvmContext);
   if (auto dataLayoutAttr =
           m->getAttr(LLVM::LLVMDialect::getDataLayoutAttrName()))
     llvmModule->setDataLayout(dataLayoutAttr.cast<StringAttr>().getValue());
+  if (auto targetTripleAttr =
+          m->getAttr(LLVM::LLVMDialect::getTargetTripleAttrName()))
+    llvmModule->setTargetTriple(targetTripleAttr.cast<StringAttr>().getValue());
 
   // Inject declarations for `malloc` and `free` functions that can be used in
   // memref allocation/deallocation coming from standard ops lowering.
   llvm::IRBuilder<> builder(llvmContext);
   llvmModule->getOrInsertFunction("malloc", builder.getInt8PtrTy(),
                                   builder.getInt64Ty());
   llvmModule->getOrInsertFunction("free", builder.getVoidTy(),
                                   builder.getInt8PtrTy());
 
   return llvmModule;
 }

mlir/test/Target/llvmir.mlir

 // CHECK: target datalayout = "E"
 module attributes {llvm.data_layout = "E"} {
   llvm.func @module_big_endian()
 }
 
 // -----
 
 // CHECK: "CodeView", i32 1
+// CHECK: x86_64-pc-windows-msvc
 module attributes {llvm.target_triple = "x86_64-pc-windows-msvc"} {}
 
 // -----
 
 // CHECK-NOT: "CodeView", i32 1
+// CHECK: aarch64-linux-android
 module attributes {llvm.target_triple = "aarch64-linux-android"} {}
 
 // -----
 
 // CHECK-NOT: "CodeView", i32 1
 module attributes {} {}