slyubomirsky commented on code in PR #15171: URL: https://github.com/apache/tvm/pull/15171#discussion_r1244389951
########## python/tvm/tir/schedule/testing.py: ########## @@ -16,14 +16,27 @@ # under the License. # pylint: disable=dangerous-default-value """Testing utilities for the TensorIR schedule API""" -from typing import Sequence, Union +from typing import Any, Sequence, Union import tvm from tvm.ir import IRModule, assert_structural_equal from tvm.tir import PrimFunc from tvm.tir.schedule import Schedule, Trace +def assert_structural_equal_gs(f1: PrimFunc, f2: PrimFunc, *args: Any, **kwargs: Any) -> None: Review Comment: I don't like the name but I didn't want it to be too verbose because this has to be used in a lot of places. Also, is there a better file where this should reside? It's used in tests outside of those related to scheduling. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
