LinalgToSDFGTranslator.h

Go to the documentation of this file.

#pragma once
 
#include <llvm/Support/LogicalResult.h>
#include "mlir/Dialect/Linalg/IR/Linalg.h"
#include "mlir/IR/Operation.h"
#include "mlir/Support/LLVM.h"
#include "mlir/Target/SDFG/SDFGTranslator.h"
 
namespace mlir {
namespace sdfg {
 
LogicalResult translateLinalgOp(SDFGTranslator& translator, Operation* op);
 
LogicalResult translateLinalgFillOp(SDFGTranslator& translator, linalg::FillOp* op);
 
LogicalResult translateLinalgMatmulOp(SDFGTranslator& translator, linalg::MatmulOp* op);
 
LogicalResult translateLinalgBatchMatmulOp(SDFGTranslator& translator, linalg::BatchMatmulOp* op);
 
LogicalResult translateLinalgTransposeOp(SDFGTranslator& translator, linalg::TransposeOp* op);
 
LogicalResult translateLinalgBroadcastOp(SDFGTranslator& translator, linalg::BroadcastOp* op);
 
LogicalResult translateLinalgConv2DNchwFchwOp(SDFGTranslator& translator, linalg::Conv2DNchwFchwOp* op);
 
LogicalResult translateLinalgDepthwiseConv2DNchwChwOp(SDFGTranslator& translator, linalg::DepthwiseConv2DNchwChwOp* op);
 
} // namespace sdfg
} // namespace mlir