torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Enable custom perf models in MemoryBalancedPartitioner

Open micrain opened this issue 7 months ago • 1 comments

Summary: We are enabling custom perf models while evaluating candidate plans in MemoryBalancedPartitioner. This will enable us to leverage the new critical path model defined in NoopCriticalPathPerfModel while trying to construct a memory balanced plan with the greedy partitioner.

Upcoming diffs will modify the greedy partitioner to leverage the new perf model.

Differential Revision: D73292939

micrain avatar Apr 19 '25 00:04 micrain

This pull request was exported from Phabricator. Differential Revision: D73292939

facebook-github-bot avatar Apr 19 '25 00:04 facebook-github-bot