ray-project / ray

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
https://ray.io
Apache License 2.0
34.24k stars 5.81k forks source link

[aDAG] Support torch profiling with configurable parameters #47745

Open ruisearch42 opened 2 months ago

ruisearch42 commented 2 months ago

Description

Example parameter:

See https://github.com/ray-project/ray/pull/47586#discussion_r1767004227

Use case

No response

ruisearch42 commented 1 month ago

This needs to wait before we have torch profiling support in the first place, which I'm planning to do.