NiccoloN
909c4acfdd
Validate Operations / validate-operations (push) Has been cancelled
remove Spatial many ops in favor of tensor ops like in pim
14 lines
421 B
C++
14 lines
421 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);
|
|
|
|
} // namespace onnx_mlir
|