MLIR
20.0.0git
lib
Target
LLVMIR
Dialect
AMX
AMXToLLVMIRTranslation.cpp
Go to the documentation of this file.
1
//===- AMXToLLVMIRTranslation.cpp - Translate AMX to LLVM IR --------------===//
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 file implements a translation between the AMX dialect and LLVM IR.
10
//
11
//===----------------------------------------------------------------------===//
12
13
#include "
mlir/Target/LLVMIR/Dialect/AMX/AMXToLLVMIRTranslation.h
"
14
#include "
mlir/Dialect/AMX/AMXDialect.h
"
15
#include "
mlir/IR/Operation.h
"
16
#include "
mlir/Target/LLVMIR/ModuleTranslation.h
"
17
18
#include "llvm/IR/IRBuilder.h"
19
#include "llvm/IR/IntrinsicsX86.h"
20
21
using namespace
mlir
;
22
using namespace
mlir::LLVM
;
23
24
namespace
{
25
/// Implementation of the dialect interface that converts operations belonging
26
/// to the AMX dialect to LLVM IR.
27
class
AMXDialectLLVMIRTranslationInterface
28
:
public
LLVMTranslationDialectInterface
{
29
public
:
30
using
LLVMTranslationDialectInterface::LLVMTranslationDialectInterface
;
31
32
/// Translates the given operation to LLVM IR using the provided IR builder
33
/// and saving the state in `moduleTranslation`.
34
LogicalResult
35
convertOperation(
Operation
*op, llvm::IRBuilderBase &builder,
36
LLVM::ModuleTranslation
&moduleTranslation)
const
final
{
37
Operation
&opInst = *op;
38
#include "mlir/Dialect/AMX/AMXConversions.inc"
39
40
return
failure();
41
}
42
};
43
}
// namespace
44
45
void
mlir::registerAMXDialectTranslation
(
DialectRegistry
®istry) {
46
registry.
insert
<amx::AMXDialect>();
47
registry.
addExtension
(+[](
MLIRContext
*ctx, amx::AMXDialect *dialect) {
48
dialect->addInterfaces<AMXDialectLLVMIRTranslationInterface>();
49
});
50
}
51
52
void
mlir::registerAMXDialectTranslation
(
MLIRContext
&context) {
53
DialectRegistry
registry;
54
registerAMXDialectTranslation
(registry);
55
context.
appendDialectRegistry
(registry);
56
}
AMXDialect.h
AMXToLLVMIRTranslation.h
Operation.h
ModuleTranslation.h
mlir::DialectRegistry
The DialectRegistry maps a dialect namespace to a constructor for the matching dialect.
Definition:
DialectRegistry.h:139
mlir::DialectRegistry::addExtension
bool addExtension(TypeID extensionID, std::unique_ptr< DialectExtensionBase > extension)
Add the given extension to the registry.
Definition:
DialectRegistry.h:210
mlir::DialectRegistry::insert
void insert()
Definition:
DialectRegistry.h:147
mlir::LLVMTranslationDialectInterface
Base class for dialect interfaces providing translation to LLVM IR.
Definition:
LLVMTranslationInterface.h:38
mlir::LLVMTranslationDialectInterface::LLVMTranslationDialectInterface
LLVMTranslationDialectInterface(Dialect *dialect)
Definition:
LLVMTranslationInterface.h:40
mlir::LLVM::ModuleTranslation
Implementation class for module translation.
Definition:
ModuleTranslation.h:58
mlir::MLIRContext
MLIRContext is the top-level object for a collection of MLIR operations.
Definition:
MLIRContext.h:60
mlir::MLIRContext::appendDialectRegistry
void appendDialectRegistry(const DialectRegistry ®istry)
Append the contents of the given dialect registry to the registry associated with this context.
Definition:
MLIRContext.cpp:419
mlir::Operation
Operation is the basic unit of execution within MLIR.
Definition:
Operation.h:88
mlir::LLVM
Definition:
ConvertFuncToLLVM.h:21
mlir
Include the generated interface declarations.
Definition:
LocalAliasAnalysis.h:20
mlir::registerAMXDialectTranslation
void registerAMXDialectTranslation(DialectRegistry ®istry)
Register the AMX dialect and the translation from it to the LLVM IR in the given registry;.
Definition:
AMXToLLVMIRTranslation.cpp:45
Generated on Tue Oct 8 2024 12:33:22 for MLIR by
1.9.1