fix(//tests/core/partitioning): Fixing some issues with the partition

narendasan · narendasan · commit ff89059fe65e · 2021-04-30T13:49:19.000-07:00
tests BUILD files

Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/tests/core/BUILD b/tests/core/BUILD
@@ -3,6 +3,6 @@ test_suite(
     tests = [
         "//tests/core/conversion:conversion_tests",
         "//tests/core/lowering:lowering_tests",
-        "//tests/core/partitioning::partitioning_tests"
+        "//tests/core/partitioning:partitioning_tests"
     ],
 )
diff --git a/tests/core/partitioning/BUILD b/tests/core/partitioning/BUILD
@@ -7,12 +7,6 @@ config_setting(
     }
 )
 
-filegroup(
-    name = "jit_models",
-    srcs = ["//tests/modules:resnet50_traced.jit.pt",
-            "//tests/modules:mobilenet_v2_traced.jit.pt"]
-)
-
 partitioning_test(
   name = "test_segmentation",
 )
@@ -41,12 +35,12 @@ cc_test(
       "//conditions:default":  ["@libtorch//:libtorch"],
   }),
   data = [
-      ":jit_models"
+      "//tests/modules:jit_models"
   ]
 )
 
 test_suite(
-    name = "partitioning_test",
+    name = "partitioning_tests",
     tests = [
         ":test_segmentation",
         ":test_shape_analysis",

Original file line number	Diff line number	Diff line change
`@@ -3,6 +3,6 @@ test_suite(`
`3`	`3`	`tests = [`
`4`	`4`	`"//tests/core/conversion:conversion_tests",`
`5`	`5`	`"//tests/core/lowering:lowering_tests",`
`6`		`- "//tests/core/partitioning::partitioning_tests"`
	`6`	`+ "//tests/core/partitioning:partitioning_tests"`
`7`	`7`	`],`
`8`	`8`	`)`
Original file line number	Diff line number	Diff line change
`@@ -7,12 +7,6 @@ config_setting(`
`7`	`7`	`}`
`8`	`8`	`)`
`9`	`9`
`10`		`-filegroup(`
`11`		`- name = "jit_models",`
`12`		`- srcs = ["//tests/modules:resnet50_traced.jit.pt",`
`13`		`- "//tests/modules:mobilenet_v2_traced.jit.pt"]`
`14`		`-)`
`15`		`-`
`16`	`10`	`partitioning_test(`
`17`	`11`	`name = "test_segmentation",`
`18`	`12`	`)`
`@@ -41,12 +35,12 @@ cc_test(`
`41`	`35`	`"//conditions:default": ["@libtorch//:libtorch"],`
`42`	`36`	`}),`
`43`	`37`	`data = [`
`44`		`- ":jit_models"`
	`38`	`+ "//tests/modules:jit_models"`
`45`	`39`	`]`
`46`	`40`	`)`
`47`	`41`
`48`	`42`	`test_suite(`
`49`		`- name = "partitioning_test",`
	`43`	`+ name = "partitioning_tests",`
`50`	`44`	`tests = [`
`51`	`45`	`":test_segmentation",`
`52`	`46`	`":test_shape_analysis",`