Skip to content

[Bug]: PostGradPassManager should not reference the compilation_config #27028

@zou3519

Description

@zou3519

Your current environment

main

🐛 Describe the bug

Inductor configs need to be serializable and deepcopy-able, the PostGradPassManager is added to the Inductor config, and the CompilationConfig holds some nn.Modules. The Inductor config gets deepcopied sometimes, plus we have observed OOMs due to nn.Modules being in the PostGradPassManager.

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    Status

    Done

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions