viable/strict/1788205825: [BE] Extend dropout support to complex numbers (#195373)
- PyTorch: 317 events in the last 90 days
- PyTorch: 314th Release in the last 90 days
- Previous: earlier the same day · viable/strict/1788204145
What happened
Summary nn.functional.dropout raised NotImplementedError for complex inputs on CPU (and CUDA), while silently "working" on MPS. Root cause: _dropout_impl allocated the Bernoulli mask ( noise ) in the input's dtype and called bernoulli_ on it. For complex inputs that requires a complex bernoulli_ , which CPU and CUDA do not implement (both dispatch via AT_DISPATCH_ALL_TYPES_AND3 , no complex). MPS only accepted it be…
Summary assembled by rule from the sources below