update

microsoft · Jan 17, 2024 · 79597fa · 79597fa
1 parent 0c27cd7
commit 79597fa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/onnxruntime/contrib_ops/cuda/diffusion/group_norm_impl_kernel.cuh b/onnxruntime/contrib_ops/cuda/diffusion/group_norm_impl_kernel.cuh
@@ -21,6 +21,7 @@
 // Licensed under the MIT License.
 #pragma once
 #include <cuda_fp16.h>
+#include <cub/cub.cuh>
 #include "core/providers/cuda/cuda_common.h"
 #include "core/providers/cuda/cu_inc/common.cuh"