pytorch Architecture Docs
Home / Class/ device_type Class — pytorch Architecture

device_type Class — pytorch Architecture

Architecture documentation for the device_type class in CachingHostAllocator.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/core/CachingHostAllocator.h lines 1077–1082

template <DeviceType device_type>
struct HostAllocatorRegistry {
  explicit HostAllocatorRegistry(HostAllocator* allocator) {
    at::setHostAllocator(device_type, allocator);
  }
};

Source

Analyze Your Own Codebase

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

Try Supermodel Free