Skip to content
Navigation Menu
Sign in
Appearance settings
Platform
AI CODE CREATION
GitHub Copilot
Write better code with AI
GitHub Copilot app
Direct agents from issue to merge
MCP Registry
Integrate external tools
DEVELOPER WORKFLOWS
Actions
Automate any workflow
Codespaces
Instant dev environments
Issues
Plan and track work
Code Review
Manage code changes
Code Quality
Enforce quality at merge
APPLICATION SECURITY
GitHub Advanced Security
Find and fix vulnerabilities
Code security
Secure your code as you build
Secret protection
Stop leaks before they start
EXPLORE
Why GitHub
Documentation
Blog
Changelog
Marketplace
View all features
Solutions
BY COMPANY SIZE
Enterprises
Small and medium teams
Startups
Nonprofits
BY USE CASE
App Modernization
DevSecOps
DevOps
CI/CD
View all use cases
BY INDUSTRY
Healthcare
Financial services
Manufacturing
Government
View all industries
View all solutions
Resources
EXPLORE BY TOPIC
AI
Software Development
DevOps
Security
View all topics
EXPLORE BY TYPE
Customer stories
Events & webinars
Ebooks & reports
Business insights
GitHub Skills
SUPPORT & SERVICES
Documentation
Customer support
Community forum
Trust center
Partners
View all resources
Open Source
COMMUNITY
GitHub Sponsors
Fund open source developers
PROGRAMS
Security Lab
Maintainer Community
Accelerator
GitHub Stars
Archive Program
REPOSITORIES
Topics
Trending
Collections
Enterprise
ENTERPRISE SOLUTIONS
Enterprise platform
AI-powered developer platform
AVAILABLE ADD-ONS
GitHub Advanced Security
Enterprise-grade security features
Copilot for Business
Enterprise-grade AI features
Premium Support
Enterprise-grade 24/7 support
Pricing
Type
/
to search
Sign in
Sign up
Appearance settings
You signed in with another tab or window.
Reload
to refresh your session.
You signed out in another tab or window.
Reload
to refresh your session.
You switched accounts on another tab or window.
Reload
to refresh your session.
Dismiss alert
{{ message }}
Uh oh!
There was an error while loading.
Please reload this page
.
NVIDIA
/
TransformerEngine
Public
Notifications
You must be signed in to change notification settings
Fork
795
Star
3.5k
Code
Issues
156
Pull requests
161
Discussions
Actions
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Discussions
Actions
Security and quality
Insights
[PyT] Update THD sink attention logic for cudnn >=9.18.0
- #2568
#2568
Merged
cyanguwa
merged 16 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
cuichenx:chcui/thd_cp_sink_attn
cuichenx/TransformerEngine:chcui/thd_cp_sink_attn
Copy head branch name to clipboard
Jan 22, 2026
Conversation
Commits
16
(16)
Checks
Files changed
Merged
[PyT] Update THD sink attention logic for cudnn >=9.18.0
#2568
cyanguwa
merged 16 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
cuichenx:chcui/thd_cp_sink_attn
cuichenx/TransformerEngine:chcui/thd_cp_sink_attn
Copy head branch name to clipboard
Commits
Commits on Jan 5, 2026
Update THD sink attention logic for newer cudnn versions
Show description for fbb0702
cuichenx
authored
fbb0702
View commit details
Copy full SHA for fbb0702
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for 01848c0
pre-commit-ci[bot]
committed
01848c0
View commit details
Copy full SHA for 01848c0
Browse repository at this point
Commits on Jan 6, 2026
Merge branch 'main' into patch-1
cyanguwa
authored
442699c
View commit details
Copy full SHA for 442699c
Browse repository at this point
update thd sink attention logic for cp>1
Show description for ab13ba0
cuichenx
committed
ab13ba0
View commit details
Copy full SHA for ab13ba0
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for c2c4341
pre-commit-ci[bot]
committed
c2c4341
View commit details
Copy full SHA for c2c4341
Browse repository at this point
Commits on Jan 7, 2026
add unit test for thd + sink attention
Show description for 392a033
cuichenx
committed
392a033
View commit details
Copy full SHA for 392a033
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for a75af44
pre-commit-ci[bot]
committed
a75af44
View commit details
Copy full SHA for a75af44
Browse repository at this point
Commits on Jan 8, 2026
Merge branch 'main' into chcui/thd_cp_sink_attn
cyanguwa
authored
4322420
View commit details
Copy full SHA for 4322420
Browse repository at this point
address comments
Show description for 200632f
cuichenx
committed
200632f
View commit details
Copy full SHA for 200632f
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for 7f4333a
pre-commit-ci[bot]
committed
7f4333a
View commit details
Copy full SHA for 7f4333a
Browse repository at this point
Commits on Jan 9, 2026
Merge branch 'main' into chcui/thd_cp_sink_attn
cyanguwa
authored
1212b75
View commit details
Copy full SHA for 1212b75
Browse repository at this point
Commits on Jan 20, 2026
do not skip thd cp sink attention test
Show description for 549888e
cuichenx
authored
549888e
View commit details
Copy full SHA for 549888e
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for f318e8b
pre-commit-ci[bot]
committed
f318e8b
View commit details
Copy full SHA for f318e8b
Browse repository at this point
Merge branch 'main' into chcui/thd_cp_sink_attn
cyanguwa
authored
2113e03
View commit details
Copy full SHA for 2113e03
Browse repository at this point
Commits on Jan 22, 2026
disable deterministic mode for sink attention
Show description for 96e5c6d
cuichenx
committed
96e5c6d
View commit details
Copy full SHA for 96e5c6d
Browse repository at this point
Merge branch 'main' into chcui/thd_cp_sink_attn
cyanguwa
authored
33434b1
View commit details
Copy full SHA for 33434b1
Browse repository at this point
You can’t perform that action at this time.