Home / Function/ _init_device() — pytorch Function Reference

_init_device() — pytorch Function Reference

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

Function python CoreTensor Dispatcher called by 1

Entity Profile

Dependency Diagram

graph TD
  89ca7f4b_c6a3_7849_81c9_ba6dfa92cd70["_init_device()"]
  1fb81840_0784_4f41_94ba_38ef34fe84e5["_benchmark_tile_reduce_single()"]
  1fb81840_0784_4f41_94ba_38ef34fe84e5 -->|calls| 89ca7f4b_c6a3_7849_81c9_ba6dfa92cd70
  style 89ca7f4b_c6a3_7849_81c9_ba6dfa92cd70 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/distributed/bench_nvshmem_tile_reduce.py lines 41–45

    def _init_device(self) -> None:
        # TODO: relieve this (seems to hang if without)
        device_module.set_device(self.device)
        # Set NVSHMEM as SymmMem backend
        symm_mem.set_backend("NVSHMEM")

Domain

Subdomains

Called By

Source

Frequently Asked Questions

What does _init_device() do?
_init_device() is a function in the pytorch codebase.
What calls _init_device()?
_init_device() is called by 1 function(s): _benchmark_tile_reduce_single.

Analyze Your Own Codebase

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

Try Supermodel Free