AITemplate
AITemplate copied to clipboard
Published
20 hours ago •
facebookincubator
facebookincubator
Relax constraint on num_inputs
Summary:
Currently we don't allow AIT's inputs to be a subset of python's inputs.
This could cause us some trouble.
e.g. https://www.internalfb.com/phabricator/paste/view/P619446970?lines=37
Here, add_14 can be deduced at a very early stage and therefore the source of add_14: cat_37 actually isn't used in AIT graph anywhere!
Therefore, we would got complaint: User passed 6 inputs, but the model expects 5. This is actually because cat_37 is not longer needed in AIT's graph!
We can relax the constraint that AIT's inputs == python's inputs, as along as we can find each AIT's input can corresponding python's input.
Differential Revision: D43205927
