pytorch
cb320482 - adding test to audit CompositeImplicitAutograd ops that do not have a batching rule (#91367)

Commit
2 years ago
adding test to audit CompositeImplicitAutograd ops that do not have a batching rule (#91367) Fixes https://github.com/pytorch/functorch/issues/1087 It looks like there are `306` rules that should be looked into ``` test/functorch/test_vmap_registrations.py .x.....xxxxxxx.x.x.x.x.x.x.x.x........xx.x.x..x.x.xxx...xxxx.x.x.x........x.........xxxxx..x..x.....xx...xx.....xxx.xxxxxxxxxxxxxxxxx.. [ 24%] .........x.x......x.xxxxxx..x..xx.x.xxx.x.......x.xxx.xx..xxx.xxx...xxxxx.x....xxxxxxxxxxxxxxx....xx.xxx.xx.x...xx...xx...xxxxxx...xxxxx..x...xxxxxxxxxxxx..xx..xx.xx.x..xxxx..xx [ 56%] .xx..x.x....xxxxxx.x.xx...xxxxx.xx...x..x.x.xx...xx.xxxxxx.xxxxxx..x........xxxxxxxx..xxxxxxxx..xx.xxxxxxxxxxxxxxxxxxxxxxx..........xxxx.xxxx.........xxxxxxxx..xxx..xxx.x.x.x.xx [ 88%] xx.xxx.x......xxx.x.xxxxxxxx....x......xxxxxxxxx.xx.x.x.x.......xx [100%] =================================================================== 249 passed, 1185 deselected, 306 xfailed in 3.17s =================================================================== ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/91367 Approved by: https://github.com/zou3519
Author
Committer
Parents
Loading