Skip to content

Distributed RPC profiling should support record_shapes #39969

@rohan-varma

Description

@rohan-varma

🚀 Feature

We currently don't carry over the record_shapes flag and return the recorded shapes over RPC when we remotely invoke the profiler, we should add support for this for parity with regular profiling.

cc @pietern @mrshenli @pritamdamania87 @zhaojuanmao @satgera @gqchen @aazzolini @rohan-varma @xush6528 @jjlilley @osalpekar @jiayisuse

Metadata

Metadata

Assignees

Labels

better-engineeringRelatively self-contained tasks for better engineering contributorsmodule: rpcRelated to RPC, distributed autograd, RRef, and distributed optimizeroncall: profilerprofiler-related issues (cpu, gpu, kineto)triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions