Skip to content

Commit 6219dad

Browse files
committed
the strategy was not propagated in the pytest
1 parent 90d760a commit 6219dad

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/pytest/test_cnn_mnist_qkeras.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def mnist_model():
4545
def hls_model(mnist_model, backend, io_type, strategy):
4646
keras_model = mnist_model
4747
hls_config = hls4ml.utils.config_from_keras_model(keras_model, granularity='name')
48-
hls_config['Model']['Strategy'] = 'Resource'
48+
hls_config['Model']['Strategy'] = strategy
4949
hls_config['LayerName']['softmax']['Strategy'] = 'Stable'
5050
output_dir = str(test_root_path / 'hls4mlprj_cnn_mnist_qkeras_{}_{}_{}'.format(backend, io_type, strategy))
5151

0 commit comments

Comments
 (0)