Implementation for complex sorting of dndarrays

[Berkant03]

Due Diligence

• General:
  • title of the PR is suitable to appear in the Release Notes
• Implementation:
  • unit tests: all split configurations tested
  • unit tests: multiple dtypes tested
  • NEW unit tests: MPS tested (1 MPI process, 1 GPU)
  • benchmarks: created for new functionality
  • benchmarks: performance improved or maintained
  • documentation updated where needed

Description

Use vectorized_sort to implement a complex sort similar to np.sort_complex.

Issue/s resolved: #2372

Changes proposed:

• Create a ht.sort_complex similar to np.sort_complex.
• Reuse row exchange fuctionality of vectorized_sort.

Type of change

• New feature (non-breaking change which adds functionality)

Does this change modify the behaviour of other functions? If so, which?

• vectorized_sort, extraction of the communication logic into a seperate function.

---

_{Stack created with GitHub Stacks CLI • Give Feedback 💬}

[brownbaerchen]

I marked the PR as draft because I couldn't find any actual implementation or tests for the new feature.

[codecov]

Codecov Report

❌ Patch coverage is 74.60317% with 16 lines in your changes missing coverage. Please review.
⚠️ Please upload report for BASE (363-vectorized-sorting@e487637). Learn more about missing BASE report.
✅ All tests successful. No failed tests found.

Files with missing lines	Patch %	Lines
heat/core/manipulations.py	74.60%	16 Missing ⚠️

Additional details and impacted files

@@                    Coverage Diff                    @@
##             363-vectorized-sorting    #2482   +/-   ##
=========================================================
  Coverage                          ?   83.68%           
=========================================================
  Files                             ?      105           
  Lines                             ?    15760           
  Branches                          ?        0           
=========================================================
  Hits                              ?    13189           
  Misses                            ?     2571           
  Partials                          ?        0           

Flag	Coverage Δ
unit	83.68% <74.60%> (?)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.