run black

apache · Apr 19, 2022 · e504536 · e504536
1 parent 7fdd4da
commit e504536
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/tvm/tir/schedule/analysis.py b/python/tvm/tir/schedule/analysis.py
@@ -67,7 +67,9 @@ class TensorizeInfo(Object):
     """Necessary information used for tensorization."""
 
 
-def get_tensorize_loop_mapping(sch: Schedule, block: BlockRV, desc_func: PrimFunc) -> Optional[TensorizeInfo]:
+def get_tensorize_loop_mapping(
+    sch: Schedule, block: BlockRV, desc_func: PrimFunc
+) -> Optional[TensorizeInfo]:
     """Establish a mapping between loops in a target block and an intrinsic description
 
     Parameters