Home / Function/ eager() — pytorch Function Reference

eager() — pytorch Function Reference

Architecture documentation for the eager() function in kernels.py from the pytorch codebase.

Function python DynamoBenchmarks Configuration calls 2

Entity Profile

Dependency Diagram

graph TD
  3a2f2893_5914_e379_2f43_005a0538e035["eager()"]
  2b63d71e_33f7_1a73_5259_ed1e703de26b["rms_norm_ref()"]
  3a2f2893_5914_e379_2f43_005a0538e035 -->|calls| 2b63d71e_33f7_1a73_5259_ed1e703de26b
  ca055034_d94c_dc37_1965_ddd3537c220d["rms_norm_ref()"]
  3a2f2893_5914_e379_2f43_005a0538e035 -->|calls| ca055034_d94c_dc37_1965_ddd3537c220d
  style 3a2f2893_5914_e379_2f43_005a0538e035 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/genai_layers/kernels.py lines 485–492

    def eager(self, args, kwargs=None) -> Any:
        if kwargs is not None:
            raise AssertionError(f"Expected kwargs to be None, but got {kwargs}")
        x, w, dy = args
        y = self.rms_norm_ref(x, w)
        return lambda: torch.autograd.grad(
            y, [x, w], grad_outputs=dy, retain_graph=True
        )

Domain

Subdomains

Calls

Source

Frequently Asked Questions

What does eager() do?
eager() is a function in the pytorch codebase.
What does eager() call?
eager() calls 2 function(s): rms_norm_ref, rms_norm_ref.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free