“The `flash_kda.fwd` kernel API currently requires the key (K) and value (V) dimensions to both be 128.”