This commit is contained in:
@@ -0,0 +1,27 @@
|
||||
#pragma once
|
||||
|
||||
#include "mlir/Support/LogicalResult.h"
|
||||
|
||||
#include <cstddef>
|
||||
|
||||
namespace onnx_mlir::spatial {
|
||||
|
||||
struct MatrixUnitShape {
|
||||
size_t rows = 0;
|
||||
size_t columns = 0;
|
||||
};
|
||||
|
||||
struct SpatialTargetResources {
|
||||
MatrixUnitShape matrixShape;
|
||||
size_t matrixUnitsPerProcessor = 0;
|
||||
size_t processorCount = 0;
|
||||
size_t vectorWidth = 0;
|
||||
|
||||
mlir::LogicalResult verify() const {
|
||||
return mlir::success(matrixShape.rows != 0 && matrixShape.columns != 0
|
||||
&& matrixUnitsPerProcessor != 0 && processorCount != 0
|
||||
&& vectorWidth != 0);
|
||||
}
|
||||
};
|
||||
|
||||
} // namespace onnx_mlir::spatial
|
||||
Reference in New Issue
Block a user