search

wilson-labs / cola

Compositional Linear Algebra
Apache License 2.0
325 stars 24 forks source link

[Bug] Deprecated PyTree Function in Pytorch 2.2 #84

Open adam-hartshorne opened 4 months ago

adam-hartshorne commented 4 months ago

The latest version of PyTorch has deprecated torch.utils._pytree._register_pytree_node in favour of torch.utils._pytree.register_pytree_node, which results in many of the following warnings,

...site-packages/cola/backends/backends.py:75: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.