hub / github.com/NVIDIA/DALI / test_named_sharding_workflow

Function test_named_sharding_workflow

dali/test/python/jax_plugin/test_multigpu.py:317–321 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

315
316
317	def test_named_sharding_workflow():
318	mesh = Mesh(jax.devices(), axis_names=("device"))
319	sharding = NamedSharding(mesh, PartitionSpec("device"))
320
321	run_sharding_test(sharding)
322
323
324	def test_positional_sharding_workflow_with_iterator():

nothing calls this directly

run_sharding_testFunction · 0.85

no test coverage detected