Home / Class/ RegisterHIPDispatch Class — pytorch Architecture

RegisterHIPDispatch Class — pytorch Architecture

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

Class c

Entity Profile

Source Code

aten/src/ATen/native/DispatchStub.h lines 361–367

template <typename DispatchStub>
struct RegisterHIPDispatch {
  RegisterHIPDispatch(DispatchStub &stub, typename DispatchStub::FnPtr value) {
    // TODO: make this point at hip_dispatch_ptr
    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