update

microsoft · Jan 24, 2024 · 8a8a7a1 · 8a8a7a1
1 parent 202c1cc
commit 8a8a7a1
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"