Searched defs:nsplits (Results 1 – 3 of 3) sorted by relevance
64 __global__ void flash_bwd_convert_dq_kernel(const Flash_bwd_params params, const int nsplits) { in flash_bwd_convert_dq_kernel()
181 inline __device__ void convert_dQ(const Params ¶ms, const int nsplits) { in convert_dQ()
941 …const int nsplits = (dprops->multiProcessorCount + batch_size * num_heads - 1) / (batch_size * num… in mha_bwd() local1172 …const int nsplits = (dprops->multiProcessorCount + batch_size * num_heads - 1) / (batch_size * num… in mha_varlen_bwd() local