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 profiling utility pytorch #2946

Merged

Conversation

quic-kyunggeu
Copy link
Contributor

Main Changes

  1. pytorch profile function will call torch.cuda.synchronize() by default for accurate speed measurement
  2. Adjusted function signature of profile such that only "label" is a mandatory argument, and output will be sent to stdout by default
  3. profile can be also used as a decorator

quic-akhobare
quic-akhobare previously approved these changes May 13, 2024
Signed-off-by: Kyunggeun Lee <quic_kyunggeu@quicinc.com>
Signed-off-by: Kyunggeun Lee <quic_kyunggeu@quicinc.com>
Signed-off-by: Kyunggeun Lee <quic_kyunggeu@quicinc.com>
Signed-off-by: Kyunggeun Lee <quic_kyunggeu@quicinc.com>
@quic-kyunggeu quic-kyunggeu merged commit e088e15 into quic:develop May 24, 2024
3 checks passed
@quic-kyunggeu quic-kyunggeu deleted the add_profiling_utility_pytorch branch May 24, 2024 21:46
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.

None yet

2 participants