Skip to content

[tk] add tk gb300 kernels#1019

Open
xuzhao9 wants to merge 2 commits intomainfrom
xz9/update-tk-b300
Open

[tk] add tk gb300 kernels#1019
xuzhao9 wants to merge 2 commits intomainfrom
xz9/update-tk-b300

Conversation

@xuzhao9
Copy link
Copy Markdown
Contributor

@xuzhao9 xuzhao9 commented Apr 15, 2026 •
edited
Loading

To use these kernels:

  1. Install tk with python install.py --tk
  2. Run the kernels:
python run.py --op blackwell_attentions --only tk_bf16_b300_mha

TODO: validate that these actually works on GB300

@meta-cla meta-cla Bot added the cla signed label Apr 15, 2026
@xuzhao9 xuzhao9 requested a review from htyu April 15, 2026 17:19
@xuzhao9 xuzhao9 force-pushed the xz9/update-tk-b300 branch from 3dc2544 to 5d602c6 Compare April 15, 2026 17:23
@xuzhao9 xuzhao9 temporarily deployed to docker-s3-upload April 15, 2026 17:23 — with GitHub Actions Inactive
@xuzhao9 xuzhao9 temporarily deployed to docker-s3-upload April 15, 2026 17:23 — with GitHub Actions Inactive
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@htyu htyu Awaiting requested review from htyu

Assignees

No one assigned

Labels

cla signed

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@xuzhao9