mirror of
https://github.com/zama-ai/concrete.git
synced 2026-04-17 03:00:54 -04:00
5d3af16617
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`.
11 lines
579 B
MLIR
11 lines
579 B
MLIR
// 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>>
|
|
}
|
|
|