Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dint4mm.cu1175 #define RUN_GEMM(WARPS, K_TILES_PER_WARP, Q_GROUP_SIZE, REDUCE_TYPE) \ in _weight_int4pack_mm_cuda() argument
1255 #define HANDLE_Q_GROUP(WARPS, K_TILES_PER_WARP, REDUCE_TYPE) \ in _weight_int4pack_mm_cuda() argument