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
798
Star
3.5k
Code
Issues
158
Pull requests
160
Discussions
Actions
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Discussions
Actions
Security and quality
Insights
[PyT] [Common] add support for enabling cuda graph under thd format in megatron.
- #2898
#2898
Merged
timmoon10
merged 12 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
HaochenYuan:thd_cuda_graph_0415
HaochenYuan/TransformerEngine:thd_cuda_graph_0415
Copy head branch name to clipboard
Jul 1, 2026
Conversation
Commits
12
(12)
Checks
Files changed
Merged
[PyT] [Common] add support for enabling cuda graph under thd format in megatron.
#2898
timmoon10
merged 12 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
HaochenYuan:thd_cuda_graph_0415
HaochenYuan/TransformerEngine:thd_cuda_graph_0415
Copy head branch name to clipboard
Commits
Commits on Jun 5, 2026
add support for THD CUDA graph
Show description for 2c58118
HaochenYuan
committed
2c58118
View commit details
Copy full SHA for 2c58118
Browse repository at this point
Commits on Jun 6, 2026
modify comment
Show description for b6aa597
HaochenYuan
committed
b6aa597
View commit details
Copy full SHA for b6aa597
Browse repository at this point
Commits on Jun 8, 2026
Merge branch 'main' into thd_cuda_graph_0415
HaochenYuan
authored
69fddf7
View commit details
Copy full SHA for 69fddf7
Browse repository at this point
Commits on Jun 9, 2026
address @timmoon10: drop FAv2-bwd alloc gate, rely on THD tail zero-fill
Show description for 805746a
HaochenYuan
committed
805746a
View commit details
Copy full SHA for 805746a
Browse repository at this point
Commits on Jun 11, 2026
Support graph-safe MoE aux loss token count
Show description for a2cc7ef
HaochenYuan
committed
a2cc7ef
View commit details
Copy full SHA for a2cc7ef
Browse repository at this point
add graph guard for one zero fill
Show description for 21d8d3a
HaochenYuan
committed
21d8d3a
View commit details
Copy full SHA for 21d8d3a
Browse repository at this point
remove redundant zero-fill
Show description for cee7b6a
HaochenYuan
committed
cee7b6a
View commit details
Copy full SHA for cee7b6a
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for 823805d
pre-commit-ci[bot]
committed
823805d
View commit details
Copy full SHA for 823805d
Browse repository at this point
Commits on Jun 30, 2026
rename & remove prelude kernel
Show description for bce6bc5
HaochenYuan
committed
bce6bc5
View commit details
Copy full SHA for bce6bc5
Browse repository at this point
Merge branch 'main' into thd_cuda_graph_0415
timmoon10
authored
edfdb78
View commit details
Copy full SHA for edfdb78
Browse repository at this point
Update warp reduction function
Show description for a69fb1b
timmoon10
authored
a69fb1b
View commit details
Copy full SHA for a69fb1b
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for 1bc41e5
pre-commit-ci[bot]
committed
1bc41e5
View commit details
Copy full SHA for 1bc41e5
Browse repository at this point
You can’t perform that action at this time.
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}
