Skip to content

Commit 80437b4

Browse files
committed
trigger pre-commit
1 parent ccc3189 commit 80437b4

2 files changed

Lines changed: 2 additions & 0 deletions

File tree

hls4ml/converters/pytorch/core.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,7 @@ def parse_batchnorm_layer(operation, layer_name, input_names, input_shapes, node
158158

159159
return layer, [shape for shape in input_shapes[0]]
160160

161+
161162
@pytorch_handler('einsum')
162163
def parse_einsum_layer(operation, layer_name, input_names, input_shapes, node, class_object, data_reader, config):
163164
assert 'einsum' in operation

test/pytest/test_pytorch_api.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -878,6 +878,7 @@ def forward(self, x):
878878
atol = 5.0e-2
879879
np.testing.assert_allclose(hls_prediction, pytorch_prediction, rtol=rtol, atol=atol)
880880

881+
881882
class EinsumOuterProduct(nn.Module):
882883
def __init__(self):
883884
super().__init__()

0 commit comments

Comments
 (0)