-
Notifications
You must be signed in to change notification settings - Fork 568
lower complex number operations (view_as_real, view_as_complex, conj, abs) #3070
Copy link
Copy link
Closed
Labels
loweringATen Operation loweringATen Operation loweringnostaleDo not consider for stalenessDo not consider for stalenesspytorch divergenceXLA behavior doesn't match Pytorch eager frontendXLA behavior doesn't match Pytorch eager frontend
Description
🚀 Feature
Multiple operations on complex numbers default to cpu (visible by aten:: in metrics report) or just do not work.
Specifically I have found:
defaulting to cpu:
- torch.real(x), x.real
- torch.imag(x) x.imag
- torch.conj(x)
not working:
- torch.abs(x) when x.requires_grad==True
prints: abs does not support automatic differentiation for outputs with complex dtype
There are probably many more that I have not found and i am happy to add to this list if other people finde more.
This is my First Issue so feel free to tell me if any Information is missing, or additional steps I can take.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
loweringATen Operation loweringATen Operation loweringnostaleDo not consider for stalenessDo not consider for stalenesspytorch divergenceXLA behavior doesn't match Pytorch eager frontendXLA behavior doesn't match Pytorch eager frontend