You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
PyTree manipulation is now widely used in PyTorch. It becomes a fairly significant perf bottleneck during torch.fx tracing using the pure-Python approach of PyTree. We are pushing to use C++-based PyTree implementation. The lack of conda package in the Anaconda default channel feels like a serious hindrance for this.
I have asked the Anaconda team to add optree in the default channel (ContinuumIO/anaconda-issues#13131). But the average response time on the anaconda issue tracker is roughly weeks. I wonder if it is possible to add optree in the pytorch channel.
PyTree manipulation is now widely used in PyTorch. It becomes a fairly significant perf bottleneck during
torch.fx
tracing using the pure-Python approach of PyTree. We are pushing to use C++-based PyTree implementation. The lack of conda package in the Anaconda default channel feels like a serious hindrance for this.I have asked the Anaconda team to add
optree
in the default channel (ContinuumIO/anaconda-issues#13131). But the average response time on the anaconda issue tracker is roughly weeks. I wonder if it is possible to addoptree
in the pytorch channel.Ref: Conda-Forge feedstock: https://github.com/conda-forge/optree-feedstock
/
The text was updated successfully, but these errors were encountered: