Skip to content

Commit 5b63e5b

Browse files
committed
OpenXLA-specific changes to fork of github.com/openai/triton/tree/llvm-head
- Add `BUILD` files. - Remove `python/` folder.
1 parent e18886f commit 5b63e5b

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

81 files changed

+679
-38677
lines changed

BUILD

Lines changed: 620 additions & 0 deletions
Large diffs are not rendered by default.

lib/Target/LLVMIR/LLVMIRTranslation.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
#include "llvm/IRReader/IRReader.h"
2626
#include "llvm/Linker/Linker.h"
2727
#include "llvm/Support/SourceMgr.h"
28+
#include "third_party/py/triton/google/find_cuda.h"
2829
#include <dlfcn.h>
2930
#include <filesystem>
3031
#include <iterator>
@@ -163,9 +164,8 @@ static std::map<std::string, std::string> getExternLibs(mlir::ModuleOp module) {
163164
}
164165
return std::filesystem::path(fileinfo.dli_fname);
165166
}();
166-
static const auto runtime_path =
167-
this_library_path.parent_path().parent_path() / "third_party" / "cuda" /
168-
"lib" / "libdevice.10.bc";
167+
static const auto runtime_path = (
168+
fs::path(PathToLibdevice()) / "libdevice.10.bc");
169169
if (fs::exists(runtime_path)) {
170170
externLibs.try_emplace(libdevice, runtime_path.string());
171171
} else {

lib/Target/PTX/PTXTranslation.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ std::string translateLLVMIRToPTX(llvm::Module &module, int cc, int version) {
4949
auto *shortPtr =
5050
static_cast<llvm::cl::opt<bool> *>(options["nvptx-short-ptr"]);
5151
assert(shortPtr);
52-
shortPtr->setValue(true);
52+
shortPtr->setValue(false);
5353
std::string sm = cc == 90 ? "sm_90a" : "sm_" + std::to_string(cc);
5454
// max PTX version
5555
int ptxMajor = maxPTX / 10;

python/MANIFEST.in

Lines changed: 0 additions & 4 deletions
This file was deleted.

python/README.md

Whitespace-only changes.

python/examples/copy_strided.py

Lines changed: 0 additions & 18 deletions
This file was deleted.

python/examples/empty.py

Lines changed: 0 additions & 13 deletions
This file was deleted.

python/pyproject.toml

Lines changed: 0 additions & 8 deletions
This file was deleted.

python/setup.py

Lines changed: 0 additions & 305 deletions
This file was deleted.

python/src/extra/cuda.ll

Lines changed: 0 additions & 17 deletions
This file was deleted.

0 commit comments

Comments
 (0)