Skip to content

lower complex number operations (view_as_real, view_as_complex, conj, abs) #3070

@pitneitemeier

Description

@pitneitemeier

🚀 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.

Metadata

Metadata

Assignees

Labels

loweringATen Operation loweringnostaleDo not consider for stalenesspytorch divergenceXLA behavior doesn't match Pytorch eager frontend

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions