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.

Sign up for GitHub

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

Jump to bottom

Enabling Kineto dumping on Pytorch #82

Closed

rashidi1saeed opened this issue Jul 26, 2023 · 2 comments

rashidi1saeed commented Jul 26, 2023

Hi,
I am interested in enabling kineto traces on Pytorch. By looking at the Param code I was able to roughly understand how to enable kineto on pytorch. But the problem is that kineto dumps on pytorch generates only a single operator. So can you guys give me some instructions on how to correctly enable kineto on pytroch models?

Author

rashidi1saeed commented Jul 26, 2023

For example, I am interested in enabling kineto and graph observer for the mlperf DLRM file:
https://github.com/mlcommons/training/blob/master/recommendation_v2/torchrec_dlrm/dlrm_main.py

I modified the DLRM main file as follows:

1 (1)

2 (1)

3 (1)

However, while graph observer works perfectly, Kineto traces have only one op. Where is my issue?

srinivas212 commented Jul 26, 2023

@rashidi1saeed this question is better suited to Kineto https://github.com/pytorch/kineto?
Can you check there?

srinivas212 closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment