mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-10 04:35:03 -05:00
3ce7c96f3f
This adds a new operation interface that allows an operation to specify that a batched version of the operation exists that applies it on the elements of a flat tensor in parallel.
53 lines
1.6 KiB
TableGen
53 lines
1.6 KiB
TableGen
#ifndef CONCRETELANG_INTERFACES_BATCHABLEINTERFACE
|
|
#define CONCRETELANG_INTERFACES_BATCHABLEINTERFACE
|
|
|
|
include "mlir/IR/OpBase.td"
|
|
|
|
def BatchableOpInterface : OpInterface<"BatchableOpInterface"> {
|
|
let description = [{
|
|
Interface for operations processing a scalar that can be batched
|
|
if invoked multiple times with different, independent operands.
|
|
}];
|
|
let cppNamespace = "::mlir::concretelang";
|
|
|
|
let methods = [
|
|
InterfaceMethod<[{
|
|
Return the scalar operand that can be batched in a tensor to
|
|
be passed to the corresponding batched operation.
|
|
}],
|
|
/*retTy=*/"::mlir::OpOperand&",
|
|
/*methodName=*/"getBatchableOperand",
|
|
/*args=*/(ins),
|
|
/*methodBody=*/"",
|
|
/*defaultImplementation=*/[{
|
|
llvm_unreachable("getBatchableOperand not implemented");
|
|
}]
|
|
>,
|
|
InterfaceMethod<[{
|
|
Return all operands that cannot be batched.
|
|
}],
|
|
/*retTy=*/"::mlir::OperandRange",
|
|
/*methodName=*/"getNonBatchableOperands",
|
|
/*args=*/(ins),
|
|
/*methodBody=*/"",
|
|
/*defaultImplementation=*/[{
|
|
llvm_unreachable("getNonBatchableOperands not implemented");
|
|
}]
|
|
>,
|
|
InterfaceMethod<[{
|
|
Create the batched operation and return it as a value.
|
|
}],
|
|
/*retTy=*/"::mlir::Value",
|
|
/*methodName=*/"createBatchedOperation",
|
|
/*args=*/(ins "::mlir::ImplicitLocOpBuilder&":$builder,
|
|
"::mlir::Value":$batchedOperands),
|
|
/*methodBody=*/"",
|
|
/*defaultImplementation=*/[{
|
|
llvm_unreachable("createBatchedOperation not implemented");
|
|
}]
|
|
>
|
|
];
|
|
}
|
|
|
|
#endif // CONCRETELANG_INTERFACES_BATCHABLEINTERFACE
|