eager() — pytorch Function Reference

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

Function python DynamoBenchmarks Microbenchmarking calls 1

Entity Profile

DynamoBenchmarks→ Microbenchmarking→ eager() — pytorch Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

benchmarks/dynamo/genai_layers/kernels.py lines 600–604

    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 = args
        return lambda: self.layernorm_ref(x, w)