()
| 2196 | |
| 2197 | |
| 2198 | def test_gpu2cpu_arg_input(): |
| 2199 | @pipeline_def(batch_size=1, num_threads=4, device_id=0, exec_dynamic=True) |
| 2200 | def pdef(): |
| 2201 | data = dali.types.Constant([42], device="gpu") |
| 2202 | resized = fn.zeros(shape=data.cpu(), dtype=types.INT32) |
| 2203 | return resized |
| 2204 | |
| 2205 | pipe = pdef() |
| 2206 | (o,) = pipe.run() |
| 2207 | assert o[0].shape() == [42] |
| 2208 | |
| 2209 | |
| 2210 | def test_gpu2cpu2mixed(): |