Skip to content

Conversation

@jpvillam-amd
Copy link

@jpvillam-amd jpvillam-amd commented Apr 12, 2024

The goal is to allow something like:

import torch
from rpdTracerControl import rpdTracerControl

for step in [1,2,3]:
    trace_file_name = f"{step}_torch.rpd"
    print(f"in {step}!!")
    with rpdTracerControl(trace_file_name, nvtx=True) as p:
        a =torch.rand(step).cuda()
        torch.add(a, 20)
    p.top_totals()
  • Make rpdTracerControl into a standalone context manager similar to the functionality of torch.profile
  • Make it all callable in a loop, regardless of the static nature of the library

@jpvillam-amd jpvillam-amd self-assigned this Apr 12, 2024
@jpvillam-amd jpvillam-amd requested a review from mwootton April 12, 2024 20:20
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