Skip to content

fwd_rng_state show up in the aot_export_joint grpah input #164559

@SherlockNoMad

Description

@SherlockNoMad

See pytorch/torchtitan#1794

P1975157784: rank0_autograd_function_0fea2786.py

Setting torch._functorch.config.graphsafe_rng_functionalization = False doesn't work.

How to avoid fwd_rng_state from showing up?

cc @chauhang @penguinwu @zou3519 @bdhirsh

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: aotdispatchumbrella label for AOTAutograd issuesmodule: pt2-dispatcherPT2 dispatcher-related issues (e.g., aotdispatch, functionalization, faketensor, custom-op,oncall: pt2

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions