Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

馃殌 perf: use half in Metal kernels #121

Merged
merged 17 commits into from
May 22, 2024
Merged

馃殌 perf: use half in Metal kernels #121

merged 17 commits into from
May 22, 2024

Conversation

jean-francoisreboud
Copy link
Collaborator

@jean-francoisreboud jean-francoisreboud commented May 22, 2024 •
edited
Loading

Changes:

  • use half in Metal kernels
  • fix when nbElems < batch size in FloatBuffer
  • FlowPrecisionTrainer
  • add tests for Float16 GPU flow
  • forceFloat to enforce Float precision

@jean-francoisreboud jean-francoisreboud self-assigned this May 22, 2024
@jean-francoisreboud jean-francoisreboud merged commit ceff714 into release_5 May 22, 2024
4 checks passed
@jean-francoisreboud jean-francoisreboud deleted the jfr/half branch May 22, 2024 16:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees

@jean-francoisreboud jean-francoisreboud

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
1 participant
@jean-francoisreboud