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

Improve kokkos trace handling. #573

Merged
merged 6 commits into from
Feb 22, 2025

Conversation

xuchen-amd
Copy link
Contributor

@xuchen-amd xuchen-amd commented Feb 20, 2025

  • refactored run_prof() to improve readability for rocprofv3 output processing
  • added --kokkos-trace output processing for rocprofv3 support
  • updated join_prof() with --hip-trace and --kokkos-trace compatibility
  • tested locally with ctest

@xuchen-amd xuchen-amd changed the title Improve kokkos trace is handling. Improve kokkos trace handling. Feb 20, 2025
results_files_csv = glob.glob(workload_dir + "/out/pmc_1/*/*_converted.csv")
elif is_timestamps:
# when the input is timestamps, we know counter csv file is not generated and will instead parse kernel trace file
results_files_csv = glob.glob(
Copy link
Contributor

Choose a reason for hiding this comment

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

Not a big deal but path lib should be better than glob

Copy link
Contributor Author

Choose a reason for hiding this comment

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

results_files_csv is returned and to expected to be looped through as string type to pass to pd.read_csv(). In future refactoring, I can address this problem to handle paths with more robustness.

Copy link
Contributor

@feizheng10 feizheng10 left a comment

Choose a reason for hiding this comment

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

Some minor questions.

Copy link
Contributor

@feizheng10 feizheng10 left a comment

Choose a reason for hiding this comment

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

Please make sure CI passed.

@xuchen-amd xuchen-amd merged commit 10e1d57 into ROCm:develop Feb 22, 2025
11 checks passed
@xuchen-amd xuchen-amd deleted the improve_kokkos_trace branch February 22, 2025 19:07
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.

2 participants