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

feat(compiler): bench, llm, big mat mul

Browse Source
This commit is contained in:
rudy
2023-09-01 17:36:19 +02:00
committed by rudy-6-4
parent a28a7bf397
commit 4907577b1c
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
compilers/concrete-compiler/compiler/tests/end_to_end_fixture/benchmarks_cpu/levelled_llm.yaml Normal file
View File

@@ -0,0 +1,18 @@
description: big_matmul
program: |
func.func @main(%arg0: !FHE.esint<22>) -> !FHE.esint<22> {
%cst = arith.constant dense<0> : tensor<19x3072xi23>
%from_elements = tensor.from_elements %arg0 : tensor<1x!FHE.esint<22>>
%0 = "FHELinalg.add_eint_int"(%from_elements, %cst) : (tensor<1x!FHE.esint<22>>, tensor<19x3072xi23>) -> tensor<19x3072x!FHE.esint<22>>
%cst_0 = arith.constant dense<5> : tensor<3072x768xi23>
%1 = "FHELinalg.matmul_eint_int"(%0, %cst_0) : (tensor<19x3072x!FHE.esint<22>>, tensor<3072x768xi23>) -> tensor<19x768x!FHE.esint<22>>
%c0 = arith.constant 0 : index
%extracted = tensor.extract %1[%c0, %c0] : tensor<19x768x!FHE.esint<22>>
return %extracted : !FHE.esint<22>
}
p-error: 6.3342483999973e-05
tests:
- inputs:
- scalar: 0
outputs:
- scalar: 0