Pull Request Overview

Opened on September 17, 2026
Status Merged
Commit count 1 with first commit September 17, 2026

Total Delta

151 Total Diff Delta

Open Days

Open 1 weekdays

Test Delta

71 Diff Delta in Test Files

Breakdown by Phase

How long has this pull request spent in each phase of its lifecycle?

Data pending calculation for pull request

Lower shard_map in partial explicit manual mode correctly by using the aval.sharding.spec instead of `in_specs` since the former contain more info (explicit mesh axes too) while in_specs only refer to axes passed in axis_names.

Lower shard_map in partial explicit manual mode correctly by using the aval.sharding.spec instead of in_specs since the former contain more info (explicit mesh axes too) while in_specs only refer to axes passed in axis_names.

Also add partial manual support to the check which checks if in_specs == aval.sharding.spec (otherwise it errors out).

PR was closed without comments.

Lower shard_map in partial explicit manual mode correctly by using the aval.sharding.spec instead of `in_specs` since the former contain more info (explicit mesh axes too) while in_specs only refer to axes passed in axis_names. #40776

Pull Request Overview

Total Delta

Open Days

Test Delta

Lower shard_map in partial explicit manual mode correctly by using the aval.sharding.spec instead of `in_specs` since the former contain more info (explicit mesh axes too) while in_specs only refer to axes passed in axis_names.