fmt

Author: oraluben

docs/conf.py

@@ -1,4 +1,3 @@
-
 # General information about the project.
 project = "Tile Language <br>"
 author = "Tile Lang Contributors"

examples/bitnet-1.58b/modeling_bitnet.py

@@ -17,6 +17,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """PyTorch LLaMA model."""
+
 from __future__ import annotations
 
 import math

examples/bitnet-1.58b/tokenization_bitnet.py

@@ -17,6 +17,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Tokenization classes for LLaMA."""
+
 from __future__ import annotations
 
 import os

tilelang/autotuner/param.py

@@ -1,4 +1,5 @@
 """The auto-tune parameters."""
+
 from __future__ import annotations
 
 import tilelang

tilelang/autotuner/tuner.py

@@ -3,6 +3,7 @@
 This module provides functionality for auto-tuning tilelang programs, including JIT compilation
 and performance optimization through configuration search.
 """
+
 from __future__ import annotations
 
 import tilelang

tilelang/cache/__init__.py

@@ -1,4 +1,5 @@
 """The cache utils with class and database persistence - Init file"""
+
 from __future__ import annotations
 
 from typing import Literal

tilelang/cache/kernel_cache.py

@@ -1,4 +1,5 @@
 """The cache utils with class and database persistence - KernelCache Class"""
+
 from __future__ import annotations
 
 import json

tilelang/carver/analysis.py

@@ -1,4 +1,5 @@
 """Analysis on TIR blocks, loops and functions."""
+
 from __future__ import annotations
 
 from typing_extensions import Literal

tilelang/carver/common_schedules.py

@@ -19,6 +19,7 @@
 # Modifications Copyright (c) Microsoft.
 # The code below is mostly copied from apache/tvm common_schedules.py in dlight.
 """Common schedule strategies for TIR."""
+
 from __future__ import annotations
 
 from typing import Callable

tilelang/carver/matmul_analysis.py

@@ -1,5 +1,6 @@
 # pylint: disable=missing-docstring, invalid-name
 """A GEMM schedule rule for GPU operators."""
+
 from __future__ import annotations
 
 from dataclasses import dataclass
@@ -570,9 +571,7 @@ def check_sm_version(arch: str) -> int:
         sm_version = arch.replace("sm_", "")
         return int(sm_version) if sm_version.isdigit() else -1
 
-    def analysis_tensorcore_tags(
-        sch: tir.Schedule, block: BlockRV, target: Target
-    ) -> bool | dict:
+    def analysis_tensorcore_tags(sch: tir.Schedule, block: BlockRV, target: Target) -> bool | dict:
         tags: dict[str, list[int] | int] = {}
         block_stmt = sch.get(block)