C10_RESTRICT Class — pytorch Architecture
Architecture documentation for the C10_RESTRICT class in SumKernel.cpp from the pytorch codebase.
Entity Profile
Source Code
aten/src/ATen/native/cpu/SumKernel.cpp lines 280–286
template <typename scalar_t, typename acc_t>
struct CastStoreAccumulate {
static void store(char * C10_RESTRICT data, int64_t stride, int64_t index, acc_t value) {
auto * ptr = reinterpret_cast<scalar_t*>(data + index * stride);
*ptr += value;
}
};
Source
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free