Home / Class/ RegisterCUDADispatch Class — pytorch Architecture

RegisterCUDADispatch Class — pytorch Architecture

Architecture documentation for the RegisterCUDADispatch class in DispatchStub.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/DispatchStub.h lines 333–338

template <typename DispatchStub>
struct RegisterCUDADispatch {
  RegisterCUDADispatch(DispatchStub &stub, typename DispatchStub::FnPtr value) {
    stub.set_cuda_dispatch_ptr(value);
  }
};

Source

Analyze Your Own Codebase

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

Try Supermodel Free