Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Fix AMP for memory optimized options#111

Merged
mihir-db merged 1 commit into
databricks:mainfrom
mvpatel2000:mvpatel2000/fix-amp
May 21, 2024
Merged

Fix AMP for memory optimized options#111
mihir-db merged 1 commit into
databricks:mainfrom
mvpatel2000:mvpatel2000/fix-amp

Conversation

@mvpatel2000

Copy link
Copy Markdown
Contributor

Correctly casts inputs/weights in right precision for AMP, which may not operate correctly otherwise

@mihir-db mihir-db merged commit 39d3e3b into databricks:main May 21, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants