fix msc testcase

mshr-h · mshr-h · commit 456c72e72b16 · 2024-09-17T13:38:35.000+09:00
diff --git a/tests/python/contrib/test_msc/test_graph_build.py b/tests/python/contrib/test_msc/test_graph_build.py
@@ -2364,9 +2364,9 @@ def forward(self, q_data, k_data, v_data):
         "outputs": [
             {
                 "name": "attention",
-                "shape": [1, seq, 8, 64],
+                "shape": [1, 8, seq, 64],
                 "dtype": "float32",
-                "layout": "ABCD",
+                "layout": "ACBD",
             }
         ],
         "nodes": {"total": 4, "input": 3, "msc.attention": 1},
@@ -2396,9 +2396,9 @@ def forward(self, q_data, k_data, v_data, mask):
         "outputs": [
             {
                 "name": "attention_bias",
-                "shape": [1, seq, 8, 64],
+                "shape": [1, 8, seq, 64],
                 "dtype": "float32",
-                "layout": "ABCD",
+                "layout": "ACBD",
             }
         ],
         "nodes": {"total": 5, "input": 4, "msc.attention": 1},

Original file line number	Diff line number	Diff line change
`@@ -2364,9 +2364,9 @@ def forward(self, q_data, k_data, v_data):`
`2364`	`2364`	`"outputs": [`
`2365`	`2365`	`{`
`2366`	`2366`	`"name": "attention",`
`2367`		`- "shape": [1, seq, 8, 64],`
	`2367`	`+ "shape": [1, 8, seq, 64],`
`2368`	`2368`	`"dtype": "float32",`
`2369`		`- "layout": "ABCD",`
	`2369`	`+ "layout": "ACBD",`
`2370`	`2370`	`}`
`2371`	`2371`	`],`
`2372`	`2372`	`"nodes": {"total": 4, "input": 3, "msc.attention": 1},`
`@@ -2396,9 +2396,9 @@ def forward(self, q_data, k_data, v_data, mask):`
`2396`	`2396`	`"outputs": [`
`2397`	`2397`	`{`
`2398`	`2398`	`"name": "attention_bias",`
`2399`		`- "shape": [1, seq, 8, 64],`
	`2399`	`+ "shape": [1, 8, seq, 64],`
`2400`	`2400`	`"dtype": "float32",`
`2401`		`- "layout": "ABCD",`
	`2401`	`+ "layout": "ACBD",`
`2402`	`2402`	`}`
`2403`	`2403`	`],`
`2404`	`2404`	`"nodes": {"total": 5, "input": 4, "msc.attention": 1},`