From 78f76df1eb1602f66cacb888a97b6b267f8600a7 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Wed, 2 Feb 2022 07:31:54 +0900 Subject: [PATCH] run black --- python/tvm/contrib/cutlass/gen_gemm.py | 1 - python/tvm/contrib/cutlass/gen_tensor_op.py | 9 ++++++++- tests/python/contrib/test_cutlass.py | 1 - 3 files changed, 8 insertions(+), 3 deletions(-) diff --git a/python/tvm/contrib/cutlass/gen_gemm.py b/python/tvm/contrib/cutlass/gen_gemm.py index ac60f6c4ebc56..f059693819077 100644 --- a/python/tvm/contrib/cutlass/gen_gemm.py +++ b/python/tvm/contrib/cutlass/gen_gemm.py @@ -270,7 +270,6 @@ def profile( profile_all_alignments=profile_all_alignments, find_first_valid=find_first_valid, use_multiprocessing=use_multiprocessing, - ) name, opdef = create_gemm_operator_with_epilogue( diff --git a/python/tvm/contrib/cutlass/gen_tensor_op.py b/python/tvm/contrib/cutlass/gen_tensor_op.py index c1ebc05a873dc..76d43834c7ae6 100644 --- a/python/tvm/contrib/cutlass/gen_tensor_op.py +++ b/python/tvm/contrib/cutlass/gen_tensor_op.py @@ -63,7 +63,14 @@ def generate_tensor_op_common( def generate_sm75_tensor_op_1688( - out_dtype, arg0_dtype, arg1_dtype, op_creator, check_align, _, profile_all_alignments=False, accumlator_dtype="float32", + out_dtype, + arg0_dtype, + arg1_dtype, + op_creator, + check_align, + _, + profile_all_alignments=False, + accumlator_dtype="float32", ): """Generate GEMM or Conv2D kernels for Turing.""" assert out_dtype in ["float32", "float16", "int32"] diff --git a/tests/python/contrib/test_cutlass.py b/tests/python/contrib/test_cutlass.py index 0975058cefeae..cdd6d66c9accc 100644 --- a/tests/python/contrib/test_cutlass.py +++ b/tests/python/contrib/test_cutlass.py @@ -828,7 +828,6 @@ def test_conv2d_transpose(): ) - @pytest.mark.skip("weird") def test_conv2d_backward_weight(): OC = 8