github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[BUILD] updated LLVM binaries (#1504)

Browse Source 
Co-authored-by: Christian Sigg <csigg@google.com>
This commit is contained in:
Philippe Tillet
2023-04-11 00:14:00 -07:00
committed by GitHub
parent 6d0ed41307
commit e0d6f5f4f5
3 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lib/Conversion/TritonGPUToLLVM/TritonGPUToLLVMBase.h
View File

@@ -77,7 +77,7 @@ protected:
auto varargsAttr = funcOp->getAttrOfType<BoolAttr>("func.varargs");
TypeConverter::SignatureConversion result(funcOp.getNumArguments());
auto llvmType = getTypeConverter()->convertFunctionSignature(
funcOp.getFunctionType(), varargsAttr && varargsAttr.getValue(),
funcOp.getFunctionType(), varargsAttr && varargsAttr.getValue(), false,
result);
if (!llvmType)
return nullptr;

8
lib/Target/LLVMIR/LLVMIRTranslation.cpp
View File

@@ -7,6 +7,7 @@
#include "mlir/IR/Dialect.h"
#include "mlir/Pass/Pass.h"
#include "mlir/Pass/PassManager.h"
#include "mlir/Target/LLVMIR/Dialect/Builtin/BuiltinToLLVMIRTranslation.h"
#include "mlir/Target/LLVMIR/Dialect/LLVMIR/LLVMToLLVMIRTranslation.h"
#include "mlir/Target/LLVMIR/Dialect/NVVM/NVVMToLLVMIRTranslation.h"
#include "mlir/Target/LLVMIR/Dialect/ROCDL/ROCDLToLLVMIRTranslation.h"
@@ -245,6 +246,7 @@ std::unique_ptr<llvm::Module>
translateLLVMToLLVMIR(llvm::LLVMContext *llvmContext, mlir::ModuleOp module,
bool isROCM) {
DialectRegistry registry;
mlir::registerBuiltinDialectTranslation(registry);
mlir::registerLLVMDialectTranslation(registry);
mlir::registerROCDLDialectTranslation(registry);
mlir::registerNVVMDialectTranslation(registry);
@@ -296,7 +298,11 @@ translateTritonGPUToLLVMIR(llvm::LLVMContext *llvmContext,
mlir::ModuleOp module, int computeCapability,
bool isROCM) {
mlir::PassManager pm(module->getContext());
applyPassManagerCLOptions(pm);
mlir::registerPassManagerCLOptions();
if (failed(applyPassManagerCLOptions(pm))) {
llvm::errs() << "failed to apply pass manager CL options\n";
return nullptr;
}
auto printingFlags = mlir::OpPrintingFlags();
printingFlags.elideLargeElementsAttrs(16);
pm.enableIRPrinting(

2
python/setup.py
View File

@@ -68,7 +68,7 @@ def get_llvm_package_info():
use_assert_enabled_llvm = check_env_flag("TRITON_USE_ASSERT_ENABLED_LLVM", "False")
release_suffix = "assert" if use_assert_enabled_llvm else "release"
name = f'llvm+mlir-17.0.0-x86_64-{system_suffix}-{release_suffix}'
version = "llvm-17.0.0-2538e550420f"
version = "llvm-17.0.0-f733b4fb9b8b"
url = f"https://github.com/ptillet/triton-llvm-releases/releases/download/{version}/{name}.tar.xz"
return Package("llvm", name, url, "LLVM_INCLUDE_DIRS", "LLVM_LIBRARY_DIR", "LLVM_SYSPATH")