Exclude MViT from cuda tests (#6573)

datumbox · web-flow · commit 52ecad8d7ce3 · 2022-09-13T14:02:34.000+01:00
diff --git a/test/test_models.py b/test/test_models.py
@@ -348,8 +348,8 @@ def _check_input_backprop(model, inputs):
 skipped_big_models = {
     "vit_h_14": {("Windows", "cpu"), ("Windows", "cuda")},
     "regnet_y_128gf": {("Windows", "cpu"), ("Windows", "cuda")},
-    "mvit_v1_b": {("Windows", "cuda")},
-    "mvit_v2_s": {("Windows", "cuda")},
+    "mvit_v1_b": {("Windows", "cuda"), ("Linux", "cuda")},
+    "mvit_v2_s": {("Windows", "cuda"), ("Linux", "cuda")},
 }
 
 

Original file line number	Diff line number	Diff line change
`@@ -348,8 +348,8 @@ def _check_input_backprop(model, inputs):`
`348`	`348`	`skipped_big_models = {`
`349`	`349`	`"vit_h_14": {("Windows", "cpu"), ("Windows", "cuda")},`
`350`	`350`	`"regnet_y_128gf": {("Windows", "cpu"), ("Windows", "cuda")},`
`351`		`- "mvit_v1_b": {("Windows", "cuda")},`
`352`		`- "mvit_v2_s": {("Windows", "cuda")},`
	`351`	`+ "mvit_v1_b": {("Windows", "cuda"), ("Linux", "cuda")},`
	`352`	`+ "mvit_v2_s": {("Windows", "cuda"), ("Linux", "cuda")},`
`353`	`353`	`}`
`354`	`354`
`355`	`355`