fix: resolve all_equal operator compilation error on iluvatar platform

tianyuxbear · tianyuxbear · commit 3f1ab67a3dca · 2025-08-11T16:43:50.000+08:00
diff --git a/src/infiniop/ops/all_equal/all_equal.h b/src/infiniop/ops/all_equal/all_equal.h
@@ -1,7 +1,6 @@
 #ifndef __ALL_EQUAL_H
 #define __ALL_EQUAL_H
 
-#include "../../devices/cpu/common_cpu.h"
 #include "../../handle.h"
 #include "../../operator.h"
 #include "info.h"
diff --git a/src/infiniop/ops/all_equal/cpu/all_equal_cpu.cc b/src/infiniop/ops/all_equal/cpu/all_equal_cpu.cc
@@ -1,4 +1,5 @@
 #include "all_equal_cpu.h"
+#include "../../../devices/cpu/common_cpu.h"
 #include <cstddef>
 
 namespace op::all_equal::cpu {
diff --git a/src/infiniop/ops/all_equal/cuda/kernel.cuh b/src/infiniop/ops/all_equal/cuda/kernel.cuh
@@ -3,7 +3,6 @@
 #include "../../../elementwise/nvidia/elementwise_nvidia.cuh"
 #include <cstddef>
 #include <cstdint>
-#include <cuda/atomic>
 
 template <unsigned int BLOCK_SIZE, typename Tdata>
 __global__ void compareKernel(size_t input_numel, size_t ndim, const bool *__restrict__ input_contiguous, const bool *__restrict__ input_broadcasted, const size_t *__restrict__ input_shapes, const ptrdiff_t *__restrict__ output_strides, const ptrdiff_t *__restrict__ input_strides, const void *const *inputs, uint8_t *flags) {

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`#include "all_equal_cpu.h"`
	`2`	`+#include "../../../devices/cpu/common_cpu.h"`
`2`	`3`	`#include <cstddef>`
`3`	`4`
`4`	`5`	`namespace op::all_equal::cpu {`