1d0d7da57c
Enables verification of attributes, independent of op, that references symbols. This enables verifying Attribute with symbol usage independent of operation attached to (e.g., the validity is on the Attribute independent of the operation). --------- Co-authored-by: Mehdi Amini <joker.eph@gmail.com>
59 lines
1.8 KiB
C++
59 lines
1.8 KiB
C++
//===- TestTypes.h - MLIR Test Dialect Types --------------------*- 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 types defined by the TestDialect for testing various
|
|
// features of MLIR.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef MLIR_TESTATTRIBUTES_H
|
|
#define MLIR_TESTATTRIBUTES_H
|
|
|
|
#include <tuple>
|
|
|
|
#include "TestTraits.h"
|
|
#include "mlir/Dialect/Ptr/IR/MemorySpaceInterfaces.h"
|
|
#include "mlir/Dialect/Utils/StructuredOpsUtils.h"
|
|
#include "mlir/IR/Attributes.h"
|
|
#include "mlir/IR/BuiltinAttributes.h"
|
|
#include "mlir/IR/Dialect.h"
|
|
#include "mlir/IR/DialectResourceBlobManager.h"
|
|
#include "mlir/IR/OpImplementation.h"
|
|
#include "mlir/IR/SymbolTable.h"
|
|
#include "mlir/IR/TensorEncoding.h"
|
|
|
|
// generated files require above includes to come first
|
|
#include "TestAttrInterfaces.h.inc"
|
|
#include "TestOpEnums.h.inc"
|
|
|
|
namespace test {
|
|
class TestDialect;
|
|
// Payload class for the CopyCountAttr.
|
|
class CopyCount {
|
|
public:
|
|
CopyCount(std::string value) : value(value) {}
|
|
CopyCount(const CopyCount &rhs);
|
|
CopyCount &operator=(const CopyCount &rhs);
|
|
CopyCount(CopyCount &&rhs) = default;
|
|
CopyCount &operator=(CopyCount &&rhs) = default;
|
|
static int counter;
|
|
std::string value;
|
|
};
|
|
llvm::raw_ostream &operator<<(llvm::raw_ostream &os,
|
|
const test::CopyCount &value);
|
|
|
|
/// A handle used to reference external elements instances.
|
|
using TestDialectResourceBlobHandle =
|
|
mlir::DialectResourceBlobHandle<TestDialect>;
|
|
} // namespace test
|
|
|
|
#define GET_ATTRDEF_CLASSES
|
|
#include "TestAttrDefs.h.inc"
|
|
|
|
#endif // MLIR_TESTATTRIBUTES_H
|