Skip to content

[maint] use complex64 in PyMUST to match the GPU implementations#31

Merged
charlesbmi merged 1 commit intomainfrom
maint/complex64-pymust
Jul 9, 2025
Merged

[maint] use complex64 in PyMUST to match the GPU implementations#31
charlesbmi merged 1 commit intomainfrom
maint/complex64-pymust

Conversation

@charlesbmi
Copy link
Copy Markdown
Collaborator

Introduction

GPU implementations use complex64 precision by default. This does affect runtime (mostly because of memory access)

Changes

  • Use complex64 in the PyMUST benchmark

Behavior

Did not update the benchmark figure yet because vbeam seems to have slowed down, so I'll track down where that comes from (or maybe the previous benchmarks were unexpectedly fast)?

  • pyMUST benchmark is down to 90ms?

Review checklist

  • All existing tests and checks pass
  • Unit tests covering the new feature or bugfix have been added
  • The documentation has been updated if necessary

@charlesbmi charlesbmi self-assigned this Jul 9, 2025
@charlesbmi charlesbmi merged commit 54b0847 into main Jul 9, 2025
6 checks passed
@charlesbmi charlesbmi deleted the maint/complex64-pymust branch July 9, 2025 18:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant