github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 12:15:09 -05:00
Code Issues Packages Projects Releases Wiki Activity

test(compiler): Add unit tests for the lowering of HLFHELinalg.zero to linalg

Browse Source
This commit is contained in:
Andi Drebes
2021-11-23 16:15:07 +01:00
committed by Quentin Bourgerie
parent 898e4157cf
commit d2b2934f1f
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
compiler/tests/Conversion/HLFHELinalgToLinalg/zero.mlir Normal file
View File

@@ -0,0 +1,14 @@
// RUN: zamacompiler %s --action=dump-midlfhe --passes hlfhe-tensor-ops-to-linalg 2>&1 | FileCheck %s
// CHECK: func @zero(%arg0: !HLFHE.eint<2>) -> tensor<3x2x!HLFHE.eint<2>> {
// CHECK-NEXT: %[[v0:.*]] = tensor.generate {
// CHECK-NEXT: ^bb0(%arg1: index, %arg2: index): // no predecessors
// CHECK-NEXT: %[[yld:.*]] = "HLFHE.zero"() : () -> !HLFHE.eint<2>
// CHECK-NEXT: tensor.yield %[[yld]] : !HLFHE.eint<2>
// CHECK-NEXT: } : tensor<3x2x!HLFHE.eint<2>>
// CHECK-NEXT: return %[[v0]] : tensor<3x2x!HLFHE.eint<2>>
// CHECK-NEXT: }
func @zero(%arg0: !HLFHE.eint<2>) -> tensor<3x2x!HLFHE.eint<2>> {
%1 = "HLFHELinalg.zero"(): () -> tensor<3x2x!HLFHE.eint<2>>
return %1 : tensor<3x2x!HLFHE.eint<2>>
}