Home
last modified time | relevance | path

Searched defs:ScalingCoefs (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/
Dkernel_forward.h514 struct ScalingCoefs { struct
515 cutlass::Array<accum_t, kQueriesPerBlock> m_prime;
516 cutlass::Array<accum_t, kQueriesPerBlock> s_prime;
517 cutlass::Array<accum_t, kQueriesPerBlock> mi;
518 cutlass::Array<accum_t, kQueriesPerBlock> out_rescale;
520 addition_storage;