mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-08 03:25:05 -05:00
fix(compiler): Annotate FHELinalg as Pure for canonicalization
This commit is contained in:
Bourgerie Quentin
committed by
Quentin Bourgerie
Quentin Bourgerie
parent
21474cc75d
commit
5a80e22cf3
@@ -26,7 +26,7 @@ def TensorBinaryEint : NativeOpTrait<"TensorBinaryEint">;
|
||||
def TensorUnaryEint : NativeOpTrait<"TensorUnaryEint">;
|
||||
|
||||
|
||||
def FHELinalg_AddEintIntOp : FHELinalg_Op<"add_eint_int", [TensorBroadcastingRules, TensorBinaryEintInt]> {
|
||||
def FHELinalg_AddEintIntOp : FHELinalg_Op<"add_eint_int", [Pure, TensorBroadcastingRules, TensorBinaryEintInt]> {
|
||||
let summary = "Returns a tensor that contains the addition of a tensor of encrypted integers and a tensor of clear integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -81,7 +81,7 @@ def FHELinalg_AddEintIntOp : FHELinalg_Op<"add_eint_int", [TensorBroadcastingRul
|
||||
let hasFolder = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_AddEintOp : FHELinalg_Op<"add_eint", [TensorBroadcastingRules, TensorBinaryEint]> {
|
||||
def FHELinalg_AddEintOp : FHELinalg_Op<"add_eint", [Pure, TensorBroadcastingRules, TensorBinaryEint]> {
|
||||
let summary = "Returns a tensor that contains the addition of two tensor of encrypted integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -133,7 +133,7 @@ def FHELinalg_AddEintOp : FHELinalg_Op<"add_eint", [TensorBroadcastingRules, Ten
|
||||
];
|
||||
}
|
||||
|
||||
def FHELinalg_SubIntEintOp : FHELinalg_Op<"sub_int_eint", [TensorBroadcastingRules, TensorBinaryIntEint]> {
|
||||
def FHELinalg_SubIntEintOp : FHELinalg_Op<"sub_int_eint", [Pure, TensorBroadcastingRules, TensorBinaryIntEint]> {
|
||||
let summary = "Returns a tensor that contains the subtraction of a tensor of clear integers and a tensor of encrypted integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -186,7 +186,7 @@ def FHELinalg_SubIntEintOp : FHELinalg_Op<"sub_int_eint", [TensorBroadcastingRul
|
||||
];
|
||||
}
|
||||
|
||||
def FHELinalg_SubEintIntOp : FHELinalg_Op<"sub_eint_int", [TensorBroadcastingRules, TensorBinaryEintInt]> {
|
||||
def FHELinalg_SubEintIntOp : FHELinalg_Op<"sub_eint_int", [Pure, TensorBroadcastingRules, TensorBinaryEintInt]> {
|
||||
let summary = "Returns a tensor that contains the subtraction of a tensor of clear integers from a tensor of encrypted integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -242,7 +242,7 @@ def FHELinalg_SubEintIntOp : FHELinalg_Op<"sub_eint_int", [TensorBroadcastingRul
|
||||
let hasFolder = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_SubEintOp : FHELinalg_Op<"sub_eint", [TensorBroadcastingRules, TensorBinaryEint]> {
|
||||
def FHELinalg_SubEintOp : FHELinalg_Op<"sub_eint", [Pure, TensorBroadcastingRules, TensorBinaryEint]> {
|
||||
let summary = "Returns a tensor that contains the subtraction of two tensor of encrypted integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -294,7 +294,7 @@ def FHELinalg_SubEintOp : FHELinalg_Op<"sub_eint", [TensorBroadcastingRules, Ten
|
||||
];
|
||||
}
|
||||
|
||||
def FHELinalg_NegEintOp : FHELinalg_Op<"neg_eint", [TensorUnaryEint]> {
|
||||
def FHELinalg_NegEintOp : FHELinalg_Op<"neg_eint", [Pure, TensorUnaryEint]> {
|
||||
let summary = "Returns a tensor that contains the negation of a tensor of encrypted integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -326,7 +326,7 @@ def FHELinalg_NegEintOp : FHELinalg_Op<"neg_eint", [TensorUnaryEint]> {
|
||||
];
|
||||
}
|
||||
|
||||
def FHELinalg_MulEintIntOp : FHELinalg_Op<"mul_eint_int", [TensorBroadcastingRules, TensorBinaryEintInt]> {
|
||||
def FHELinalg_MulEintIntOp : FHELinalg_Op<"mul_eint_int", [Pure, TensorBroadcastingRules, TensorBinaryEintInt]> {
|
||||
let summary = "Returns a tensor that contains the multiplication of a tensor of encrypted integers and a tensor of clear integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -377,7 +377,7 @@ def FHELinalg_MulEintIntOp : FHELinalg_Op<"mul_eint_int", [TensorBroadcastingRul
|
||||
let hasCanonicalizer = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_MulEintOp : FHELinalg_Op<"mul_eint", [TensorBroadcastingRules, TensorBinaryEint]> {
|
||||
def FHELinalg_MulEintOp : FHELinalg_Op<"mul_eint", [Pure, TensorBroadcastingRules, TensorBinaryEint]> {
|
||||
let summary = "Returns a tensor that contains the multiplication of two tensor of encrypted integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -429,7 +429,7 @@ def FHELinalg_MulEintOp : FHELinalg_Op<"mul_eint", [TensorBroadcastingRules, Ten
|
||||
];
|
||||
}
|
||||
|
||||
def FHELinalg_ApplyLookupTableEintOp : FHELinalg_Op<"apply_lookup_table", []> {
|
||||
def FHELinalg_ApplyLookupTableEintOp : FHELinalg_Op<"apply_lookup_table", [Pure]> {
|
||||
let summary = "Returns a tensor that contains the result of the lookup on a table.";
|
||||
|
||||
let description = [{
|
||||
@@ -465,7 +465,7 @@ def FHELinalg_ApplyLookupTableEintOp : FHELinalg_Op<"apply_lookup_table", []> {
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_ApplyMultiLookupTableEintOp : FHELinalg_Op<"apply_multi_lookup_table", []> {
|
||||
def FHELinalg_ApplyMultiLookupTableEintOp : FHELinalg_Op<"apply_multi_lookup_table", [Pure]> {
|
||||
let summary = "Returns a tensor that contains the result of the lookup on a table, using a different lookup table for each element.";
|
||||
|
||||
let description = [{
|
||||
@@ -512,7 +512,7 @@ def FHELinalg_ApplyMultiLookupTableEintOp : FHELinalg_Op<"apply_multi_lookup_tab
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_ApplyMappedLookupTableEintOp : FHELinalg_Op<"apply_mapped_lookup_table", []> {
|
||||
def FHELinalg_ApplyMappedLookupTableEintOp : FHELinalg_Op<"apply_mapped_lookup_table", [Pure]> {
|
||||
let summary = "Returns a tensor that contains the result of the lookup on a table, using a different lookup table for each element, specified by a map.";
|
||||
|
||||
let description = [{
|
||||
@@ -566,7 +566,7 @@ def FHELinalg_ApplyMappedLookupTableEintOp : FHELinalg_Op<"apply_mapped_lookup_t
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_Dot : FHELinalg_Op<"dot_eint_int"> {
|
||||
def FHELinalg_Dot : FHELinalg_Op<"dot_eint_int", [Pure]> {
|
||||
let summary = "Returns the encrypted dot product between a vector of encrypted integers and a vector of clean integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -590,7 +590,7 @@ def FHELinalg_Dot : FHELinalg_Op<"dot_eint_int"> {
|
||||
}
|
||||
|
||||
|
||||
def FHELinalg_DotEint : FHELinalg_Op<"dot_eint_eint"> {
|
||||
def FHELinalg_DotEint : FHELinalg_Op<"dot_eint_eint", [Pure]> {
|
||||
let summary = "Returns the encrypted dot product between two vectors of encrypted integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -614,7 +614,7 @@ def FHELinalg_DotEint : FHELinalg_Op<"dot_eint_eint"> {
|
||||
}
|
||||
|
||||
|
||||
def FHELinalg_MatMulEintIntOp : FHELinalg_Op<"matmul_eint_int", [TensorBinaryEintInt]> {
|
||||
def FHELinalg_MatMulEintIntOp : FHELinalg_Op<"matmul_eint_int", [Pure, TensorBinaryEintInt]> {
|
||||
let summary = "Returns a tensor that contains the result of the matrix multiplication of a matrix of encrypted integers and a matrix of clear integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -759,7 +759,7 @@ def FHELinalg_MatMulEintIntOp : FHELinalg_Op<"matmul_eint_int", [TensorBinaryEin
|
||||
}];
|
||||
}
|
||||
|
||||
def FHELinalg_MatMulIntEintOp : FHELinalg_Op<"matmul_int_eint", [TensorBinaryIntEint]> {
|
||||
def FHELinalg_MatMulIntEintOp : FHELinalg_Op<"matmul_int_eint", [Pure, TensorBinaryIntEint]> {
|
||||
let summary = "Returns a tensor that contains the result of the matrix multiplication of a matrix of clear integers and a matrix of encrypted integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -905,7 +905,7 @@ def FHELinalg_MatMulIntEintOp : FHELinalg_Op<"matmul_int_eint", [TensorBinaryInt
|
||||
}
|
||||
|
||||
|
||||
def FHELinalg_MatMulEintEintOp : FHELinalg_Op<"matmul_eint_eint", [TensorBinaryEint]> {
|
||||
def FHELinalg_MatMulEintEintOp : FHELinalg_Op<"matmul_eint_eint", [Pure, TensorBinaryEint]> {
|
||||
let summary = "Returns a tensor that contains the result of the matrix multiplication of a matrix of encrypted integers and a second matrix of encrypted integers.";
|
||||
|
||||
let description = [{
|
||||
@@ -1042,7 +1042,7 @@ def FHELinalg_MatMulEintEintOp : FHELinalg_Op<"matmul_eint_eint", [TensorBinaryE
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_SumOp : FHELinalg_Op<"sum", [TensorUnaryEint]> {
|
||||
def FHELinalg_SumOp : FHELinalg_Op<"sum", [Pure, TensorUnaryEint]> {
|
||||
let summary = "Returns the sum of elements of a tensor of encrypted integers along specified axes.";
|
||||
|
||||
let description = [{
|
||||
@@ -1125,7 +1125,7 @@ def FHELinalg_SumOp : FHELinalg_Op<"sum", [TensorUnaryEint]> {
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_ConcatOp : FHELinalg_Op<"concat"> {
|
||||
def FHELinalg_ConcatOp : FHELinalg_Op<"concat", [Pure]> {
|
||||
let summary = "Concatenates a sequence of tensors along an existing axis.";
|
||||
|
||||
let description = [{
|
||||
@@ -1167,7 +1167,7 @@ def FHELinalg_ConcatOp : FHELinalg_Op<"concat"> {
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_Conv2dOp : FHELinalg_Op<"conv2d", []> {
|
||||
def FHELinalg_Conv2dOp : FHELinalg_Op<"conv2d", [Pure]> {
|
||||
let summary = "Returns the 2D convolution of a tensor in the form NCHW with weights in the form FCHW";
|
||||
let arguments = (ins
|
||||
Type<And<[TensorOf<[FHE_AnyEncryptedInteger]>.predicate, HasStaticShapePred]>>:$input,
|
||||
@@ -1183,7 +1183,7 @@ def FHELinalg_Conv2dOp : FHELinalg_Op<"conv2d", []> {
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_Maxpool2dOp : FHELinalg_Op<"maxpool2d", []> {
|
||||
def FHELinalg_Maxpool2dOp : FHELinalg_Op<"maxpool2d", [Pure]> {
|
||||
let summary = "Returns the 2D maxpool of a tensor in the form NCHW";
|
||||
let arguments = (ins
|
||||
Type<And<[TensorOf<[FHE_AnyEncryptedInteger]>.predicate, HasStaticShapePred]>>:$input,
|
||||
@@ -1195,7 +1195,7 @@ def FHELinalg_Maxpool2dOp : FHELinalg_Op<"maxpool2d", []> {
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_TransposeOp : FHELinalg_Op<"transpose", []> {
|
||||
def FHELinalg_TransposeOp : FHELinalg_Op<"transpose", [Pure]> {
|
||||
let summary = "Returns a tensor that contains the transposition of the input tensor.";
|
||||
|
||||
let description = [{
|
||||
@@ -1234,7 +1234,7 @@ def FHELinalg_TransposeOp : FHELinalg_Op<"transpose", []> {
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_FromElementOp : FHELinalg_Op<"from_element", []> {
|
||||
def FHELinalg_FromElementOp : FHELinalg_Op<"from_element", [Pure]> {
|
||||
let summary = "Creates a tensor with a single element.";
|
||||
|
||||
let description = [{
|
||||
@@ -1251,7 +1251,7 @@ def FHELinalg_FromElementOp : FHELinalg_Op<"from_element", []> {
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_ToSignedOp : FHELinalg_Op<"to_signed", []> {
|
||||
def FHELinalg_ToSignedOp : FHELinalg_Op<"to_signed", [Pure]> {
|
||||
let summary = "Cast an unsigned integer tensor to a signed one";
|
||||
|
||||
let description = [{
|
||||
@@ -1281,7 +1281,7 @@ def FHELinalg_ToSignedOp : FHELinalg_Op<"to_signed", []> {
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_ToUnsignedOp : FHELinalg_Op<"to_unsigned", []> {
|
||||
def FHELinalg_ToUnsignedOp : FHELinalg_Op<"to_unsigned", [Pure]> {
|
||||
let summary = "Cast a signed integer tensor to an unsigned one";
|
||||
|
||||
let description = [{
|
||||
@@ -1311,7 +1311,7 @@ def FHELinalg_ToUnsignedOp : FHELinalg_Op<"to_unsigned", []> {
|
||||
let hasVerifier = 1;
|
||||
}
|
||||
|
||||
def FHELinalg_RoundOp : FHELinalg_Op<"round", [TensorUnaryEint]> {
|
||||
def FHELinalg_RoundOp : FHELinalg_Op<"round", [Pure, TensorUnaryEint]> {
|
||||
let summary = "Rounds a tensor of ciphertexts into a smaller precision.";
|
||||
|
||||
let description = [{
|
||||
|
||||
@@ -379,7 +379,7 @@ func.func @matmul_eint_int_cst_p_2_n_1(%arg0: tensor<3x2x!FHE.eint<2>>) -> tenso
|
||||
|
||||
// -----
|
||||
|
||||
func.func @matmul_eint_int_cst(%0: tensor<4x3x!FHE.eint<7>>) -> tensor<4x3x!FHE.eint<7>> {
|
||||
func.func @matmul_eint_int_cst(%0: tensor<4x3x!FHE.eint<7>>) -> (tensor<4x!FHE.eint<7>>, tensor<4x2x!FHE.eint<7>>, tensor<5x4x2x!FHE.eint<7>>, tensor<2x5x4x2x!FHE.eint<7>>) {
|
||||
|
||||
// ===============================
|
||||
|
||||
@@ -549,12 +549,12 @@ func.func @matmul_eint_int_cst(%0: tensor<4x3x!FHE.eint<7>>) -> tensor<4x3x!FHE.
|
||||
|
||||
// ===============================
|
||||
|
||||
return %0 : tensor<4x3x!FHE.eint<7>>
|
||||
return %2, %4, %6, %8 : tensor<4x!FHE.eint<7>>, tensor<4x2x!FHE.eint<7>>, tensor<5x4x2x!FHE.eint<7>>, tensor<2x5x4x2x!FHE.eint<7>>
|
||||
}
|
||||
|
||||
// -----
|
||||
|
||||
func.func @matmul_eint_int_cst_different_operand_manp() -> tensor<4x3x!FHE.eint<7>> {
|
||||
func.func @matmul_eint_int_cst_different_operand_manp() -> tensor<4x!FHE.eint<7>> {
|
||||
// CHECK: {MANP = 1 : ui{{[0-9]+}}}
|
||||
%z = "FHE.zero_tensor"() : () -> tensor<4x3x!FHE.eint<7>>
|
||||
%a = arith.constant dense<[[4, 6, 5], [2, 6, 3], [5, 6, 1], [5, 5, 3]]> : tensor<4x3xi8>
|
||||
@@ -574,7 +574,7 @@ func.func @matmul_eint_int_cst_different_operand_manp() -> tensor<4x3x!FHE.eint<
|
||||
|
||||
// ===============================
|
||||
|
||||
return %0 : tensor<4x3x!FHE.eint<7>>
|
||||
return %2 : tensor<4x!FHE.eint<7>>
|
||||
}
|
||||
|
||||
/////////////////////////////////////////////////
|
||||
@@ -662,7 +662,7 @@ func.func @matmul_int_eint_cst_p_2_n_1(%arg0: tensor<2x3x!FHE.eint<2>>) -> tenso
|
||||
|
||||
// -----
|
||||
|
||||
func.func @matmul_int_eint_cst(%0: tensor<3x2x!FHE.eint<7>>) -> tensor<3x2x!FHE.eint<7>> {
|
||||
func.func @matmul_int_eint_cst(%0: tensor<3x2x!FHE.eint<7>>) -> (tensor<2x!FHE.eint<7>>, tensor<2x2x!FHE.eint<7>>, tensor<5x2x2x!FHE.eint<7>>, tensor<2x5x2x2x!FHE.eint<7>>) {
|
||||
|
||||
// ===============================
|
||||
|
||||
@@ -816,12 +816,12 @@ func.func @matmul_int_eint_cst(%0: tensor<3x2x!FHE.eint<7>>) -> tensor<3x2x!FHE.
|
||||
|
||||
// ===============================
|
||||
|
||||
return %0 : tensor<3x2x!FHE.eint<7>>
|
||||
return %2, %4, %6, %8 : tensor<2x!FHE.eint<7>>, tensor<2x2x!FHE.eint<7>>, tensor<5x2x2x!FHE.eint<7>>, tensor<2x5x2x2x!FHE.eint<7>>
|
||||
}
|
||||
|
||||
// -----
|
||||
|
||||
func.func @matmul_int_eint_cst_different_operand_manp() -> tensor<3x2x!FHE.eint<7>> {
|
||||
func.func @matmul_int_eint_cst_different_operand_manp() -> tensor<2x!FHE.eint<7>> {
|
||||
%z = "FHE.zero_tensor"() : () -> tensor<3x2x!FHE.eint<7>>
|
||||
%a = arith.constant dense<[[4, 6], [2, 6], [5, 6]]> : tensor<3x2xi8>
|
||||
|
||||
@@ -840,190 +840,10 @@ func.func @matmul_int_eint_cst_different_operand_manp() -> tensor<3x2x!FHE.eint<
|
||||
|
||||
// ===============================
|
||||
|
||||
return %0 : tensor<3x2x!FHE.eint<7>>
|
||||
return %2 : tensor<2x!FHE.eint<7>>
|
||||
}
|
||||
|
||||
// -----
|
||||
|
||||
func.func @sum() -> !FHE.eint<7> {
|
||||
%0 = "FHE.zero_tensor"() : () -> tensor<5x3x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 11 : ui{{[0-9]+}}
|
||||
%1 = "FHELinalg.sum"(%0) : (tensor<5x3x4x2x!FHE.eint<7>>) -> !FHE.eint<7>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%2 = "FHELinalg.sum"(%0) { axes = [0] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<3x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%3 = "FHELinalg.sum"(%0) { axes = [1] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%4 = "FHELinalg.sum"(%0) { axes = [2] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%5 = "FHELinalg.sum"(%0) { axes = [3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x4x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%6 = "FHELinalg.sum"(%0) { axes = [0, 1] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 5 : ui{{[0-9]+}}
|
||||
%7 = "FHELinalg.sum"(%0) { axes = [0, 2] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<3x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%8 = "FHELinalg.sum"(%0) { axes = [0, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<3x4x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%9 = "FHELinalg.sum"(%0) { axes = [1, 2] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%10 = "FHELinalg.sum"(%0) { axes = [1, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x4x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%11 = "FHELinalg.sum"(%0) { axes = [2, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 8 : ui{{[0-9]+}}
|
||||
%12 = "FHELinalg.sum"(%0) { axes = [0, 1, 2] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 6 : ui{{[0-9]+}}
|
||||
%13 = "FHELinalg.sum"(%0) { axes = [0, 1, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<4x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 7 : ui{{[0-9]+}}
|
||||
%14 = "FHELinalg.sum"(%0) { axes = [0, 2, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 5 : ui{{[0-9]+}}
|
||||
%15 = "FHELinalg.sum"(%0) { axes = [1, 2, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 11 : ui{{[0-9]+}}
|
||||
%16 = "FHELinalg.sum"(%0) { axes = [0, 1, 2, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> !FHE.eint<7>
|
||||
|
||||
// CHECK: MANP = 11 : ui{{[0-9]+}}
|
||||
%17 = "FHELinalg.sum"(%0) { keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x1x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%18 = "FHELinalg.sum"(%0) { axes = [0], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x3x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%19 = "FHELinalg.sum"(%0) { axes = [1], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x1x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%20 = "FHELinalg.sum"(%0) { axes = [2], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x1x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%21 = "FHELinalg.sum"(%0) { axes = [3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x4x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%22 = "FHELinalg.sum"(%0) { axes = [0, 1], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x1x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 5 : ui{{[0-9]+}}
|
||||
%23 = "FHELinalg.sum"(%0) { axes = [0, 2], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x3x1x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%24 = "FHELinalg.sum"(%0) { axes = [0, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x3x4x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%25 = "FHELinalg.sum"(%0) { axes = [1, 2], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x1x1x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%26 = "FHELinalg.sum"(%0) { axes = [1, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x1x4x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%27 = "FHELinalg.sum"(%0) { axes = [2, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 8 : ui{{[0-9]+}}
|
||||
%28 = "FHELinalg.sum"(%0) { axes = [0, 1, 2], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x1x1x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 6 : ui{{[0-9]+}}
|
||||
%29 = "FHELinalg.sum"(%0) { axes = [0, 1, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x1x4x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 7 : ui{{[0-9]+}}
|
||||
%30 = "FHELinalg.sum"(%0) { axes = [0, 2, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x3x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 5 : ui{{[0-9]+}}
|
||||
%31 = "FHELinalg.sum"(%0) { axes = [1, 2, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x1x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 11 : ui{{[0-9]+}}
|
||||
%32 = "FHELinalg.sum"(%0) { axes = [0, 1, 2, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x1x1x1x!FHE.eint<7>>
|
||||
|
||||
// ===============================
|
||||
|
||||
%35 = "FHE.zero_tensor"() : () -> tensor<2x0x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%36 = "FHELinalg.sum"(%35) : (tensor<2x0x3x!FHE.eint<7>>) -> !FHE.eint<7>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%37 = "FHELinalg.sum"(%35) { axes = [0] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<0x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%38 = "FHELinalg.sum"(%35) { axes = [1] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%39 = "FHELinalg.sum"(%35) { axes = [2] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x0x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%40 = "FHELinalg.sum"(%35) { axes = [0, 1] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%41 = "FHELinalg.sum"(%35) { axes = [0, 2] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<0x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%42 = "FHELinalg.sum"(%35) { axes = [1, 2] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%43 = "FHELinalg.sum"(%35) { axes = [0, 1 ,2] } : (tensor<2x0x3x!FHE.eint<7>>) -> !FHE.eint<7>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%44 = "FHELinalg.sum"(%35) { keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<1x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%45 = "FHELinalg.sum"(%35) { axes = [0], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<1x0x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%46 = "FHELinalg.sum"(%35) { axes = [1], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x1x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%47 = "FHELinalg.sum"(%35) { axes = [2], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x0x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%48 = "FHELinalg.sum"(%35) { axes = [0, 1], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<1x1x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%49 = "FHELinalg.sum"(%35) { axes = [0, 2], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<1x0x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%50 = "FHELinalg.sum"(%35) { axes = [1, 2], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%51 = "FHELinalg.sum"(%35) { axes = [0, 1 ,2], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<1x1x1x!FHE.eint<7>>
|
||||
|
||||
return %1 : !FHE.eint<7>
|
||||
}
|
||||
|
||||
// -----
|
||||
|
||||
func.func @concat() -> tensor<3x!FHE.eint<7>> {
|
||||
%0 = "FHE.zero_tensor"() : () -> tensor<4x!FHE.eint<7>>
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%1 = "FHELinalg.sum"(%0) { keep_dims = true } : (tensor<4x!FHE.eint<7>>) -> tensor<1x!FHE.eint<7>>
|
||||
|
||||
%2 = "FHE.zero_tensor"() : () -> tensor<5x!FHE.eint<7>>
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%3 = "FHELinalg.sum"(%2) { keep_dims = true } : (tensor<5x!FHE.eint<7>>) -> tensor<1x!FHE.eint<7>>
|
||||
|
||||
%4 = "FHE.zero_tensor"() : () -> tensor<10x!FHE.eint<7>>
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%5 = "FHELinalg.sum"(%4) { keep_dims = true } : (tensor<10x!FHE.eint<7>>) -> tensor<1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%6 = "FHELinalg.concat"(%1, %3) : (tensor<1x!FHE.eint<7>>, tensor<1x!FHE.eint<7>>) -> tensor<2x!FHE.eint<7>>
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%7 = "FHELinalg.concat"(%1, %5) : (tensor<1x!FHE.eint<7>>, tensor<1x!FHE.eint<7>>) -> tensor<2x!FHE.eint<7>>
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%8 = "FHELinalg.concat"(%3, %5) : (tensor<1x!FHE.eint<7>>, tensor<1x!FHE.eint<7>>) -> tensor<2x!FHE.eint<7>>
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%9 = "FHELinalg.concat"(%1, %3, %5) : (tensor<1x!FHE.eint<7>>, tensor<1x!FHE.eint<7>>, tensor<1x!FHE.eint<7>>) -> tensor<3x!FHE.eint<7>>
|
||||
|
||||
return %9 : tensor<3x!FHE.eint<7>>
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -0,0 +1,182 @@
|
||||
// RUN: concretecompiler --passes MANP --passes ConcreteOptimizer --action=dump-fhe --split-input-file %s 2>&1 | FileCheck %s
|
||||
|
||||
func.func @sum() -> !FHE.eint<7> {
|
||||
%0 = "FHE.zero_tensor"() : () -> tensor<5x3x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 11 : ui{{[0-9]+}}
|
||||
%1 = "FHELinalg.sum"(%0) : (tensor<5x3x4x2x!FHE.eint<7>>) -> !FHE.eint<7>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%2 = "FHELinalg.sum"(%0) { axes = [0] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<3x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%3 = "FHELinalg.sum"(%0) { axes = [1] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%4 = "FHELinalg.sum"(%0) { axes = [2] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%5 = "FHELinalg.sum"(%0) { axes = [3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x4x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%6 = "FHELinalg.sum"(%0) { axes = [0, 1] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 5 : ui{{[0-9]+}}
|
||||
%7 = "FHELinalg.sum"(%0) { axes = [0, 2] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<3x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%8 = "FHELinalg.sum"(%0) { axes = [0, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<3x4x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%9 = "FHELinalg.sum"(%0) { axes = [1, 2] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%10 = "FHELinalg.sum"(%0) { axes = [1, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x4x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%11 = "FHELinalg.sum"(%0) { axes = [2, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 8 : ui{{[0-9]+}}
|
||||
%12 = "FHELinalg.sum"(%0) { axes = [0, 1, 2] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 6 : ui{{[0-9]+}}
|
||||
%13 = "FHELinalg.sum"(%0) { axes = [0, 1, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<4x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 7 : ui{{[0-9]+}}
|
||||
%14 = "FHELinalg.sum"(%0) { axes = [0, 2, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 5 : ui{{[0-9]+}}
|
||||
%15 = "FHELinalg.sum"(%0) { axes = [1, 2, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 11 : ui{{[0-9]+}}
|
||||
%16 = "FHELinalg.sum"(%0) { axes = [0, 1, 2, 3] } : (tensor<5x3x4x2x!FHE.eint<7>>) -> !FHE.eint<7>
|
||||
|
||||
// CHECK: MANP = 11 : ui{{[0-9]+}}
|
||||
%17 = "FHELinalg.sum"(%0) { keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x1x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%18 = "FHELinalg.sum"(%0) { axes = [0], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x3x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%19 = "FHELinalg.sum"(%0) { axes = [1], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x1x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%20 = "FHELinalg.sum"(%0) { axes = [2], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x1x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%21 = "FHELinalg.sum"(%0) { axes = [3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x4x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%22 = "FHELinalg.sum"(%0) { axes = [0, 1], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x1x4x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 5 : ui{{[0-9]+}}
|
||||
%23 = "FHELinalg.sum"(%0) { axes = [0, 2], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x3x1x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%24 = "FHELinalg.sum"(%0) { axes = [0, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x3x4x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%25 = "FHELinalg.sum"(%0) { axes = [1, 2], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x1x1x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%26 = "FHELinalg.sum"(%0) { axes = [1, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x1x4x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%27 = "FHELinalg.sum"(%0) { axes = [2, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x3x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 8 : ui{{[0-9]+}}
|
||||
%28 = "FHELinalg.sum"(%0) { axes = [0, 1, 2], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x1x1x2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 6 : ui{{[0-9]+}}
|
||||
%29 = "FHELinalg.sum"(%0) { axes = [0, 1, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x1x4x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 7 : ui{{[0-9]+}}
|
||||
%30 = "FHELinalg.sum"(%0) { axes = [0, 2, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x3x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 5 : ui{{[0-9]+}}
|
||||
%31 = "FHELinalg.sum"(%0) { axes = [1, 2, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<5x1x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 11 : ui{{[0-9]+}}
|
||||
%32 = "FHELinalg.sum"(%0) { axes = [0, 1, 2, 3], keep_dims = true } : (tensor<5x3x4x2x!FHE.eint<7>>) -> tensor<1x1x1x1x!FHE.eint<7>>
|
||||
|
||||
// ===============================
|
||||
|
||||
%35 = "FHE.zero_tensor"() : () -> tensor<2x0x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%36 = "FHELinalg.sum"(%35) : (tensor<2x0x3x!FHE.eint<7>>) -> !FHE.eint<7>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%37 = "FHELinalg.sum"(%35) { axes = [0] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<0x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%38 = "FHELinalg.sum"(%35) { axes = [1] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%39 = "FHELinalg.sum"(%35) { axes = [2] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x0x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%40 = "FHELinalg.sum"(%35) { axes = [0, 1] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%41 = "FHELinalg.sum"(%35) { axes = [0, 2] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<0x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%42 = "FHELinalg.sum"(%35) { axes = [1, 2] } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%43 = "FHELinalg.sum"(%35) { axes = [0, 1 ,2] } : (tensor<2x0x3x!FHE.eint<7>>) -> !FHE.eint<7>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%44 = "FHELinalg.sum"(%35) { keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<1x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%45 = "FHELinalg.sum"(%35) { axes = [0], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<1x0x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%46 = "FHELinalg.sum"(%35) { axes = [1], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x1x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%47 = "FHELinalg.sum"(%35) { axes = [2], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x0x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%48 = "FHELinalg.sum"(%35) { axes = [0, 1], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<1x1x3x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%49 = "FHELinalg.sum"(%35) { axes = [0, 2], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<1x0x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%50 = "FHELinalg.sum"(%35) { axes = [1, 2], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<2x1x1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 1 : ui{{[0-9]+}}
|
||||
%51 = "FHELinalg.sum"(%35) { axes = [0, 1 ,2], keep_dims = true } : (tensor<2x0x3x!FHE.eint<7>>) -> tensor<1x1x1x!FHE.eint<7>>
|
||||
|
||||
return %1 : !FHE.eint<7>
|
||||
}
|
||||
|
||||
// -----
|
||||
|
||||
func.func @concat() -> tensor<3x!FHE.eint<7>> {
|
||||
%0 = "FHE.zero_tensor"() : () -> tensor<4x!FHE.eint<7>>
|
||||
// CHECK: MANP = 2 : ui{{[0-9]+}}
|
||||
%1 = "FHELinalg.sum"(%0) { keep_dims = true } : (tensor<4x!FHE.eint<7>>) -> tensor<1x!FHE.eint<7>>
|
||||
|
||||
%2 = "FHE.zero_tensor"() : () -> tensor<5x!FHE.eint<7>>
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%3 = "FHELinalg.sum"(%2) { keep_dims = true } : (tensor<5x!FHE.eint<7>>) -> tensor<1x!FHE.eint<7>>
|
||||
|
||||
%4 = "FHE.zero_tensor"() : () -> tensor<10x!FHE.eint<7>>
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%5 = "FHELinalg.sum"(%4) { keep_dims = true } : (tensor<10x!FHE.eint<7>>) -> tensor<1x!FHE.eint<7>>
|
||||
|
||||
// CHECK: MANP = 3 : ui{{[0-9]+}}
|
||||
%6 = "FHELinalg.concat"(%1, %3) : (tensor<1x!FHE.eint<7>>, tensor<1x!FHE.eint<7>>) -> tensor<2x!FHE.eint<7>>
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%7 = "FHELinalg.concat"(%1, %5) : (tensor<1x!FHE.eint<7>>, tensor<1x!FHE.eint<7>>) -> tensor<2x!FHE.eint<7>>
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%8 = "FHELinalg.concat"(%3, %5) : (tensor<1x!FHE.eint<7>>, tensor<1x!FHE.eint<7>>) -> tensor<2x!FHE.eint<7>>
|
||||
// CHECK: MANP = 4 : ui{{[0-9]+}}
|
||||
%9 = "FHELinalg.concat"(%1, %3, %5) : (tensor<1x!FHE.eint<7>>, tensor<1x!FHE.eint<7>>, tensor<1x!FHE.eint<7>>) -> tensor<3x!FHE.eint<7>>
|
||||
|
||||
return %9 : tensor<3x!FHE.eint<7>>
|
||||
}
|
||||
Reference in New Issue
Block a user