1 files changed, 143 insertions, 0 deletions
diff --git a/src/gallium/drivers/radeon/radeon_llvm_emit.cpp b/src/gallium/drivers/radeon/radeon_llvm_emit.cpp
new file mode 100644
index 00000000000..f5e357a392f
--- /dev/null
+++ b/src/gallium/drivers/radeon/radeon_llvm_emit.cpp
@@ -0,0 +1,143 @@
+/*
+ * Copyright 2011 Advanced Micro Devices, Inc.
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ *
+ * Authors: Tom Stellard <thomas.stellard@amd.com>
+ *
+ */
+#include "radeon_llvm.h"
+
+#include <llvm/Module.h>
+#include <llvm/PassManager.h>
+#include <llvm/ADT/Triple.h>
+#include <llvm/Support/FormattedStream.h>
+#include <llvm/Support/Host.h>
+#include <llvm/Support/TargetRegistry.h>
+#include <llvm/Support/TargetSelect.h>
+#include <llvm/Target/TargetData.h>
+#include <llvm/Target/TargetMachine.h>
+
+#include <llvm/Transforms/Scalar.h>
+
+#include <llvm-c/Target.h>
+
+#include <iostream>
+#include <stdlib.h>
+#include <stdio.h>
+
+using namespace llvm;
+
+#ifndef EXTERNAL_LLVM
+#include "AMDISATargetMachine.h"
+Target TheAMDISATarget;
+#endif
+
+
+/**
+ * Compile an LLVM module to machine code.
+ *
+ * @param bytes This function allocates memory for the byte stream, it is the
+ * caller's responsibility to free it.
+ */
+extern "C" unsigned
+radeon_llvm_compile(LLVMModuleRef M, unsigned char ** bytes,
+                 unsigned * byte_count, const char * gpu_family,
+                 unsigned dump) {
+
+#ifdef EXTERNAL_LLVM
+   const Target * AMDISATarget = NULL;
+
+   /* XXX: Can we just initialize the AMDISA target here? */
+   InitializeAllTargets();
+   Triple::ArchType Arch = Triple::getArchTypeForLLVMName("amdisa");
+   if (Arch == Triple::UnknownArch) {
+      fprintf(stderr, "Unknown Arch\n");
+   }
+   std::string AMDISAArchName = "amdisa";
+   AMDISATriple.setArch(Arch);
+   for (TargetRegistry::iterator it = TargetRegistry::begin(),
+           ie = TargetRegistry::end(); it != ie; ++it) {
+      if (it->getName() == AMDISAArchName) {
+         AMDISATarget = &*it;
+         break;
+      }
+   }
+
+   if(!AMDISATarget) {
+      fprintf(stderr, "Can't find target\n");
+      return 1;
+   }
+#else
+   RegisterTargetMachine<AMDISATargetMachine> Y(TheAMDISATarget);
+   RegisterMCAsmInfoFn A(TheAMDISATarget, createMCAsmInfo);
+#endif
+
+   Module * mod = unwrap(M);
+   Triple AMDISATriple(sys::getHostTriple());
+
+
+   std::string FS = gpu_family;
+
+#ifdef EXTERNAL_LLVM
+   std::auto_ptr<TargetMachine> tm(AMDISATarget->createTargetMachine(
+                     AMDISATriple.getTriple(), FS));
+   TargetMachine &AMDISATargetMachine = *tm.get();
+#else
+   AMDISATargetMachine * tm = new AMDISATargetMachine(TheAMDISATarget,
+            AMDISATriple.getTriple(), gpu_family, "", Reloc::Default,
+            CodeModel::Default);
+   TargetMachine &AMDISATargetMachine = *tm;
+   /* XXX: Use TargetMachine.Options in 3.0 */
+   if (dump) {
+      tm->dumpCode();
+   }
+#endif
+   const TargetData * AMDISAData = AMDISATargetMachine.getTargetData();
+   PassManager PM;
+   PM.add(new TargetData(*AMDISAData));
+   PM.add(createPromoteMemoryToRegisterPass());
+   AMDISATargetMachine.setAsmVerbosityDefault(true);
+
+   std::string CodeString;
+   raw_string_ostream oStream(CodeString);
+   formatted_raw_ostream out(oStream);
+
+   /* Optional extra paramater true / false to disable verify */
+   if (AMDISATargetMachine.addPassesToEmitFile(PM, out, TargetMachine::CGFT_AssemblyFile,
+                                               CodeGenOpt::Default, true)){
+      fprintf(stderr, "AddingPasses failed.\n");
+      return 1;
+   }
+   PM.run(*mod);
+
+   out.flush();
+   std::string &data = oStream.str();
+
+   *bytes = (unsigned char*)malloc(data.length() * sizeof(unsigned char));
+   memcpy(*bytes, data.c_str(), data.length() * sizeof(unsigned char));
+   *byte_count = data.length();
+
+#ifndef EXTERNAL_LLVM
+   delete tm;
+#endif
+
+   return 0;
+}