Skip to content

Commit

Permalink
remove changes to test files
Browse files Browse the repository at this point in the history
  • Loading branch information
hypercubestart committed Mar 12, 2020
1 parent d0d9f54 commit 1783a45
Showing 1 changed file with 35 additions and 2 deletions.
37 changes: 35 additions & 2 deletions tests/python/relay/test_ir_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
from tvm import te
from tvm import relay
from tvm.relay.analysis import graph_equal, assert_graph_equal
from tvm.relay.analysis import alpha_equal, assert_alpha_equal
import pytest
from numpy import isclose
from typing import Union
Expand Down Expand Up @@ -866,10 +867,42 @@ def test_extern_adt_defn():
""",
mod
)

def test_import_grad():
mod = tvm.IRModule()
mod.import_from_std("gradient.rly")

if __name__ == "__main__":
pytest.main([__file__])
test_comments()
test_int_literal()
test_float_literal()
test_bool_literal()
test_negative()
test_bin_op()
test_parens()
test_op_assoc()
test_let()
test_seq()
test_graph()
test_tuple()
test_func()
test_defn()
test_recursive_call()
test_ifelse()
test_call()
test_incomplete_type()
test_builtin_types()
test_tensor_type()
test_function_type()
test_tuple_type()
test_adt_defn()
test_empty_adt_defn()
test_multiple_cons_defn()
test_multiple_type_param_defn()
test_match()
test_adt_cons_expr()
test_duplicate_adt_defn()
test_duplicate_adt_cons()
test_duplicate_adt_cons_defn()
test_duplicate_global_var()
test_extern_adt_defn()
test_import_grad()

0 comments on commit 1783a45

Please sign in to comment.