From ece13d84dc4f7b039a65bec574faf7192e59c383 Mon Sep 17 00:00:00 2001 From: Ye Wang <52801275+wangyems@users.noreply.github.com> Date: Tue, 28 May 2024 10:46:24 -0700 Subject: [PATCH] fix a build error with cuda 12.5 (#20770) ### Motivation and Context https://github.com/microsoft/onnxruntime/issues/20765 --- onnxruntime/contrib_ops/cuda/moe/ft_moe/moe_kernel.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/onnxruntime/contrib_ops/cuda/moe/ft_moe/moe_kernel.cu b/onnxruntime/contrib_ops/cuda/moe/ft_moe/moe_kernel.cu index 360c0aacd9c7a..39ce6aec90e1a 100644 --- a/onnxruntime/contrib_ops/cuda/moe/ft_moe/moe_kernel.cu +++ b/onnxruntime/contrib_ops/cuda/moe/ft_moe/moe_kernel.cu @@ -17,6 +17,7 @@ // Licensed under the MIT License. #include +#include #include #include #include