fix

PaddlePaddle · Sep 14, 2024 · 934f536 · 934f536
1 parent 10630ec
commit 934f536
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/...mple_kernels/top_p_sampling_from_probs.cu → ...u/sample_kernels/top_p_sampling_reject.cu b/...mple_kernels/top_p_sampling_from_probs.cu → ...u/sample_kernels/top_p_sampling_reject.cu
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_from_probs.cu",
+    "./gpu/sample_kernels/top_p_sampling_reject",
 ]
 
 cutlass_dir = "third_party/cutlass"