eager() — pytorch Function Reference

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

Function python DynamoBenchmarks Configuration calls 1

Entity Profile

DynamoBenchmarks→ Configuration→ eager() — pytorch Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

benchmarks/dynamo/genai_layers/kernels.py lines 389–393

    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.rms_norm_ref(x, w)