mlir/lib/Target/LLVMIR/Dialect/AMX/AMXToLLVMIRTranslation.cpp

   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 &registry) {
  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 }