pytorch
095c328d - Add supported backward_dtype to OpInfo (#56156)

Commit
3 years ago
Add supported backward_dtype to OpInfo (#56156) Summary: Related to https://github.com/pytorch/pytorch/issues/55601. - [x] removed complex autograd checker in `test_supported_backward` - [x] created `backward_dtype[If<Device>]` that inherits from normal `dtype[If<Device>]` by default - [x] removed all skip for backward test, instead added backward dtype - [x] change complex autograd to a function call: `support_complex_autograd(device_type)` that depends on `backward_dtype*` since they essentially mean the same thing for complex types TODO for next PR - add `test_unsupported_backward` to verify they are actually unsupported. Pull Request resolved: https://github.com/pytorch/pytorch/pull/56156 Reviewed By: mruberry Differential Revision: D27926717 Pulled By: walterddr fbshipit-source-id: 9a4af8612278ca44a97b6f1510b6b175852c893b
Author
Rong Rong (AI Infra)
Parents
Loading