Skip to content

[RL][Feature Request] Tun on torch.compile + cudagraphs for trainer definition #2508

@Lucaskabela

Description

@Lucaskabela

With Generator sped up by vllm's support_torch_compile, the new bottleneck is trainer.

Let's enable torch.compile and cudagraph there to get similar speedups

Metadata

Metadata

Assignees

Labels

No labels
No labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions