Code

[MLIR][mesh] Mesh fixes#124724

Code

Merged

fschlimb merged 10 commits intollvm:mainllvm/llvm-project:mainfrom fschlimb:mesh-fixesfschlimb/llvm-project:mesh-fixesCopy head branch name to clipboardFeb 12, 2025

+525-89

Commits

Commits on Feb 12, 2025

• Allowing constant-like operands to ShardingInterface ops

  Show description for 300f788
  fschlimb

  committed
  300f788

  Copy full SHA for 300f788

• better handling of replicated tensors

  fschlimb

  committed
  f41780f

  Copy full SHA for f41780f

• canonicalize ShardOp and ShardingOp

  fschlimb

  committed
  c1324d3

  Copy full SHA for c1324d3

• sharding propagation: add only one shardop for each result

  fschlimb

  committed
  1d86186

  Copy full SHA for 1d86186

• Adding sharding extraction operation and op tests

  Show description for 70fb9a5
  fschlimb

  committed
  70fb9a5

  Copy full SHA for 70fb9a5

• comments

  Show description for 6f42ee5
  fschlimb

  committed
  6f42ee5

  Copy full SHA for 6f42ee5

• assert expected ArrayRerf argument size

  fschlimb

  committed
  ec4a18e

  Copy full SHA for ec4a18e

• added sharding exmpample

  fschlimb

  committed
  e8ccad1

  Copy full SHA for e8ccad1

• comments and nicer code (from review)

  fschlimb

  committed
  99cf24e

  Copy full SHA for 99cf24e

• maybeInsertTargetShardingAnnotation accepting reference only

  fschlimb

  committed
  3729a86

  Copy full SHA for 3729a86