()
| 37 | |
| 38 | @pytest.mark.skipif(cuda_driver is None, reason="Requires cuda-python") |
| 39 | def test_cuda_driver_stream() -> None: |
| 40 | assert cuda_driver is not None |
| 41 | echo = tvm_ffi.get_global_func("testing.echo") |
| 42 | stream = cuda_driver.CUstream(0) |
| 43 | y = echo(stream) |
| 44 | assert y is not None |
| 45 | z = echo(cuda_driver.CUstream(1)) |
| 46 | assert isinstance(z, ctypes.c_void_p) |
| 47 | assert z.value == 1 |
| 48 | |
| 49 | |
| 50 | def gen_check_stream_mod() -> tvm_ffi.Module: |