[Typo] Rename warp_source into wrap_source(#440)

* Fix typo

* bugfix

---------
Co-authored-by: LeiWang1999 <leiwang1999@outlook.com>

testing/python/transform/test_tilelang_transform_annotate_device_regions.py

 import tilelang
 import tilelang.testing
-from tilelang import language as T
+from tvm.script import tir as T
 
 
 class BaseCompare(tilelang.testing.CompareBeforeAfter):
@@ -10,12 +10,12 @@ class BaseCompare(tilelang.testing.CompareBeforeAfter):
 class TestAnnotateThreadExtent(BaseCompare):
     """Annotation inserted at the "thread_extent" attribute"""
 
-    def before(A: T.Tensor(16, "float32")):
+    def before(A: T.Buffer(16, "float32")):
         T.func_attr({"target": T.target("cuda", host="llvm")})
         i = T.launch_thread("threadIdx.x", 16)
         A[i] = 0.0
 
-    def expected(A: T.Tensor(16, "float32")):
+    def expected(A: T.Buffer(16, "float32")):
         T.func_attr({"target": T.target("cuda", host="llvm")})
         T.attr(T.target("cuda"), "target", 0)
         i = T.launch_thread("threadIdx.x", 16)
@@ -25,12 +25,12 @@ class TestAnnotateThreadExtent(BaseCompare):
 class TestAnnotateDeviceScope(BaseCompare):
     """Annotation inserted at the "device_scope" attribute"""
 
-    def before(A: T.Tensor(1, "float32")):
+    def before(A: T.Buffer(1, "float32")):
         T.func_attr({"target": T.target("cuda", host="llvm")})
         T.attr(0, "device_scope", 0)
         A[0] = 0.0
 
-    def expected(A: T.Tensor(1, "float32")):
+    def expected(A: T.Buffer(1, "float32")):
         T.func_attr({"target": T.target("cuda", host="llvm")})
         T.attr(T.target("cuda"), "target", 0)
         T.attr(0, "device_scope", 0)

tilelang/cache/kernel_cache.py

@@ -16,7 +16,7 @@ import logging
 from tilelang.env import TILELANG_CACHE_DIR, is_cache_enabled
 
 KERNEL_PATH = "kernel.cu"
-WRAPPED_KERNEL_PATH = "warpped_kernel.cu"
+WRAPPED_KERNEL_PATH = "wrapped_kernel.cu"
 KERNEL_LIB_PATH = "kernel_lib.so"
 PARAMS_PATH = "params.pkl"
 
@@ -26,7 +26,7 @@ class KernelCache:
     Caches compiled kernels using a class and database persistence to avoid redundant compilation.
     Cache files:
         kernel.cu: The compiled kernel source code
-        warpped_kernel.cu: The compiled wrapped kernel source code
+        wrapped_kernel.cu: The compiled wrapped kernel source code
         kernel_lib.so: The compiled kernel library
         params.pkl: The compiled kernel parameters
     """