Home / Function/ eager() — pytorch Function Reference

eager() — pytorch Function Reference

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

Entity Profile

Dependency Diagram

graph TD
  fc3c62fa_4ffc_e8d4_5ee1_c74572a1d012["eager()"]
  7f08e9ae_5ab7_6b54_e9d5_a34f8e037133["layernorm_ref()"]
  fc3c62fa_4ffc_e8d4_5ee1_c74572a1d012 -->|calls| 7f08e9ae_5ab7_6b54_e9d5_a34f8e037133
  e5a28a86_38cc_d5f9_6155_e07365660a34["layernorm_ref()"]
  fc3c62fa_4ffc_e8d4_5ee1_c74572a1d012 -->|calls| e5a28a86_38cc_d5f9_6155_e07365660a34
  style fc3c62fa_4ffc_e8d4_5ee1_c74572a1d012 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/genai_layers/kernels.py lines 680–687

    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.layernorm_ref(x, w)
        return lambda: torch.autograd.grad(
            y, [x, w], grad_outputs=dy, retain_graph=True
        )

Subdomains

Frequently Asked Questions

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

Analyze Your Own Codebase

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

Try Supermodel Free