Merge pull request #17 from facebookresearch/concat-test

Add test case for concat operation

Author: prigoyal

test/test_execution_engine.cc

@@ -45,6 +45,24 @@ struct ATenCompilationUnitTest : public ::testing::Test {
   }
 };
 
+TEST_F(ATenCompilationUnitTest, DISABLED_Concat) {
+  at::Tensor a = at::CUDA(at::kFloat).rand({32, 16});
+  at::Tensor b = at::CUDA(at::kFloat).rand({32, 16});
+  std::vector<at::Tensor> inputs = {a, b};
+  std::vector<at::Tensor> outputs;
+
+  Check(
+      R"(
+      def concat(float(M, N) A, float(M, N) B) -> (O1, O2) {
+        O1(n, i, m) = i == 0 ? A(m, n) : B(m, n) where i in 0:2
+      }
+    )",
+      "concat",
+      tc::MappingOptions::makeNaiveMappingOptions(),
+      inputs,
+      outputs);
+}
+
 TEST_F(ATenCompilationUnitTest, Indexing) {
   at::Tensor a = at::CUDA(at::kFloat).rand({3, 4});
   at::Tensor b = at::CUDA(at::kInt).ones({2});