Boian Petkantchin 7a4c49756d [mlir][mesh] Use one type for mesh axis (#76830) ...

Make all ops and attributes use the types MeshAxis and MeshAxesAttr
instead of int16_t, int32_t, DenseI16ArrayAttr and DenseI32ArrayAttr.

2024-01-03 15:47:11 -08:00

..

CMakeLists.txt

[mlir][mesh] Add resharding spmdization on a 1D device mesh (#76179)

2024-01-02 15:50:07 -08:00

ShardingPropagation.cpp

[MLIR][Mesh] Add sharding propagation pass (#71261)

2023-11-03 21:07:31 -07:00

Simplifications.cpp

[mlir][mesh] Add endomorphism simplification for all-reduce (#73150)

2023-12-12 10:21:52 -08:00

Spmdization.cpp

[mlir][mesh] Use one type for mesh axis (#76830)

2024-01-03 15:47:11 -08:00