Skip to content

Commit e767f91

Browse files
gflegarloislo
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 d8efd15 commit e767f91

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
#include "third_party/proton/dialect/include/TritonProtonToLLVM/PatternTritonProtonOpToLLVM.h"
3132

@@ -88,6 +89,15 @@ struct ConvertTritonAMDGPUToLLVM
8889
mod.emitError("unsupported target: '") << this->arch.getValue() << "'";
8990
return signalPassFailure();
9091
}
92+
llvm::StringRef chipset =
93+
llvm::AMDGPU::getArchNameAMDGCN(targetInfo.getGPUKind());
94+
llvm::FailureOr<mlir::amdgpu::Chipset> maybeChipset =
95+
mlir::amdgpu::Chipset::parse(chipset);
96+
if (failed(maybeChipset)) {
97+
mlir::emitError(mlir::UnknownLoc::get(&getContext()),
98+
"Invalid chipset name: " + chipset);
99+
return signalPassFailure();
100+
}
91101

92102
mlir::LowerToLLVMOptions option(context);
93103
option.overrideIndexBitwidth(32);

0 commit comments

Comments
 (0)