issues
search
openxla
/
shardy
MLIR-based partitioning system
Apache License 2.0
45
stars
7
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
#sdy make Sdy ops adhere to `InferTypeOpInterface` and expose various utility functions to get ready to make `xla/mlir_hlo/stablehlo_ext` shape refinement code able to support shardy.
#246
copybara-service[bot]
opened
2 days ago
0
Reserve backing data for axis list ref tests.
#245
copybara-service[bot]
closed
2 days ago
0
#sdy cleanup: remove unused pipeline creation function and add `Sdy` namespace to ops in `ops.td`.
#244
copybara-service[bot]
closed
3 days ago
0
Add iterator for AxisListRef.
#243
copybara-service[bot]
closed
4 days ago
0
Fix MacOS breakage due to structured binding being captured by reference round 2.
#242
copybara-service[bot]
closed
4 days ago
0
#sdy bump version due to JAX MacOS breakage round 2.
#241
copybara-service[bot]
closed
4 days ago
0
Explicitly handle ManualComputationOp in sharding walker to avoid removing/adding manual axes from shardings.
#240
copybara-service[bot]
closed
5 days ago
0
#sdy make `ManualComputationOp` adhere to the `ShardableDataFlowOpInterface`.
#239
copybara-service[bot]
closed
5 days ago
0
Rename AxesWithTail to AxisListRef and move to its own file on shardy/dialect/sdy/ir.
#238
copybara-service[bot]
closed
4 days ago
0
#sdy #debug Add test for `ManualComputationOp` with manual axes in source sharding debugging.
#237
copybara-service[bot]
closed
4 days ago
0
#sdy bump version due to JAX MacOS breakage
#236
copybara-service[bot]
closed
5 days ago
0
Fix MacOS breakage due to structured binding being captured by reference.
#235
copybara-service[bot]
closed
5 days ago
0
Use a vector instead of map when iterating to find the best candidates and expanding the current assignments.
#234
copybara-service[bot]
closed
6 days ago
0
Add getLocalTypeFromValue utility
#233
copybara-service[bot]
closed
6 days ago
0
#sdy Implement `ShardableDataFlowOpInterface`'s `setOpResultEdgeOwnerSharding` and `setBlockArgumentEdgeOwnerSharding` through the plural functions.
#232
copybara-service[bot]
closed
6 days ago
0
Instead of re-creating factor-axes pair candidates at each iteration, keep a map of factor-axes pairs to candidates, and maintain.
#231
copybara-service[bot]
closed
6 days ago
0
Add getLocalTensorType method to TensorShardingAttr
#230
copybara-service[bot]
closed
6 days ago
0
#sdy Introduce `transformEdgeSharding` to the `ShardableDataFlowOpInterface` and use it during propagation.
#229
copybara-service[bot]
closed
6 days ago
0
Choose the axes with larger size when the counts and factor indexes are the same.
#228
copybara-service[bot]
closed
6 days ago
0
Add reshard to all_gather pass
#227
copybara-service[bot]
closed
1 week ago
0
Tie break factor-axes pairs on sharded tensor size when the counts are the same, favor the larger one.
#226
copybara-service[bot]
closed
5 days ago
0
Trim axes to use the largest prefix that does not overlap with the best, instead of erasing.
#225
copybara-service[bot]
closed
4 days ago
0
Build system support
#224
nsmithtt
opened
1 week ago
2
Consider all prefixes of a tensor sharding so that largest available prefix can be assigned to factors.
#223
copybara-service[bot]
closed
1 week ago
0
[SDY] fix JAX layouts tests for Shardy.
#222
copybara-service[bot]
closed
1 week ago
0
#sdy Add debugging tool to show the origin shardings on the MLIR module.
#221
copybara-service[bot]
closed
6 days ago
0
Add an option to skip converting sharding_constraint to reshard
#220
copybara-service[bot]
opened
1 week ago
0
Use FactorAxesPair instead of FactorAxesAssignmentCandidate to hold the best one.
#219
copybara-service[bot]
closed
1 week ago
0
Count factor-axes candidates accurately for when one axes is prefix of another, and extend from existing assignment.
#218
copybara-service[bot]
closed
1 week ago
0
Introduce and use AxesWithTail struct to represent axes with a tailing AxisRefAttr.
#217
copybara-service[bot]
closed
1 week ago
0
Update tie-break so that larger axes based on AxesWithTail comparator has a preference.
#216
copybara-service[bot]
closed
1 week ago
0
Factor out finding factor-axes pairs and their counts.
#215
copybara-service[bot]
closed
1 week ago
0
Refactor to introduce and use AxesWithTail struct to represent axes with a tailing AxisRefAttr.
#214
copybara-service[bot]
closed
1 week ago
0
Refactor to use FactorAxesPair.overlap api when finding and invalidating factor-axes pairs with overlapping axes with the picked one.
#213
copybara-service[bot]
closed
2 weeks ago
0
Add a tailing AxisRefAttr to FactorAxesPair, so that the FactorAxesPair can represent any axes, excluding its most-minor axis, that is a prefix of an existing factor sharding.
#212
copybara-service[bot]
closed
2 weeks ago
0
Comment third_party BUILD files
#211
copybara-service[bot]
closed
2 weeks ago
0
Refactor to use FactorAxesAssignmentCandidate.assignTo api for assigning the picked axes to the picked factor.
#210
copybara-service[bot]
closed
2 weeks ago
0
Refactor to use FactorAxesPair.overlaps api for invalidating factor-axes pairs with the picked factor or with an overlapping axes.
#209
copybara-service[bot]
closed
2 weeks ago
0
Find common axes deterministically by not depending on DenseMap iteration order to find the best factor-axes pair.
#208
copybara-service[bot]
closed
2 weeks ago
0
[SDY:nit] update summary of sharding group op to be more descriptive.
#207
copybara-service[bot]
closed
2 weeks ago
0
Change sharding rule of @PartialReduce custom call so we don't propagate sideways between the operands or results along the reduction dim.
#206
copybara-service[bot]
closed
2 weeks ago
0
Use AxisRefsWithTail instead of ArrayRef of AxisRefs, so the tailing AxisRefAttr can be modified on factor-axis pairs.
#205
copybara-service[bot]
closed
2 weeks ago
0
Fix typos in docs.
#204
copybara-service[bot]
closed
2 weeks ago
0
Fix Shardy ToC for website
#203
copybara-service[bot]
closed
2 weeks ago
0
Fix links in docs
#202
copybara-service[bot]
closed
2 weeks ago
0
Use DenseMap of factor-axis pairs instead of a vector whose elements correspond to factors.
#201
copybara-service[bot]
closed
2 weeks ago
0
Refactor to use a std::pair to hold factor index and axis refs.
#200
copybara-service[bot]
closed
2 weeks ago
0
PJRT doc typo fix for website
#199
copybara-service[bot]
closed
2 weeks ago
0
Use FactorAxesAssignmentCandidate struct to simplify the logic that picks the best factor axes pair.
#198
copybara-service[bot]
closed
2 weeks ago
0
Use majority vote heuristic to find common axes for factors, for multiple axes.
#197
copybara-service[bot]
closed
2 weeks ago
0
Next