├── .bazelrc ├── .bazelversion ├── .github └── workflows │ ├── build_test_bazel.yml │ └── buildifier.yml ├── CONTRIBUTING.md ├── LICENSE ├── MODULE.bazel ├── README.md ├── WORKSPACE ├── docs ├── _toc.yaml ├── automatic_partitioner.md ├── compiler_api.md ├── dialect_agnostic_sharding.md ├── getting_started_jax.ipynb ├── images │ ├── constant_splitting.png │ ├── op_priorities.png │ ├── projected_sharding.png │ ├── propagation.png │ ├── propagation_algorithm.png │ └── sharding_representation.png ├── mpmd │ ├── mpmd_common_passes.md │ ├── mpmd_dialect.md │ ├── mpmd_export_passes.md │ ├── mpmd_import_passes.md │ ├── mpmd_optimize_passes.md │ └── mpmd_sharding_propagation_passes.md ├── overview.md ├── propagation.md ├── sdy_dialect.md ├── sdy_export_passes.md ├── sdy_import_passes.md ├── sdy_op_interfaces.md ├── sdy_propagation_passes.md └── sharding_representation.md ├── rfcs ├── 2024-03-14-shardy-partitioner-rfc.md ├── 2025-06-18-mpmd-rfc-read-only-colab.ipynb ├── 2025-06-18-mpmd-rfc.md └── images │ ├── 2024-03-14-shardy-partitioner-rfc │ └── sdy0.png │ └── 2025-06-18-mpmd-rfc │ ├── mpmd_1f1b_xprof.png │ ├── mpmd_circular_xprof.png │ ├── mpmd_gpipe_xprof.png │ ├── mpmd_intro.jpg │ ├── pipelined_transformer.jpg │ ├── sharding_propagation.jpg │ └── spmd_xprof.png ├── shardy ├── BUILD ├── common │ ├── BUILD │ ├── file_utils.cc │ ├── file_utils.h │ ├── logging.cc │ ├── logging.h │ ├── save_module_op.cc │ └── save_module_op.h ├── dialect │ ├── mpmd │ │ ├── README │ │ ├── ir │ │ │ ├── BUILD │ │ │ ├── attrs.td │ │ │ ├── canonicalization.td │ │ │ ├── dialect.cc │ │ │ ├── dialect.h │ │ │ ├── dialect.td │ │ │ ├── enums.td │ │ │ ├── fragment_arg_res_attrs.h │ │ │ ├── fragment_execution_rules.cc │ │ │ ├── fragment_execution_rules.h │ │ │ ├── fragment_execution_rules_test.cc │ │ │ ├── ops.td │ │ │ ├── register.cc │ │ │ ├── register.h │ │ │ ├── test │ │ │ │ ├── BUILD │ │ │ │ ├── broadcast_canonicalization.mlir │ │ │ │ ├── broadcast_parse_and_print.mlir │ │ │ │ ├── call_err.mlir │ │ │ │ ├── call_op_validation.mlir │ │ │ │ ├── call_print_and_parse.mlir │ │ │ │ ├── cannot_inline_calls.mlir │ │ │ │ ├── fragment_call_parse_and_print.mlir │ │ │ │ ├── fragment_canonicalization.mlir │ │ │ │ ├── fragment_parse_and_print.mlir │ │ │ │ ├── memory_kind_parse_and_print.mlir │ │ │ │ ├── origin_attrs_parse_and_print.mlir │ │ │ │ ├── reduce_canonicalization.mlir │ │ │ │ ├── reduce_parse_and_print.mlir │ │ │ │ ├── reshard_fragment.mlir │ │ │ │ ├── transfer_canonicalization.mlir │ │ │ │ ├── transfer_parse_and_print.mlir │ │ │ │ ├── transfer_w_memory_kind_verify.mlir │ │ │ │ ├── verify_mesh_tensor_type.mlir │ │ │ │ └── verify_reduce_op.mlir │ │ │ ├── types.td │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ └── utils_test.cc │ │ └── transforms │ │ │ ├── BUILD │ │ │ ├── common │ │ │ ├── BUILD │ │ │ ├── absorb_inferred_fragments.cc │ │ │ ├── call_rewrites.cc │ │ │ ├── copy_constants.cc │ │ │ ├── distributed_function_pass.cc │ │ │ ├── distributed_function_pass.h │ │ │ ├── fragment_dce.cc │ │ │ ├── fragment_dedup.cc │ │ │ ├── merge_fragments.cc │ │ │ ├── merge_fragments.h │ │ │ ├── merge_transfers.cc │ │ │ ├── passes.h │ │ │ ├── passes.td │ │ │ ├── remove_transfer_cycles.cc │ │ │ ├── rule_based_merge.cc │ │ │ ├── scheduler_preprocess.cc │ │ │ ├── scheduler_preprocess.h │ │ │ ├── simplify_region_op_base.cc │ │ │ ├── simplify_region_op_base.h │ │ │ ├── split_bwd_fragments.cc │ │ │ ├── test │ │ │ │ ├── BUILD │ │ │ │ ├── absorb_inferred_fragments.mlir │ │ │ │ ├── call_inline.mlir │ │ │ │ ├── copy_constants.mlir │ │ │ │ ├── custom_merge_forward_with_backward.mlir │ │ │ │ ├── erase_unused_callee_block_arguments.mlir │ │ │ │ ├── fragment_dce.mlir │ │ │ │ ├── fragment_dedup.mlir │ │ │ │ ├── fragment_dedup_and_dce.mlir │ │ │ │ ├── from_unroll_to_call_counter.mlir │ │ │ │ ├── merge_inferred_any_consumer.mlir │ │ │ │ ├── merge_inferred_fragments.mlir │ │ │ │ ├── merge_inferred_fragments_sideways.mlir │ │ │ │ ├── merge_inferred_fragments_with_cloning.mlir │ │ │ │ ├── merge_transfers.mlir │ │ │ │ ├── merge_user_fragments_into_schedule_units.mlir │ │ │ │ ├── merged_stages_verification.mlir │ │ │ │ ├── merged_stages_verification_failure.mlir │ │ │ │ ├── remove_transfer_cycles.mlir │ │ │ │ ├── rule_based_merge.mlir │ │ │ │ ├── sink_negligible_ops_into_call_op.mlir │ │ │ │ ├── split_bwd_fragments.mlir │ │ │ │ ├── split_fragment_prioritize_transfer_independent_computations.mlir │ │ │ │ ├── uniquify_function_inputs_outputs_with_reshard.mlir │ │ │ │ ├── uniquify_function_inputs_outputs_with_transfer.mlir │ │ │ │ └── unroll_for_loops.mlir │ │ │ ├── testing_utils.h │ │ │ ├── uniquify_function_inputs_outputs.cc │ │ │ ├── unroll_for_loops.cc │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ └── utils_test.cc │ │ │ ├── export │ │ │ ├── BUILD │ │ │ ├── export_pipeline.cc │ │ │ ├── lower_to_fragment_calls.cc │ │ │ ├── mark_aliasing_and_donation.cc │ │ │ ├── mark_fragment_reserved_memory.cc │ │ │ ├── mark_input_output_with_layouts.cc │ │ │ ├── mark_offloaded_input_output.cc │ │ │ ├── naming_utils.cc │ │ │ ├── naming_utils.h │ │ │ ├── naming_utils_test.cc │ │ │ ├── passes.h │ │ │ ├── passes.td │ │ │ ├── reschedule_ops.cc │ │ │ ├── test │ │ │ │ ├── BUILD │ │ │ │ ├── delay_inferred_fragments.mlir │ │ │ │ ├── delay_transfers_from_cpu.mlir │ │ │ │ ├── export_pipeline.mlir │ │ │ │ ├── lower_to_fragment_calls.mlir │ │ │ │ ├── lower_to_fragment_calls_group_across_meshes.mlir │ │ │ │ ├── lower_to_fragment_calls_naming.mlir │ │ │ │ ├── lower_to_fragment_calls_pretty_print_origin.mlir │ │ │ │ ├── mark_aliasing_and_donation.mlir │ │ │ │ ├── mark_fragment_reserved_memory.mlir │ │ │ │ ├── mark_input_output_with_layouts.mlir │ │ │ │ ├── mark_offloaded_input_output.mlir │ │ │ │ └── mark_offloaded_input_output_failure.mlir │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ └── utils_test.cc │ │ │ ├── import │ │ │ ├── BUILD │ │ │ ├── copy_topology_from_main.cc │ │ │ ├── enforce_equisharding.cc │ │ │ ├── enforce_equisharding_test.cc │ │ │ ├── generate_sdy_meshes_from_topology_pass.cc │ │ │ ├── import_pipeline.cc │ │ │ ├── infer_mesh_assignment.cc │ │ │ ├── infer_mesh_assignment.h │ │ │ ├── infer_mesh_validation.cc │ │ │ ├── insert_nameless_clones_of_negligible_ops.cc │ │ │ ├── introduce_transfers.cc │ │ │ ├── map_input_output_to_mesh.cc │ │ │ ├── map_named_ops_to_mpmd_ops.cc │ │ │ ├── mesh_assignment_map.cc │ │ │ ├── mesh_assignment_map.h │ │ │ ├── mesh_inference_origins.cc │ │ │ ├── mesh_inference_origins.h │ │ │ ├── mesh_inference_utils.cc │ │ │ ├── mesh_inference_utils.h │ │ │ ├── mesh_inference_utils_test.cc │ │ │ ├── meshes_with_origins.cc │ │ │ ├── meshes_with_origins.h │ │ │ ├── meshes_with_origins_test.cc │ │ │ ├── passes.h │ │ │ ├── passes.td │ │ │ ├── sharding_constraints.cc │ │ │ ├── sharding_constraints.h │ │ │ ├── simplify_named_computations.cc │ │ │ ├── test │ │ │ │ ├── BUILD │ │ │ │ ├── call_op_set_topology.mlir │ │ │ │ ├── convert_none_reduce.mlir │ │ │ │ ├── generate_sdy_meshes_from_topology.mlir │ │ │ │ ├── import_pipeline.mlir │ │ │ │ ├── import_pipeline_with_heterogeneous_meshes.mlir │ │ │ │ ├── import_with_validation_errors.mlir │ │ │ │ ├── infer_mesh_assign_mesh_for_func_op_leaves.mlir │ │ │ │ ├── infer_mesh_assign_mesh_func_leaves_errors.mlir │ │ │ │ ├── infer_mesh_assign_using_input_output_constraints.mlir │ │ │ │ ├── infer_mesh_assignment_validate_no_additional_transfers.mlir │ │ │ │ ├── infer_mesh_assignment_validate_no_additional_transfers_failures.mlir │ │ │ │ ├── infer_mesh_assignment_validate_src_set_not_empty_failures.mlir │ │ │ │ ├── infer_mesh_convert_reduce_ops.mlir │ │ │ │ ├── infer_mesh_pipeline.mlir │ │ │ │ ├── infer_mesh_pipeline_broadcast_e2e.mlir │ │ │ │ ├── infer_mesh_pipeline_call_op.mlir │ │ │ │ ├── infer_mesh_pipeline_call_op_failures.mlir │ │ │ │ ├── infer_mesh_pipeline_create_reductions.mlir │ │ │ │ ├── infer_mesh_pipeline_create_transfers.mlir │ │ │ │ ├── infer_mesh_pipeline_failures.mlir │ │ │ │ ├── infer_mesh_populate_src_set.mlir │ │ │ │ ├── infer_mesh_populate_src_set_and_infer_reductions.mlir │ │ │ │ ├── infer_mesh_populate_use_set.mlir │ │ │ │ ├── infer_mesh_rewrite_using_analysis.mlir │ │ │ │ ├── infer_mesh_rewrite_using_analysis_failures.mlir │ │ │ │ ├── infer_mesh_rewrite_with_cloning.mlir │ │ │ │ ├── inline_nested_annotations.mlir │ │ │ │ ├── inline_nested_annotations_failed.mlir │ │ │ │ ├── insert_nameless_clones_of_negligible_ops.mlir │ │ │ │ ├── introduce_transfers.mlir │ │ │ │ ├── map_computations_to_stages.mlir │ │ │ │ ├── map_input_output_to_mesh.mlir │ │ │ │ ├── map_io_to_mesh_with_memory_kinds.mlir │ │ │ │ ├── map_named_computations_to_meshes.mlir │ │ │ │ ├── map_named_ops_to_meshes_with_memory_kinds.mlir │ │ │ │ ├── map_named_tensors_to_meshes.mlir │ │ │ │ ├── mesh_inference_finalize.mlir │ │ │ │ ├── mesh_inference_finalize_with_memory_kinds.mlir │ │ │ │ ├── mesh_inference_validation_error_limit.mlir │ │ │ │ ├── push_unassign_to_call_op.mlir │ │ │ │ ├── simplify_named_computations.mlir │ │ │ │ └── validate_no_named_ops.mlir │ │ │ └── validate_named_ops_in_mpmd_func.cc │ │ │ ├── optimize │ │ │ ├── BUILD │ │ │ ├── optimize_pipeline.cc │ │ │ ├── passes.h │ │ │ ├── passes.td │ │ │ ├── pipeline_schedule.cc │ │ │ ├── pipeline_schedule.h │ │ │ ├── remat_fragment.cc │ │ │ ├── rule_based_schedule.cc │ │ │ ├── scheduler.cc │ │ │ ├── scheduler.h │ │ │ ├── scheduling_and_rule_based_merge.cc │ │ │ ├── test │ │ │ │ ├── BUILD │ │ │ │ ├── fragment_remat.mlir │ │ │ │ ├── fragment_remat_merged_mode.mlir │ │ │ │ ├── import_and_optimize_pipeline.mlir │ │ │ │ ├── import_and_optimize_pipeline_merge_after_scheduling.mlir │ │ │ │ ├── move_transfers_to_producer.mlir │ │ │ │ ├── rule_based_schedule_conflicting_dependency.mlir │ │ │ │ ├── rule_based_schedule_missing_fragment.mlir │ │ │ │ ├── rule_based_schedule_three_fragments.mlir │ │ │ │ ├── rule_based_schedule_two_fragments.mlir │ │ │ │ └── verify_scheduling_units.mlir │ │ │ ├── utils.cc │ │ │ ├── utils.h │ │ │ └── utils_test.cc │ │ │ ├── passes.cc │ │ │ ├── passes.h │ │ │ ├── sharding_propagation │ │ │ ├── BUILD │ │ │ ├── convert_sdy_constants.cc │ │ │ ├── convert_sdy_shardings_to_mpmd_types.cc │ │ │ ├── enforce_user_shardings.cc │ │ │ ├── extract_reshards_from_inter_mesh_transfers.cc │ │ │ ├── passes.h │ │ │ ├── passes.td │ │ │ ├── sharding_propagation_pipeline.cc │ │ │ ├── simplify_program.cc │ │ │ ├── test │ │ │ │ ├── BUILD │ │ │ │ ├── add_sdy_data_flow_edges.mlir │ │ │ │ ├── convert_sdy_shardings_to_mpmd_types.mlir │ │ │ │ ├── convert_sdy_shardings_to_mpmd_types_failures.mlir │ │ │ │ ├── enforce_user_shardings.mlir │ │ │ │ ├── extract_reshard_from_inter_mesh_transfer.mlir │ │ │ │ ├── sharding_propagation_pipeline.mlir │ │ │ │ ├── simplify_program.mlir │ │ │ │ └── sink_sdy_data_flow_edges.mlir │ │ │ └── utils.h │ │ │ └── test │ │ │ ├── BUILD │ │ │ └── e2e_pipeline.mlir │ └── sdy │ │ ├── ir │ │ ├── BUILD │ │ ├── attrs.td │ │ ├── axis_list_ref.cc │ │ ├── axis_list_ref.h │ │ ├── axis_list_ref_test.cc │ │ ├── bytecode.cc │ │ ├── bytecode.h │ │ ├── bytecode.td │ │ ├── canonicalization.cc │ │ ├── canonicalization.td │ │ ├── compatibility.cc │ │ ├── compatibility.h │ │ ├── constants.h │ │ ├── dialect.cc │ │ ├── dialect.h │ │ ├── dialect.td │ │ ├── dialect_test.cc │ │ ├── enums.h │ │ ├── enums.td │ │ ├── extensions │ │ │ ├── stablehlo_extensions.cc │ │ │ └── stablehlo_extensions.h │ │ ├── macros.h │ │ ├── op_interface.td │ │ ├── ops.td │ │ ├── parsers.cc │ │ ├── parsers.h │ │ ├── printers.cc │ │ ├── printers.h │ │ ├── register.cc │ │ ├── register.h │ │ ├── test │ │ │ ├── BUILD │ │ │ ├── collective_canonicalization.mlir │ │ │ ├── collective_parse_print.mlir │ │ │ ├── collective_verification.mlir │ │ │ ├── constant_parse_print.mlir │ │ │ ├── constant_verification.mlir │ │ │ ├── data_flow_edge_verification.mlir │ │ │ ├── edge_sharding_parse_print.mlir │ │ │ ├── edge_sharding_verification.mlir │ │ │ ├── manual_computation_canonicalization.mlir │ │ │ ├── manual_computation_parse_print.mlir │ │ │ ├── manual_computation_verification.mlir │ │ │ ├── mesh_parse_print.mlir │ │ │ ├── mesh_verification.mlir │ │ │ ├── named_computation_parse_print.mlir │ │ │ ├── named_computation_verification.mlir │ │ │ ├── propagation_barrier_parse_print.mlir │ │ │ ├── propagation_barrier_verification.mlir │ │ │ ├── reshard_canonicalization.mlir │ │ │ ├── reshard_verification.mlir │ │ │ ├── sharding_constraint_verification.mlir │ │ │ ├── sharding_group_parse_print.mlir │ │ │ ├── sharding_rule_parse_print.mlir │ │ │ ├── sharding_rule_parsing_failure.mlir │ │ │ ├── sharding_rule_verification.mlir │ │ │ ├── tensor_sharding_parse_print.mlir │ │ │ ├── tensor_sharding_parsing_failure.mlir │ │ │ └── tensor_sharding_verification.mlir │ │ ├── testing_utils.h │ │ ├── utils.cc │ │ ├── utils.h │ │ ├── utils_test.cc │ │ ├── verifiers.cc │ │ └── verifiers.h │ │ └── transforms │ │ ├── BUILD │ │ ├── common │ │ ├── BUILD │ │ ├── op_properties.cc │ │ ├── op_properties.h │ │ ├── op_properties_test.cc │ │ ├── propagation_options.h │ │ ├── sharding_walker.cc │ │ └── sharding_walker.h │ │ ├── export │ │ ├── BUILD │ │ ├── close_shardings.cc │ │ ├── constant_or_scalar_merger.cc │ │ ├── drop_sharding_rules.cc │ │ ├── explicit_reshards_util.cc │ │ ├── explicit_reshards_util.h │ │ ├── export_pipeline.cc │ │ ├── insert_explicit_reshards.cc │ │ ├── passes.h │ │ ├── passes.td │ │ ├── remove_ag_rs_for_cmv1.cc │ │ ├── remove_propagation_debug_info.cc │ │ ├── remove_sharding_groups.cc │ │ ├── reshard_to_collectives.cc │ │ ├── sharding_constraint_to_reshard.cc │ │ ├── sink_data_flow_edges.cc │ │ ├── test │ │ │ ├── BUILD │ │ │ ├── close_shardings.mlir │ │ │ ├── constant_or_scalar_merger.mlir │ │ │ ├── drop_sharding_rules.mlir │ │ │ ├── export_pipeline.mlir │ │ │ ├── export_pipeline_explicit_collectives.mlir │ │ │ ├── insert_explicit_reshards.mlir │ │ │ ├── insert_explicit_reshards │ │ │ │ ├── bitcast_convert.mlir │ │ │ │ ├── broadcast_in_dim.mlir │ │ │ │ ├── cholesky_triangular_solve.mlir │ │ │ │ ├── clamp_select.mlir │ │ │ │ ├── collective_ops.mlir │ │ │ │ ├── concatenate.mlir │ │ │ │ ├── convolution.mlir │ │ │ │ ├── custom_call.mlir │ │ │ │ ├── data_flow_ops.mlir │ │ │ │ ├── dot_dot_general.mlir │ │ │ │ ├── dynamic_slice_dynamic_update_slice.mlir │ │ │ │ ├── elementwise_ops.mlir │ │ │ │ ├── fft.mlir │ │ │ │ ├── func_inputs_outputs.mlir │ │ │ │ ├── gather_scatter.mlir │ │ │ │ ├── manual_computation.mlir │ │ │ │ ├── meshes.mlir │ │ │ │ ├── pad_slice.mlir │ │ │ │ ├── reduce.mlir │ │ │ │ ├── reduce_window_select_and_scatter.mlir │ │ │ │ ├── reshape.mlir │ │ │ │ ├── reverse.mlir │ │ │ │ ├── rng_bit_generator.mlir │ │ │ │ ├── sort.mlir │ │ │ │ └── unreduced.mlir │ │ │ ├── insert_explicit_reshards_avoid_reshards_on_named_computations.mlir │ │ │ ├── remove_ag_rs_for_cmv1.mlir │ │ │ ├── remove_propagation_debug_info.mlir │ │ │ ├── remove_sharding_groups.mlir │ │ │ ├── reshard_to_collectives.mlir │ │ │ ├── reshard_to_collectives_keep_redundant_reshards_true.mlir │ │ │ ├── sharding_constraint_to_reshard.mlir │ │ │ ├── sink_data_flow_edges.mlir │ │ │ └── update_non_divisible_input_output_shardings.mlir │ │ └── update_non_divisible_input_output_shardings.cc │ │ ├── import │ │ ├── BUILD │ │ ├── add_data_flow_edges.cc │ │ ├── apply_sharding_constraints.cc │ │ ├── constant_or_scalar_splitter.cc │ │ ├── import_pipeline.cc │ │ ├── inline_meshes.cc │ │ ├── lift_inlined_meshes.cc │ │ ├── manual_axes_cleanup.cc │ │ ├── passes.h │ │ ├── passes.td │ │ ├── remove_size_one_axes.cc │ │ ├── sharding_group_import.cc │ │ └── test │ │ │ ├── BUILD │ │ │ ├── add_data_flow_edges.mlir │ │ │ ├── apply_sharding_constraints.mlir │ │ │ ├── constant_or_scalar_splitter.mlir │ │ │ ├── import_pipeline.mlir │ │ │ ├── inline_meshes.mlir │ │ │ ├── lift_inlined_meshes.mlir │ │ │ ├── manual_axes_cleanup.mlir │ │ │ ├── manual_axes_cleanup_failures.mlir │ │ │ ├── remove_size_one_axes.mlir │ │ │ ├── sharding_group_constraints.mlir │ │ │ └── sharding_group_import.mlir │ │ ├── passes.cc │ │ ├── passes.h │ │ └── propagation │ │ ├── BUILD │ │ ├── aggressive_factor_propagation.cc │ │ ├── aggressive_factor_propagation.h │ │ ├── aggressive_factor_propagation_test.cc │ │ ├── aggressive_propagation.cc │ │ ├── aggressive_propagation.h │ │ ├── auto_partitioner_registry.cc │ │ ├── auto_partitioner_registry.h │ │ ├── auto_partitioner_registry_test.cc │ │ ├── basic_factor_propagation.cc │ │ ├── basic_factor_propagation.h │ │ ├── basic_factor_propagation_test.cc │ │ ├── basic_propagation.cc │ │ ├── basic_propagation.h │ │ ├── debugging │ │ ├── BUILD │ │ ├── source_sharding.cc │ │ ├── source_sharding.h │ │ └── test │ │ │ ├── BUILD │ │ │ ├── edge_shardings.mlir │ │ │ └── sharding_origins.mlir │ │ ├── factor_propagation.h │ │ ├── op_priority_propagation.cc │ │ ├── op_priority_propagation.h │ │ ├── op_sharding_rule_builder.cc │ │ ├── op_sharding_rule_builder.h │ │ ├── op_sharding_rule_registry.cc │ │ ├── op_sharding_rule_registry.h │ │ ├── passes.h │ │ ├── passes.td │ │ ├── populate_op_sharding_rules.cc │ │ ├── propagation_pipeline.cc │ │ ├── sharding_group_map.cc │ │ ├── sharding_group_map.h │ │ ├── sharding_projection.cc │ │ ├── sharding_projection.h │ │ ├── sharding_projection_test.cc │ │ ├── test │ │ ├── BUILD │ │ ├── aggressive_propagation.mlir │ │ ├── basic_propagation.mlir │ │ ├── basic_propagation_conservative.mlir │ │ ├── basic_propagation_data_flow_edges.mlir │ │ ├── basic_propagation_keep_sharding_rules.mlir │ │ ├── basic_propagation_manual_computation.mlir │ │ ├── op_priority_propagation.mlir │ │ ├── op_sharding_rule_registry.mlir │ │ ├── op_sharding_rule_registry_conservative.mlir │ │ ├── op_sharding_rule_registry_failures.mlir │ │ ├── propagation_pipeline.mlir │ │ ├── propagation_pipeline_data_flow_edges.mlir │ │ ├── sharding_group_propagation.mlir │ │ └── user_priority_propagation.mlir │ │ ├── testing_utils.h │ │ ├── user_priority_propagation.cc │ │ ├── user_priority_propagation.h │ │ ├── utils.cc │ │ └── utils.h ├── integrations │ ├── c │ │ ├── BUILD │ │ ├── attributes.cc │ │ ├── attributes.h │ │ ├── dialect.cc │ │ ├── dialect.h │ │ ├── mpmd │ │ │ ├── BUILD │ │ │ ├── attributes.cc │ │ │ ├── attributes.h │ │ │ ├── dialect.cc │ │ │ ├── dialect.h │ │ │ ├── passes.cc │ │ │ └── passes.h │ │ ├── passes.cc │ │ └── passes.h │ └── python │ │ ├── BUILD │ │ ├── ir │ │ ├── BUILD │ │ ├── __init__.py │ │ ├── _ods_common.py │ │ ├── mpmd │ │ │ ├── BUILD │ │ │ ├── mpmd.py │ │ │ ├── mpmd_module.cc │ │ │ └── mpmd_ops.td │ │ ├── sdy.py │ │ ├── sdy_enums.td │ │ ├── sdy_module.cc │ │ └── sdy_ops.td │ │ └── jax │ │ └── mpmd │ │ ├── __init__.py │ │ ├── api.py │ │ ├── api_test.py │ │ ├── jax_mpmd_test.py │ │ ├── jaxlib │ │ ├── BUILD │ │ ├── mpmd_program.cc │ │ ├── mpmd_program.h │ │ └── throw_error.h │ │ ├── jit.py │ │ ├── jit_test.py │ │ ├── ops.py │ │ ├── ops_test.py │ │ ├── partitioning_options.py │ │ ├── stages.py │ │ ├── stages_test.py │ │ ├── test_utils.py │ │ ├── types.py │ │ ├── types_test.py │ │ ├── utils.py │ │ └── utils_test.py ├── lit.bzl ├── lit.cfg.py ├── lit.site.cfg.py.in └── tools │ ├── BUILD │ ├── mpmd_opt_main.cc │ ├── sdy_opt_main.cc │ └── sdy_translate_main.cc └── third_party ├── BUILD ├── llvm ├── BUILD ├── build.patch ├── generated.patch ├── lit_test.patch ├── llvm.BUILD ├── mathextras.patch ├── run_lit.sh ├── setup.bzl ├── toolchains.patch ├── workspace.bzl └── zstd.patch ├── py ├── BUILD ├── mlir │ ├── BUILD │ └── _mlir_libs │ │ └── BUILD ├── py_extension.bzl └── symlink_files.bzl ├── remote_config ├── BUILD ├── BUILD.tpl ├── common.bzl └── remote_platform_configure.bzl ├── repo.bzl └── stablehlo ├── BUILD ├── temporary.patch └── workspace.bzl /.bazelrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/.bazelrc -------------------------------------------------------------------------------- /.bazelversion: -------------------------------------------------------------------------------- 1 | 6.5.0 2 | -------------------------------------------------------------------------------- /.github/workflows/build_test_bazel.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/.github/workflows/build_test_bazel.yml -------------------------------------------------------------------------------- /.github/workflows/buildifier.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/.github/workflows/buildifier.yml -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/LICENSE -------------------------------------------------------------------------------- /MODULE.bazel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/MODULE.bazel -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/README.md -------------------------------------------------------------------------------- /WORKSPACE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/WORKSPACE -------------------------------------------------------------------------------- /docs/_toc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/_toc.yaml -------------------------------------------------------------------------------- /docs/automatic_partitioner.md: -------------------------------------------------------------------------------- 1 | # Automatic partitioner 2 | -------------------------------------------------------------------------------- /docs/compiler_api.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/compiler_api.md -------------------------------------------------------------------------------- /docs/dialect_agnostic_sharding.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/dialect_agnostic_sharding.md -------------------------------------------------------------------------------- /docs/getting_started_jax.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/getting_started_jax.ipynb -------------------------------------------------------------------------------- /docs/images/constant_splitting.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/images/constant_splitting.png -------------------------------------------------------------------------------- /docs/images/op_priorities.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/images/op_priorities.png -------------------------------------------------------------------------------- /docs/images/projected_sharding.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/images/projected_sharding.png -------------------------------------------------------------------------------- /docs/images/propagation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/images/propagation.png -------------------------------------------------------------------------------- /docs/images/propagation_algorithm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/images/propagation_algorithm.png -------------------------------------------------------------------------------- /docs/images/sharding_representation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/images/sharding_representation.png -------------------------------------------------------------------------------- /docs/mpmd/mpmd_common_passes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/mpmd/mpmd_common_passes.md -------------------------------------------------------------------------------- /docs/mpmd/mpmd_dialect.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/mpmd/mpmd_dialect.md -------------------------------------------------------------------------------- /docs/mpmd/mpmd_export_passes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/mpmd/mpmd_export_passes.md -------------------------------------------------------------------------------- /docs/mpmd/mpmd_import_passes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/mpmd/mpmd_import_passes.md -------------------------------------------------------------------------------- /docs/mpmd/mpmd_optimize_passes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/mpmd/mpmd_optimize_passes.md -------------------------------------------------------------------------------- /docs/mpmd/mpmd_sharding_propagation_passes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/mpmd/mpmd_sharding_propagation_passes.md -------------------------------------------------------------------------------- /docs/overview.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/overview.md -------------------------------------------------------------------------------- /docs/propagation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/propagation.md -------------------------------------------------------------------------------- /docs/sdy_dialect.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/sdy_dialect.md -------------------------------------------------------------------------------- /docs/sdy_export_passes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/sdy_export_passes.md -------------------------------------------------------------------------------- /docs/sdy_import_passes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/sdy_import_passes.md -------------------------------------------------------------------------------- /docs/sdy_op_interfaces.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/sdy_op_interfaces.md -------------------------------------------------------------------------------- /docs/sdy_propagation_passes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/sdy_propagation_passes.md -------------------------------------------------------------------------------- /docs/sharding_representation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/docs/sharding_representation.md -------------------------------------------------------------------------------- /rfcs/2024-03-14-shardy-partitioner-rfc.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/2024-03-14-shardy-partitioner-rfc.md -------------------------------------------------------------------------------- /rfcs/2025-06-18-mpmd-rfc-read-only-colab.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/2025-06-18-mpmd-rfc-read-only-colab.ipynb -------------------------------------------------------------------------------- /rfcs/2025-06-18-mpmd-rfc.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/2025-06-18-mpmd-rfc.md -------------------------------------------------------------------------------- /rfcs/images/2024-03-14-shardy-partitioner-rfc/sdy0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/images/2024-03-14-shardy-partitioner-rfc/sdy0.png -------------------------------------------------------------------------------- /rfcs/images/2025-06-18-mpmd-rfc/mpmd_1f1b_xprof.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/images/2025-06-18-mpmd-rfc/mpmd_1f1b_xprof.png -------------------------------------------------------------------------------- /rfcs/images/2025-06-18-mpmd-rfc/mpmd_circular_xprof.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/images/2025-06-18-mpmd-rfc/mpmd_circular_xprof.png -------------------------------------------------------------------------------- /rfcs/images/2025-06-18-mpmd-rfc/mpmd_gpipe_xprof.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/images/2025-06-18-mpmd-rfc/mpmd_gpipe_xprof.png -------------------------------------------------------------------------------- /rfcs/images/2025-06-18-mpmd-rfc/mpmd_intro.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/images/2025-06-18-mpmd-rfc/mpmd_intro.jpg -------------------------------------------------------------------------------- /rfcs/images/2025-06-18-mpmd-rfc/pipelined_transformer.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/images/2025-06-18-mpmd-rfc/pipelined_transformer.jpg -------------------------------------------------------------------------------- /rfcs/images/2025-06-18-mpmd-rfc/sharding_propagation.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/images/2025-06-18-mpmd-rfc/sharding_propagation.jpg -------------------------------------------------------------------------------- /rfcs/images/2025-06-18-mpmd-rfc/spmd_xprof.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/rfcs/images/2025-06-18-mpmd-rfc/spmd_xprof.png -------------------------------------------------------------------------------- /shardy/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/BUILD -------------------------------------------------------------------------------- /shardy/common/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/common/BUILD -------------------------------------------------------------------------------- /shardy/common/file_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/common/file_utils.cc -------------------------------------------------------------------------------- /shardy/common/file_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/common/file_utils.h -------------------------------------------------------------------------------- /shardy/common/logging.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/common/logging.cc -------------------------------------------------------------------------------- /shardy/common/logging.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/common/logging.h -------------------------------------------------------------------------------- /shardy/common/save_module_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/common/save_module_op.cc -------------------------------------------------------------------------------- /shardy/common/save_module_op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/common/save_module_op.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/README -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/attrs.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/attrs.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/canonicalization.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/canonicalization.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/dialect.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/dialect.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/dialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/dialect.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/dialect.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/dialect.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/enums.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/enums.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/fragment_arg_res_attrs.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/fragment_arg_res_attrs.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/fragment_execution_rules.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/fragment_execution_rules.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/fragment_execution_rules.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/fragment_execution_rules.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/fragment_execution_rules_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/fragment_execution_rules_test.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/ops.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/ops.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/register.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/register.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/register.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/register.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/broadcast_canonicalization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/broadcast_canonicalization.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/broadcast_parse_and_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/broadcast_parse_and_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/call_err.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/call_err.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/call_op_validation.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/call_op_validation.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/call_print_and_parse.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/call_print_and_parse.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/cannot_inline_calls.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/cannot_inline_calls.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/fragment_call_parse_and_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/fragment_call_parse_and_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/fragment_canonicalization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/fragment_canonicalization.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/fragment_parse_and_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/fragment_parse_and_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/memory_kind_parse_and_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/memory_kind_parse_and_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/origin_attrs_parse_and_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/origin_attrs_parse_and_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/reduce_canonicalization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/reduce_canonicalization.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/reduce_parse_and_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/reduce_parse_and_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/reshard_fragment.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/reshard_fragment.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/transfer_canonicalization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/transfer_canonicalization.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/transfer_parse_and_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/transfer_parse_and_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/transfer_w_memory_kind_verify.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/transfer_w_memory_kind_verify.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/verify_mesh_tensor_type.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/verify_mesh_tensor_type.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/test/verify_reduce_op.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/test/verify_reduce_op.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/types.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/types.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/utils.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/utils.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/ir/utils_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/ir/utils_test.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/absorb_inferred_fragments.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/absorb_inferred_fragments.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/call_rewrites.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/call_rewrites.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/copy_constants.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/copy_constants.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/distributed_function_pass.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/distributed_function_pass.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/distributed_function_pass.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/distributed_function_pass.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/fragment_dce.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/fragment_dce.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/fragment_dedup.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/fragment_dedup.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/merge_fragments.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/merge_fragments.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/merge_fragments.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/merge_fragments.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/merge_transfers.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/merge_transfers.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/passes.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/passes.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/remove_transfer_cycles.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/remove_transfer_cycles.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/rule_based_merge.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/rule_based_merge.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/scheduler_preprocess.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/scheduler_preprocess.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/scheduler_preprocess.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/scheduler_preprocess.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/simplify_region_op_base.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/simplify_region_op_base.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/simplify_region_op_base.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/simplify_region_op_base.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/split_bwd_fragments.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/split_bwd_fragments.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/absorb_inferred_fragments.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/absorb_inferred_fragments.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/call_inline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/call_inline.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/copy_constants.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/copy_constants.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/custom_merge_forward_with_backward.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/custom_merge_forward_with_backward.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/erase_unused_callee_block_arguments.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/erase_unused_callee_block_arguments.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/fragment_dce.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/fragment_dce.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/fragment_dedup.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/fragment_dedup.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/fragment_dedup_and_dce.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/fragment_dedup_and_dce.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/from_unroll_to_call_counter.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/from_unroll_to_call_counter.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/merge_inferred_any_consumer.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/merge_inferred_any_consumer.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/merge_inferred_fragments.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/merge_inferred_fragments.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/merge_inferred_fragments_sideways.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/merge_inferred_fragments_sideways.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/merge_inferred_fragments_with_cloning.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/merge_inferred_fragments_with_cloning.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/merge_transfers.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/merge_transfers.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/merge_user_fragments_into_schedule_units.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/merge_user_fragments_into_schedule_units.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/merged_stages_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/merged_stages_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/merged_stages_verification_failure.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/merged_stages_verification_failure.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/remove_transfer_cycles.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/remove_transfer_cycles.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/rule_based_merge.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/rule_based_merge.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/sink_negligible_ops_into_call_op.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/sink_negligible_ops_into_call_op.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/split_bwd_fragments.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/split_bwd_fragments.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/split_fragment_prioritize_transfer_independent_computations.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/split_fragment_prioritize_transfer_independent_computations.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/uniquify_function_inputs_outputs_with_reshard.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/uniquify_function_inputs_outputs_with_reshard.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/uniquify_function_inputs_outputs_with_transfer.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/uniquify_function_inputs_outputs_with_transfer.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/test/unroll_for_loops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/test/unroll_for_loops.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/testing_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/testing_utils.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/uniquify_function_inputs_outputs.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/uniquify_function_inputs_outputs.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/unroll_for_loops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/unroll_for_loops.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/utils.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/utils.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/common/utils_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/common/utils_test.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/export_pipeline.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/export_pipeline.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/lower_to_fragment_calls.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/lower_to_fragment_calls.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/mark_aliasing_and_donation.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/mark_aliasing_and_donation.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/mark_fragment_reserved_memory.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/mark_fragment_reserved_memory.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/mark_input_output_with_layouts.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/mark_input_output_with_layouts.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/mark_offloaded_input_output.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/mark_offloaded_input_output.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/naming_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/naming_utils.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/naming_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/naming_utils.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/naming_utils_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/naming_utils_test.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/passes.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/passes.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/reschedule_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/reschedule_ops.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/delay_inferred_fragments.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/delay_inferred_fragments.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/delay_transfers_from_cpu.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/delay_transfers_from_cpu.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/export_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/export_pipeline.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/lower_to_fragment_calls.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/lower_to_fragment_calls.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/lower_to_fragment_calls_group_across_meshes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/lower_to_fragment_calls_group_across_meshes.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/lower_to_fragment_calls_naming.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/lower_to_fragment_calls_naming.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/lower_to_fragment_calls_pretty_print_origin.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/lower_to_fragment_calls_pretty_print_origin.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/mark_aliasing_and_donation.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/mark_aliasing_and_donation.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/mark_fragment_reserved_memory.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/mark_fragment_reserved_memory.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/mark_input_output_with_layouts.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/mark_input_output_with_layouts.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/mark_offloaded_input_output.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/mark_offloaded_input_output.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/test/mark_offloaded_input_output_failure.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/test/mark_offloaded_input_output_failure.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/utils.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/utils.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/export/utils_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/export/utils_test.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/copy_topology_from_main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/copy_topology_from_main.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/enforce_equisharding.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/enforce_equisharding.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/enforce_equisharding_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/enforce_equisharding_test.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/generate_sdy_meshes_from_topology_pass.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/generate_sdy_meshes_from_topology_pass.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/import_pipeline.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/import_pipeline.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/infer_mesh_assignment.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/infer_mesh_assignment.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/infer_mesh_assignment.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/infer_mesh_assignment.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/infer_mesh_validation.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/infer_mesh_validation.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/insert_nameless_clones_of_negligible_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/insert_nameless_clones_of_negligible_ops.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/introduce_transfers.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/introduce_transfers.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/map_input_output_to_mesh.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/map_input_output_to_mesh.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/map_named_ops_to_mpmd_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/map_named_ops_to_mpmd_ops.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/mesh_assignment_map.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/mesh_assignment_map.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/mesh_assignment_map.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/mesh_assignment_map.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/mesh_inference_origins.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/mesh_inference_origins.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/mesh_inference_origins.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/mesh_inference_origins.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/mesh_inference_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/mesh_inference_utils.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/mesh_inference_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/mesh_inference_utils.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/mesh_inference_utils_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/mesh_inference_utils_test.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/meshes_with_origins.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/meshes_with_origins.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/meshes_with_origins.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/meshes_with_origins.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/meshes_with_origins_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/meshes_with_origins_test.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/passes.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/passes.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/sharding_constraints.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/sharding_constraints.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/sharding_constraints.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/sharding_constraints.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/simplify_named_computations.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/simplify_named_computations.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/call_op_set_topology.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/call_op_set_topology.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/convert_none_reduce.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/convert_none_reduce.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/generate_sdy_meshes_from_topology.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/generate_sdy_meshes_from_topology.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/import_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/import_pipeline.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/import_pipeline_with_heterogeneous_meshes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/import_pipeline_with_heterogeneous_meshes.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/import_with_validation_errors.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/import_with_validation_errors.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_assign_mesh_for_func_op_leaves.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_assign_mesh_for_func_op_leaves.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_assign_mesh_func_leaves_errors.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_assign_mesh_func_leaves_errors.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_assign_using_input_output_constraints.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_assign_using_input_output_constraints.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_assignment_validate_no_additional_transfers.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_assignment_validate_no_additional_transfers.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_assignment_validate_no_additional_transfers_failures.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_assignment_validate_no_additional_transfers_failures.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_assignment_validate_src_set_not_empty_failures.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_assignment_validate_src_set_not_empty_failures.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_convert_reduce_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_convert_reduce_ops.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_broadcast_e2e.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_broadcast_e2e.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_call_op.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_call_op.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_call_op_failures.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_call_op_failures.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_create_reductions.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_create_reductions.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_create_transfers.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_create_transfers.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_failures.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_pipeline_failures.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_populate_src_set.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_populate_src_set.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_populate_src_set_and_infer_reductions.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_populate_src_set_and_infer_reductions.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_populate_use_set.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_populate_use_set.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_rewrite_using_analysis.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_rewrite_using_analysis.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_rewrite_using_analysis_failures.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_rewrite_using_analysis_failures.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/infer_mesh_rewrite_with_cloning.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/infer_mesh_rewrite_with_cloning.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/inline_nested_annotations.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/inline_nested_annotations.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/inline_nested_annotations_failed.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/inline_nested_annotations_failed.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/insert_nameless_clones_of_negligible_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/insert_nameless_clones_of_negligible_ops.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/introduce_transfers.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/introduce_transfers.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/map_computations_to_stages.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/map_computations_to_stages.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/map_input_output_to_mesh.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/map_input_output_to_mesh.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/map_io_to_mesh_with_memory_kinds.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/map_io_to_mesh_with_memory_kinds.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/map_named_computations_to_meshes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/map_named_computations_to_meshes.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/map_named_ops_to_meshes_with_memory_kinds.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/map_named_ops_to_meshes_with_memory_kinds.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/map_named_tensors_to_meshes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/map_named_tensors_to_meshes.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/mesh_inference_finalize.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/mesh_inference_finalize.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/mesh_inference_finalize_with_memory_kinds.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/mesh_inference_finalize_with_memory_kinds.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/mesh_inference_validation_error_limit.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/mesh_inference_validation_error_limit.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/push_unassign_to_call_op.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/push_unassign_to_call_op.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/simplify_named_computations.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/simplify_named_computations.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/test/validate_no_named_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/test/validate_no_named_ops.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/import/validate_named_ops_in_mpmd_func.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/import/validate_named_ops_in_mpmd_func.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/optimize_pipeline.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/optimize_pipeline.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/passes.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/passes.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/pipeline_schedule.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/pipeline_schedule.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/pipeline_schedule.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/pipeline_schedule.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/remat_fragment.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/remat_fragment.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/rule_based_schedule.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/rule_based_schedule.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/scheduler.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/scheduler.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/scheduler.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/scheduler.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/scheduling_and_rule_based_merge.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/scheduling_and_rule_based_merge.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/fragment_remat.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/fragment_remat.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/fragment_remat_merged_mode.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/fragment_remat_merged_mode.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/import_and_optimize_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/import_and_optimize_pipeline.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/import_and_optimize_pipeline_merge_after_scheduling.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/import_and_optimize_pipeline_merge_after_scheduling.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/move_transfers_to_producer.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/move_transfers_to_producer.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/rule_based_schedule_conflicting_dependency.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/rule_based_schedule_conflicting_dependency.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/rule_based_schedule_missing_fragment.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/rule_based_schedule_missing_fragment.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/rule_based_schedule_three_fragments.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/rule_based_schedule_three_fragments.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/rule_based_schedule_two_fragments.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/rule_based_schedule_two_fragments.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/test/verify_scheduling_units.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/test/verify_scheduling_units.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/utils.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/utils.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/optimize/utils_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/optimize/utils_test.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/passes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/passes.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/passes.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/convert_sdy_constants.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/convert_sdy_constants.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/convert_sdy_shardings_to_mpmd_types.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/convert_sdy_shardings_to_mpmd_types.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/enforce_user_shardings.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/enforce_user_shardings.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/extract_reshards_from_inter_mesh_transfers.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/extract_reshards_from_inter_mesh_transfers.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/passes.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/passes.td -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/sharding_propagation_pipeline.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/sharding_propagation_pipeline.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/simplify_program.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/simplify_program.cc -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/test/add_sdy_data_flow_edges.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/test/add_sdy_data_flow_edges.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/test/convert_sdy_shardings_to_mpmd_types.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/test/convert_sdy_shardings_to_mpmd_types.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/test/convert_sdy_shardings_to_mpmd_types_failures.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/test/convert_sdy_shardings_to_mpmd_types_failures.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/test/enforce_user_shardings.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/test/enforce_user_shardings.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/test/extract_reshard_from_inter_mesh_transfer.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/test/extract_reshard_from_inter_mesh_transfer.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/test/sharding_propagation_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/test/sharding_propagation_pipeline.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/test/simplify_program.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/test/simplify_program.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/test/sink_sdy_data_flow_edges.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/test/sink_sdy_data_flow_edges.mlir -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/sharding_propagation/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/sharding_propagation/utils.h -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/mpmd/transforms/test/e2e_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/mpmd/transforms/test/e2e_pipeline.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/attrs.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/attrs.td -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/axis_list_ref.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/axis_list_ref.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/axis_list_ref.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/axis_list_ref.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/axis_list_ref_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/axis_list_ref_test.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/bytecode.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/bytecode.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/bytecode.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/bytecode.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/bytecode.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/bytecode.td -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/canonicalization.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/canonicalization.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/canonicalization.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/canonicalization.td -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/compatibility.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/compatibility.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/compatibility.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/compatibility.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/constants.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/constants.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/dialect.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/dialect.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/dialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/dialect.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/dialect.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/dialect.td -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/dialect_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/dialect_test.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/enums.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/enums.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/enums.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/enums.td -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/extensions/stablehlo_extensions.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/extensions/stablehlo_extensions.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/extensions/stablehlo_extensions.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/extensions/stablehlo_extensions.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/macros.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/macros.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/op_interface.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/op_interface.td -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/ops.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/ops.td -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/parsers.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/parsers.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/parsers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/parsers.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/printers.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/printers.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/printers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/printers.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/register.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/register.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/register.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/register.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/collective_canonicalization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/collective_canonicalization.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/collective_parse_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/collective_parse_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/collective_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/collective_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/constant_parse_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/constant_parse_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/constant_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/constant_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/data_flow_edge_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/data_flow_edge_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/edge_sharding_parse_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/edge_sharding_parse_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/edge_sharding_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/edge_sharding_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/manual_computation_canonicalization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/manual_computation_canonicalization.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/manual_computation_parse_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/manual_computation_parse_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/manual_computation_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/manual_computation_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/mesh_parse_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/mesh_parse_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/mesh_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/mesh_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/named_computation_parse_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/named_computation_parse_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/named_computation_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/named_computation_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/propagation_barrier_parse_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/propagation_barrier_parse_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/propagation_barrier_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/propagation_barrier_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/reshard_canonicalization.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/reshard_canonicalization.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/reshard_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/reshard_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/sharding_constraint_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/sharding_constraint_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/sharding_group_parse_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/sharding_group_parse_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/sharding_rule_parse_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/sharding_rule_parse_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/sharding_rule_parsing_failure.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/sharding_rule_parsing_failure.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/sharding_rule_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/sharding_rule_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/tensor_sharding_parse_print.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/tensor_sharding_parse_print.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/tensor_sharding_parsing_failure.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/tensor_sharding_parsing_failure.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/test/tensor_sharding_verification.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/test/tensor_sharding_verification.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/testing_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/testing_utils.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/utils.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/utils.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/utils_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/utils_test.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/verifiers.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/verifiers.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/ir/verifiers.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/ir/verifiers.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/common/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/common/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/common/op_properties.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/common/op_properties.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/common/op_properties.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/common/op_properties.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/common/op_properties_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/common/op_properties_test.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/common/propagation_options.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/common/propagation_options.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/common/sharding_walker.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/common/sharding_walker.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/common/sharding_walker.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/common/sharding_walker.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/close_shardings.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/close_shardings.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/constant_or_scalar_merger.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/constant_or_scalar_merger.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/drop_sharding_rules.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/drop_sharding_rules.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/explicit_reshards_util.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/explicit_reshards_util.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/explicit_reshards_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/explicit_reshards_util.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/export_pipeline.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/export_pipeline.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/insert_explicit_reshards.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/insert_explicit_reshards.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/passes.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/passes.td -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/remove_ag_rs_for_cmv1.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/remove_ag_rs_for_cmv1.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/remove_propagation_debug_info.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/remove_propagation_debug_info.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/remove_sharding_groups.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/remove_sharding_groups.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/reshard_to_collectives.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/reshard_to_collectives.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/sharding_constraint_to_reshard.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/sharding_constraint_to_reshard.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/sink_data_flow_edges.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/sink_data_flow_edges.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/close_shardings.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/close_shardings.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/constant_or_scalar_merger.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/constant_or_scalar_merger.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/drop_sharding_rules.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/drop_sharding_rules.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/export_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/export_pipeline.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/export_pipeline_explicit_collectives.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/export_pipeline_explicit_collectives.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/bitcast_convert.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/bitcast_convert.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/broadcast_in_dim.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/broadcast_in_dim.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/cholesky_triangular_solve.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/cholesky_triangular_solve.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/clamp_select.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/clamp_select.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/collective_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/collective_ops.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/concatenate.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/concatenate.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/convolution.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/convolution.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/custom_call.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/custom_call.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/data_flow_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/data_flow_ops.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/dot_dot_general.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/dot_dot_general.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/dynamic_slice_dynamic_update_slice.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/dynamic_slice_dynamic_update_slice.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/elementwise_ops.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/elementwise_ops.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/fft.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/fft.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/func_inputs_outputs.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/func_inputs_outputs.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/gather_scatter.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/gather_scatter.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/manual_computation.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/manual_computation.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/meshes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/meshes.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/pad_slice.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/pad_slice.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/reduce.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/reduce.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/reduce_window_select_and_scatter.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/reduce_window_select_and_scatter.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/reshape.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/reshape.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/reverse.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/reverse.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/rng_bit_generator.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/rng_bit_generator.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/sort.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/sort.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/unreduced.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards/unreduced.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards_avoid_reshards_on_named_computations.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/insert_explicit_reshards_avoid_reshards_on_named_computations.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/remove_ag_rs_for_cmv1.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/remove_ag_rs_for_cmv1.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/remove_propagation_debug_info.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/remove_propagation_debug_info.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/remove_sharding_groups.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/remove_sharding_groups.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/reshard_to_collectives.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/reshard_to_collectives.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/reshard_to_collectives_keep_redundant_reshards_true.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/reshard_to_collectives_keep_redundant_reshards_true.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/sharding_constraint_to_reshard.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/sharding_constraint_to_reshard.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/sink_data_flow_edges.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/sink_data_flow_edges.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/test/update_non_divisible_input_output_shardings.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/test/update_non_divisible_input_output_shardings.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/export/update_non_divisible_input_output_shardings.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/export/update_non_divisible_input_output_shardings.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/add_data_flow_edges.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/add_data_flow_edges.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/apply_sharding_constraints.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/apply_sharding_constraints.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/constant_or_scalar_splitter.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/constant_or_scalar_splitter.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/import_pipeline.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/import_pipeline.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/inline_meshes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/inline_meshes.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/lift_inlined_meshes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/lift_inlined_meshes.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/manual_axes_cleanup.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/manual_axes_cleanup.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/passes.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/passes.td -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/remove_size_one_axes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/remove_size_one_axes.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/sharding_group_import.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/sharding_group_import.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/add_data_flow_edges.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/add_data_flow_edges.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/apply_sharding_constraints.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/apply_sharding_constraints.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/constant_or_scalar_splitter.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/constant_or_scalar_splitter.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/import_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/import_pipeline.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/inline_meshes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/inline_meshes.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/lift_inlined_meshes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/lift_inlined_meshes.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/manual_axes_cleanup.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/manual_axes_cleanup.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/manual_axes_cleanup_failures.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/manual_axes_cleanup_failures.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/remove_size_one_axes.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/remove_size_one_axes.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/sharding_group_constraints.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/sharding_group_constraints.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/import/test/sharding_group_import.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/import/test/sharding_group_import.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/passes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/passes.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/passes.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/aggressive_factor_propagation.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/aggressive_factor_propagation.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/aggressive_factor_propagation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/aggressive_factor_propagation.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/aggressive_factor_propagation_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/aggressive_factor_propagation_test.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/aggressive_propagation.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/aggressive_propagation.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/aggressive_propagation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/aggressive_propagation.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/auto_partitioner_registry.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/auto_partitioner_registry.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/auto_partitioner_registry.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/auto_partitioner_registry.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/auto_partitioner_registry_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/auto_partitioner_registry_test.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/basic_factor_propagation.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/basic_factor_propagation.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/basic_factor_propagation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/basic_factor_propagation.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/basic_factor_propagation_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/basic_factor_propagation_test.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/basic_propagation.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/basic_propagation.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/basic_propagation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/basic_propagation.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/debugging/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/debugging/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/debugging/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/debugging/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/debugging/test/edge_shardings.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/debugging/test/edge_shardings.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/debugging/test/sharding_origins.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/debugging/test/sharding_origins.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/factor_propagation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/factor_propagation.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/op_priority_propagation.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/op_priority_propagation.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/op_priority_propagation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/op_priority_propagation.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/op_sharding_rule_builder.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/op_sharding_rule_builder.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/op_sharding_rule_builder.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/op_sharding_rule_builder.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/op_sharding_rule_registry.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/op_sharding_rule_registry.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/op_sharding_rule_registry.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/op_sharding_rule_registry.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/passes.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/passes.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/passes.td -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/populate_op_sharding_rules.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/populate_op_sharding_rules.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/propagation_pipeline.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/propagation_pipeline.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/sharding_group_map.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/sharding_group_map.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/sharding_group_map.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/sharding_group_map.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/sharding_projection.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/sharding_projection.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/sharding_projection.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/sharding_projection.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/sharding_projection_test.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/sharding_projection_test.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/BUILD -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/aggressive_propagation.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/aggressive_propagation.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/basic_propagation.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/basic_propagation.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/basic_propagation_conservative.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/basic_propagation_conservative.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/basic_propagation_data_flow_edges.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/basic_propagation_data_flow_edges.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/basic_propagation_keep_sharding_rules.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/basic_propagation_keep_sharding_rules.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/basic_propagation_manual_computation.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/basic_propagation_manual_computation.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/op_priority_propagation.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/op_priority_propagation.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/op_sharding_rule_registry.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/op_sharding_rule_registry.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/op_sharding_rule_registry_conservative.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/op_sharding_rule_registry_conservative.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/op_sharding_rule_registry_failures.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/op_sharding_rule_registry_failures.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/propagation_pipeline.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/propagation_pipeline.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/propagation_pipeline_data_flow_edges.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/propagation_pipeline_data_flow_edges.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/sharding_group_propagation.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/sharding_group_propagation.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/test/user_priority_propagation.mlir: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/test/user_priority_propagation.mlir -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/testing_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/testing_utils.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/user_priority_propagation.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/user_priority_propagation.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/user_priority_propagation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/user_priority_propagation.h -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/utils.cc -------------------------------------------------------------------------------- /shardy/dialect/sdy/transforms/propagation/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/dialect/sdy/transforms/propagation/utils.h -------------------------------------------------------------------------------- /shardy/integrations/c/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/BUILD -------------------------------------------------------------------------------- /shardy/integrations/c/attributes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/attributes.cc -------------------------------------------------------------------------------- /shardy/integrations/c/attributes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/attributes.h -------------------------------------------------------------------------------- /shardy/integrations/c/dialect.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/dialect.cc -------------------------------------------------------------------------------- /shardy/integrations/c/dialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/dialect.h -------------------------------------------------------------------------------- /shardy/integrations/c/mpmd/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/mpmd/BUILD -------------------------------------------------------------------------------- /shardy/integrations/c/mpmd/attributes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/mpmd/attributes.cc -------------------------------------------------------------------------------- /shardy/integrations/c/mpmd/attributes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/mpmd/attributes.h -------------------------------------------------------------------------------- /shardy/integrations/c/mpmd/dialect.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/mpmd/dialect.cc -------------------------------------------------------------------------------- /shardy/integrations/c/mpmd/dialect.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/mpmd/dialect.h -------------------------------------------------------------------------------- /shardy/integrations/c/mpmd/passes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/mpmd/passes.cc -------------------------------------------------------------------------------- /shardy/integrations/c/mpmd/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/mpmd/passes.h -------------------------------------------------------------------------------- /shardy/integrations/c/passes.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/passes.cc -------------------------------------------------------------------------------- /shardy/integrations/c/passes.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/c/passes.h -------------------------------------------------------------------------------- /shardy/integrations/python/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/BUILD -------------------------------------------------------------------------------- /shardy/integrations/python/ir/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/BUILD -------------------------------------------------------------------------------- /shardy/integrations/python/ir/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/__init__.py -------------------------------------------------------------------------------- /shardy/integrations/python/ir/_ods_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/_ods_common.py -------------------------------------------------------------------------------- /shardy/integrations/python/ir/mpmd/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/mpmd/BUILD -------------------------------------------------------------------------------- /shardy/integrations/python/ir/mpmd/mpmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/mpmd/mpmd.py -------------------------------------------------------------------------------- /shardy/integrations/python/ir/mpmd/mpmd_module.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/mpmd/mpmd_module.cc -------------------------------------------------------------------------------- /shardy/integrations/python/ir/mpmd/mpmd_ops.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/mpmd/mpmd_ops.td -------------------------------------------------------------------------------- /shardy/integrations/python/ir/sdy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/sdy.py -------------------------------------------------------------------------------- /shardy/integrations/python/ir/sdy_enums.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/sdy_enums.td -------------------------------------------------------------------------------- /shardy/integrations/python/ir/sdy_module.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/sdy_module.cc -------------------------------------------------------------------------------- /shardy/integrations/python/ir/sdy_ops.td: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/ir/sdy_ops.td -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/__init__.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/api.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/api_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/api_test.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/jax_mpmd_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/jax_mpmd_test.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/jaxlib/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/jaxlib/BUILD -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/jaxlib/mpmd_program.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/jaxlib/mpmd_program.cc -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/jaxlib/mpmd_program.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/jaxlib/mpmd_program.h -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/jaxlib/throw_error.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/jaxlib/throw_error.h -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/jit.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/jit_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/jit_test.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/ops.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/ops_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/ops_test.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/partitioning_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/partitioning_options.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/stages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/stages.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/stages_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/stages_test.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/test_utils.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/types.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/types_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/types_test.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/utils.py -------------------------------------------------------------------------------- /shardy/integrations/python/jax/mpmd/utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/integrations/python/jax/mpmd/utils_test.py -------------------------------------------------------------------------------- /shardy/lit.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/lit.bzl -------------------------------------------------------------------------------- /shardy/lit.cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/lit.cfg.py -------------------------------------------------------------------------------- /shardy/lit.site.cfg.py.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/lit.site.cfg.py.in -------------------------------------------------------------------------------- /shardy/tools/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/tools/BUILD -------------------------------------------------------------------------------- /shardy/tools/mpmd_opt_main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/tools/mpmd_opt_main.cc -------------------------------------------------------------------------------- /shardy/tools/sdy_opt_main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/tools/sdy_opt_main.cc -------------------------------------------------------------------------------- /shardy/tools/sdy_translate_main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/shardy/tools/sdy_translate_main.cc -------------------------------------------------------------------------------- /third_party/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/llvm/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/llvm/BUILD -------------------------------------------------------------------------------- /third_party/llvm/build.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/llvm/build.patch -------------------------------------------------------------------------------- /third_party/llvm/generated.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/llvm/generated.patch -------------------------------------------------------------------------------- /third_party/llvm/lit_test.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/llvm/lit_test.patch -------------------------------------------------------------------------------- /third_party/llvm/llvm.BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/llvm/mathextras.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/llvm/mathextras.patch -------------------------------------------------------------------------------- /third_party/llvm/run_lit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/llvm/run_lit.sh -------------------------------------------------------------------------------- /third_party/llvm/setup.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/llvm/setup.bzl -------------------------------------------------------------------------------- /third_party/llvm/toolchains.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/llvm/toolchains.patch -------------------------------------------------------------------------------- /third_party/llvm/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/llvm/workspace.bzl -------------------------------------------------------------------------------- /third_party/llvm/zstd.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/llvm/zstd.patch -------------------------------------------------------------------------------- /third_party/py/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/py/mlir/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/py/mlir/BUILD -------------------------------------------------------------------------------- /third_party/py/mlir/_mlir_libs/BUILD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/py/mlir/_mlir_libs/BUILD -------------------------------------------------------------------------------- /third_party/py/py_extension.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/py/py_extension.bzl -------------------------------------------------------------------------------- /third_party/py/symlink_files.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/py/symlink_files.bzl -------------------------------------------------------------------------------- /third_party/remote_config/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/remote_config/BUILD.tpl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/remote_config/BUILD.tpl -------------------------------------------------------------------------------- /third_party/remote_config/common.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/remote_config/common.bzl -------------------------------------------------------------------------------- /third_party/remote_config/remote_platform_configure.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/remote_config/remote_platform_configure.bzl -------------------------------------------------------------------------------- /third_party/repo.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/repo.bzl -------------------------------------------------------------------------------- /third_party/stablehlo/BUILD: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /third_party/stablehlo/temporary.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/stablehlo/temporary.patch -------------------------------------------------------------------------------- /third_party/stablehlo/workspace.bzl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/openxla/shardy/HEAD/third_party/stablehlo/workspace.bzl --------------------------------------------------------------------------------