(process_id)
| 90 | |
| 91 | |
| 92 | def test_positional_sharding_workflow(process_id): |
| 93 | mesh = Mesh(jax.devices(), axis_names=("device",)) |
| 94 | sharding = NamedSharding(mesh, PartitionSpec("device")) |
| 95 | |
| 96 | run_distributed_sharing_test(sharding=sharding, process_id=process_id) |
| 97 | |
| 98 | log.info("Passed positional sharding workflow test") |
| 99 | |
| 100 | |
| 101 | def test_named_sharding_workflow(process_id): |
no test coverage detected