Skip to content

Commit 824393e

Browse files
committed
fix(metax): skip trap asm on MetaX GPU to fix compile error
1 parent 91ca3d1 commit 824393e

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

custom_ops/gpu_ops/append_attn/mla_cache_kernel.cuh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,9 @@ __global__ void prefill_absorb_cache_kernel(
217217
printf("block_idx1 %d != block_idx %d\n", block_idx1, block_idx);
218218
printf("token_idx %d\n", token_idx);
219219
printf("slot_mapping %d\n", slot_mapping[token_idx]);
220-
asm volatile("trap;");
220+
#ifndef PADDLE_WITH_CUSTOM_DEVICE_METAX_GPU
221+
asm volatile("trap;");
222+
#endif
221223
}
222224

223225
if (bias < nope_hidden_size) { // pe

0 commit comments

Comments
 (0)