mirror of https://github.com/llvm/torch-mlir
[RefBackend] Rename RefBackend dialect to Refback
I now realize that VerboseCamelCase is not the best choice for dialect directory/file names and C++ identifiers (take e.g. "Linalg", "Basicpy", etc. as prior art here; not LinearAlgebra or BasicPython). If I had to name the convention it seems to be "Shortword" (or of course just acronym dialects like LLVM, SCF, etc.). This rename also has the side benefit of differentiating RefBackend directories, which now refer to the actual backend itself, from Refback/Refbackrt, which are the dialects which happen to be used by that backend.pull/73/head
parent
bf99a82832
commit
7edb5f3641
|
@ -1,7 +1,7 @@
|
||||||
add_subdirectory(ATen)
|
add_subdirectory(ATen)
|
||||||
add_subdirectory(Basicpy)
|
add_subdirectory(Basicpy)
|
||||||
add_subdirectory(Numpy)
|
add_subdirectory(Numpy)
|
||||||
add_subdirectory(RefBackend)
|
add_subdirectory(Refback)
|
||||||
add_subdirectory(Refbackrt)
|
add_subdirectory(Refbackrt)
|
||||||
add_subdirectory(TCF)
|
add_subdirectory(TCF)
|
||||||
add_subdirectory(TCP)
|
add_subdirectory(TCP)
|
||||||
|
|
|
@ -1 +0,0 @@
|
||||||
add_mlir_dialect(RefBackendOps refback)
|
|
|
@ -0,0 +1 @@
|
||||||
|
add_mlir_dialect(RefbackOps refback)
|
|
@ -6,12 +6,12 @@
|
||||||
//
|
//
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
|
|
||||||
#ifndef REFBACKEND_BASE
|
#ifndef REFBACK_BASE
|
||||||
#define REFBACKEND_BASE
|
#define REFBACK_BASE
|
||||||
|
|
||||||
include "mlir/IR/OpBase.td"
|
include "mlir/IR/OpBase.td"
|
||||||
|
|
||||||
def RefBackend_Dialect : Dialect {
|
def Refback_Dialect : Dialect {
|
||||||
let name = "refback";
|
let name = "refback";
|
||||||
let cppNamespace = "::mlir::NPCOMP::refback";
|
let cppNamespace = "::mlir::NPCOMP::refback";
|
||||||
let description = [{
|
let description = [{
|
||||||
|
@ -20,4 +20,4 @@ def RefBackend_Dialect : Dialect {
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
#endif // REFBACKEND_BASE
|
#endif // REFBACK_BASE
|
|
@ -6,11 +6,11 @@
|
||||||
//
|
//
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
|
|
||||||
#ifndef NPCOMP_DIALECT_REFBACKEND_IR_REFBACKENDDIALECT_H
|
#ifndef NPCOMP_DIALECT_REFBACK_IR_REFBACKDIALECT_H
|
||||||
#define NPCOMP_DIALECT_REFBACKEND_IR_REFBACKENDDIALECT_H
|
#define NPCOMP_DIALECT_REFBACK_IR_REFBACKDIALECT_H
|
||||||
|
|
||||||
#include "mlir/IR/Dialect.h"
|
#include "mlir/IR/Dialect.h"
|
||||||
|
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOpsDialect.h.inc"
|
#include "npcomp/Dialect/Refback/IR/RefbackOpsDialect.h.inc"
|
||||||
|
|
||||||
#endif // NPCOMP_DIALECT_REFBACKEND_IR_REFBACKENDDIALECT_H
|
#endif // NPCOMP_DIALECT_REFBACK_IR_REFBACKDIALECT_H
|
|
@ -6,8 +6,8 @@
|
||||||
//
|
//
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
|
|
||||||
#ifndef NPCOMP_DIALECT_REFBACKEND_IR_REFBACKENDOPS_H
|
#ifndef NPCOMP_DIALECT_REFBACK_IR_REFBACKOPS_H
|
||||||
#define NPCOMP_DIALECT_REFBACKEND_IR_REFBACKENDOPS_H
|
#define NPCOMP_DIALECT_REFBACK_IR_REFBACKOPS_H
|
||||||
|
|
||||||
#include "mlir/Dialect/Shape/IR/Shape.h"
|
#include "mlir/Dialect/Shape/IR/Shape.h"
|
||||||
#include "mlir/IR/OpDefinition.h"
|
#include "mlir/IR/OpDefinition.h"
|
||||||
|
@ -17,6 +17,6 @@
|
||||||
#include "mlir/Interfaces/SideEffectInterfaces.h"
|
#include "mlir/Interfaces/SideEffectInterfaces.h"
|
||||||
|
|
||||||
#define GET_OP_CLASSES
|
#define GET_OP_CLASSES
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.h.inc"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.h.inc"
|
||||||
|
|
||||||
#endif // NPCOMP_DIALECT_REFBACKEND_IR_REFBACKENDOPS_H
|
#endif // NPCOMP_DIALECT_REFBACK_IR_REFBACKOPS_H
|
|
@ -6,21 +6,21 @@
|
||||||
//
|
//
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
|
|
||||||
#ifndef REFBACKEND_OPS
|
#ifndef REFBACK_OPS
|
||||||
#define REFBACKEND_OPS
|
#define REFBACK_OPS
|
||||||
|
|
||||||
include "npcomp/Dialect/RefBackend/IR/RefBackendBase.td"
|
include "npcomp/Dialect/Refback/IR/RefbackBase.td"
|
||||||
include "mlir/Dialect/Shape/IR/ShapeBase.td"
|
include "mlir/Dialect/Shape/IR/ShapeBase.td"
|
||||||
include "mlir/Interfaces/SideEffectInterfaces.td"
|
include "mlir/Interfaces/SideEffectInterfaces.td"
|
||||||
include "mlir/Interfaces/InferTypeOpInterface.td"
|
include "mlir/Interfaces/InferTypeOpInterface.td"
|
||||||
include "mlir/Interfaces/ControlFlowInterfaces.td"
|
include "mlir/Interfaces/ControlFlowInterfaces.td"
|
||||||
include "mlir/IR/SymbolInterfaces.td"
|
include "mlir/IR/SymbolInterfaces.td"
|
||||||
|
|
||||||
class RefBackend_Op<string mnemonic, list<OpTrait> traits = []>
|
class Refback_Op<string mnemonic, list<OpTrait> traits = []>
|
||||||
: Op<RefBackend_Dialect, mnemonic, traits> {
|
: Op<Refback_Dialect, mnemonic, traits> {
|
||||||
}
|
}
|
||||||
|
|
||||||
def RefBackend_GlobalOp : RefBackend_Op<"global", [Symbol]> {
|
def Refback_GlobalOp : Refback_Op<"global", [Symbol]> {
|
||||||
let summary = "Represents a global variable";
|
let summary = "Represents a global variable";
|
||||||
let description = [{
|
let description = [{
|
||||||
Represents a global variable.
|
Represents a global variable.
|
||||||
|
@ -38,11 +38,11 @@ def RefBackend_GlobalOp : RefBackend_Op<"global", [Symbol]> {
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
// Ops related to tensor->memref conversion.
|
// Ops related to tensor->memref conversion.
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
// TODO: These ops probably belong in a "RefBackend on memrefs" dialect analogous
|
// TODO: These ops probably belong in a "Refback on memrefs" dialect analogous
|
||||||
// to `lmhlo`
|
// to `lmhlo`
|
||||||
|
|
||||||
// TODO: Use TypesMatchWith to verify this better.
|
// TODO: Use TypesMatchWith to verify this better.
|
||||||
def RefBackend_TensorToMemrefOp : RefBackend_Op<"tensor_to_memref", [NoSideEffect]> {
|
def Refback_TensorToMemrefOp : Refback_Op<"tensor_to_memref", [NoSideEffect]> {
|
||||||
let summary = "Converts a tensor to a memref";
|
let summary = "Converts a tensor to a memref";
|
||||||
let description = [{
|
let description = [{
|
||||||
This op is used to materialize conversions to allow incremental lowering of
|
This op is used to materialize conversions to allow incremental lowering of
|
||||||
|
@ -55,7 +55,7 @@ def RefBackend_TensorToMemrefOp : RefBackend_Op<"tensor_to_memref", [NoSideEffec
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: Use TypesMatchWith to verify this better.
|
// TODO: Use TypesMatchWith to verify this better.
|
||||||
def RefBackend_MemrefToTensorOp : RefBackend_Op<"memref_to_tensor", [NoSideEffect]> {
|
def Refback_MemrefToTensorOp : Refback_Op<"memref_to_tensor", [NoSideEffect]> {
|
||||||
let summary = "Converts a memref to a tensor";
|
let summary = "Converts a memref to a tensor";
|
||||||
let description = [{
|
let description = [{
|
||||||
This op is used to materialize conversions to allow incremental lowering of
|
This op is used to materialize conversions to allow incremental lowering of
|
||||||
|
@ -66,7 +66,7 @@ def RefBackend_MemrefToTensorOp : RefBackend_Op<"memref_to_tensor", [NoSideEffec
|
||||||
let assemblyFormat = "attr-dict $memref `:` type($memref) `->` type($tensor)";
|
let assemblyFormat = "attr-dict $memref `:` type($memref) `->` type($tensor)";
|
||||||
}
|
}
|
||||||
|
|
||||||
def RefBackend_AllocMemRefOp : RefBackend_Op<"alloc_memref", []> {
|
def Refback_AllocMemRefOp : Refback_Op<"alloc_memref", []> {
|
||||||
let summary = "Allocates a memref of the given shape.";
|
let summary = "Allocates a memref of the given shape.";
|
||||||
let description = [{
|
let description = [{
|
||||||
Allocates a memref of the given shape.
|
Allocates a memref of the given shape.
|
||||||
|
@ -79,7 +79,7 @@ def RefBackend_AllocMemRefOp : RefBackend_Op<"alloc_memref", []> {
|
||||||
let assemblyFormat = "$shape attr-dict `:` type($memref)";
|
let assemblyFormat = "$shape attr-dict `:` type($memref)";
|
||||||
}
|
}
|
||||||
|
|
||||||
def RefBackend_GetGlobalMemrefOp : RefBackend_Op<"get_global_memref"> {
|
def Refback_GetGlobalMemrefOp : Refback_Op<"get_global_memref"> {
|
||||||
let summary = "Obtain a memref pointing at the given global";
|
let summary = "Obtain a memref pointing at the given global";
|
||||||
let description = [{
|
let description = [{
|
||||||
Obtain a memref pointing at the given global.
|
Obtain a memref pointing at the given global.
|
||||||
|
@ -95,7 +95,7 @@ def RefBackend_GetGlobalMemrefOp : RefBackend_Op<"get_global_memref"> {
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
// TODO: These belong in a shape-related dialect.
|
// TODO: These belong in a shape-related dialect.
|
||||||
|
|
||||||
def RefBackend_ShapedResultsOp : RefBackend_Op<"shaped_results", [
|
def Refback_ShapedResultsOp : Refback_Op<"shaped_results", [
|
||||||
DeclareOpInterfaceMethods<RegionBranchOpInterface>,
|
DeclareOpInterfaceMethods<RegionBranchOpInterface>,
|
||||||
SingleBlockImplicitTerminator<"YieldOp">,
|
SingleBlockImplicitTerminator<"YieldOp">,
|
||||||
RecursiveSideEffects,
|
RecursiveSideEffects,
|
||||||
|
@ -136,12 +136,12 @@ def RefBackend_ShapedResultsOp : RefBackend_Op<"shaped_results", [
|
||||||
let parser = [{ return ::parse$cppClass(parser, result); }];
|
let parser = [{ return ::parse$cppClass(parser, result); }];
|
||||||
}
|
}
|
||||||
|
|
||||||
def RefBackend_YieldOp : RefBackend_Op<"yield", [NoSideEffect, ReturnLike, Terminator,
|
def Refback_YieldOp : Refback_Op<"yield", [NoSideEffect, ReturnLike, Terminator,
|
||||||
ParentOneOf<["ShapedResultsOp"]>]> {
|
ParentOneOf<["ShapedResultsOp"]>]> {
|
||||||
let summary = "Yield-like terminator for RefBackend dialect";
|
let summary = "Yield-like terminator for Refback dialect";
|
||||||
let description = "See scf.yield";
|
let description = "See scf.yield";
|
||||||
let arguments = (ins Variadic<AnyType>:$operands);
|
let arguments = (ins Variadic<AnyType>:$operands);
|
||||||
let assemblyFormat = "attr-dict ($operands^ `:` type($operands))?";
|
let assemblyFormat = "attr-dict ($operands^ `:` type($operands))?";
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif // REFBACKEND_OPS
|
#endif // REFBACK_OPS
|
|
@ -39,7 +39,7 @@ add_mlir_library(NPCOMPInitAll
|
||||||
PUBLIC
|
PUBLIC
|
||||||
# Local depends
|
# Local depends
|
||||||
NPCOMPRefBackend
|
NPCOMPRefBackend
|
||||||
NPCOMPRefBackendDialect
|
NPCOMPRefbackDialect
|
||||||
NPCOMPTCPDialect
|
NPCOMPTCPDialect
|
||||||
NPCOMPTCFDialect
|
NPCOMPTCFDialect
|
||||||
NPCOMPTorchDialect
|
NPCOMPTorchDialect
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
add_subdirectory(ATen)
|
add_subdirectory(ATen)
|
||||||
add_subdirectory(Basicpy)
|
add_subdirectory(Basicpy)
|
||||||
add_subdirectory(Numpy)
|
add_subdirectory(Numpy)
|
||||||
add_subdirectory(RefBackend)
|
add_subdirectory(Refback)
|
||||||
add_subdirectory(Refbackrt)
|
add_subdirectory(Refbackrt)
|
||||||
add_subdirectory(TCF)
|
add_subdirectory(TCF)
|
||||||
add_subdirectory(TCP)
|
add_subdirectory(TCP)
|
||||||
|
|
|
@ -1,19 +0,0 @@
|
||||||
add_mlir_dialect_library(NPCOMPRefBackendDialect
|
|
||||||
RefBackendDialect.cpp
|
|
||||||
RefBackendOps.cpp
|
|
||||||
|
|
||||||
ADDITIONAL_HEADER_DIRS
|
|
||||||
${PROJECT_SOURCE_DIR}/include/npcomp/Dialect/RefBackend
|
|
||||||
|
|
||||||
DEPENDS
|
|
||||||
MLIRRefBackendOpsIncGen
|
|
||||||
|
|
||||||
LINK_COMPONENTS
|
|
||||||
Core
|
|
||||||
|
|
||||||
LINK_LIBS PUBLIC
|
|
||||||
MLIRIR
|
|
||||||
MLIRSupport
|
|
||||||
MLIRSideEffectInterfaces
|
|
||||||
MLIRShape
|
|
||||||
)
|
|
|
@ -0,0 +1,19 @@
|
||||||
|
add_mlir_dialect_library(NPCOMPRefbackDialect
|
||||||
|
RefbackDialect.cpp
|
||||||
|
RefbackOps.cpp
|
||||||
|
|
||||||
|
ADDITIONAL_HEADER_DIRS
|
||||||
|
${PROJECT_SOURCE_DIR}/include/npcomp/Dialect/Refback
|
||||||
|
|
||||||
|
DEPENDS
|
||||||
|
MLIRRefbackOpsIncGen
|
||||||
|
|
||||||
|
LINK_COMPONENTS
|
||||||
|
Core
|
||||||
|
|
||||||
|
LINK_LIBS PUBLIC
|
||||||
|
MLIRIR
|
||||||
|
MLIRSupport
|
||||||
|
MLIRSideEffectInterfaces
|
||||||
|
MLIRShape
|
||||||
|
)
|
|
@ -6,19 +6,19 @@
|
||||||
//
|
//
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
|
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendDialect.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackDialect.h"
|
||||||
#include "mlir/Transforms/InliningUtils.h"
|
#include "mlir/Transforms/InliningUtils.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.h"
|
||||||
|
|
||||||
using namespace mlir;
|
using namespace mlir;
|
||||||
using namespace mlir::NPCOMP::refback;
|
using namespace mlir::NPCOMP::refback;
|
||||||
|
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
// RefBackendDialect Dialect Interfaces
|
// RefbackDialect Dialect Interfaces
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
struct RefBackendInlinerInterface : public DialectInlinerInterface {
|
struct RefbackInlinerInterface : public DialectInlinerInterface {
|
||||||
using DialectInlinerInterface::DialectInlinerInterface;
|
using DialectInlinerInterface::DialectInlinerInterface;
|
||||||
bool isLegalToInline(Region *dest, Region *src,
|
bool isLegalToInline(Region *dest, Region *src,
|
||||||
BlockAndValueMapping &valueMapping) const final {
|
BlockAndValueMapping &valueMapping) const final {
|
||||||
|
@ -41,10 +41,10 @@ struct RefBackendInlinerInterface : public DialectInlinerInterface {
|
||||||
};
|
};
|
||||||
} // end anonymous namespace
|
} // end anonymous namespace
|
||||||
|
|
||||||
void RefBackendDialect::initialize() {
|
void RefbackDialect::initialize() {
|
||||||
addOperations<
|
addOperations<
|
||||||
#define GET_OP_LIST
|
#define GET_OP_LIST
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.cpp.inc"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.cpp.inc"
|
||||||
>();
|
>();
|
||||||
addInterfaces<RefBackendInlinerInterface>();
|
addInterfaces<RefbackInlinerInterface>();
|
||||||
}
|
}
|
|
@ -6,7 +6,7 @@
|
||||||
//
|
//
|
||||||
//===----------------------------------------------------------------------===//
|
//===----------------------------------------------------------------------===//
|
||||||
|
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.h"
|
||||||
#include "mlir/Dialect/Shape/IR/Shape.h"
|
#include "mlir/Dialect/Shape/IR/Shape.h"
|
||||||
#include "mlir/IR/TypeUtilities.h"
|
#include "mlir/IR/TypeUtilities.h"
|
||||||
#include "llvm/ADT/STLExtras.h"
|
#include "llvm/ADT/STLExtras.h"
|
||||||
|
@ -123,4 +123,4 @@ static LogicalResult verifyGetGlobalMemrefOp(GetGlobalMemrefOp op) {
|
||||||
}
|
}
|
||||||
|
|
||||||
#define GET_OP_CLASSES
|
#define GET_OP_CLASSES
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.cpp.inc"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.cpp.inc"
|
|
@ -12,10 +12,10 @@
|
||||||
#include "npcomp/Dialect/ATen/ATenPasses.h"
|
#include "npcomp/Dialect/ATen/ATenPasses.h"
|
||||||
#include "npcomp/Dialect/Basicpy/IR/BasicpyDialect.h"
|
#include "npcomp/Dialect/Basicpy/IR/BasicpyDialect.h"
|
||||||
#include "npcomp/Dialect/Basicpy/Transforms/Passes.h"
|
#include "npcomp/Dialect/Basicpy/Transforms/Passes.h"
|
||||||
#include "npcomp/Dialect/Refbackrt/IR/RefbackrtDialect.h"
|
|
||||||
#include "npcomp/Dialect/Numpy/IR/NumpyDialect.h"
|
#include "npcomp/Dialect/Numpy/IR/NumpyDialect.h"
|
||||||
#include "npcomp/Dialect/Numpy/Transforms/Passes.h"
|
#include "npcomp/Dialect/Numpy/Transforms/Passes.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendDialect.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackDialect.h"
|
||||||
|
#include "npcomp/Dialect/Refbackrt/IR/RefbackrtDialect.h"
|
||||||
#include "npcomp/Dialect/TCF/IR/TCFDialect.h"
|
#include "npcomp/Dialect/TCF/IR/TCFDialect.h"
|
||||||
#include "npcomp/Dialect/TCF/Transforms/Passes.h"
|
#include "npcomp/Dialect/TCF/Transforms/Passes.h"
|
||||||
#include "npcomp/Dialect/TCP/IR/TCPDialect.h"
|
#include "npcomp/Dialect/TCP/IR/TCPDialect.h"
|
||||||
|
@ -74,7 +74,7 @@ void mlir::NPCOMP::registerAllDialects(mlir::DialectRegistry ®istry) {
|
||||||
Basicpy::BasicpyDialect,
|
Basicpy::BasicpyDialect,
|
||||||
Numpy::NumpyDialect,
|
Numpy::NumpyDialect,
|
||||||
refbackrt::RefbackrtDialect,
|
refbackrt::RefbackrtDialect,
|
||||||
refback::RefBackendDialect,
|
refback::RefbackDialect,
|
||||||
tcf::TCFDialect,
|
tcf::TCFDialect,
|
||||||
tcp::TCPDialect,
|
tcp::TCPDialect,
|
||||||
mlir::NPCOMP::Torch::TorchDialect>();
|
mlir::NPCOMP::Torch::TorchDialect>();
|
||||||
|
|
|
@ -10,8 +10,8 @@
|
||||||
#include "mlir/Dialect/Linalg/IR/LinalgOps.h"
|
#include "mlir/Dialect/Linalg/IR/LinalgOps.h"
|
||||||
#include "mlir/Dialect/Shape/IR/Shape.h"
|
#include "mlir/Dialect/Shape/IR/Shape.h"
|
||||||
#include "mlir/Dialect/StandardOps/IR/Ops.h"
|
#include "mlir/Dialect/StandardOps/IR/Ops.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendDialect.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackDialect.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.h"
|
||||||
#include "npcomp/Dialect/TCP/IR/TCPOps.h"
|
#include "npcomp/Dialect/TCP/IR/TCPOps.h"
|
||||||
#include "npcomp/RefBackend/RefBackend.h"
|
#include "npcomp/RefBackend/RefBackend.h"
|
||||||
|
|
||||||
|
@ -49,7 +49,7 @@ namespace {
|
||||||
// to be lowered by LowerShapedResults.
|
// to be lowered by LowerShapedResults.
|
||||||
class BypassShapes : public BypassShapesBase<BypassShapes> {
|
class BypassShapes : public BypassShapesBase<BypassShapes> {
|
||||||
void getDependentDialects(DialectRegistry ®istry) const override {
|
void getDependentDialects(DialectRegistry ®istry) const override {
|
||||||
registry.insert<shape::ShapeDialect, refback::RefBackendDialect>();
|
registry.insert<shape::ShapeDialect, refback::RefbackDialect>();
|
||||||
}
|
}
|
||||||
|
|
||||||
void runOnOperation() override {
|
void runOnOperation() override {
|
||||||
|
|
|
@ -14,9 +14,9 @@
|
||||||
#include "mlir/IR/Verifier.h"
|
#include "mlir/IR/Verifier.h"
|
||||||
#include "mlir/Transforms/DialectConversion.h"
|
#include "mlir/Transforms/DialectConversion.h"
|
||||||
|
|
||||||
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.h"
|
||||||
#include "npcomp/Dialect/Refbackrt/IR/RefbackrtDialect.h"
|
#include "npcomp/Dialect/Refbackrt/IR/RefbackrtDialect.h"
|
||||||
#include "npcomp/Dialect/Refbackrt/IR/RefbackrtOps.h"
|
#include "npcomp/Dialect/Refbackrt/IR/RefbackrtOps.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.h"
|
|
||||||
|
|
||||||
using namespace mlir;
|
using namespace mlir;
|
||||||
using namespace mlir::NPCOMP;
|
using namespace mlir::NPCOMP;
|
||||||
|
|
|
@ -27,7 +27,7 @@
|
||||||
#include "mlir/Transforms/DialectConversion.h"
|
#include "mlir/Transforms/DialectConversion.h"
|
||||||
#include "mlir/Transforms/Passes.h"
|
#include "mlir/Transforms/Passes.h"
|
||||||
#include "npcomp/Conversion/TCFToTCP/TCFToTCP.h"
|
#include "npcomp/Conversion/TCFToTCP/TCFToTCP.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.h"
|
||||||
#include "npcomp/Dialect/TCP/IR/TCPDialect.h"
|
#include "npcomp/Dialect/TCP/IR/TCPDialect.h"
|
||||||
#include "npcomp/Dialect/TCP/IR/TCPOps.h"
|
#include "npcomp/Dialect/TCP/IR/TCPOps.h"
|
||||||
|
|
||||||
|
|
|
@ -17,8 +17,8 @@
|
||||||
#include "mlir/Pass/Pass.h"
|
#include "mlir/Pass/Pass.h"
|
||||||
#include "mlir/Pass/PassRegistry.h"
|
#include "mlir/Pass/PassRegistry.h"
|
||||||
#include "mlir/Transforms/DialectConversion.h"
|
#include "mlir/Transforms/DialectConversion.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendDialect.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackDialect.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.h"
|
||||||
|
|
||||||
using namespace mlir;
|
using namespace mlir;
|
||||||
using namespace mlir::NPCOMP;
|
using namespace mlir::NPCOMP;
|
||||||
|
@ -81,7 +81,7 @@ namespace {
|
||||||
class LowerConstantTensorsToMemref
|
class LowerConstantTensorsToMemref
|
||||||
: public LowerConstantTensorsToMemrefBase<LowerConstantTensorsToMemref> {
|
: public LowerConstantTensorsToMemrefBase<LowerConstantTensorsToMemref> {
|
||||||
void getDependentDialects(DialectRegistry ®istry) const override {
|
void getDependentDialects(DialectRegistry ®istry) const override {
|
||||||
registry.insert<refback::RefBackendDialect>();
|
registry.insert<refback::RefbackDialect>();
|
||||||
}
|
}
|
||||||
|
|
||||||
void runOnOperation() override {
|
void runOnOperation() override {
|
||||||
|
|
|
@ -19,7 +19,7 @@
|
||||||
#include "mlir/Transforms/DialectConversion.h"
|
#include "mlir/Transforms/DialectConversion.h"
|
||||||
#include "mlir/Transforms/InliningUtils.h"
|
#include "mlir/Transforms/InliningUtils.h"
|
||||||
#include "npcomp/Conversion/TCFToTCP/TCFToTCP.h"
|
#include "npcomp/Conversion/TCFToTCP/TCFToTCP.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.h"
|
||||||
#include "npcomp/Dialect/TCP/IR/TCPDialect.h"
|
#include "npcomp/Dialect/TCP/IR/TCPDialect.h"
|
||||||
#include "npcomp/Dialect/TCP/IR/TCPOps.h"
|
#include "npcomp/Dialect/TCP/IR/TCPOps.h"
|
||||||
|
|
||||||
|
|
|
@ -13,8 +13,8 @@
|
||||||
#include "mlir/Pass/Pass.h"
|
#include "mlir/Pass/Pass.h"
|
||||||
#include "mlir/Pass/PassRegistry.h"
|
#include "mlir/Pass/PassRegistry.h"
|
||||||
#include "mlir/Transforms/DialectConversion.h"
|
#include "mlir/Transforms/DialectConversion.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendDialect.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackDialect.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.h"
|
||||||
|
|
||||||
using namespace mlir;
|
using namespace mlir;
|
||||||
using namespace mlir::NPCOMP;
|
using namespace mlir::NPCOMP;
|
||||||
|
@ -88,7 +88,7 @@ namespace {
|
||||||
// TODO: Upstream this.
|
// TODO: Upstream this.
|
||||||
class LowerStdToMemref : public LowerStdToMemrefBase<LowerStdToMemref> {
|
class LowerStdToMemref : public LowerStdToMemrefBase<LowerStdToMemref> {
|
||||||
void getDependentDialects(DialectRegistry ®istry) const override {
|
void getDependentDialects(DialectRegistry ®istry) const override {
|
||||||
registry.insert<refback::RefBackendDialect>();
|
registry.insert<refback::RefbackDialect>();
|
||||||
}
|
}
|
||||||
|
|
||||||
void runOnOperation() override {
|
void runOnOperation() override {
|
||||||
|
|
|
@ -14,7 +14,7 @@
|
||||||
#include "mlir/Pass/Pass.h"
|
#include "mlir/Pass/Pass.h"
|
||||||
#include "mlir/Pass/PassRegistry.h"
|
#include "mlir/Pass/PassRegistry.h"
|
||||||
#include "mlir/Transforms/DialectConversion.h"
|
#include "mlir/Transforms/DialectConversion.h"
|
||||||
#include "npcomp/Dialect/RefBackend/IR/RefBackendOps.h"
|
#include "npcomp/Dialect/Refback/IR/RefbackOps.h"
|
||||||
|
|
||||||
using namespace mlir;
|
using namespace mlir;
|
||||||
using namespace mlir::NPCOMP;
|
using namespace mlir::NPCOMP;
|
||||||
|
|
Loading…
Reference in New Issue