Skip to content

Conversation

@allenwang28
Copy link
Contributor

Summary:
This is the first step to supporting individual installs for Monarch capabilities, i.e. core/rdma/tensor_engine.

Changes:

  • Modifies monarch_extension to use core, rdma, tensor_engine, full feature sets.
  • Switches setup.py to use MONARCH_FEATURES accepting strings rather than the 0/1 boolean only USE_TENSOR_ENGINE (also makes all the downstream changes needed to reflect it)
  • Renames the default wheel name from monarch to torchmonarch, to better align with what we're officially publishing to PyPI
  • Slight clean up to setup.py workflow

Differential Revision: D87884695

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Dec 1, 2025
@meta-codesync
Copy link

meta-codesync bot commented Dec 1, 2025

@allenwang28 has exported this pull request. If you are a Meta employee, you can view the originating Diff in D87884695.

@allenwang28 allenwang28 changed the title Enable Monarch extra builds [Builds] Add separate Rust build features for core, rdma and tensor_engine Dec 1, 2025
Summary:
This is the first step to supporting individual installs for Monarch capabilities, i.e. core/rdma/tensor_engine.

Changes:
- Modifies monarch_extension to use core, rdma, tensor_engine, full feature sets.
- Switches setup.py to use MONARCH_FEATURES accepting strings rather than the 0/1 boolean only USE_TENSOR_ENGINE (also makes all the downstream changes needed to reflect it)
- Renames the default wheel name from `monarch` to `torchmonarch`, to better align with what we're officially publishing to PyPI
- Slight clean up to setup.py workflow

Differential Revision: D87884695
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot. fb-exported meta-exported

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant