conditional_data_ptr Class — pytorch Architecture

Architecture documentation for the conditional_data_ptr class in FlashAttentionKernel.cpp from the pytorch codebase.

Class cpp

Source Code

aten/src/ATen/native/cpu/FlashAttentionKernel.cpp lines 160–164

template <typename scalar_t>
inline scalar_t* conditional_data_ptr(scalar_t* ptr, scalar_t* ptr2) {
  TORCH_CHECK(ptr2 == nullptr);
  return ptr;
}

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