black

masahi · Feb 5, 2022 · 040eab0 · 040eab0
1 parent e5a07c2
commit 040eab0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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