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.

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

Already on GitHub? Sign in to your account

Add Apple MPS training test #9960

Open
wants to merge 10 commits into
base: main
Choose a base branch
from
Open

Add Apple MPS training test #9960

wants to merge 10 commits into from

Conversation

glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented Apr 10, 2024

πŸ› οΈ PR Summary

Made with ❀️ by Ultralytics Actions

🌟 Summary

Enhanced testing suite for Ultralytics on Apple MPS devices πŸπŸ’».

πŸ“Š Key Changes

  • Imported new identifiers (ARM64, MACOS) to improve device-specific checks.
  • Added a new test to validate YOLO training specifically on Apple MPS (Metal Performance Shaders) devices.

🎯 Purpose & Impact

  • Purpose: To ensure Ultralytics' YOLO models train effectively on devices leveraging Apple's MPS, catering specifically to ARM64 and macOS environments. This inclusivity ensures broader compatibility and performance optimization.
  • Impact: Users with Apple MPS devices can expect more reliable performance, with testing now covering these environments. This update promises a smoother experience for a significant user segment utilizing ARM64-based macOS devices for machine learning tasks.

Copy link

codecov bot commented Apr 10, 2024

Codecov Report

All modified and coverable lines are covered by tests βœ…

Project coverage is 70.61%. Comparing base (972f615) to head (f75e17a).

Additional details and impacted files
@@            Coverage Diff             @@
##             main    #9960      +/-   ##
==========================================
- Coverage   70.61%   70.61%   -0.01%     
==========================================
  Files         122      122              
  Lines       15584    15584              
==========================================
- Hits        11005    11004       -1     
- Misses       4579     4580       +1     
Flag Coverage Ξ”
Benchmarks 35.61% <ΓΈ> (ΓΈ)
GPU 37.37% <ΓΈ> (-0.02%) ⬇️
Tests 66.63% <ΓΈ> (-0.09%) ⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

β˜” View full report in Codecov by Sentry.
πŸ“’ Have feedback on the report? Share it here.

@glenn-jocher
Copy link
Member Author

glenn-jocher commented Apr 10, 2024

Seems like CI runners are preventing access to MPS memory, so test results here are not valid.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants