Home / Function/ requires_nvshmem() — pytorch Function Reference

requires_nvshmem() — pytorch Function Reference

Architecture documentation for the requires_nvshmem() function in bench_nvshmem_tile_reduce.py from the pytorch codebase.

Function python CoreTensor Boxing

Entity Profile

Relationship Graph

Source Code

benchmarks/distributed/bench_nvshmem_tile_reduce.py lines 26–30

def requires_nvshmem():
    return skip_but_pass_in_sandcastle_if(
        not symm_mem.is_nvshmem_available(),
        "bench_nvshmem_tile_reduce requires NVSHMEM, skipping benchmark",
    )

Domain

Subdomains

Source

Analyze Your Own Codebase

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

Try Supermodel Free