From 8a8a7a1a6f9bb047c68f830e08ffc09f2ceed6cd Mon Sep 17 00:00:00 2001
From: peixuanzuo <peixuanzuo@microsoft.com>
Date: Wed, 17 Jan 2024 08:30:41 +0000
Subject: [PATCH] update

---
 .../contrib_ops/cuda/diffusion/group_norm_impl_kernel.cuh        | 1 +
 1 file changed, 1 insertion(+)

diff --git a/onnxruntime/contrib_ops/cuda/diffusion/group_norm_impl_kernel.cuh b/onnxruntime/contrib_ops/cuda/diffusion/group_norm_impl_kernel.cuh
index cfeeac2049355..ecd06315e3708 100644
--- 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"