Skip to content

Copy the imaginary halves of the inputs in MmaPlanarComplex - #3575

Open
darkdi wants to merge 1 commit into
NVIDIA:mainfrom
darkdi:planar-complex-imag
Open

Copy the imaginary halves of the inputs in MmaPlanarComplex#3575
darkdi wants to merge 1 commit into
NVIDIA:mainfrom
darkdi:planar-complex-imag

Conversation

@darkdi

@darkdi darkdi commented Sep 1, 2026

Copy link
Copy Markdown

Was reading mma_planar_complex.h and stopped on frag_A.imag = frag_A.imag; in the non-conjugate branch. Grepping the rest of the file, A_in.imag and B_in.imag do not appear anywhere, so the imaginary halves of both input fragments never reach the local ones. ArrayPlanarComplex holds two plain cutlass::Array members and nothing clears them, so frag_A.imag is whatever was on the stack, and the conjugate branch negates that same uninitialized value rather than the input. Three of the four real-valued mmas below then consume it.

I have not run this on hardware and could not find a test that reaches this specialization, so it is worth a second opinion on whether it is still on a live path before this goes in.

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