Register audit for kernel flash_attn_bc128 (kernels/attention/flash_attention/flash_br16_bc128.sm_86.cubin)
Registers per thread: 158
Block size: 128
Registers per block: 20224
Static shared memory (KB): 0
Dynamic shared memory: FALSE
Local bytes: 0
Register spill: FALSE
Constant bytes: 396
Theoretical blocks per SM: 3
Warps per SM: 12