Improved test infrastructure. #205
GitHub Actions / TT-XLA Tests
failed
Dec 18, 2024 in 0s
365 tests run, 342 passed, 22 skipped, 1 failed.
Annotations
Check failure on line 21 in tests/TTIR/test_device.py
github-actions / TT-XLA Tests
test_device.test_to_device
AssertionError: assert 'cpu' == 'wormhole'
- wormhole
+ cpu
Raw output
def test_to_device():
cpu_array = random_input_tensor((32, 32))
device = jax.devices()[0]
tt_array = jax.device_put(cpu_array, device)
> assert tt_array.device.device_kind == "wormhole"
E AssertionError: assert 'cpu' == 'wormhole'
E
E - wormhole
E + cpu
tests/TTIR/test_device.py:21: AssertionError
Loading