device() — pytorch Function Reference

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

Function python CoreTensor Registration

Entity Profile

CoreTensor→ Registration→ device() — pytorch Function Reference

Relationship Graph

Source Code

benchmarks/distributed/bench_nvshmem_tile_reduce.py lines 48–49

    def device(self) -> torch.device:
        return torch.device(device_type, self.rank)

Domain

CoreTensor

Subdomains

Registration

Source

View on GitHub

Analyze Your Own Codebase

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

Try Supermodel Free