[DCU] update dcu paddlenlp_ops (#9433)

PaddlePaddle · Nov 15, 2024 · 81a2d43 · 81a2d43
1 parent 921fc44
commit 81a2d43
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/csrc/setup_hip.py b/csrc/setup_hip.py
@@ -50,6 +50,8 @@ def update_git_submodule():
             "./gpu/quant_int8.cu",
             "./gpu/dequant_int8.cu",
             "./gpu/flash_attn_bwd.cc",
+            "./gpu/update_inputs_v2.cu",
+            "./gpu/set_preids_token_penalty_multi_scores.cu",
         ],
         extra_compile_args={
             "cxx": ["-O3"],