Skip to content

Add optional Kineto profiler for DeepPotPT#8

Open
caic99 wants to merge 1 commit intodevelfrom
codex/implement-profiling-logic-in-deeppotpt.cc
Open

Add optional Kineto profiler for DeepPotPT#8
caic99 wants to merge 1 commit intodevelfrom
codex/implement-profiling-logic-in-deeppotpt.cc

Conversation

@caic99
Copy link
Owner

@caic99 caic99 commented Aug 5, 2025

Summary

  • add AutoProfiler class to enable PyTorch Kineto profiling via DEEPMP_PROFILE
  • capture CPU and CUDA activities and save trace to deeppotpt_profile.json
  • start profiling in DeepPotPT compute methods when enabled

Testing

  • python -m pre_commit run --files source/api_cc/src/DeepPotPT.cc
  • pytest source/tests/pt/test_utils.py::TestCvt::test_to_numpy -q

https://chatgpt.com/codex/tasks/task_b_689071c50dac83329af18f8b921f30c8

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.

1 participant