From 67974aee8afe3cf7a91f0fe9da72a636480b44fa Mon Sep 17 00:00:00 2001 From: Hardy <100662313+wanghailu0717@users.noreply.github.com> Date: Mon, 27 Nov 2023 14:18:12 +0800 Subject: [PATCH] Fix https://github.com/InfiniTensor/InfiniTensor/pull/160 (#185) Co-authored-by: wanghailu --- src/kernels/bang/reduce_mean.cc | 2 +- src/kernels/kunlun/reduce_mean.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/kernels/bang/reduce_mean.cc b/src/kernels/bang/reduce_mean.cc index 2dd77e10..1b55c2ca 100644 --- a/src/kernels/bang/reduce_mean.cc +++ b/src/kernels/bang/reduce_mean.cc @@ -1,6 +1,6 @@ -#include "operators/reduce_mean.h" #include "bang/bang_kernel_without_config.h" #include "bang/bang_runtime.h" +#include "operators/reduce.h" namespace infini { class ReduceMeanCnnl : public BangKernelWithoutConfig { diff --git a/src/kernels/kunlun/reduce_mean.cc b/src/kernels/kunlun/reduce_mean.cc index 08a01fd6..c7cf19ac 100644 --- a/src/kernels/kunlun/reduce_mean.cc +++ b/src/kernels/kunlun/reduce_mean.cc @@ -1,6 +1,6 @@ -#include "operators/reduce_mean.h" #include "kunlun/kunlun_kernel_without_config.h" #include "kunlun/kunlun_runtime.h" +#include "operators/reduce.h" namespace infini { class ReduceMeanXdnn : public KUNLUNKernelWithoutConfig {