diff --git a/examples/developer_mode/gelu_mul_developer.py b/examples/developer_mode/gelu_mul_developer.py index 65eb800bf..7c0e7b910 100644 --- a/examples/developer_mode/gelu_mul_developer.py +++ b/examples/developer_mode/gelu_mul_developer.py @@ -73,6 +73,6 @@ def main(A: T.Tensor((M, N), dtype), B: T.Tensor((M, N // 2), dtype)): a1, a2 = torch.split(a, N // 2, dim=1) ref_b = gelu(a1) * a2 torch.testing.assert_close(b.cpu(), ref_b.cpu(), rtol=1e-2, atol=1e-2) - print("Test passed!") + print("Test passed") print("Kernel Output Match!") diff --git a/testing/python/language/test_tilelang_ascend_language_parallel.py b/testing/python/language/test_tilelang_ascend_language_parallel.py index 076a7e78b..953ff7c96 100644 --- a/testing/python/language/test_tilelang_ascend_language_parallel.py +++ b/testing/python/language/test_tilelang_ascend_language_parallel.py @@ -97,7 +97,7 @@ def main( @staticmethod @tilelang.jit(out_idx=[-1], pass_configs=pass_configs) def binary_op_kernel_int(M, N, block_M, block_N, op_func, dtype="int16"): - """Generic binary operation kernel for int types""" + """Generic binary operation kernel for int type""" m_num = M // block_M n_num = N // block_N VEC_NUM = 2