fix the degenerated case - SemanticDiff

Commit

1 year ago

fix the degenerated case

References

#25699 - [shmap/partial-auto] Fixes lowering for jax.lax.axis_index in shard_map for degenerated shmaps.

#194 - CI: 01/07/25 upstream sync

Author

yliu120

yliu120

Committer

yliu120

yliu120

Parents

Loading