Commit c9ff10d
authored
1 file changed
+1
-1
lines changed- include/ttmlir/Conversion/StableHLOToTTIR/ShardingUtils.h+71-9
- include/ttmlir/Conversion/StableHLOToTTIR/ShardyToTTIR.h+22
- include/ttmlir/Dialect/TT/IR/TTOpsTypes.h+33
- include/ttmlir/Dialect/TT/IR/TTOpsTypes.td+19
- include/ttmlir/Dialect/TT/Utils/Mesh.h+66
- include/ttmlir/Dialect/TTIR/IR/TTIROps.td+3-2
- include/ttmlir/Dialect/TTNN/IR/TTNNOps.td+11
- include/ttmlir/Dialect/TTNN/IR/TTNNWorkarounds.h+5
- include/ttmlir/Dialect/TTNN/Utils/Utils.h-2
- lib/Conversion/StableHLOToTTIR/CMakeLists.txt+1
- lib/Conversion/StableHLOToTTIR/ShardingUtils.cpp+188-109
- lib/Conversion/StableHLOToTTIR/ShardyToTTIRPatterns.cpp+225
- lib/Conversion/StableHLOToTTIR/StableHLOToTTIRPass.cpp+19-2
- lib/Conversion/StableHLOToTTIR/StableHLOToTTIRPatterns.cpp+141-134
- lib/Conversion/TTIRToTTIRDecomposition/TTIRToTTIRDecomposition.cpp+93-92
- lib/Conversion/TTIRToTTIRDecomposition/TTIRToTTIRDecompositionPass.cpp+1-1
- lib/Conversion/TTIRToTTNN/TTIRToTTNN.cpp+4-4
- lib/Dialect/TT/IR/TTOpsTypes.cpp+1-29
- lib/Dialect/TTIR/IR/TTIROps.cpp+4-6
- lib/Dialect/TTIR/Transforms/Utility.cpp+7-1
- lib/Dialect/TTNN/Analysis/LegalLayoutAnalysis.cpp+3-3
- lib/Dialect/TTNN/IR/TTNNOps.cpp+3-4
- lib/Dialect/TTNN/IR/TTNNWorkarounds.cpp+31-1
- lib/Dialect/TTNN/Transforms/TTNNDecomposeLayouts.cpp+4-7
- lib/Dialect/TTNN/Transforms/TTNNLayout.cpp+2-2
- lib/Dialect/TTNN/Transforms/Workarounds/TTNNWorkarounds.cpp+1-1
- lib/Dialect/TTNN/Utils/TransformUtils.cpp+2-2
- lib/Dialect/TTNN/Utils/Utils.cpp+1-33
- runtime/lib/common/system_desc.cpp+4-2
- runtime/lib/ttmetal/runtime.cpp+3-1
- runtime/lib/ttnn/operations/ccl/mesh_shard.cpp+14-4
- runtime/lib/ttnn/runtime.cpp+3-1
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/all_reduce.mlir+26
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/ccl_ops_gspmd.mlir+65-1
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/ccl_ops_shardy.mlir+332
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/e2e_dp_gspmd.mlir+192
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/e2e_dp_shardy.mlir+163
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/e2e_fsdp_gspmd.mlir+205
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/e2e_fsdp_shardy.mlir+175
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/e2e_fsdp_tp_gspmd.mlir+166
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/e2e_fsdp_tp_shardy.mlir+136
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/e2e_tp_gspmd.mlir+157
- test/ttmlir/Conversion/StableHLOToTTIR/ccl/e2e_tp_shardy.mlir+103
- test/ttmlir/Decomposition/TTIR/pooling/max_pool2d.mlir+103
- test/ttmlir/Dialect/TTNN/Transforms/Workarounds/where_workaround.mlir+29
- test/ttmlir/Dialect/TTNN/ccl/mesh_shard.mlir+6-6
- test/ttmlir/Dialect/TTNN/typecast.mlir+17-14
- test/ttmlir/Silicon/StableHLO/n150/maxpool2d_op.mlir+125-12
- test/ttmlir/Silicon/StableHLO/n150/select_op.mlir+29
- third_party/CMakeLists.txt+1-1
0 commit comments