Skip to content

Commit b055233

Browse files
committed
catapult and oneapi tests
1 parent e73b3d3 commit b055233

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

hls4ml/backends/fpga/passes/inplace_stream_flatten.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ def match(self, node):
1515
if not (isinstance(node, Reshape)):
1616
# Reshape with multiple outputs will be kept as is, or repack cannot handle different shapes
1717
return False
18-
if len(node.get_output_variable().shape) + node.name in node.model.outputs != 1:
18+
if len(node.get_output_variable().shape) + (node.name in node.model.outputs) != 1:
1919
return False
2020
io_type = node.model.config.get_config_value('IOType')
2121
return io_type == 'io_stream'

test/pytest/test_multiout_network.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ def test_multi_output_nn(model, data, backend: str, io_type: str):
6767
assert np.allclose(r_hls[1], r_keras[1], atol=1e-5, rtol=0)
6868

6969

70-
@pytest.mark.parametrize('backend', ['Vivado', 'Quartus', 'Vitis'])
70+
@pytest.mark.parametrize('backend', ['Vivado', 'Quartus', 'Vitis', 'Catapult', 'OneAPI'])
7171
@pytest.mark.parametrize('io_type', ['io_parallel', 'io_stream'])
7272
@pytest.mark.parametrize('strategy', ['latency', 'resource'])
7373
def test_multi_output_nn_2(model2, data2, backend: str, io_type: str, strategy: str):

0 commit comments

Comments
 (0)