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
800
Star
3.5k
Code
Issues
159
Pull requests
164
Discussions
Actions
Security and quality
0
Insights
Additional navigation options
Code
Issues
Pull requests
Discussions
Actions
Security and quality
Insights
Update FP8 scale-inverse in kernels with FP8 output
- #1083
#1083
Merged
timmoon10
merged 20 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
timmoon10:fuse-cast-and-scale-inv-update
timmoon10/TransformerEngine:fuse-cast-and-scale-inv-update
Copy head branch name to clipboard
Aug 21, 2024
Conversation
Commits
20
(20)
Checks
Files changed
Merged
Update FP8 scale-inverse in kernels with FP8 output
#1083
timmoon10
merged 20 commits into
NVIDIA:main
NVIDIA/TransformerEngine:main
from
timmoon10:fuse-cast-and-scale-inv-update
timmoon10/TransformerEngine:fuse-cast-and-scale-inv-update
Copy head branch name to clipboard
Commits
Commits on Aug 1, 2024
Perform scale-inv update in cast-transpose kernels
Show description for 711b77e
timmoon10
committed
711b77e
View commit details
Copy full SHA for 711b77e
Browse repository at this point
Commits on Aug 2, 2024
Perform scale-inv update in cast and activation kernels
Show description for c53bf0d
timmoon10
committed
c53bf0d
View commit details
Copy full SHA for c53bf0d
Browse repository at this point
Commits on Aug 5, 2024
Perform sclae-inv update in LayerNorm and RMSNorm kernels
Show description for 0a12bff
timmoon10
committed
0a12bff
View commit details
Copy full SHA for 0a12bff
Browse repository at this point
Perform scale-inv update after FP8 GEMMs
Show description for 20bed0b
timmoon10
committed
20bed0b
View commit details
Copy full SHA for 20bed0b
Browse repository at this point
Commits on Aug 6, 2024
Fuse casts and scale-inv updates in linear module
Show description for f65b3d1
timmoon10
committed
f65b3d1
View commit details
Copy full SHA for f65b3d1
Browse repository at this point
Fuse casts and scale-inv updates in layernorm-linear module
Show description for fa65672
timmoon10
committed
fa65672
View commit details
Copy full SHA for fa65672
Browse repository at this point
Simplify kernel to update FP8 scale-inv
Show description for a3c00ec
timmoon10
committed
a3c00ec
View commit details
Copy full SHA for a3c00ec
Browse repository at this point
Fix typos
Show description for 1182e3e
timmoon10
committed
1182e3e
View commit details
Copy full SHA for 1182e3e
Browse repository at this point
Commits on Aug 7, 2024
Debug amax update in layernorm kernels
Show description for c548e8a
timmoon10
committed
c548e8a
View commit details
Copy full SHA for c548e8a
Browse repository at this point
[pre-commit.ci] auto fixes from pre-commit.com hooks
Show description for a265eae
pre-commit-ci[bot]
committed
a265eae
View commit details
Copy full SHA for a265eae
Browse repository at this point
Merge branch 'main' into fuse-cast-and-scale-inv-update
timmoon10
authored
a47500d
View commit details
Copy full SHA for a47500d
Browse repository at this point
Commits on Aug 8, 2024
Debug test failures
Show description for 35b5a1b
timmoon10
committed
35b5a1b
View commit details
Copy full SHA for 35b5a1b
Browse repository at this point
Commits on Aug 9, 2024
Debug ONNX export
Show description for ad7b1f6
timmoon10
committed
ad7b1f6
View commit details
Copy full SHA for ad7b1f6
Browse repository at this point
Merge branch 'main' into fuse-cast-and-scale-inv-update
timmoon10
authored
9798cf8
View commit details
Copy full SHA for 9798cf8
Browse repository at this point
Commits on Aug 12, 2024
Merge branch 'main' into fuse-cast-and-scale-inv-update
Show description for e550929
timmoon10
committed
e550929
View commit details
Copy full SHA for e550929
Browse repository at this point
Commits on Aug 14, 2024
Merge branch 'main' into fuse-cast-and-scale-inv-update
timmoon10
authored
dca23c5
View commit details
Copy full SHA for dca23c5
Browse repository at this point
Commits on Aug 17, 2024
Review suggestion from @ptrendx
Show description for 4909a50
timmoon10
authored
4909a50
View commit details
Copy full SHA for 4909a50
Browse repository at this point
Merge branch 'main' into fuse-cast-and-scale-inv-update
timmoon10
authored
0d45a2e
View commit details
Copy full SHA for 0d45a2e
Browse repository at this point
Commits on Aug 19, 2024
Debug mismatched dtypes
Show description for 66ff01d
timmoon10
committed
66ff01d
View commit details
Copy full SHA for 66ff01d
Browse repository at this point
Merge branch 'main' into fuse-cast-and-scale-inv-update
timmoon10
committed
87a4309
View commit details
Copy full SHA for 87a4309
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}