Skip to content

Commit 31219e3

Browse files
committed
update pytest to new default setting
1 parent 9d8edbc commit 31219e3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/pytest/test_pytorch_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -498,7 +498,7 @@ def test_pooling(pooling, padds, backend):
498498
model.eval()
499499
pytorch_prediction = model(torch.Tensor(X_input)).detach().numpy()
500500

501-
config = config_from_pytorch_model(model, input_shape_forHLS)
501+
config = config_from_pytorch_model(model, input_shape_forHLS, transpose_outputs=True)
502502
output_dir = str(test_root_path / f'hls4mlprj_pytorch_api_pooling_{pooling.__name__}_padds_{padds}_backend_{backend}')
503503
hls_model = convert_from_pytorch_model(model, hls_config=config, output_dir=output_dir, backend=backend)
504504
hls_model.compile()

0 commit comments

Comments
 (0)