MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / test_float_lastheavy

Function test_float_lastheavy

tests/unit/runtime/utils/test_partition.py:170–175  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

168
169@pytest.mark.skip(reason="Variance-minimizing partitioning returns different result.")
170def test_float_lastheavy():
171 weights = [0., 1.1, 1.9, 3., 30.]
172 P = 2
173 parts = partition_balanced(weights, P)
174 assert_valid_partition(weights, parts, P)
175 assert parts == [0, 4, 5]
176
177
178def test_float_midheavy():

Callers

nothing calls this directly

Calls 2

partition_balancedFunction · 0.90
assert_valid_partitionFunction · 0.85

Tested by

no test coverage detected