Add test case for reported failure in halide2isl

See #124, #125.

Author: ftynse

test/test_tc_mapper_bugs.cc

@@ -696,6 +696,27 @@ TEST(LayerNorm, ReferenceBelongsToTwoGroups) {
   atCompl.compile("layernorm", inputs, options);
 }
 
+// #124
+TEST(Halide2Isl, MinInUpperBound) {
+  at::Tensor mat1 = at::CUDA(at::kFloat).rand({1, 100, 184, 184});
+  at::Tensor mat1_pad = at::CUDA(at::kFloat).rand({1, 100, 186, 186});
+  at::Tensor mat2 = at::CUDA(at::kFloat).rand({3, 3});
+  std::vector<at::Tensor> inputs = {mat1, mat1_pad, mat2};
+
+  static constexpr auto TC = R"TC(
+    def graph2(float(N, C, H, W) I, float(N, C, R, T) J, float(KH, KW) W1) -> (O, Out) {
+        O(n, c, h, w) +=! J(n, c, h + kh, w + kw) * W1(kh, kw)
+        Out(i, j) +=! I(n, i, h, w) * O(n, j, h, w)
+    }
+  )TC";
+  auto options = tc::MappingOptions::makeNaiveMappingOptions();
+
+  tc::ATenCompilationUnit atCompl;
+  atCompl.define(TC);
+  EXPECT_THROW(
+      atCompl.compile("graph2", inputs, options), isl::exception_invalid);
+}
+
 int main(int argc, char** argv) {
   ::testing::InitGoogleTest(&argc, argv);
   ::gflags::ParseCommandLineFlags(&argc, &argv, true);