NiccoloN
17 lines
700 B
C++
17 lines
700 B
C++
#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);
|
|
mlir::LogicalResult withScalarCoreFromBatchLanes(pim::PimCoreBatchOp coreBatchOp,
|
|
llvm::ArrayRef<unsigned> lanes,
|
|
llvm::function_ref<mlir::LogicalResult(pim::PimCoreOp)> callback);
|
|
|
|
} // namespace onnx_mlir
|