[TEST] Test DpasLayout in test_local_load_store_mma (#3047)

whitneywhtsang · web-flow · commit 13725c19f351 · 2024-12-19T08:53:25.000-05:00
Signed-off-by: Whitney Tsang &lt;whitney.tsang@intel.com&gt;
diff --git a/python/test/unit/language/test_core.py b/python/test/unit/language/test_core.py
@@ -5595,6 +5595,8 @@ def test_local_load_store(M, N, K, dist_layout, shared_layout, device, tmp_path:
     MmaLayout((3, 0), [4, 2], [1, 1], [1, 1], [0, 1], [16, 128, 16]),  # multiple warps on the row
     MmaLayout((3, 0), [4, 2], [1, 1], [1, 1], [0, 1], [16, 64, 16]),  # small instrN
     MmaLayout((3, 0), [8, 4], [1, 1], [1, 1], [0, 1], [16, 64, 16]),  # large number of warps
+    DpasLayout(repeatCount=8, systolic_depth=8, execution_size=8, ops_per_chan=1, threads_per_warp=32,
+               warps_per_cta=[4, 1], rep_cluster=[1, 1]),
 ]
 
 shared_layouts = [

Original file line number	Diff line number	Diff line change
`@@ -5595,6 +5595,8 @@ def test_local_load_store(M, N, K, dist_layout, shared_layout, device, tmp_path:`
`5595`	`5595`	`MmaLayout((3, 0), [4, 2], [1, 1], [1, 1], [0, 1], [16, 128, 16]), # multiple warps on the row`
`5596`	`5596`	`MmaLayout((3, 0), [4, 2], [1, 1], [1, 1], [0, 1], [16, 64, 16]), # small instrN`
`5597`	`5597`	`MmaLayout((3, 0), [8, 4], [1, 1], [1, 1], [0, 1], [16, 64, 16]), # large number of warps`
	`5598`	`+ DpasLayout(repeatCount=8, systolic_depth=8, execution_size=8, ops_per_chan=1, threads_per_warp=32,`
	`5599`	`+ warps_per_cta=[4, 1], rep_cluster=[1, 1]),`
`5598`	`5600`	`]`
`5599`	`5601`
`5600`	`5602`	`shared_layouts = [`