#pragma once

#include <optional>

#include "mlir/IR/PatternMatch.h"
#include "mlir/Support/LogicalResult.h"

#include "src/Accelerators/PIM/Dialect/Spatial/SpatialOps.hpp"

namespace onnx_mlir {

mlir::FailureOr<mlir::Value>
lowerSelectedConv2DPlan(spatial::SpatConv2DPlanOp planOp,
                        std::optional<mlir::Value> rowStripInput,
                        bool emitRowStripLayout,
                        mlir::PatternRewriter& rewriter);

mlir::LogicalResult canLowerConvPlanToRowStrip(spatial::SpatConv2DPlanOp planOp);
mlir::LogicalResult canConsumeAndProduceRowStrip(spatial::SpatConv2DPlanOp planOp);

} // namespace onnx_mlir