openxla/shardy issues and pull requests

#100 - Fix the case for ShardableDataFlowOpInterface` in `Value getDataFlowEdgeOwner(OpOperand& source)`

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

#100 - Fix the case for ShardableDataFlowOpInterface` in `Value getDataFlowEdgeOwner(OpOperand& source)`

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

#99 - Add SDY prefix to custom macros

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

#99 - Add SDY prefix to custom macros

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

#98 - [SDY] Add validation to assert that sharding group values cannot cross in between (and out of) manual computation ops.

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

#98 - [SDY] Add validation to assert that sharding group values cannot cross in between (and out of) manual computation ops.

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

#97 - #sdy Add sdy-round-trip-shard-map-export Pass.

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

#97 - #sdy Add sdy-round-trip-shard-map-export Pass.

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

#96 - #sdy Move shard map import pass to mhlo_round_trip.

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

#96 - #sdy Move shard map import pass to mhlo_round_trip.

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

#95 - #sdy Add support for exporting nested ManualComputations in MHLO export.

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

#95 - #sdy Add support for exporting nested ManualComputations in MHLO export.

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

#94 - #sdy Add C/Pybind API for manual axes and add `ManualComputationOp` Python test.

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

#94 - #sdy Add C/Pybind API for manual axes and add `ManualComputationOp` Python test.

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

#93 - Sink ShardableDataFlowEdgeOpInterface data flow edges

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

#93 - Sink ShardableDataFlowEdgeOpInterface data flow edges

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

#92 - Handle when JAX saves the manual axes on the `CallOp` of the `shmap_body`.

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

#92 - Handle when JAX saves the manual axes on the `CallOp` of the `shmap_body`.

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

#91 - Support JAX shard map lowering with Shardy enabled.

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

#91 - Support JAX shard map lowering with Shardy enabled.

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

#90 - Small cleanup of the AddDataFlowEdgePass, no functionality change

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

#90 - Small cleanup of the AddDataFlowEdgePass, no functionality change

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

#89 - Support mlir::ShapedType during import

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

#89 - Support mlir::ShapedType during import

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

#88 - Update doc of `ShardableDataFlowOpInterface` to include an example

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

#88 - Update doc of `ShardableDataFlowOpInterface` to include an example

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

#87 - Add an op interface to allow custom sharding rules

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

#87 - Add an op interface to allow custom sharding rules

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

#86 - Use data flow edge utils provided by `ShardableDataFlowOpInterface` during propagation

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

#86 - Use data flow edge utils provided by `ShardableDataFlowOpInterface` during propagation

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

#85 - Rename root in data flow edges to owner

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

#85 - Rename root in data flow edges to owner

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

#84 - Add interface functions needed for data flow edges

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

#84 - Add interface functions needed for data flow edges

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

#83 - Add helper functions to get/set all arg shardings

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

#83 - Add helper functions to get/set all arg shardings

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

#82 - [SDY] Refactor `propagateTensorSharding` methods to an external interface.

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

#82 - [SDY] Refactor `propagateTensorSharding` methods to an external interface.

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

#81 - #sdy make build/test bazel CI run faster

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

#81 - #sdy make build/test bazel CI run faster

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

#80 - #sdy remove timeout for build-test-bazel

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

#80 - #sdy remove timeout for build-test-bazel

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

#79 - failing build test

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

#79 - failing build test

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

#78 - failing test test

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

#78 - failing test test

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

#77 - Add `build-test-bazel` CI

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

#77 - Add `build-test-bazel` CI

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

#76 - #sdy add a workflow or building/testing with bazel

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

#76 - #sdy add a workflow or building/testing with bazel

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

#75 - Small cleanup to build file

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

#75 - Small cleanup to build file

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

#74 - #sdy add a `registerAllDialects` method and use it.

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

#74 - #sdy add a `registerAllDialects` method and use it.

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

#73 - #sdy Convert sdy.sharding=n to maximal sharding during import.

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

#73 - #sdy Convert sdy.sharding=n to maximal sharding during import.

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

#72 - #sdy Support the use of integer as sdy.sharding attribute in operations.

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

#72 - #sdy Support the use of integer as sdy.sharding attribute in operations.

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

#71 - #sdy add op interfaces doc

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

#71 - #sdy add op interfaces doc

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

#70 - #sdy apply a sharding constraint that isn't dangling and has multiple uses

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

#70 - #sdy apply a sharding constraint that isn't dangling and has multiple uses

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

#69 - modified op_sharding_rule_registry to not error on terminator ops

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

#69 - modified op_sharding_rule_registry to not error on terminator ops

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

#68 - #sdy Support device ID list in export.

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

#68 - #sdy Support device ID list in export.

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

#67 - [SDY] README.md nit: r/functionality/functional.

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

#67 - [SDY] README.md nit: r/functionality/functional.

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

#66 - #sdy Put MeshOp axes inside a pair of square brackets.

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

#66 - #sdy Put MeshOp axes inside a pair of square brackets.

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

#65 - Introduce OpInterface to allow sdy to get and set shardings of the op

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

#65 - Introduce OpInterface to allow sdy to get and set shardings of the op

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

#64 - #sdy Change device_id=n to device_ids=[n] in the MeshOp assembly format.

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

#64 - #sdy Change device_id=n to device_ids=[n] in the MeshOp assembly format.

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

#63 - #sdy Support a mesh with an arbitrary order of devices.

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

#63 - #sdy Support a mesh with an arbitrary order of devices.

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

#62 - #sdy Fix an error message.

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

#62 - #sdy Fix an error message.

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

#61 - /s/unique/common in failure msg

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

#61 - /s/unique/common in failure msg

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

#60 - [SDY] Dedupe the Sharding Group Ops within the same block which share the same input and group id. These duplicate ops can be introduced organically, but most likely will occur as part of the shard-group unification import pass. For example this will rewrite the ops in block

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

#60 - [SDY] Dedupe the Sharding Group Ops within the same block which share the same input and group id. These duplicate ops can be introduced organically, but most likely will occur as part of the shard-group unification import pass. For example this will rewrite the ops in block

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

#59 - Never use intptr_t* in C API --- prefer int64_t* instead.

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

#59 - Never use intptr_t* in C API --- prefer int64_t* instead.

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

#58 - #sdy add @MoveToDevice/Host custom calls to registry.

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

#58 - #sdy add @MoveToDevice/Host custom calls to registry.

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

#57 - Remove grp.h

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

#57 - Remove grp.h

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

#56 - Reverts 88e11f6dbbb8f286b8d9a05e97e80fdba32ec547

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

#56 - Reverts 88e11f6dbbb8f286b8d9a05e97e80fdba32ec547

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

#55 - #sdy Do not propagate for an op using more than one meshes in its operands and

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

#55 - #sdy Do not propagate for an op using more than one meshes in its operands and

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

#54 - #sdy Allow for empty meshes.

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

#54 - #sdy Allow for empty meshes.

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

#53 - [Shardy] Replace int64_t with intptr_t to match the C API types

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

#53 - [Shardy] Replace int64_t with intptr_t to match the C API types

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

#52 - #sdy don't copy sharding constraints with open dimensions to their input to match GSPMD propagation.

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

#52 - #sdy don't copy sharding constraints with open dimensions to their input to match GSPMD propagation.

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

#51 - [Shardy] Replace int64_t with intptr_t to match the C API types

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

#51 - [Shardy] Replace int64_t with intptr_t to match the C API types

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

GitHub / openxla/shardy issues and pull requests