Skip to content

Insert the **profiler_kwargs from pytorchprofiler into usual args  #19410

@gardiens

Description

@gardiens

📚 Documentation

Hello,
I was reading about lightning profiling which looks useful and especially how it can be linked with the pytorch profiler.
However, in the doc ( https://lightning.ai/docs/pytorch/stable/api/lightning.pytorch.profilers.PyTorchProfiler.html#pytorchprofiler ), there an argument called **profiler_kwargs with doc: Keyword arguments for the PyTorch profiler. This depends on your PyTorch version.
Apart from the fact that this is vague, These kwargs contain most of the important function information you would expect from a profiler:

  • use_cpu:bool=True to say that we want to monitor the CPU ,
  • use_cuda to indicate whether you want to monitor cuda via the profiler.

cc @Borda @carmocca

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions