black

masahi · masahi · commit b724f446d070 · 2021-12-12T08:06:03.000+09:00
diff --git a/python/tvm/contrib/cutlass/gen_gemm.py b/python/tvm/contrib/cutlass/gen_gemm.py
@@ -24,7 +24,7 @@
     ProfilerEngine,
     generate_sm75_tensor_op_1688,
     generate_sm80_tensor_op_16816,
-    GENERATOR_FUNC_TABLE
+    GENERATOR_FUNC_TABLE,
 )
 from .library import (
     EpilogueFunctor,
@@ -133,8 +133,6 @@ def create_gemm_operator(
     return ret
 
 
-
-
 # TODO(masahi): A sensible way to pick reasonable default kernels
 DEFAULT_KERNELS = {
     75: {
diff --git a/python/tvm/contrib/cutlass/gen_tensor_op.py b/python/tvm/contrib/cutlass/gen_tensor_op.py
@@ -164,6 +164,7 @@ def get_tile_descriptions(math_inst):
     80: generate_sm80_tensor_op_16816,
 }
 
+
 class ProfilerEngine:
     """Compile and run a given profiler executable."""
 

Original file line number	Diff line number	Diff line change
`@@ -164,6 +164,7 @@ def get_tile_descriptions(math_inst):`
`164`	`164`	`80: generate_sm80_tensor_op_16816,`
`165`	`165`	`}`
`166`	`166`
	`167`	`+`
`167`	`168`	`class ProfilerEngine:`
`168`	`169`	`"""Compile and run a given profiler executable."""`
`169`	`170`