[llvm-branch-commits] [mlir] 40ca3b5 - Amend op parsing test case.

[Wen-Heng Chung via llvm-branch-commits]

Author: Wen-Heng (Jack) Chung
Date: 2020-06-05T22:18:19-05:00
New Revision: 40ca3b593063c2628082dbdff6b31b8e728cb926

URL: https://github.com/llvm/llvm-project/commit/40ca3b593063c2628082dbdff6b31b8e728cb926
DIFF: https://github.com/llvm/llvm-project/commit/40ca3b593063c2628082dbdff6b31b8e728cb926.diff

LOG: Amend op parsing test case.

Added the following attributes:
- source_layout
- output_layout
- gridwise_gemm_argument_position

Added: 
    

Modified: 
    mlir/test/Dialect/MIOpen/ops.mlir

Removed: 
    


################################################################################
diff  --git a/mlir/test/Dialect/MIOpen/ops.mlir b/mlir/test/Dialect/MIOpen/ops.mlir
index 9b3b2e3db27e..1fc8bca38f6b 100644
--- a/mlir/test/Dialect/MIOpen/ops.mlir
+++ b/mlir/test/Dialect/MIOpen/ops.mlir
@@ -50,7 +50,9 @@ func @miopen_transform_1_to_1(%memref: memref<?x?x?x?xf32>) {
         source_dimensions = [3],
         source_names = ["wi"]
       }
-    ]
+    ],
+    source_layout = ["n", "c", "hi", "wi"],
+    output_layout = ["n", "c", "hipad", "wipad"]
   } : memref<?x?x?x?xf32> to memref<?x?x?x?xf32>
   return
 }
@@ -73,9 +75,12 @@ func @miopen_transform_n_to_1(%memref : memref<?x?x?x?xf32>) {
         names = ["gemmM"],
         transformation = "passthrough",
         source_dimensions = [0],
-        source_names = ["n"]
+        source_names = ["k"]
       }
-    ]
+    ],
+    source_layout = ["k", "c", "y", "x"],
+    output_layout = ["gemmK", "gemmM"],
+    gridwise_gemm_argument_pos = 0
   } : memref<?x?x?x?xf32> to memref<?x?xf32>
   return
 }
@@ -116,7 +121,9 @@ func @miopen_transform_1_to_n(%memref : memref<?x?x?x?xf32>) {
         source_dimensions = [3],
         source_names = ["wipad"]
       }
-    ]
+    ],
+    intermediate_layout = ["n", "c", "hipad", "wipad"],
+    output_layout = ["n", "c", "y", "ho", "x", "wo"]
   } : memref<?x?x?x?xf32> to memref<?x?x?x?x?x?xf32>
   return
 }