Skip to content

Draft: Add custom kernel implementation#11

Open
nsarka wants to merge 2 commits into
samnordmann:mainfrom
nsarka:nsarka/tp-rowwise-kernel
Open

Draft: Add custom kernel implementation#11
nsarka wants to merge 2 commits into
samnordmann:mainfrom
nsarka:nsarka/tp-rowwise-kernel

Conversation

@nsarka
Copy link
Copy Markdown
Collaborator

@nsarka nsarka commented Jan 30, 2026

JIT compiles .cu kernel implementations for TPRowwise.

Allows the "option" field to point to a .cu file. I'm not 100% sure of this, but maybe it would be useful. WDYT?

@nsarka nsarka requested a review from samnordmann January 30, 2026 02:46
@nsarka nsarka self-assigned this Jan 30, 2026
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