Skip to content

A tracking utility for gathering the compile and/or runtime time, size, profiling and other statistics #4777

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

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

AndreyPavlenko
Copy link
Contributor

@AndreyPavlenko AndreyPavlenko commented Jul 25, 2025

To enable the tracking, set the environment variable TRITON_TRACK_DUMP to either 1, true, yes, on, y or a path to a directory where the tracking reports will be dumped.
To add the profiling statistics to the reports, set the TRITON_TRACK_PROFILE environment variable.
To track the kernel launches, set the TRITON_TRACK_RUN environment variable.

Link #4716

@AndreyPavlenko AndreyPavlenko force-pushed the AndreyPavlenko/track branch 3 times, most recently from 41015d0 to 1216480 Compare July 25, 2025 20:48
@AndreyPavlenko AndreyPavlenko changed the title Implemented compile time/size tracking and profiling utility A tracking utility for gathering the compile and/or runtime time, size, profiling and other statistics Jul 25, 2025
@AndreyPavlenko AndreyPavlenko force-pushed the AndreyPavlenko/track branch 2 times, most recently from 7843958 to 9752167 Compare July 29, 2025 13:44
@AndreyPavlenko AndreyPavlenko marked this pull request as ready for review July 29, 2025 18:26
Comment on lines -268 to -269
},
py::call_guard<py::gil_scoped_release>());
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why removed?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It doesn't allow calling the callback function.

@anmyachev
Copy link
Contributor

I would also add tests for this utility so that the code does not become outdated unexpectedly.

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.

3 participants