ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Consolidate review tests

Browse source
This commit is contained in:
Daniel Han 2026-04-21 00:34:27 +00:00
commit e169ff2cd7
1 changed files with 0 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

11
test_infer_buffer_only_multi_device_recurses.py
View file

@ -79,14 +79,3 @@ def test_buffer_only_subtree_with_multi_device_recurses():
assert dm.get("a") == torch.device("cuda", 0), dm
assert dm.get("b") == torch.device("cuda", 1), dm
assert "" not in dm, dm
def test_buffer_only_subtree_with_single_device_collapses():
"""Param-less subtree with buffers all on one device still collapses to a
single-entry prefix; no accidental recursion when not needed."""
infer, _ = _load_fns()
a = _BufMod(buffers = [("cache", "cuda:1")])
b = _BufMod(buffers = [("cache", "cuda:1")])
root = _BufMod(children = [("a", a), ("b", b)])
dm = infer(root)
assert dm == {"": torch.device("cuda", 1)}, dm