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 } _______________________________________________ llvm-branch-commits mailing list llvm-branch-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits
