feat: add convolution extension

extend the current tracing and compilation with convolution, which
should compile to the FHELinalg.conv2d operation from the compiler
This commit is contained in:
youben11
2022-02-22 11:14:23 +01:00
committed by Ayoub Benaissa
parent 8d12a53651
commit 98bec17050
13 changed files with 663 additions and 58 deletions

View File

@@ -45,7 +45,8 @@ pygraphviz = { version = "^1.7", optional = true }
Pillow = "^9.0.0"
loguru = "^0.5.3"
setuptools = "*"
concrete-compiler = "^0.2.0"
concrete-compiler = "^0.3.1"
torch = "^1.10.2"
[tool.poetry.extras]
full = ["pygraphviz"]