Skip to content

Commit c739c0e

Browse files
authored
Merge pull request #511 from vloncar/qkeras_dep
Add QKeras as a package dependency
2 parents 83024d3 + 41a212a commit c739c0e

File tree

4 files changed

+10
-4
lines changed

4 files changed

+10
-4
lines changed

Jenkinsfile

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
pipeline {
22
agent {
33
docker {
4-
image 'vivado-el7:1'
4+
image 'vivado-el7:2'
55
args '-v /data/Xilinx:/data/Xilinx'
66
}
77
}
@@ -14,9 +14,8 @@ pipeline {
1414
steps {
1515
dir(path: 'test') {
1616
sh '''#!/bin/bash --login
17-
conda activate hls4ml-py36
17+
conda activate hls4ml-py37
1818
pip install tensorflow
19-
pip install git+git://github.com/google/[email protected]#egg=qkeras
2019
pip install -U ../ --user
2120
./convert-keras-models.sh -x -f keras-models.txt
2221
pip uninstall hls4ml -y'''

docs/setup.rst

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,9 @@ http://www.h5py.org
1818

1919
https://pypi.python.org/pypi/PyYAML
2020

21+
**QKeras**\ : for working with quantized models
22+
23+
https://github.com/google/qkeras
2124

2225
**PyTorch**\ : for reading in Torch models
2326

setup.cfg

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
[metadata]
22
description-file = README.md
33

4+
[options]
5+
python_requires = >=3.7
6+
47
[options.entry_points]
58
pytest_randomly.random_seeder =
69
hls4ml = hls4ml:reseed

setup.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,8 @@ def get_version(rel_path):
3333
'h5py',
3434
'onnx>=1.4.0',
3535
'calmjs.parse',
36-
'tabulate'
36+
'tabulate',
37+
'qkeras',
3738
],
3839
extras_require={
3940
'profiling': [

0 commit comments

Comments
 (0)