fix

Author: co63oc

backends/gcu/common/gcu_funcs.h

@@ -23,6 +23,8 @@
 namespace custom_kernel {
 // using Tensor = phi::DenseTensor;
 // using Context = phi::CustomContext;
+// using DataType = phi::DataType;
+// using DataLayout = phi::DataLayout;
 /**
  * CPU -> GCU
  * GCU -> CPU

backends/gcu/custom_engine/custom_engine_interface.cc

@@ -240,7 +240,7 @@ C_Status CustomEngineOpLower(C_CustomEngineLowerParams* lower_param) {
   pir::Block* op_block = &(region.front());
 
   // process subgraph block
-  paddle::dialect::ProcessBlock(
+  pir::ProcessBlock(
       *place, sub_graph_block, op_block, ctx, map_op_pair, map_value_pair);
 
   if (VLOG_IS_ON(3)) {

backends/gcu/tests/unittests/test_cross.py

@@ -69,6 +69,8 @@ def prepare_data(self):
     def forward_with_dtype(self, dtype):
         x = paddle.to_tensor(self.data_x, dtype=dtype)
         y = paddle.to_tensor(self.data_y, dtype=dtype)
+        if self.axis is None:
+            return paddle.cross(x, y)
         return paddle.cross(x, y, self.axis)
 
     def forward(self):

backends/gcu/tests/unittests/test_layer_norm.py

@@ -99,8 +99,8 @@ def test_check_output(
         self.with_weight = with_weight
         self.with_bias = with_bias
         self.weight_bias_shape = [np.prod(self.normalized_shape)]
-        rtol = 1e-5
-        atol = 1e-5
+        rtol = 1e-4
+        atol = 1e-4
         if dtype == np.float16:
             rtol = 1e-3
             atol = 1e-3

backends/gcu/tests/unittests/test_rnn.py

@@ -84,7 +84,7 @@ def forward(self):
             self.input_size,
             self.hidden_size,
             self.num_layers,
-            self.direction,
+            direction=self.direction,
             time_major=False,
             weight_ih_attr=weight_ih_attr,
             weight_hh_attr=weight_hh_attr,