OpenXLA-specific changes to fork of github.com/openai/triton/tree/llv…

…m-head - Add `BUILD` files. - Remove `python/` folder.
openxla · May 4, 2023 · 5b63e5b · 5b63e5b
1 parent e18886f
commit 5b63e5b
Show file tree

Hide file tree

Showing 81 changed files with 679 additions and 38,677 deletions.
diff --git a/BUILD b/BUILD
diff --git a/lib/Target/LLVMIR/LLVMIRTranslation.cpp b/lib/Target/LLVMIR/LLVMIRTranslation.cpp
@@ -25,6 +25,7 @@
 #include "llvm/IRReader/IRReader.h"
 #include "llvm/Linker/Linker.h"
 #include "llvm/Support/SourceMgr.h"
+#include "third_party/py/triton/google/find_cuda.h"
 #include <dlfcn.h>
 #include <filesystem>
 #include <iterator>
@@ -163,9 +164,8 @@ static std::map<std::string, std::string> getExternLibs(mlir::ModuleOp module) {
       }
       return std::filesystem::path(fileinfo.dli_fname);
     }();
-    static const auto runtime_path =
-        this_library_path.parent_path().parent_path() / "third_party" / "cuda" /
-        "lib" / "libdevice.10.bc";
+    static const auto runtime_path = (
+        fs::path(PathToLibdevice()) / "libdevice.10.bc");
     if (fs::exists(runtime_path)) {
       externLibs.try_emplace(libdevice, runtime_path.string());
     } else {

diff --git a/lib/Target/PTX/PTXTranslation.cpp b/lib/Target/PTX/PTXTranslation.cpp
@@ -49,7 +49,7 @@ std::string translateLLVMIRToPTX(llvm::Module &module, int cc, int version) {
   auto *shortPtr =
       static_cast<llvm::cl::opt<bool> *>(options["nvptx-short-ptr"]);
   assert(shortPtr);
-  shortPtr->setValue(true);
+  shortPtr->setValue(false);
   std::string sm = cc == 90 ? "sm_90a" : "sm_" + std::to_string(cc);
   // max PTX version
   int ptxMajor = maxPTX / 10;

diff --git a/python/MANIFEST.in b/python/MANIFEST.in
diff --git a/python/README.md b/python/README.md
diff --git a/python/examples/copy_strided.py b/python/examples/copy_strided.py
diff --git a/python/examples/empty.py b/python/examples/empty.py
diff --git a/python/pyproject.toml b/python/pyproject.toml
diff --git a/python/setup.py b/python/setup.py
diff --git a/python/src/extra/cuda.ll b/python/src/extra/cuda.ll