There was an error while loading. Please reload this page.
1 parent f76a0c3 commit bf1a2e8Copy full SHA for bf1a2e8
paddle/pten/kernels/gpu/reduce.h
@@ -45,7 +45,7 @@ namespace cub = hipcub;
45
#include "paddle/pten/api/ext/dispatch.h"
46
#include "paddle/pten/backends/gpu/gpu_context.h"
47
#include "paddle/pten/core/dense_tensor.h"
48
-#include "paddle/pten/kernels/elementwise.h"
+#include "paddle/pten/kernels/gpu/elementwise.h"
49
50
// Reduce split or not, Whether to use ReduceHigherDim
51
#define REDUCE_SPLIT_BOUNDARY 512
0 commit comments