Home / Class/ ThrustAllocator Class — pytorch Architecture

ThrustAllocator Class — pytorch Architecture

Architecture documentation for the ThrustAllocator class in ThrustAllocator.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/cuda/ThrustAllocator.h lines 10–21

class ThrustAllocator {
public:
  typedef char value_type;

  char* allocate(std::ptrdiff_t size) {
    return static_cast<char*>(c10::cuda::CUDACachingAllocator::raw_alloc(size));
  }

  void deallocate(char* p, size_t size) {
    c10::cuda::CUDACachingAllocator::raw_delete(p);
  }
};

Source

Analyze Your Own Codebase

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

Try Supermodel Free