Skip to content

Commit 8def5e0

Browse files
gflegarolegshyshkov
authored andcommitted
[BACKEND] Update LLVM version to llvm/llvm-project@7752e0a (triton-lang#6735)
Removed `addArgumentMaterialization` since that method was removed from MLIR in llvm/llvm-project@23e3cbb
1 parent d3a8a5a commit 8def5e0

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

third_party/amd/lib/TritonAMDGPUToLLVM/TritonGPUToLLVM.cpp

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
#include "triton/Dialect/Triton/IR/Dialect.h"
2727
#include "triton/Dialect/TritonGPU/IR/Dialect.h"
2828
#include "triton/Dialect/TritonNvidiaGPU/IR/Dialect.h"
29+
#include "llvm/TargetParser/TargetParser.h"
2930

3031
namespace mlir::triton {
3132
#define GEN_PASS_DEF_CONVERTTRITONAMDGPUTOLLVM
@@ -86,6 +87,15 @@ struct ConvertTritonAMDGPUToLLVM
8687
mod.emitError("unsupported target: '") << this->arch.getValue() << "'";
8788
return signalPassFailure();
8889
}
90+
llvm::StringRef chipset =
91+
llvm::AMDGPU::getArchNameAMDGCN(targetInfo.getGPUKind());
92+
llvm::FailureOr<mlir::amdgpu::Chipset> maybeChipset =
93+
mlir::amdgpu::Chipset::parse(chipset);
94+
if (failed(maybeChipset)) {
95+
mlir::emitError(mlir::UnknownLoc::get(&getContext()),
96+
"Invalid chipset name: " + chipset);
97+
return signalPassFailure();
98+
}
8999

90100
mlir::LowerToLLVMOptions option(context);
91101
option.overrideIndexBitwidth(32);

0 commit comments

Comments
 (0)