pytorch Architecture Docs
Home / Class/ bgemm Class — pytorch Architecture

bgemm Class — pytorch Architecture

Architecture documentation for the bgemm class in CUDABlas.cpp from the pytorch codebase.

Class cpp

Entity Profile

Source Code

aten/src/ATen/cuda/CUDABlas.cpp lines 963–972

template <>
void bgemm<double>(CUDABLAS_BGEMM_ARGTYPES(double)) {
  auto tuning_ctx = at::cuda::tunable::getTuningContext();
  if (tuning_ctx->IsTunableOpEnabled()) {
    bgemm_tunable<double>(CUDABLAS_BGEMM_ARGS(double));
  }
  else {
    bgemm_internal<double>(CUDABLAS_BGEMM_ARGS(double));
  }
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free