Merge pull request PaddlePaddle#43 from Superjomn/remove/placeholder-…

…in-schedule remove placeholder in schedule
thisjiang · Feb 27, 2020 · 1033f4f · 1033f4f
2 parents 83439e8 + a162473
commit 1033f4f
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/cinn/lang/lower.cc b/cinn/lang/lower.cc
@@ -96,7 +96,7 @@ std::vector<LoweredFunc> Lower(const std::string& name, const std::vector<Tensor
     }
 
     Expr group_expr = LowerGroup(group, tuple_to_expr);
-    VLOG(3) << "group expr: " << group_expr;
+    VLOG(3) << "group expr:\n" << group_expr;
     exprs.push_back(group_expr);
   }
 

diff --git a/cinn/lang/lower_test.cc b/cinn/lang/lower_test.cc
@@ -22,7 +22,8 @@ TEST(lower, basic) {
 
   LOG(INFO) << "lower_size " << lower_funcs.size();
 
-#define TEST_SOUTPUT(x, out)  LOG(INFO) << "\n" << x; \
+#define TEST_SOUTPUT(x, out) \
+  LOG(INFO) << "\n" << x;    \
   EXPECT_EQ(utils::GetStreamCnt(x), utils::Trim(out));
 
   auto out = R"ROC(
@@ -40,5 +41,25 @@ TEST(lower, basic) {
   TEST_SOUTPUT(lower_funcs.front().body, out);
 }
 
+TEST(lower, more_complex) {
+  const int M = 100;
+  const int N = 15;
+  const int K = 200;
+
+  Placeholder<float> A("A", {Expr(M), Expr(N)});
+  Placeholder<float> B("B", {Expr(N), Expr(K)});
+
+  auto C = Compute(
+      {M, N, K}, [=](Var i, Var j, Var k) -> Expr { return A(i, j) * B(j, k); }, "C");
+
+  auto lower_funcs = Lower("cal_C", {A, B, C});
+
+  LOG(INFO) << "lower_size " << lower_funcs.size();
+
+#define TEST_SOUTPUT(x, out) \
+  LOG(INFO) << "\n" << x;    \
+  EXPECT_EQ(utils::GetStreamCnt(x), utils::Trim(out));
+}
+
 }  // namespace lang
 }  // namespace cinn