Merge pull request #17 from narendasan/master

Rename tests/models to tests/modules

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";