search

openxla / shardy

MLIR-based partitioning system
Apache License 2.0
45 stars 7 forks source link

Explicitly handle ManualComputationOp in sharding walker to avoid removing/adding manual axes from shardings. #240

Closed copybara-service[bot] closed 5 days ago

copybara-service[bot] commented 5 days ago

Explicitly handle ManualComputationOp in sharding walker to avoid removing/adding manual axes from shardings.