fix

PaddlePaddle · Sep 20, 2024 · e6585ec · e6585ec
1 parent 907ad20
commit e6585ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/csrc/setup_cuda.py b/csrc/setup_cuda.py
@@ -107,7 +107,7 @@ def get_gencode_flags():
     "./gpu/dequant_int8.cu",
     "./gpu/flash_attn_bwd.cc",
     "./gpu/tune_cublaslt_gemm.cu",
-    "./gpu/sample_kernels/top_p_sampling_reject",
+    "./gpu/sample_kernels/top_p_sampling_reject.cu",
 ]
 
 cutlass_dir = "third_party/cutlass"