[Language] Make TL scripts friendly to Python syntax highlights (#1466)

* Language] Make TL scripts friendly to Python syntax highlights

* add comments

* fix submodule

examples/gemm_sp/example_custom_compress.py

-# Copyright (c) Tile-AI Corporation.
-# Licensed under the MIT License.
 import argparse
 
 import tilelang

examples/gemm_sp/example_gemm_sp.py

-# Copyright (c) Tile-AI Corporation.
-# Licensed under the MIT License.
 import argparse
 
 import tilelang

tilelang/language/__init__.py

@@ -58,12 +58,12 @@ from .allocate import (
     alloc_tcgen05_instr_desc,  # noqa: F401
     empty,  # noqa: F401
 )
-from .copy import copy, c2d_im2col  # noqa: F401
+from .copy_op import copy, c2d_im2col  # noqa: F401
 from tilelang.tileop.base import GemmWarpPolicy  # noqa: F401
-from .gemm import gemm, gemm_v1, gemm_v2  # noqa: F401
+from .gemm_op import gemm, gemm_v1, gemm_v2  # noqa: F401
 from .experimental.gemm_sp import gemm_sp, gemm_sp_v2  # noqa: F401
-from .fill import fill, clear  # noqa: F401
-from .reduce import (
+from .fill_op import fill, clear  # noqa: F401
+from .reduce_op import (
     reduce,  # noqa: F401
     reduce_max,  # noqa: F401
     reduce_min,  # noqa: F401
@@ -81,7 +81,7 @@ from .reduce import (
     warp_reduce_bitand,  # noqa: F401
     warp_reduce_bitor,  # noqa: F401
 )
-from .print import print, device_assert  # noqa: F401
+from .print_op import print, device_assert  # noqa: F401
 from .customize import (
     atomic_max,  # noqa: F401
     atomic_min,  # noqa: F401

tilelang/language/atomic.py

-# Copyright (c) Tile-AI Corporation.
-# Licensed under the MIT License.
-"""Atomic operations for tilelang."""
+"""Atomic operations exposed on the TileLang language surface."""
 
 from __future__ import annotations
 

tilelang/language/builtin.py

-"""The language interface for tl programs."""
+"""Builtin operations exposed on the TileLang language surface."""
 
 from __future__ import annotations
 

tilelang/language/copy.py → tilelang/language/copy_op.py

-"""The language interface for tl programs."""
+"""Copy operations exposed on the TileLang language surface."""
 
 from __future__ import annotations
 from typing import Literal

tilelang/language/customize.py

-"""The language interface for tl programs."""
+"""Some customized operations frequently used in tensor programming, exposed on the TileLang language surface."""
 
 from __future__ import annotations
 import tilelang.language as T

tilelang/language/fastmath.py

+"""Fast math operations exposed on the TileLang language surface."""
+
 from tvm import tir
 
 

tilelang/language/fill.py → tilelang/language/fill_op.py

-"""The language interface for tl programs."""
+"""Fill operations exposed on the TileLang language surface."""
 
 from __future__ import annotations
 from tvm import tir

tilelang/language/gemm.py → tilelang/language/gemm_op.py

-"""The language interface for tl programs."""
+"""GEMM (General Matrix Multiplication) operators exposed on the TileLang language surface."""
 
 from __future__ import annotations
 from tilelang.tileop.base import GemmWarpPolicy

tilelang/language/kernel.py

-"""The language interface for tl programs."""
+"""Kernel launching language interface in TileLang."""
 
 from __future__ import annotations
 from collections import deque

tilelang/language/logical.py

-"""The language interface for tl programs."""
+"""Logical operations exposed on the TileLang language surface."""
 
 from __future__ import annotations
 

tilelang/language/loop.py

-"""The language interface for tl programs."""
+"""Loop related language interfaces in TileLang."""
 
 from __future__ import annotations
 from typing import Any
@@ -175,5 +175,12 @@ def unroll(
         return UnrollForWithStep(start, stop, step, annotations=annotations)
 
 
-Serial = serial
-Unroll = unroll
+# "Serial" and "Unroll" are aliases of "T.serial" and "T.unroll". We use uppercase to emphasize that they are tile-level loops.
+
+
+def Serial(*args, **kwargs):
+    return serial(*args, **kwargs)
+
+
+def Unroll(*args, **kwargs):
+    return unroll(*args, **kwargs)

tilelang/language/math_intrinsics.py

+"""Common math intrinsics exposed on the TileLang language surface."""
+
 from tvm import tir
 
 

tilelang/language/proxy.py

-"""The language interface for tl programs."""
+"""Buffer/Tensor proxy in TileLang."""
 
 from __future__ import annotations
 

tilelang/language/reduce.py → tilelang/language/reduce_op.py

-"""The language interface for tl programs."""
+"""Reduce operations exposed on the TileLang language surface."""
 
 from __future__ import annotations
 from tvm import tir

tilelang/quantize/quantization.py

@@ -16,8 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 #
-# Copyright (c) Tile-AI Corporation.
-# Licensed under the MIT License.
+
 # The code below is mostly copied from mlc.ai quantization.py in mlc-llm.
 # pylint: disable=invalid-name,missing-function-docstring,unused-variable
 """TIR computation utilities for quantization."""