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

…m-head - Add `BUILD` files. - Remove `python/` folder.
openxla · Apr 14, 2023 · 121d8e9 · 121d8e9
1 parent 1e8882e
commit 121d8e9
Show file tree

Hide file tree

Showing 43 changed files with 679 additions and 22,543 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/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/setup.cfg b/python/setup.cfg
diff --git a/python/src/main.cc b/python/src/main.cc
diff --git a/python/test/unit/language/assert_helper.py b/python/test/unit/language/assert_helper.py
diff --git a/python/test/unit/language/print_helper.py b/python/test/unit/language/print_helper.py
diff --git a/python/test/unit/language/test_block_pointer.py b/python/test/unit/language/test_block_pointer.py