Home / Class/ Alignment Class — pytorch Architecture

Alignment Class — pytorch Architecture

Architecture documentation for the Alignment class in IndexKernelUtils.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/cuda/IndexKernelUtils.h lines 29–32

template <int64_t Alignment, typename index_t>
void vectorized_gather_kernel_launch(char * out, char * inp, index_t * idx, int num_ind,
                                     int64_t slice_size_in_bytes, int64_t ind_dim_size, int64_t inp_stride_bytes, int64_t out_stride_bytes,
                                     bool allow_neg_indices=false);

Source

Analyze Your Own Codebase

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

Try Supermodel Free