Home / Function/ cast_to_bf16() — pytorch Function Reference

cast_to_bf16() — pytorch Function Reference

Architecture documentation for the cast_to_bf16() function in common.py from the pytorch codebase.

Function python DynamoBenchmarks Microbenchmarking calls 1 called by 1

Entity Profile

Dependency Diagram

graph TD
  e789c0d4_b1f7_d69d_736c_fb5c020174d2["cast_to_bf16()"]
  da8c81f0_c44b_4b89_914d_318c84a98a42["cast_based_on_args()"]
  da8c81f0_c44b_4b89_914d_318c84a98a42 -->|calls| e789c0d4_b1f7_d69d_736c_fb5c020174d2
  bbf77550_9fd0_5fa7_61ba_9de1b1cde65f["cast_to()"]
  e789c0d4_b1f7_d69d_736c_fb5c020174d2 -->|calls| bbf77550_9fd0_5fa7_61ba_9de1b1cde65f
  style e789c0d4_b1f7_d69d_736c_fb5c020174d2 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/common.py lines 1698–1699

def cast_to_bf16(model, inputs):
    return cast_to(torch.bfloat16, model, inputs)

Domain

Subdomains

Calls

Called By

Source

Frequently Asked Questions

What does cast_to_bf16() do?
cast_to_bf16() is a function in the pytorch codebase.
What does cast_to_bf16() call?
cast_to_bf16() calls 1 function(s): cast_to.
What calls cast_to_bf16()?
cast_to_bf16() is called by 1 function(s): cast_based_on_args.

Analyze Your Own Codebase

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

Try Supermodel Free