[Intel MKL] SparseAdam: an optimizer to improve the accuracy of LazyAdam #24788
Closed
mpjlu wants to merge 7 commits intotensorflow:masterfrom
Closed
[Intel MKL] SparseAdam: an optimizer to improve the accuracy of LazyAdam #24788mpjlu wants to merge 7 commits intotensorflow:masterfrom
mpjlu wants to merge 7 commits intotensorflow:masterfrom
Conversation
* Temorarily commented out Conv2D fusion * Commented out Conv2D fusion tests.
Change-Id: I6a842c30c5838f7bc701156df0129cf2bf73abd2
Change-Id: I3274f135231d89f956a252f4e82d4f8bc6931ee8
)" This reverts commit 81b2969.
ymodak
added
awaiting review
Contributor
|
Normally I'd accept this, but we're in the process of transitioning out of contrib and into the new addons SIG ( https://groups.google.com/a/tensorflow.org/forum/#!forum/addons for more information). So until that release is done I'd like to soft-freeze contrib to make the migration easier. |
copybara-service bot
pushed a commit
that referenced
this pull request
Apr 8, 2025
Reason for the update: - The current HERMETIC_CUDA_VERSION used by XLA is 12.1.1 (released April 2023), which is now 2 years old. - The current HERMETIC_CUDNN_VERSION in XLA is 8.6 (released October 2022), making it 2.5 years old. - JAX currently uses HERMETIC CUDA 12.8.0 and cuDNN 9.8.0, indicating it's already on a newer stack. - I tested upgrading XLA to CUDA 12.6.3 and cuDNN 9.3.0 - all GitHub XLA PR tests passed: PR #24788. Release dates: - CUDA 12.6.3: November 2024 - cuDNN 9.3.0: August 2024 PiperOrigin-RevId: 744868619
copybara-service bot
pushed a commit
that referenced
this pull request
Apr 9, 2025
Reason for the update: - The current HERMETIC_CUDA_VERSION used by XLA is 12.1.1 (released April 2023), which is now 2 years old. - The current HERMETIC_CUDNN_VERSION in XLA is 8.6 (released October 2022), making it 2.5 years old. - JAX currently uses HERMETIC CUDA 12.8.0 and cuDNN 9.8.0, indicating it's already on a newer stack. - I tested upgrading XLA to CUDA 12.6.3 and cuDNN 9.3.0 - all GitHub XLA PR tests passed: PR #24788. Release dates: - CUDA 12.6.3: November 2024 - cuDNN 9.3.0: August 2024 PiperOrigin-RevId: 744868619
copybara-service bot
pushed a commit
that referenced
this pull request
Apr 9, 2025
Reason for the update: - The current HERMETIC_CUDA_VERSION used by XLA is 12.1.1 (released April 2023), which is now 2 years old. - The current HERMETIC_CUDNN_VERSION in XLA is 8.6 (released October 2022), making it 2.5 years old. - JAX currently uses HERMETIC CUDA 12.8.0 and cuDNN 9.8.0, indicating it's already on a newer stack. - I tested upgrading XLA to CUDA 12.6.3 and cuDNN 9.3.0 - all GitHub XLA PR tests passed: PR #24788. Release dates: - CUDA 12.6.3: November 2024 - cuDNN 9.3.0: August 2024 PiperOrigin-RevId: 745637141
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
5 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
No description provided.