openxla/shardy issues and pull requests

#50 - #sdy Add custom-call sharding rule registry.

Pull Request - State: open - Opened by copybara-service[bot] 4 months ago

#50 - #sdy Add custom-call sharding rule registry.

Pull Request - State: open - Opened by copybara-service[bot] 4 months ago

#49 - Exposed sdy's `verifyTensorShardingAttr` and added an option for checking divisibility

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#49 - Exposed sdy's `verifyTensorShardingAttr` and added an option for checking divisibility

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#48 - Change to a warning if an op is not known to the sharding rule registry

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#48 - Change to a warning if an op is not known to the sharding rule registry

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#47 - Add jax_test configs for shardy and enable it for pjit_test.py and fix any tests.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#47 - Add jax_test configs for shardy and enable it for pjit_test.py and fix any tests.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#46 - [SDY] add sdy shard-group unification pass. This cl unifies shard group ids by "merging" groups which contain tensors that belong to more than one group across the module. Additionally it reindexes the sharding group ids to not have any gaps after merging. For example

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#46 - [SDY] add sdy shard-group unification pass. This cl unifies shard group ids by "merging" groups which contain tensors that belong to more than one group across the module. Additionally it reindexes the sharding group ids to not have any gaps after merging. For example

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#45 - #sdy support `ShapedTensor` in Shardy as long as it has a rank and static shape.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#45 - #sdy support `ShapedTensor` in Shardy as long as it has a rank and static shape.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#44 - #sdy remove IdentityOp as it's no longer needed.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#44 - #sdy remove IdentityOp as it's no longer needed.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#43 - `propagateFactorShardings` should be the only public method in factor propagation.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#43 - `propagateFactorShardings` should be the only public method in factor propagation.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#42 - #sdy Export pass to make function inputs/outputs evenly sharded.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#42 - #sdy Export pass to make function inputs/outputs evenly sharded.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#41 - Cleanup. Remove unused code in aggressive_factor_propagation_test.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#41 - Cleanup. Remove unused code in aggressive_factor_propagation_test.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#40 - #sdy Add Python binding for DimMapping, TensorMapping and OpShardingRule.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#40 - #sdy Add Python binding for DimMapping, TensorMapping and OpShardingRule.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#39 - Introduce `isFullyReplicated()` for TensorShardingAttr

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#39 - Introduce `isFullyReplicated()` for TensorShardingAttr

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#38 - #sdy remove uses of designated initializers for Windows support

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#38 - #sdy remove uses of designated initializers for Windows support

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#37 - #sdy Fix sdy_test.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#37 - #sdy Fix sdy_test.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#36 - reference to local binding

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#36 - reference to local binding

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#35 - test2

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#35 - test2

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#34 - test

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#34 - test

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#33 - #sdy remove use of designated initializers for windows support.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#33 - #sdy remove use of designated initializers for windows support.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#32 - #sdy add missing != operators

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#32 - #sdy add missing != operators

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#31 - Fixed compilation on Windows.

Pull Request - State: open - Opened by eaplatanios 4 months ago - 5 comments

#31 - Fixed compilation on Windows.

Pull Request - State: open - Opened by eaplatanios 4 months ago - 5 comments

#30 - Add `strictPrefixOf` method to AxisRefAttr.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#30 - Add `strictPrefixOf` method to AxisRefAttr.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#29 - Add getSharedSize for DimensionShardingAttr

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#29 - Add getSharedSize for DimensionShardingAttr

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#28 - Improve aggressive factor propagation strategy in Shardy. There are two main differences from `BasicFactorPropagation`.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#28 - Improve aggressive factor propagation strategy in Shardy. There are two main differences from `BasicFactorPropagation`.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#27 - remove full path import

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#27 - remove full path import

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#26 - Don't copy sharding constraints to `sdy.data_flow_edge` since they hold the sharding of multiple targets, not just the root.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#26 - Don't copy sharding constraints to `sdy.data_flow_edge` since they hold the sharding of multiple targets, not just the root.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#25 - #sdy add a canonicalization pattern to `sdy.manual_computation` that removes used block args and their corresponding operand.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#25 - #sdy add a canonicalization pattern to `sdy.manual_computation` that removes used block args and their corresponding operand.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#24 - #sdy Make `-sharding-constraint-to-reshard` pass simple by always replacing an `sdy.sharding_constraint` with an `sdy.reshard`.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#24 - #sdy Make `-sharding-constraint-to-reshard` pass simple by always replacing an `sdy.sharding_constraint` with an `sdy.reshard`.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#23 - #sdy fix sharding rule of @PartialReduce/ApproxTopK custom call.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#23 - #sdy fix sharding rule of @PartialReduce/ApproxTopK custom call.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#22 - #sdy Apply input shardings of `ManualComputationOp` in `-apply-sharding-constraints` pass, this is because they are in essense sharding constraints given by the user.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#22 - #sdy Apply input shardings of `ManualComputationOp` in `-apply-sharding-constraints` pass, this is because they are in essense sharding constraints given by the user.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#21 - #sdy Add canonicalization rewrite pattern for ReshardOp.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#21 - #sdy Add canonicalization rewrite pattern for ReshardOp.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#20 - Reverts 3885a17bf4277d8747c67324176dc956bb325a13

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#20 - Reverts 3885a17bf4277d8747c67324176dc956bb325a13

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#19 - Add yml rule for BUILD file lint checking.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#19 - Add yml rule for BUILD file lint checking.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#18 - Add linter yml rule for BUILD files.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#18 - Add linter yml rule for BUILD files.

Pull Request - State: closed - Opened by copybara-service[bot] 4 months ago

#17 - avoid copy in lambda capture and fix another reference to local binding

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#17 - avoid copy in lambda capture and fix another reference to local binding

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#16 - Fix another reference to local binding

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#16 - Fix another reference to local binding

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#15 - Fix another reference to local binding

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#15 - Fix another reference to local binding

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#14 - fix local binding

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#14 - fix local binding

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#13 - Add missing dependency.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#13 - Add missing dependency.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#12 - #sdy update dialect doc

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#12 - #sdy update dialect doc

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#11 - #sdy add SDY pass docs

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#11 - #sdy add SDY pass docs

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#10 - #sdy add SDY dialect doc

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#9 - #sdy add a pass to apply sharding constraints that dictate the sharding of their input.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#8 - Use explicit capture in lambda.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#7 - [SDY] Add equi-sharding op `ShardingGroupOp` parsing and printing.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#6 - Default `TensorShardingAttr` `replicated_axes` field of constructor to empty list.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#5 - #shardy. Fix dynamic_update_slice sharding rule.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#4 - remove TODO in registry

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#3 - delete shardy:friends

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#2 - Add support for propagating through gather/scatter batching dims.

Pull Request - State: closed - Opened by copybara-service[bot] 5 months ago

#1 - Publish additional changes

Pull Request - State: closed - Opened by tomnatan30 5 months ago

GitHub / openxla/shardy issues and pull requests