[XPU] fix axis(-1) in the GatherCompute function (PaddlePaddle#10387)

hehe1112 · Sep 11, 2023 · 67c77cc · 67c77cc
1 parent f030cc3
commit 67c77cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lite/kernels/xpu/gather_compute.cc b/lite/kernels/xpu/gather_compute.cc
@@ -43,7 +43,7 @@ void GatherCompute<DataType, IndexType, PType>::Run() {
         << " xpu only support axis int32 type";
     auto* axis_data = param.Axis->template data<int>();
     axis = axis_data[0];
-  } else {
+  } else if (param.axis != -1) {
     axis = param.axis;
   }
   std::vector<int> x_dims(x->dims().data().begin(), x->dims().data().end());