p09823145
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix https://github.com/InfiniTensor/InfiniTensor/pull/160 (#185) 

Co-authored-by: wanghailu <wanghailu0717@163.com>
This commit is contained in:
Hardy 2023-11-27 14:18:12 +08:00 committed by GitHub
parent 3ead20a23a
commit 67974aee8a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/kernels/bang/reduce_mean.cc
View File

@ -1,6 +1,6 @@
#include "operators/reduce_mean.h"
#include "bang/bang_kernel_without_config.h" #include "bang/bang_kernel_without_config.h"
#include "bang/bang_runtime.h" #include "bang/bang_runtime.h"
#include "operators/reduce.h"
namespace infini { namespace infini {
class ReduceMeanCnnl : public BangKernelWithoutConfig { class ReduceMeanCnnl : public BangKernelWithoutConfig {

2
src/kernels/kunlun/reduce_mean.cc
View File

@ -1,6 +1,6 @@
#include "operators/reduce_mean.h"
#include "kunlun/kunlun_kernel_without_config.h" #include "kunlun/kunlun_kernel_without_config.h"
#include "kunlun/kunlun_runtime.h" #include "kunlun/kunlun_runtime.h"
#include "operators/reduce.h"
namespace infini { namespace infini {
class ReduceMeanXdnn : public KUNLUNKernelWithoutConfig { class ReduceMeanXdnn : public KUNLUNKernelWithoutConfig {