test(tests/core/converters): Added test for plugins

Signed-off-by: Abhiram Iyer <[email protected]>

Signed-off-by: Abhiram Iyer <[email protected]>

Author: abhi-iyer

tests/core/converters/test_interpolate.cpp

@@ -91,7 +91,7 @@ TEST(Converters, ATenUpsampleNearest3dConvertsCorrectly3dOutputSize) {
     ASSERT_TRUE(trtorch::tests::util::almostEqual(jit_results[0], trt, 2e-6));
 }
 
-TEST(Converters, ATenUpsampleLinear1dConvertsCorrectly) {
+TEST(Converters, ATenUpsampleLinear1dConvertsCorrectlyWithAlignCorners) {
     const auto graph = R"IR(
       graph(%0 : Tensor):
         %2 : int = prim::Constant[value=10]()
@@ -121,7 +121,37 @@ TEST(Converters, ATenUpsampleLinear1dConvertsCorrectly) {
     ASSERT_TRUE(trtorch::tests::util::almostEqual(jit_results[0], trt, 2e-6));
 }
 
-TEST(Converters, ATenUpsampleBilinear2dConvertsCorrectly2dOutputSize) {
+TEST(Converters, ATenUpsampleLinear1dConvertsCorrectlyWithoutAlignCorners) {
+    const auto graph = R"IR(
+      graph(%0 : Tensor):
+        %2 : int = prim::Constant[value=10]()
+        %3 : int[] = prim::ListConstruct(%2)
+        %4 : bool = prim::Constant[value=0]()
+        %5 : None = prim::Constant()
+        %6 : Tensor = aten::upsample_linear1d(%0, %3, %4, %5)
+        return (%6))IR";
+    
+    auto g = std::make_shared<torch::jit::Graph>();
+
+    torch::jit::parseIR(graph, &*g);
+
+    // Input Tensor needs to be 3D for TensorRT upsample_linear1d
+    auto in = at::randint(1, 10, {10, 2, 2}, {at::kCUDA});
+
+    auto jit_in = at::clone(in);
+    auto params = trtorch::core::conversion::get_named_params(g->inputs(), {});
+    auto jit_results = trtorch::tests::util::RunGraph(g, params, {jit_in});
+
+    auto trt_in = at::clone(in);
+    params = trtorch::core::conversion::get_named_params(g->inputs(), {});
+    auto trt_results = trtorch::tests::util::RunGraphEngine(g, params, {trt_in});
+
+    auto trt = trt_results[0].reshape(jit_results[0].sizes());
+
+    ASSERT_TRUE(trtorch::tests::util::almostEqual(jit_results[0], trt, 2e-6));
+}
+
+TEST(Converters, ATenUpsampleBilinear2dConvertsCorrectly2dOutputSizeWithAlignCorners) {
     const auto graph = R"IR(
       graph(%0 : Tensor):
         %2 : int = prim::Constant[value=10]()
@@ -151,7 +181,37 @@ TEST(Converters, ATenUpsampleBilinear2dConvertsCorrectly2dOutputSize) {
     ASSERT_TRUE(trtorch::tests::util::almostEqual(jit_results[0], trt, 2e-6));
 }
 
-TEST(Converters, ATenUpsampleTrilinear3dConvertsCorrectly3dOutputSize) {
+TEST(Converters, ATenUpsampleBilinear2dConvertsCorrectly2dOutputSizeWithoutAlignCorners) {
+    const auto graph = R"IR(
+      graph(%0 : Tensor):
+        %2 : int = prim::Constant[value=10]()
+        %3 : int[] = prim::ListConstruct(%2, %2)
+        %4 : bool = prim::Constant[value=0]()
+        %5 : None = prim::Constant()
+        %6 : Tensor = aten::upsample_bilinear2d(%0, %3, %4, %5, %5)
+        return (%6))IR";
+    
+    auto g = std::make_shared<torch::jit::Graph>();
+
+    torch::jit::parseIR(graph, &*g);
+
+    // Input Tensor needs to be 4D for TensorRT upsample_bilinear2d
+    auto in = at::randint(1, 10, {10, 2, 2, 2}, {at::kCUDA});
+
+    auto jit_in = at::clone(in);
+    auto params = trtorch::core::conversion::get_named_params(g->inputs(), {});
+    auto jit_results = trtorch::tests::util::RunGraph(g, params, {jit_in});
+
+    auto trt_in = at::clone(in);
+    params = trtorch::core::conversion::get_named_params(g->inputs(), {});
+    auto trt_results = trtorch::tests::util::RunGraphEngine(g, params, {trt_in});
+
+    auto trt = trt_results[0].reshape(jit_results[0].sizes());
+
+    ASSERT_TRUE(trtorch::tests::util::almostEqual(jit_results[0], trt, 2e-6));
+}
+
+TEST(Converters, ATenUpsampleTrilinear3dConvertsCorrectly3dOutputSizeWithAlignCorners) {
     const auto graph = R"IR(
       graph(%0 : Tensor):
         %2 : int = prim::Constant[value=10]()
@@ -178,5 +238,35 @@ TEST(Converters, ATenUpsampleTrilinear3dConvertsCorrectly3dOutputSize) {
 
     auto trt = trt_results[0].reshape(jit_results[0].sizes());
 
+    ASSERT_TRUE(trtorch::tests::util::almostEqual(jit_results[0], trt, 2e-6));
+}
+
+TEST(Converters, ATenUpsampleTrilinear3dConvertsCorrectly3dOutputSizeWithoutAlignCorners) {
+    const auto graph = R"IR(
+      graph(%0 : Tensor):
+        %2 : int = prim::Constant[value=10]()
+        %3 : int[] = prim::ListConstruct(%2, %2, %2)
+        %4 : bool = prim::Constant[value=0]()
+        %5 : None = prim::Constant()
+        %6 : Tensor = aten::upsample_trilinear3d(%0, %3, %4, %5, %5, %5)
+        return (%6))IR";
+    
+    auto g = std::make_shared<torch::jit::Graph>();
+
+    torch::jit::parseIR(graph, &*g);
+
+    // Input Tensor needs to be 5D for TensorRT upsample_trilinear3d
+    auto in = at::randint(1, 10, {10, 2, 2, 2, 2}, {at::kCUDA});
+
+    auto jit_in = at::clone(in);
+    auto params = trtorch::core::conversion::get_named_params(g->inputs(), {});
+    auto jit_results = trtorch::tests::util::RunGraph(g, params, {jit_in});
+
+    auto trt_in = at::clone(in);
+    params = trtorch::core::conversion::get_named_params(g->inputs(), {});
+    auto trt_results = trtorch::tests::util::RunGraphEngine(g, params, {trt_in});
+
+    auto trt = trt_results[0].reshape(jit_results[0].sizes());
+
     ASSERT_TRUE(trtorch::tests::util::almostEqual(jit_results[0], trt, 2e-6));
 }