test(): Added engine serialization testing for interpolation model

abhi-iyer · abhi-iyer · commit 549ca38dcc80 · 2020-06-18T18:22:41.000-07:00
Signed-off-by: Abhiram Iyer &lt;abhirami@nvidia.com

Signed-off-by: Abhiram Iyer &lt;abhi.iyer.ai@gmail.com&gt;
diff --git a/tests/modules/test_serialization.cpp b/tests/modules/test_serialization.cpp
@@ -31,4 +31,6 @@ INSTANTIATE_TEST_SUITE_P(CompiledModuleForwardIsCloseSuite,
                          ModuleTests,
                          testing::Values(
                             PathAndInSize({"tests/modules/resnet18_traced.jit.pt",
-                                          {{1,3,224,224}}})));
+                                          {{1,3,224,224}}}),
+                            PathAndInSize({"tests/modules/interpolate_traced.jit.pt",
+                                          {{1,3,5,5,5}}})));
