File tree Expand file tree Collapse file tree 1 file changed +10
-0
lines changed
third_party/amd/lib/TritonAMDGPUToLLVM Expand file tree Collapse file tree 1 file changed +10
-0
lines changed Original file line number Diff line number Diff line change 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 );
You can’t perform that action at this time.
0 commit comments