pytorch
36fa9d82 - [functorch] Replace addr_decomp with the one in PyTorch Core, issue pytorch/functorch#833 (pytorch/functorch#836)

Commit
2 years ago
[functorch] Replace addr_decomp with the one in PyTorch Core, issue pytorch/functorch#833 (pytorch/functorch#836) * Replace addr_decomp with the one in PyTorch Core, issue pytorch/functorch#833 * only allow OpOverload to be passed to _register_jit_decomposition_bypass_script * change variable name to vmap_decompositions_lib * change function name to _register_python_decomposition_vmap
Author
Committer
Parents
Loading