Code

Mesh#962

Code

Draft

fschlimb wants to merge 24 commits intointel:mainintel/mlir-extensions:mainfrom fschlimb:meshfschlimb/mlir-extensions:meshCopy head branch name to clipboard

+2,867-11,452

Commits

Commits on Nov 15, 2024

• Using upstream mesh dialect, initial support for subview, insert_slice

  Show description for 447bdbb
  fschlimb

  committed
  447bdbb

  Copy full SHA for 447bdbb

• make Xe/GPU stuff compile (not necessarily work) with newer llvm

  fschlimb

  committed
  dc03026

  Copy full SHA for dc03026

• cleanup: remove dist dialect and its transforms/conversions, reducescope of ndarray and eliminate NDArrayType

  fschlimb

  committed
  d9032f8

  Copy full SHA for d9032f8

• cleaning

  fschlimb

  committed
  84ee293

  Copy full SHA for 84ee293

• fixing distruntime tests

  fschlimb

  committed
  84626f9

  Copy full SHA for 84626f9

• fixing ndarray tests

  fschlimb

  committed
  4396b53

  Copy full SHA for 4396b53

• new env syntax

  fschlimb

  committed
  631a8e3

  Copy full SHA for 631a8e3

Commits on Nov 18, 2024

• fixing AddGpuRegions

  fschlimb

  committed
  53d715c

  Copy full SHA for 53d715c

• fixing DropRegions

  fschlimb

  committed
  78c37b9

  Copy full SHA for 78c37b9

• fixing runner tests

  fschlimb

  committed
  f9341fc

  Copy full SHA for f9341fc

Commits on Nov 20, 2024

• removing ndarray.create; fixing NDArrayToLinalg and tests

  fschlimb

  committed
  6e8f8a6

  Copy full SHA for 6e8f8a6

Commits on Nov 21, 2024

• adding LinspaceShardingInterface; minor refactoring

  fschlimb

  committed
  a6cce35

  Copy full SHA for a6cce35

• adding test/Dialect/NDArray/Extensions/mesh-spmdization.mlir

  fschlimb

  committed
  02f0e54

  Copy full SHA for 02f0e54

Commits on Nov 26, 2024

• adding spmdization test for subview and insert_slice

  fschlimb

  committed
  677ac45

  Copy full SHA for 677ac45

• adding shardinginterfaces to CopyOp, DeleteOp and CastElemTypeOp

  fschlimb

  committed
  0f459cf

  Copy full SHA for 0f459cf

Commits on Nov 27, 2024

• removing ImmutableInsertSliceOp and ExtractSlice from ndarray

  fschlimb

  committed
  3365b9a

  Copy full SHA for 3365b9a

Commits on Nov 28, 2024

• removing region transforms; might need to enable one-shot-bufferize

  fschlimb

  committed
  65b37e1

  Copy full SHA for 65b37e1

Commits on Dec 4, 2024

• insertSlice 0d into nd works. Some refactoring;

  fschlimb

  committed
  ea582cb

  Copy full SHA for ea582cb

Commits on Dec 6, 2024

• copy constant tensors if inplace semantics needed. skipping coalescing if incomplete/no sharding info

  fschlimb

  committed
  1b0686d

  Copy full SHA for 1b0686d

Commits on Dec 9, 2024

• handle block arguments without sharding

  fschlimb

  committed
  bf4ba36

  Copy full SHA for bf4ba36

Commits on Dec 11, 2024

• moving bufferization of insertslice and subview from ndarraytolinalg to bufferizableinterface; add shardinginterface to reshape (incomplete)

  fschlimb

  committed
  f474abb

  Copy full SHA for f474abb

Commits on Dec 13, 2024

• fixes for insert_slice

  fschlimb

  committed
  6ee3453

  Copy full SHA for 6ee3453

Commits on Dec 17, 2024

• fixing sharding propagation and bufferization

  fschlimb

  committed
  2353eba

  Copy full SHA for 2353eba

Commits on Dec 19, 2024

• fixing shard sizes computation in coalescing and sharding interface

  fschlimb

  committed
  4653f79

  Copy full SHA for 4653f79