diff --git a/python/tvm/contrib/cutlass/conv2d_profiler.py b/python/tvm/contrib/cutlass/conv2d_profiler.py index a3c83bf889f13..14a63256ab71c 100644 --- a/python/tvm/contrib/cutlass/conv2d_profiler.py +++ b/python/tvm/contrib/cutlass/conv2d_profiler.py @@ -19,6 +19,7 @@ from .library import DataTypeTag + class Conv2dProfilerEmitter(object): """Emit a C++ source for profiling CUTLASS kernels.""" @@ -208,6 +209,6 @@ def emit(self, op_def, op_name, element_output, split_k_slices=1): OperatorName=op_name, ElementOutput=DataTypeTag[element_output], SplitK=split_k_slices, - Reduction=self.reduction if split_k_slices > 1 else "" + Reduction=self.reduction if split_k_slices > 1 else "", ) return src