[all-commits] [llvm/llvm-project] 944e03: [mlir][mesh] Use tensor shape notation for the sha...

Boian Petkantchin via All-commits all-commits at lists.llvm.org
Fri Dec 8 11:34:57 PST 2023


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 944e031e36d9515b68b320f611edfc97d5460259
      https://github.com/llvm/llvm-project/commit/944e031e36d9515b68b320f611edfc97d5460259
  Author: Boian Petkantchin <boian.petkantchin at amd.com>
  Date:   2023-12-08 (Fri, 08 Dec 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Mesh/IR/MeshOps.td
    M mlir/include/mlir/IR/OpImplementation.h
    M mlir/lib/Dialect/Mesh/IR/MeshOps.cpp
    M mlir/lib/IR/AsmPrinter.cpp
    M mlir/test/Dialect/Mesh/canonicalization.mlir
    M mlir/test/Dialect/Mesh/invalid.mlir
    M mlir/test/Dialect/Mesh/ops.mlir
    M mlir/test/Dialect/Mesh/sharding-propagation.mlir
    A mlir/test/IR/custom-print-parse.mlir
    A mlir/test/IR/invalid-custom-print-parse.mlir
    M mlir/test/lib/Dialect/Test/TestOps.td

  Log Message:
  -----------
  [mlir][mesh] Use tensor shape notation for the shape of a cluster (#73826)

Examle:

substitute
mesh.cluster @mesh0(rank = 2, dim_sizes = [0, 4])

with
mesh.cluster @mesh0(rank = 2, dim_sizes = ?x4)

Same as tensor/memref shapes. The only difference is for 0-rank shapes.
With tensors you would have something like `tensor<f32>`. Here to avoid
matching an empty string a 0-rank shape is denoted by `[]`.




More information about the All-commits mailing list