#pragma once

#include "llvm/ADT/STLFunctionalExtras.h"

#include "src/Accelerators/PIM/Dialect/Pim/PimOps.hpp"

namespace onnx_mlir {

mlir::LogicalResult withScalarCoreFromBatchLane(pim::PimCoreBatchOp coreBatchOp,
                                                unsigned lane,
                                                llvm::function_ref<mlir::LogicalResult(pim::PimCoreOp)> callback);

} // namespace onnx_mlir