Skip to content

gpu-p2p small fixes - #307

Open
ShawnL00 wants to merge 1 commit into
inducer:mainfrom
ShawnL00:p2p-gpu
Open

gpu-p2p small fixes#307
ShawnL00 wants to merge 1 commit into
inducer:mainfrom
ShawnL00:p2p-gpu

Conversation

@ShawnL00

Copy link
Copy Markdown
Collaborator
  • Read the source dtype from the explicit sources parameter, which is not part of kwargs.
  • Read the strength dtype from the first strength array rather than from the containing sequence.
  • Keep one source-axis entry after concatenating the two local arrays to ensure the strict zip has matching lengths.

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