Merge pull request #401 from nicolasvasilache/pr/tc-with-newlines

Add test for TC with arbitrary newlines

Author: ftynse

test/cuda/test_corner_cases.cc

@@ -282,6 +282,44 @@ TEST(TestCornerCases, E23) {
       at::Scalar(d[0]).toFloat());
 }
 
+// This tests that the TC parser functions in the presence of arbitrary newlines
+TEST(TestCornerCases, E24) {
+  auto a = F(1);
+  auto b = F(1);
+  auto c = F(1);
+  auto d = F(1);
+  Succeed(
+      R"TC(
+
+def f(float(1)
+a, float(1
+) b, float
+(1) c) -> (d) { d(
+i)
+ =
+
+ min(
+a(i), max(
+b(i), c(
+i)))
+
+d(i) = d(i) + max(b(i),
+
+c(
+i))
+
+}
+)TC",
+      {a, b, c},
+      {d});
+  CHECK_EQ(
+      fminf(
+          at::Scalar(a[0]).toFloat(),
+          fmaxf(at::Scalar(b[0]).toFloat(), at::Scalar(c[0]).toFloat())) +
+          fmaxf(at::Scalar(b[0]).toFloat(), at::Scalar(c[0]).toFloat()),
+      at::Scalar(d[0]).toFloat());
+}
+
 int main(int argc, char** argv) {
   ::testing::InitGoogleTest(&argc, argv);
   ::gflags::ParseCommandLineFlags(&argc, &argv, true);