failing test for unbound _device_num err in BEAM (#17326)

* min failing test

* switch to cpu

* err
This commit is contained in:
qazal
2026-07-31 20:24:44 +09:00
committed by GitHub
parent 93b74c75fc
commit 6d2700f0b7
+8
View File
@@ -76,6 +76,14 @@ class TestMultiTensor(unittest.TestCase):
run_linear(linear)
self.assertEqual(len(set(names)), 1, "function was relinearized")
@unittest.expectedFailure
def test_shard_beam(self):
cpu_2 = ("CPU:1", "CPU:2")
src = Tensor.ones(16).shard(cpu_2, 0).realize()
pad = src.to(cpu_2[::-1]).schedule_linear().src[0]
with Context(BEAM=1, IGNORE_BEAM_CACHE=1): prg = compile_linear(UOp(Ops.LINEAR, src=(pad,))).src[0].src[0]
self.assertNotEqual(prg.src[0].arg.applied_opts, ())
def test_shard_same_device(self):
X = Tensor.ones(256).contiguous().realize()
X.shard_((d1, X.device), 0)