2019-10-09 07:25:59 +08:00
|
|
|
#include "llvm/ADT/SetVector.h"
|
|
|
|
#include "llvm/ADT/SmallBitVector.h"
|
|
|
|
#include "mlir/IR/Block.h"
|
|
|
|
#include "mlir/IR/Builders.h"
|
|
|
|
#include "mlir/IR/Function.h"
|
|
|
|
#include "mlir/IR/IntegerSet.h"
|
|
|
|
#include "mlir/IR/Matchers.h"
|
|
|
|
#include "mlir/IR/OpImplementation.h"
|
|
|
|
#include "mlir/IR/PatternMatch.h"
|
|
|
|
|
|
|
|
#include "knl_ops.hpp"
|
|
|
|
|
|
|
|
namespace mlir {
|
|
|
|
KnlOpsDialect::KnlOpsDialect(MLIRContext* context)
|
|
|
|
: Dialect(getDialectNamespace(), context) {
|
|
|
|
addOperations<
|
|
|
|
#define GET_OP_LIST
|
2019-11-02 05:09:48 +08:00
|
|
|
#include "src/compiler/knl.cpp.inc"
|
2019-10-09 07:25:59 +08:00
|
|
|
>();
|
|
|
|
}
|
|
|
|
} // namespace mlir
|
|
|
|
|
|
|
|
namespace onnf {}
|