feat: add MLU torch profiler support - #9
Conversation
Add Cambricon MLU profiler integration that monkey-patches verl's torch profiler to support MLU device activities. The patch adds 'mlu' to allowed contents in TorchProfilerToolConfig and uses torch.profiler.ProfilerActivity.MLU when available. Changes: - Add verl_hardware_plugin/profilers/ module with torch_profile_mlu - Register profiler patches in plugin init - Add profiling guide to MLU user docs - Fix directory reference (user_guide_flagos -> user_guide_mlu) in README Co-Authored-By: Claude <noreply@anthropic.com>
|
Caution The consumer version of Gemini Code Assist on GitHub has been sunset. All code review activity has officially ceased. |
physics31415926
left a comment
There was a problem hiding this comment.
please polish the pr description and attach a test result
|
PROFILER=( |
done |
| @@ -0,0 +1,29 @@ | |||
| """Profiler integrations via monkey-patching verl's profiler utilities.""" | |||
There was a problem hiding this comment.
please add more discription with copyeight, referring to verl_hardware_plugin\platforms_init_.py
Added copyright and license information to the file.
Updated licensing information and added copyright notice.


Add Cambricon MLU profiler integration that monkey-patches verl's torch profiler to support MLU device activities. The patch adds 'mlu' to allowed contents in TorchProfilerToolConfig and uses torch.profiler.ProfilerActivity.MLU when available.
Changes:
Checklist
pre-commitchecks