Function _assert_run_result_same

tests/python/relax/test_training_optimizer_numeric.py:61–64

(tvm_func: Callable, np_func: Callable, np_inputs: list)

def _assert_run_result_same(tvm_func: Callable, np_func: Callable, np_inputs: list):
    result = _tvm_to_numpy(tvm_func(*[_numpy_to_tvm(i) for i in np_inputs]))
    expected = np_func(*np_inputs)
    _assert_allclose_nested(result, expected)


@tvm.testing.parametrize_targets("llvm")