pytorch
d5169e71 - Use a stable ordering for saved values in functorch.default_partition (#100111)

Commit
1 year ago
Use a stable ordering for saved values in functorch.default_partition (#100111) Previously, due to the use of the Python set data structure, the ordering of saved values (and how they would appear in the graph) was unstable and changed across runs, making it hard to debug downstream applications. Here we use a dict (with insertion-ordering semantics) to deduplicate values in a way that preserves ordering Pull Request resolved: https://github.com/pytorch/pytorch/pull/100111 Approved by: https://github.com/Skylion007
Author
Committer
Parents
Loading