get_memory_bytes() — pytorch Function Reference

Architecture documentation for the get_memory_bytes() function in utils.py from the pytorch codebase.

Function python DynamoBenchmarks Configuration called by 1

Entity Profile

DynamoBenchmarks→ Configuration→ get_memory_bytes() — pytorch Function Reference

Dependency Diagram

graph TD
  f0f1f380_c551_5bfd_3a89_2f91a2eabca0["get_memory_bytes()"]
  3d4e92c5_a13f_fad7_1514_52cd7fc93a06["benchmark_single_shape_for_backend()"]
  3d4e92c5_a13f_fad7_1514_52cd7fc93a06 -->|calls| f0f1f380_c551_5bfd_3a89_2f91a2eabca0
  style f0f1f380_c551_5bfd_3a89_2f91a2eabca0 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/genai_layers/utils.py lines 57–59

    def get_memory_bytes(self, args, kwargs) -> int:
        # Get the necessary memory access in bytes for the kernelßß
        raise NotImplementedError