AndrewZhaoLuo commented on code in PR #11672: URL: https://github.com/apache/tvm/pull/11672#discussion_r906293136
########## tests/python/integration/test_lower.py: ########## @@ -15,41 +15,52 @@ # specific language governing permissions and limitations # under the License. # pylint: disable=invalid-name, too-many-locals, too-many-statements, unused-argument -"""Test workload for lowering and build""" +"""Test workload for lowering and build.""" +import numpy as np + import tvm -from tvm import tir -from tvm.script import tir as T import tvm.testing -import numpy as np +from tvm.script import tir as T @T.prim_func -def tensorcore_gemm(a: T.handle, b: T.handle, c: T.handle) -> None: +def tensorcore_gemm(handle_a: T.handle, handle_b: T.handle, handle_c: T.handle) -> None: + # pylint: disable=missing-function-docstring Review Comment: this is tir script or whatever it's called, we can't actually insert a docstring since the parser does not handle it -- 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]
