Enable Training Pybindings in OSS

[JacobSzwejbka]

Enable the pybindings for the training lib in oss

Test plan

Open up a python terminal and try importing and using the optimizer from the bindings. (Also double checked that I didnt break the normal core pybindings)

build locally with

./install_executorch.sh --pybind training

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/8073

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

⏳ No Failures, 6 Pending

As of commit 4677709 with merge base 6b58e2e ():
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[dbort]

If you're adding a new python extension, you'll need to let setup.py know about it so it can include it in the wheel. You'll at least need an entry at

executorch/setup.py

Line 671 in 0733973

if ShouldBuild.pybindings():

, but you'll also need to tell cmake to build it for you like

executorch/setup.py

Line 572 in 0733973

if ShouldBuild.pybindings():

[dbort]

Once you add it to the wheel, you can test it by running install_executorch.sh into a conda env and then running a .py file that tries importing and using the new APIs

[JacobSzwejbka]

I moved these here to better match the pattern in the rest of the file and also it makes the shared dependencies between training and pybind a little easier to handle

[JacobSzwejbka]

This weirdly broke apple builds so undid

[facebook-github-bot]

@JacobSzwejbka has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

[facebook-github-bot]

@JacobSzwejbka has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.