2ea6bb645e
The revision consistently uses the AliasAnalysisOp and AccessGroupOp interfaces in the translation from MLIR to LLVMIR. It thus drops the last string based lookups of alias scope and access group attributes. Depends on D144851 Reviewed By: Dinistro Differential Revision: https://reviews.llvm.org/D145037
77 lines
3.0 KiB
C++
77 lines
3.0 KiB
C++
//===- LoopAnnotationTranslation.h ------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the translation between an MLIR loop annotations and
|
|
// the corresponding LLVMIR metadata representation.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef MLIR_LIB_TARGET_LLVMIR_LOOPANNOTATIONTRANSLATION_H_
|
|
#define MLIR_LIB_TARGET_LLVMIR_LOOPANNOTATIONTRANSLATION_H_
|
|
|
|
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
|
|
#include "mlir/Target/LLVMIR/ModuleTranslation.h"
|
|
|
|
namespace mlir {
|
|
namespace LLVM {
|
|
namespace detail {
|
|
|
|
/// A helper class that converts LoopAnnotationAttrs and AccessGroupMetadataOps
|
|
/// into a corresponding llvm::MDNodes.
|
|
class LoopAnnotationTranslation {
|
|
public:
|
|
LoopAnnotationTranslation(Operation *mlirModule, llvm::Module &llvmModule)
|
|
: mlirModule(mlirModule), llvmModule(llvmModule) {}
|
|
|
|
llvm::MDNode *translateLoopAnnotation(LoopAnnotationAttr attr, Operation *op);
|
|
|
|
/// Traverses the global access group metadata operation in the `mlirModule`
|
|
/// and creates corresponding LLVM metadata nodes.
|
|
LogicalResult createAccessGroupMetadata();
|
|
|
|
/// Returns the LLVM metadata corresponding to a symbol reference to an mlir
|
|
/// LLVM dialect access group operation.
|
|
llvm::MDNode *getAccessGroup(Operation *op,
|
|
SymbolRefAttr accessGroupRef) const;
|
|
|
|
/// Returns the LLVM metadata corresponding to the access group operations
|
|
/// referenced by the AccessGroupOpInterface or null if there are none.
|
|
llvm::MDNode *getAccessGroups(AccessGroupOpInterface op) const;
|
|
|
|
private:
|
|
/// Returns the LLVM metadata corresponding to a llvm loop metadata attribute.
|
|
llvm::MDNode *lookupLoopMetadata(Attribute options) const {
|
|
return loopMetadataMapping.lookup(options);
|
|
}
|
|
|
|
void mapLoopMetadata(Attribute options, llvm::MDNode *metadata) {
|
|
auto result = loopMetadataMapping.try_emplace(options, metadata);
|
|
(void)result;
|
|
assert(result.second &&
|
|
"attempting to map loop options that was already mapped");
|
|
}
|
|
|
|
/// Mapping from an attribute describing loop metadata to its LLVM metadata.
|
|
/// The metadata is attached to Latch block branches with this attribute.
|
|
DenseMap<Attribute, llvm::MDNode *> loopMetadataMapping;
|
|
|
|
/// Mapping from an access group metadata operation to its LLVM metadata.
|
|
/// This map is populated on module entry and is used to annotate loops (as
|
|
/// identified via their branches) and contained memory accesses.
|
|
DenseMap<Operation *, llvm::MDNode *> accessGroupMetadataMapping;
|
|
|
|
Operation *mlirModule;
|
|
llvm::Module &llvmModule;
|
|
};
|
|
|
|
} // namespace detail
|
|
} // namespace LLVM
|
|
} // namespace mlir
|
|
|
|
#endif // MLIR_LIB_TARGET_LLVMIR_LOOPANNOTATIONTRANSLATION_H_
|