fix bugs

PaddlePaddle · Sep 18, 2022 · 23138a1 · 23138a1
1 parent 0cccaf5
commit 23138a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cinn/hlir/op/contrib/sort_test.cc b/cinn/hlir/op/contrib/sort_test.cc
@@ -79,7 +79,7 @@ TEST(GenerateCode_Cpu, Sort) {
   ir::Expr h(28);
 
   lang::Placeholder<int32_t> in("in", {n, h});
-  auto stages    = CreateStages({in});
+  auto stages    = poly::CreateStages({in});
   ir::Tensor out = Sort(in, target, stages, 1, true, stages, "test_sort_out");
   stages->InsertLazily(out);
   std::vector<ir::LoweredFunc> funcs =