Skip to content

Commit

Permalink
Raise error in choose conv_2d op instead of return (#962)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #962

As title

Reviewed By: aakhundov

Differential Revision: D50866759

fbshipit-source-id: bfcf79c99205072b67be719e0e5ddc28279e782e
  • Loading branch information
henryhu6 authored and facebook-github-bot committed Nov 4, 2023
1 parent 0c10262 commit e0faa23
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion fx2ait/fx2ait/converters/ait_converters.py
Original file line number Diff line number Diff line change
Expand Up @@ -1298,7 +1298,7 @@ def _choose_conv2d_op(
weight = pad_last_dim(len(weight._attrs["shape"]), 8)(weight)
x = pad_last_dim(len(x._attrs["shape"]), 8)(x)
elif last_dim % 2 != 0:
return RuntimeError(
raise RuntimeError(
f"Conv2d is not implemented for input channel dim {last_dim}: it needs to be aligned to a multiple of 2/4/8"
)
if bias:
Expand Down

0 comments on commit e0faa23

Please sign in to comment.