black

ylc · Dec 11, 2021 · b724f44 · b724f44
1 parent fe4687b
commit b724f44
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
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 change
@@ Expand Up / @@ -164,6 +164,7 @@ def get_tile_descriptions(math_inst): @@
 : generate_sm80_tensor_op_16816,
     }
     class ProfilerEngine:
         """Compile and run a given profiler executable."""
@@ Expand Down @@