[mshr-h]

Summary

Models using torch.complex(real, imag) currently fail the default to_edge() Core ATen check. Allowing the operator through that check still leaves CPU execution blocked by a missing aten::complex.out kernel.

This PR enables default export and CPU execution for matching FP16/32/64 inputs, producing complex32/64/128 outputs. The portable kernel supports broadcasting and output resizing, and requires default contiguous dimension order. Edge dtype mapping now includes complex32.

Note: aten.complex.default is temporarily allowed by the verifier while keeping IR validation enabled. A separate PyTorch change is needed to formally include it in Core ATen. This exception can be removed once all supported PyTorch versions include that change.

Fixes #24328

Test plan

Rebuild the native bindings and C++ tests from this branch, then

python -m unittest \
  executorch.extension.pybindings.test.test_pybindings.PybindingsTest.test_complex_construction \
  executorch.extension.pybindings.test.test_pybindings.PybindingsTest.test_complex_numpy_input \
  executorch.extension.pybindings.test.test_pybindings.PybindingsTest.test_complex_half_dlpack_input \
  executorch.exir.verification.test.test_verifier \
  executorch.exir.dialects.edge.test.test_edge_ops \
  executorch.exir.dialects.edge.test.test_edge_yaml

./cmake-out/kernels/test/portable_kernels_test \
  --gtest_filter='OpComplexOutTest.*'

ctest --test-dir cmake-out \
  -R '^portable_kernels_test$' --output-on-failure

Result: Python unittest reported 24 tests passed; GoogleTest reported 21/21 complex kernel tests passed; CTest reported 1/1 test target passed. No failures.

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/24330

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

⚠️ 25 Awaiting Approval

As of commit c2763d9 with merge base 87d6190 ():

AWAITING APPROVAL - The following workflows need approval before CI can run:

• Build Aarch64 Linux CUDA Wheels (gh)
• Build Aarch64 Linux Wheels (gh)
• Build documentation (gh)
• Build Linux CUDA Wheels (gh)
• Build Linux Wheels (gh)
• Build macOS Wheels (gh)
• Build Presets (gh)
• Build Windows CUDA Wheels (gh)
• Build Windows Wheels (gh)
• Cadence Build & Test (gh)
• Lint (gh)
• periodic (gh)
• pull (gh)
• Test ARM Backend (gh)
• Test CoreML Backend (gh)
• Test Cortex-M Backend (gh)
• Test CUDA Builds (gh)
• Test Native Engines (gh)
• Test NXP Backend (gh)
• Test QNN Backend (gh)
• Test QNN Windows MSVC build (gh)
• Test Vulkan Backend (gh)
• Test Vulkan Backend (specialized runners) (gh)
• Test WebGPU Backend (gh)
• Test XNNPACK Backend (gh)

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[Copilot]

🟢 Approval recommended

No active review issues are blocking approval of the complex operator support.

No open issues 🎉

What changed in this PR

Enables torch.complex(real, imag) through default Edge export and CPU execution, addressing #24328 for matching FP16, FP32, and FP64 inputs.

Changes:

• Adds a portable complex.out kernel with broadcasting, output resizing, and default dimension-order validation.
• Registers the operator and adds complex32 dtype mapping plus a temporary Core ATen verifier exception.
• Adds kernel and Python integration tests covering supported types, broadcasting, and invalid inputs.

File	Description
shim_et/​xplat/​executorch/​kernels/​portable/​op_registration_util.bzl	Adds the portable kernel build target and broadcast dependency.
kernels/​test/​targets.bzl	Registers complex tests for ATen and portable kernels.
kernels/​test/​op_complex_test.cpp	Tests types, broadcasting, special values, resizing, and validation.
kernels/​test/​CMakeLists.txt	Includes complex kernel tests in CMake builds.
kernels/​portable/​functions.yaml	Registers the portable complex.out implementation.
kernels/​portable/​cpu/​op_complex.cpp	Implements complex tensor construction on CPU.
kernels/​aten/​functions.yaml	Registers complex.out for ATen builds.
extension/​pybindings/​test/​test_pybindings.py	Adds end-to-end export and execution coverage.
exir/​verification/​verifier.py	Temporarily allows complex.default through Core ATen validation.
exir/​dialects/​edge/​dtype/​supported.py	Maps complex32 to ComplexHalf.

_{Review effort: Balanced}

[Copilot]

🟢 Approval recommended

There are no active review issues blocking the complex-construction support change.

No open issues 🎉

_{Review effort: Balanced}

[mshr-h]

@pytorchbot label "release notes: ops & kernels"