refactor(//tests/models): renaming models tests to modules

- Adds a README for the module tests
- Removes an unecessary script
- Adds the multiple engines test to the test suite

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

tests/models/simple_net.py

@@ -8,6 +8,7 @@ test_suite(
     tests = [
         ":test_modules_as_engines",
         ":test_compiled_modules",
+	":test_multiple_registered_engines"
     ]
 )
 

tests/models/BUILD renamed to tests/modules/BUILD

@@ -0,0 +1,9 @@
+# Module Level Tests
+
+These tests focus on integration testing of the backend, using the public apis. They require TorchScript modules saved to files. You can download the set of modules used by default in the tests by running hub.py (you need PyTorch 1.4.0) installed. 
+
+## Running Tests
+
+``` shell
+bazel test //tests/modules:test_modules --compilation_mode=dbg --test_output=errors
+```

tests/modules/README.md

@@ -28,9 +28,9 @@ TEST_P(ModuleTests, CompiledModuleIsClose) {
 INSTANTIATE_TEST_SUITE_P(CompiledModuleForwardIsCloseSuite,
                          ModuleTests,
                          testing::Values(
-                             PathAndInSize({"tests/models/lenet.jit.pt",
+                             PathAndInSize({"tests/modules/lenet.jit.pt",
                                             {{1,1,28,28}}}),
-                             PathAndInSize({"tests/models/resnet18.jit.pt",
+                             PathAndInSize({"tests/modules/resnet18.jit.pt",
                                              {{1,3,224,224}}}),
-                             PathAndInSize({"tests/models/resnet50.jit.pt",
+                             PathAndInSize({"tests/modules/resnet50.jit.pt",
                                             {{1,3,224,224}}})));

tests/models/hub.py renamed to tests/modules/hub.py

@@ -19,9 +19,9 @@ TEST_P(ModuleTests, ModuleAsEngineIsClose) {
 INSTANTIATE_TEST_SUITE_P(ModuleAsEngineForwardIsCloseSuite,
                          ModuleTests,
                          testing::Values(
-                             PathAndInSize({"tests/models/lenet.jit.pt",
+                             PathAndInSize({"tests/modules/lenet.jit.pt",
                                             {{1,1,28,28}}}),
-                             PathAndInSize({"tests/models/resnet18.jit.pt",
+                             PathAndInSize({"tests/modules/resnet18.jit.pt",
                                             {{1,3,224,224}}}),
-                             PathAndInSize({"tests/models/resnet50.jit.pt",
+                             PathAndInSize({"tests/modules/resnet50.jit.pt",
                                             {{1,3,224,224}}})));

tests/models/module_test.h renamed to tests/modules/module_test.h

@@ -8,8 +8,8 @@ TEST(ModuleTests, CanRunMultipleEngines) {
     torch::jit::script::Module mod1;
     torch::jit::script::Module mod2;
     try {
-        mod1 = torch::jit::load("tests/models/resnet50.jit.pt");
-        mod2 = torch::jit::load("tests/models/resnet18.jit.pt");
+        mod1 = torch::jit::load("tests/modules/resnet50.jit.pt");
+        mod2 = torch::jit::load("tests/modules/resnet18.jit.pt");
     }
     catch (const c10::Error& e) {
         std::cerr << "error loading the model\n";