Home / Class/ pin_memory Class — pytorch Architecture

pin_memory Class — pytorch Architecture

Architecture documentation for the pin_memory class in MiscUtils.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/cuda/MiscUtils.h lines 20–29

template<class T>
static inline Storage pin_memory(int64_t size) {
  auto* allocator = cuda::getPinnedMemoryAllocator();
  int64_t adjusted_size = size * sizeof(T);
  return Storage(
      Storage::use_byte_size_t(),
      adjusted_size,
      allocator,
      /*resizable=*/false);
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free