RegisterMPSDispatch Class — pytorch Architecture
Architecture documentation for the RegisterMPSDispatch class in DispatchStub.h from the pytorch codebase.
Entity Profile
Source Code
aten/src/ATen/native/DispatchStub.h lines 354–359
template <typename DispatchStub>
struct RegisterMPSDispatch {
RegisterMPSDispatch(DispatchStub &stub, typename DispatchStub::FnPtr value) {
stub.set_mps_dispatch_ptr(value);
}
};
Source
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free