Home
last modified time | relevance | path

Searched refs:PRINT_WARP_ID (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/
H A Ddebug_utils.h27 #define PRINT_WARP_ID 0 macro
31 threadIdx.x == PRINT_LANE_ID && threadIdx.y == PRINT_WARP_ID && \
36 if (threadIdx.x == PRINT_LANE_ID && threadIdx.y == PRINT_WARP_ID && \