Skip to content

Improving performance of Legendre transformation in SHT#155

Merged
azrael417 merged 2 commits intomainfrom
tkurth/performance-updates
Mar 18, 2026
Merged

Improving performance of Legendre transformation in SHT#155
azrael417 merged 2 commits intomainfrom
tkurth/performance-updates

Conversation

@azrael417
Copy link
Collaborator

This MR is a general version of MR #150 by Jeremy McGibbon. This one covers vector and distributed SHT as well.

adding .contiguous after torch.stack for vector sht outputs

more asserts

Co-authored-by: Jeremy McGibbon <jeremy.mcgibbon@gmail.com>
@azrael417 azrael417 requested a review from bonevbs March 12, 2026 10:23
@azrael417 azrael417 self-assigned this Mar 12, 2026
@azrael417 azrael417 mentioned this pull request Mar 12, 2026
@azrael417 azrael417 marked this pull request as draft March 12, 2026 12:33
@azrael417 azrael417 marked this pull request as ready for review March 12, 2026 12:44
Copy link
Collaborator

@bonevbs bonevbs left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@azrael417 azrael417 merged commit e043ad8 into main Mar 18, 2026
1 check passed
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.

2 participants