(session_kind)
| 212 | |
| 213 | @pytest.mark.parametrize("session_kind", _all_session_kinds) |
| 214 | def test_shape_tuple(session_kind): |
| 215 | num_workers = 4 |
| 216 | sess = session_kind(num_workers=num_workers) |
| 217 | func: di.DPackedFunc = sess.get_global_func("tests.disco.shape_tuple") |
| 218 | result: di.DRef = func(Shape([1, 2, 3])) |
| 219 | for i in range(num_workers): |
| 220 | value = result.debug_get_from_remote(i) |
| 221 | assert isinstance(value, Shape) |
| 222 | assert list(value) == [1, 2, 3, 4, 5] |
| 223 | |
| 224 | |
| 225 | @pytest.mark.parametrize("session_kind", _all_session_kinds) |
nothing calls this directly
no test coverage detected
searching dependent graphs…