From 5d3af16617743890bccbce482d20528420ed333c Mon Sep 17 00:00:00 2001
From: Andi Drebes <andi.drebes@zama.ai>
Date: Thu, 9 Feb 2023 11:19:04 +0100
Subject: [PATCH] test(compiler): Add check for omp.parallel after lowering a
 parallelizable FHELinalg op

This test ensures that at least one parallel region is generated for
an FHELinalg operation that is guaranteed to result in a parallel loop
when `concretecompiler` is invoked with `--parallelize`.
---
 .../check_tests/Conversion/OpenMP/check_parallel.mlir  | 10 ++++++++++
 1 file changed, 10 insertions(+)
 create mode 100644 compiler/tests/check_tests/Conversion/OpenMP/check_parallel.mlir

diff --git a/compiler/tests/check_tests/Conversion/OpenMP/check_parallel.mlir b/compiler/tests/check_tests/Conversion/OpenMP/check_parallel.mlir
new file mode 100644
index 000000000..efa9aa271
--- /dev/null
+++ b/compiler/tests/check_tests/Conversion/OpenMP/check_parallel.mlir
@@ -0,0 +1,10 @@
+// RUN: concretecompiler %s --action=dump-llvm-dialect --parallelize 2>&1| FileCheck %s
+
+// Check that at some point the compilation pipeline generates a parallel region
+// CHECK: omp.parallel
+func.func @apply_lookup_table(%arg0: tensor<2x3x4x!FHE.eint<2>>) -> tensor<2x3x4x!FHE.eint<2>> {
+  %arg1 = arith.constant dense<"0x0000000000000000000000000000000100000000000000020000000000000003"> : tensor<4xi64>
+  %1 = "FHELinalg.apply_lookup_table"(%arg0, %arg1): (tensor<2x3x4x!FHE.eint<2>>, tensor<4xi64>) -> (tensor<2x3x4x!FHE.eint<2>>)
+  return %1: tensor<2x3x4x!FHE.eint<2>>
+}
+