1 | //===- AMXToLLVMIRTranslation.h - AMX to LLVM IR ----------------*- C++ -*-===// |
2 | // |
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
4 | // See https://llvm.org/LICENSE.txt for license information. |
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
6 | // |
7 | //===----------------------------------------------------------------------===// |
8 | // |
9 | // This provides registration calls for AMX dialect to LLVM IR translation. |
10 | // |
11 | //===----------------------------------------------------------------------===// |
12 | |
13 | #ifndef MLIR_TARGET_LLVMIR_DIALECT_AMX_AMXTOLLVMIRTRANSLATION_H |
14 | #define MLIR_TARGET_LLVMIR_DIALECT_AMX_AMXTOLLVMIRTRANSLATION_H |
15 | |
16 | namespace mlir { |
17 | |
18 | class DialectRegistry; |
19 | class MLIRContext; |
20 | |
21 | /// Register the AMX dialect and the translation from it to the LLVM IR |
22 | /// in the given registry; |
23 | void registerAMXDialectTranslation(DialectRegistry ®istry); |
24 | |
25 | /// Register the AMX dialect and the translation from it in the registry |
26 | /// associated with the given context. |
27 | void registerAMXDialectTranslation(MLIRContext &context); |
28 | |
29 | } // namespace mlir |
30 | |
31 | #endif // MLIR_TARGET_LLVMIR_DIALECT_AMX_AMXTOLLVMIRTRANSLATION_H |
32 | |