mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 20:25:34 -05:00
9f3615513b
This adds a new dialect called "SDFG" for data flow graphs. An SDFG data flow graph is composed of a set of processes, connected through data streams. Special streams allow for data to be injected into and to be retrieved from the data flow graph. The dialect is intended to be lowered to API calls that allow for offloading of the graph on hardware accelerators.
92 lines
2.7 KiB
C++
92 lines
2.7 KiB
C++
// Part of the Concrete Compiler Project, under the BSD3 License with Zama
|
|
// Exceptions. See
|
|
// https://github.com/zama-ai/concrete-compiler-internal/blob/main/LICENSE.txt
|
|
// for license information.
|
|
|
|
#include "mlir/IR/Builders.h"
|
|
|
|
#include "concretelang/Dialect/SDFG/IR/SDFGOps.h"
|
|
#include "concretelang/Dialect/SDFG/IR/SDFGTypes.h"
|
|
|
|
#include "concretelang/Dialect/SDFG/IR/SDFGEnums.cpp.inc"
|
|
#include <mlir/Support/LogicalResult.h>
|
|
|
|
#define GET_OP_CLASSES
|
|
#include "concretelang/Dialect/SDFG/IR/SDFGOps.cpp.inc"
|
|
|
|
namespace mlir {
|
|
namespace concretelang {
|
|
namespace SDFG {
|
|
mlir::LogicalResult Put::verify() {
|
|
mlir::Type streamElementType =
|
|
stream().getType().cast<StreamType>().getElementType();
|
|
mlir::Type elementType = data().getType();
|
|
|
|
if (streamElementType != elementType) {
|
|
emitError()
|
|
<< "The type " << elementType
|
|
<< " of the element to be written does not match the element type "
|
|
<< streamElementType << " of the stream.";
|
|
return mlir::failure();
|
|
}
|
|
|
|
return mlir::success();
|
|
}
|
|
|
|
mlir::LogicalResult MakeProcess::checkStreams(size_t numIn, size_t numOut) {
|
|
mlir::OperandRange streams = this->streams();
|
|
|
|
if (streams.size() != numIn + numOut) {
|
|
emitError() << "Process `" << stringifyProcessKind(type())
|
|
<< "` expects 3 streams, but " << streams.size()
|
|
<< " were given.";
|
|
return mlir::failure();
|
|
}
|
|
|
|
for (size_t i = 0; i < numIn; i++) {
|
|
MakeStream in = dyn_cast_or_null<MakeStream>(streams[i].getDefiningOp());
|
|
|
|
if (in && !in.createsInputStream()) {
|
|
emitError() << "Stream #" << (i + 1) << " of process `"
|
|
<< stringifyProcessKind(type())
|
|
<< "` must be an input stream.";
|
|
return mlir::failure();
|
|
}
|
|
}
|
|
|
|
for (size_t i = numIn; i < numIn + numOut; i++) {
|
|
MakeStream out = dyn_cast_or_null<MakeStream>(streams[i].getDefiningOp());
|
|
|
|
if (out && !out.createsOutputStream()) {
|
|
emitError() << "Stream #" << (i + 1) << " of process `"
|
|
<< stringifyProcessKind(type())
|
|
<< "` must be an output stream.";
|
|
return mlir::failure();
|
|
}
|
|
}
|
|
|
|
return mlir::success();
|
|
}
|
|
|
|
mlir::LogicalResult MakeProcess::verify() {
|
|
switch (type()) {
|
|
case ProcessKind::add_eint:
|
|
return checkStreams(2, 1);
|
|
case ProcessKind::add_eint_int:
|
|
return checkStreams(2, 1);
|
|
case ProcessKind::mul_eint_int:
|
|
return checkStreams(2, 1);
|
|
case ProcessKind::neg_eint:
|
|
return checkStreams(1, 1);
|
|
case ProcessKind::keyswitch:
|
|
return checkStreams(1, 1);
|
|
case ProcessKind::bootstrap:
|
|
return checkStreams(2, 1);
|
|
}
|
|
|
|
return mlir::failure();
|
|
}
|
|
} // namespace SDFG
|
|
} // namespace concretelang
|
|
} // namespace mlir
|