We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 91ca3d1 commit 824393eCopy full SHA for 824393e
1 file changed
custom_ops/gpu_ops/append_attn/mla_cache_kernel.cuh
@@ -217,7 +217,9 @@ __global__ void prefill_absorb_cache_kernel(
217
printf("block_idx1 %d != block_idx %d\n", block_idx1, block_idx);
218
printf("token_idx %d\n", token_idx);
219
printf("slot_mapping %d\n", slot_mapping[token_idx]);
220
- asm volatile("trap;");
+ #ifndef PADDLE_WITH_CUSTOM_DEVICE_METAX_GPU
221
+ asm volatile("trap;");
222
+ #endif
223
}
224
225
if (bias < nope_hidden_size) { // pe
0 commit comments