pytorch Architecture Docs
Home / Class/ __device__ Class — pytorch Architecture

__device__ Class — pytorch Architecture

Architecture documentation for the __device__ class in ck_gemm_template.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/hip/ck_gemm_template.h lines 49–50

  template <typename C, typename AB>
  __host__ __device__ constexpr void operator()(C& c, const AB& ab) const;

Source

Analyze Your Own Codebase

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

Try Supermodel Free