update tests

google · Dec 15, 2024 · 220e714 · 220e714
1 parent 446e672
commit 220e714
Show file tree

Hide file tree

Showing 19 changed files with 46 additions and 225 deletions.
diff --git a/tests/Dialect/PISA/BUILD → tests/Dialect/PISA/IR/BUILD b/tests/Dialect/PISA/BUILD → tests/Dialect/PISA/IR/BUILD
diff --git a/tests/Dialect/PISA/ops.mlir → tests/Dialect/PISA/IR/ops.mlir b/tests/Dialect/PISA/ops.mlir → tests/Dialect/PISA/IR/ops.mlir
diff --git a/tests/Dialect/PISA/end_to_end/BUILD b/tests/Dialect/PISA/end_to_end/BUILD
diff --git a/tests/Dialect/PISA/end_to_end/README.md b/tests/Dialect/PISA/end_to_end/README.md
diff --git a/tests/Dialect/PISA/end_to_end/basic.cpp b/tests/Dialect/PISA/end_to_end/basic.cpp
diff --git a/tests/Dialect/PISA/end_to_end/basic.csv b/tests/Dialect/PISA/end_to_end/basic.csv
diff --git a/tests/Dialect/PISA/end_to_end/basic.mlir b/tests/Dialect/PISA/end_to_end/basic.mlir
diff --git a/tests/Dialect/PISA/end_to_end/basic_bgv.mlir b/tests/Dialect/PISA/end_to_end/basic_bgv.mlir
diff --git a/tests/Dialect/PISA/end_to_end/basic_lwe.mlir b/tests/Dialect/PISA/end_to_end/basic_lwe.mlir
diff --git a/tests/Dialect/PISA/end_to_end/basic_openfhe.mlir b/tests/Dialect/PISA/end_to_end/basic_openfhe.mlir
diff --git a/tests/Dialect/PISA/end_to_end/basic_pisa.mlir b/tests/Dialect/PISA/end_to_end/basic_pisa.mlir
diff --git a/tests/Dialect/PISA/end_to_end/basic_poly.mlir b/tests/Dialect/PISA/end_to_end/basic_poly.mlir
diff --git a/tests/Dialect/PISA/end_to_end/basic_spoly.mlir b/tests/Dialect/PISA/end_to_end/basic_spoly.mlir
diff --git a/tests/Dialect/PISA/end_to_end/test.bzl b/tests/Dialect/PISA/end_to_end/test.bzl
diff --git a/tests/Dialect/Polynomial/Conversions/polynomial_to_pisa/end_to_end.mlir b/tests/Dialect/Polynomial/Conversions/polynomial_to_pisa/end_to_end.mlir
@@ -0,0 +1,16 @@
+// RUN: heir-opt --mlir-to-bgv='ciphertext-degree=8192' --bgv-to-lwe --lwe-to-polynomial --convert-elementwise-to-affine --full-loop-unroll --convert-tensor-to-scalars --polynomial-to-pisa %s
+
+// FIXME: ADD FILECHECK
+!t = tensor<8192xi16>
+
+func.func @basic_test(%x: !t {secret.secret}, %y: !t {secret.secret}) -> !t {
+  %r0 = tensor.empty() : !t
+  %r = affine.for %i = 0 to 8192 iter_args(%r = %r0) -> !t {
+    %xi = tensor.extract %x[%i] : !t
+    %yi = tensor.extract %y[%i] : !t
+    %si = arith.addi %xi, %yi : i16
+    %ri = tensor.insert %si into %r[%i] : !t
+    affine.yield %ri : !t
+  }
+  return %r : !t
+}