mirror of https://github.com/llvm/torch-mlir
48 lines
1.7 KiB
C++
48 lines
1.7 KiB
C++
//===- IrHelpers.h - Helpers for bridging analysis and IR types -----------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef NPCOMP_TYPING_SUPPORT_CPA_IR_HELPERS_H
|
|
#define NPCOMP_TYPING_SUPPORT_CPA_IR_HELPERS_H
|
|
|
|
#include "mlir/IR/BuiltinTypes.h"
|
|
#include "npcomp/Typing/Analysis/CPA/Types.h"
|
|
|
|
namespace mlir {
|
|
namespace NPCOMP {
|
|
namespace Typing {
|
|
namespace CPA {
|
|
|
|
/// Creates an array object type with a possibly unknown element type.
|
|
/// By convention, arrays have a single type slot for the element type
|
|
/// named 'e'.
|
|
ObjectValueType *newArrayType(Context &context,
|
|
ObjectValueType::IrTypeConstructor irCtor,
|
|
Identifier *typeIdentifier,
|
|
llvm::Optional<TypeNode *> elementType);
|
|
|
|
/// Gets the TypeNode associated with the element type for an array allocated
|
|
/// via newArrayType.
|
|
TypeNode *getArrayElementType(ObjectValueType *arrayType);
|
|
|
|
/// Creates an ObjectValueType for the given TensorType. The result will
|
|
/// reconstruct the original TensorType's structure but with the resolved
|
|
/// element type.
|
|
ObjectValueType *createTensorLikeArrayType(Context &context,
|
|
TensorType tensorType);
|
|
|
|
/// Creates a default IR type map hook which supports built-in MLIR types
|
|
/// that do not implement the analysis interfaces.
|
|
Context::IrTypeMapHook createDefaultTypeMapHook();
|
|
|
|
} // namespace CPA
|
|
} // namespace Typing
|
|
} // namespace NPCOMP
|
|
} // namespace mlir
|
|
|
|
#endif // NPCOMP_TYPING_SUPPORT_CPA_IR_HELPERS_H
|