update mlu_baseop.h

Author: co63oc

backends/mlu/kernels/compare_kernel.cc

@@ -18,10 +18,10 @@ namespace custom_kernel {
 
 template <typename T, typename Context>
 void EqualRawKernel(const Context& dev_ctx,
-                    const phi::DenseTensor& x,
-                    const phi::DenseTensor& y,
+                    const DenseTensor& x,
+                    const DenseTensor& y,
                     int axis,
-                    phi::DenseTensor* out) {
+                    DenseTensor* out) {
   MLULogicOp(dev_ctx, x, y, "equal", out);
 }
 

backends/mlu/kernels/funcs/mlu_baseop.h

@@ -25,6 +25,7 @@
 namespace custom_kernel {
 
 using Tensor = phi::DenseTensor;
+using DenseTensor = phi::DenseTensor;
 using Context = phi::CustomContext;
 using DataType = phi::DataType;
 using DataLayout = phi::DataLayout;