#pragma once

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

#include <cstdint>

namespace onnx_mlir::spatial {

enum class ConvLoweringStrategy : uint8_t {
  Auto,
  Legacy,
  Depthwise,
  PackedIm2Col,
  StreamedPatch,
  StreamedPacked,
  OutputChannelTiled,
  InputKTiled,
  Tiled2D,
};

} // namespace onnx_mlir::spatial

namespace onnx_mlir {

struct ONNXToSpatialPlanningOptions {
  uint64_t convIm2colMaxElements = 0;
  uint64_t convStreamChunkPositions = 0;
  spatial::ConvLoweringStrategy forcedConvStrategy = spatial::ConvLoweringStrategy::Auto;
  bool reportConvLowering = true;
};

} // namespace onnx_mlir