()
| 48 | |
| 49 | |
| 50 | def gen_check_stream_mod() -> tvm_ffi.Module: |
| 51 | return tvm_ffi.cpp.load_inline( |
| 52 | name="check_stream", |
| 53 | cpp_sources=""" |
| 54 | void check_stream(int device_type, int device_id, uint64_t stream) { |
| 55 | uint64_t cur_stream = reinterpret_cast<uint64_t>(TVMFFIEnvGetStream(device_type, device_id)); |
| 56 | TVM_FFI_ICHECK_EQ(cur_stream, stream); |
| 57 | } |
| 58 | """, |
| 59 | functions=["check_stream"], |
| 60 | ) |
| 61 | |
| 62 | |
| 63 | def test_raw_stream() -> None: |
no outgoing calls
no test coverage detected