diff --git a/hyperactor_mesh/src/lib.rs b/hyperactor_mesh/src/lib.rs index cc21cbb4c..10aaa9ffd 100644 --- a/hyperactor_mesh/src/lib.rs +++ b/hyperactor_mesh/src/lib.rs @@ -35,11 +35,25 @@ pub mod test_utils; mod testresource; pub mod v1; +// Generated build information module +#[cfg(fbcode_build)] +#[path = "../build_info.rs"] +pub mod build_info; + pub use actor_mesh::RootActorMesh; pub use actor_mesh::SlicedActorMesh; pub use bootstrap::Bootstrap; pub use bootstrap::bootstrap; pub use bootstrap::bootstrap_or_die; +// Re-export build info attributes for easy access +#[cfg(fbcode_build)] +pub use build_info::BUILD_COMMIT; +#[cfg(fbcode_build)] +pub use build_info::BUILD_HOST; +#[cfg(fbcode_build)] +pub use build_info::BUILD_TIMESTAMP; +#[cfg(fbcode_build)] +pub use build_info::BUILD_USER; pub use comm::CommActor; pub use dashmap; pub use hyperactor_mesh_macros::sel; diff --git a/python/monarch/gradient/_gradient_generator.cpp b/python/monarch/gradient/_gradient_generator.cpp index 01ea6eed3..a45b38542 100644 --- a/python/monarch/gradient/_gradient_generator.cpp +++ b/python/monarch/gradient/_gradient_generator.cpp @@ -420,12 +420,20 @@ struct GradientGenerator { DEBUG_PRINT( "// add: " << node->node->name() << ", input_nr=" << static_cast(input_nr) << "\n"); +#if !defined(FBCODE_BUILD) + realInputBuffer(node).add( + input_nr, + check_and_reduce(node->node, input_nr, std::move(t)), + std::nullopt, + std::nullopt); +#else realInputBuffer(node).add( input_nr, check_and_reduce(node->node, input_nr, std::move(t)), std::nullopt, std::nullopt, node->node); +#endif } InputBuffer& realInputBuffer(NodeState* state) {