fix bugs

PaddlePaddle · Sep 18, 2022 · 948ad2e · 948ad2e
1 parent fddcdd1
commit 948ad2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cinn/hlir/op/contrib/sort_test.cc b/cinn/hlir/op/contrib/sort_test.cc
@@ -79,8 +79,8 @@ TEST(GenerateCode_Cpu, Sort) {
   ir::Expr h(28);
 
   lang::Placeholder<int32_t> in("in", {n, h});
-  auto stages    = CreateStages({tensor_A});
-  ir::Tensor out = Sort(tensor_A, target, stages, 1, true, stages, "test_sort_out");
+  auto stages    = CreateStages({in});
+  ir::Tensor out = Sort(in, target, stages, 1, true, stages, "test_sort_out");
   stages->InsertLazily(out);
   std::vector<ir::LoweredFunc> funcs =
       lang::LowerVec("TestGenerateCodeCpu_Sort", stages, {in, index, out}, {}, {}, nullptr, target, true);