Skip to content

Fix #944: Add GPU metrics and optimizations to testspeed benchmark#950

Closed
Tanisha-Sharma005 wants to merge 5 commits intogoogle-deepmind:mainfrom
Tanisha-Sharma005:fix-testspeed-info-944
Closed

Fix #944: Add GPU metrics and optimizations to testspeed benchmark#950
Tanisha-Sharma005 wants to merge 5 commits intogoogle-deepmind:mainfrom
Tanisha-Sharma005:fix-testspeed-info-944

Conversation

@Tanisha-Sharma005
Copy link
Copy Markdown

Summary

Enhanced testspeed.py benchmark with GPU device information, memory usage reporting, and ray_test.py performance optimizations for better debugging and profiling.

Changes

  • GPU Metrics: Reports device name and memory usage (used/total GB)
  • Benchmark Summary: Structured performance metrics output
  • ray_test.py: Optimized array operations and reduced allocations
  • Trace Events: Step-by-step timing for performance analysis

Testing

Verified on NVIDIA RTX 3060 with comprehensive GPU/CPU benchmark comparison.

Closes #944

✅ GPU device name + memory usage (used/total GB)
✅ Better benchmark summary
✅ ray_test.py optimizations

Closes google-deepmind#944
@google-cla
Copy link
Copy Markdown

google-cla Bot commented Dec 29, 2025

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

@Tanisha-Sharma005
Copy link
Copy Markdown
Author

@thowell

CLA signed with tanishaasharma20@gmail.comhttps://cla.developers.google.com
Git email fixed. Ready for maintainer approval.

Please review & approve!

Thanks!

@thowell
Copy link
Copy Markdown
Collaborator

thowell commented Jan 29, 2026

@Tanisha-Sharma005 thank you for contributing to mujoco warp! closing this pr since we are not planning to merge the pr. if you are interested in contributing further to the project please follow up on issues marked 'good first issue'. thanks!

@thowell thowell closed this Jan 29, 2026
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.

add more info to testspeed results

2 participants