Commit b587fc9
committed
Mark Qwen distribution tests as multi_device for CI collection
test_distribution and test_layout_map_mesh_shapes exercise real
DeviceMesh/ModelParallel sharding but were missing the multi_device
marker, so the dedicated CI job (-m "multi_device and not
kaggle_key_required and not extra_large") added in keras-team#2809 would
collect zero tests from this file.
For test_layout_map_mesh_shapes, the marker must go directly on the
function, below @parameterized.named_parameters(...): absl's
named_parameters expands each parameterization into a fresh bound
function via functools.wraps(test_method), which only carries over
attributes already present on the original function's __dict__, not
attributes attached afterward to the _ParameterizedTestIter wrapper
object that an outermost decorator would target. Placing the pytest
mark outermost silently drops it from every expanded test case.
test_layout_map_live_presets is already marked and intentionally
excluded by this job's own kaggle_key_required/extra_large filters,
so it is left untouched.1 parent 7c2c235 commit b587fc9
1 file changed
Lines changed: 2 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
171 | 171 | | |
172 | 172 | | |
173 | 173 | | |
| 174 | + | |
174 | 175 | | |
175 | 176 | | |
176 | 177 | | |
| |||
206 | 207 | | |
207 | 208 | | |
208 | 209 | | |
| 210 | + | |
209 | 211 | | |
210 | 212 | | |
211 | 213 | | |
| |||
0 commit comments