NiccoloN
33 lines
648 B
C++
33 lines
648 B
C++
#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
|