Fix CI problems

microsoft · Oct 16, 2023 · 6e131a6 · 6e131a6
1 parent 364b9df
commit 6e131a6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/csrc/transformer/inference/includes/inference_cuda_layers.h b/csrc/transformer/inference/includes/inference_cuda_layers.h
@@ -208,7 +208,8 @@ void launch_bias_add_transform_0213(T* outputs,
                                     bool rotate_every_two,
                                     cudaStream_t stream,
                                     int trans_count,
-                                    int max_out_tokens);
+                                    int max_out_tokens,
+                                    float rope_theta);
 template <typename T>
 void pad_data(T* padded_output,
               T* output,