mirror of https://github.com/llvm/torch-mlir
88 lines
2.4 KiB
C++
88 lines
2.4 KiB
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 contains data structures (which we typically call "descriptors")
|
|
// that are emitted by the compiler and must be kept in sync with the compiler
|
|
// code that creates them in LowerToLLVM.cpp.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef NPCOMP_LIB_RUNTIME_COMPILERDATASTRUCTURES_H
|
|
#define NPCOMP_LIB_RUNTIME_COMPILERDATASTRUCTURES_H
|
|
|
|
#include <cstdint>
|
|
|
|
namespace refbackrt {
|
|
|
|
// All arguments are packed into this type-erased form for being invoked. See
|
|
// LowerToLLVM.cpp for more details.
|
|
typedef void ABIFunc(void **, void **);
|
|
|
|
enum class ABIArgType : std::uint32_t {
|
|
kNone = 0,
|
|
kMemref,
|
|
kF32,
|
|
kF64,
|
|
};
|
|
|
|
enum class ABIElementType : std::uint32_t {
|
|
kNone = 0,
|
|
kF32,
|
|
};
|
|
|
|
struct InputDescriptor {
|
|
ABIArgType abiType;
|
|
ABIElementType elementType;
|
|
|
|
std::int32_t rank;
|
|
std::int32_t* extents;
|
|
|
|
// TODO(brycearden): Change to bool at ABI boundary
|
|
// std::int32_t isStatic;
|
|
};
|
|
|
|
struct OutputDescriptor {
|
|
ABIArgType abiType;
|
|
ABIElementType elementType;
|
|
|
|
std::int32_t rank;
|
|
std::int32_t* extents;
|
|
|
|
// TODO(brycearden): Change to bool at ABI boundary
|
|
//std::int32_t isStatic;
|
|
};
|
|
|
|
struct FuncDescriptor {
|
|
// The length of the function name.
|
|
std::int32_t nameLen;
|
|
// The name of the function, to allow lookup.
|
|
const char *name;
|
|
// This is a raw function pointer to the function's entry point as
|
|
// emitted by the compiler.
|
|
ABIFunc *functionPtr;
|
|
// The number of inputs to the function.
|
|
std::int32_t numInputs;
|
|
// The number of outputs of the function.
|
|
std::int32_t numOutputs;
|
|
// TODO: Add shape checking to arg / result descriptor(s)
|
|
InputDescriptor *inputDescriptors;
|
|
OutputDescriptor *outputDescriptors;
|
|
};
|
|
|
|
// The top-level entry point of the module metadata emitted by the
|
|
// compiler. Unlike all the other descriptors here, external code does handle
|
|
// this type (albeit through an opaque pointer).
|
|
struct ModuleDescriptor {
|
|
std::int32_t numFuncDescriptors;
|
|
FuncDescriptor *functionDescriptors;
|
|
};
|
|
|
|
} // namespace refbackrt
|
|
|
|
#endif // NPCOMP_LIB_RUNTIME_COMPILERDATASTRUCTURES_H
|