Merge branch 'main' into main

Author: xwhzz

CODE_OF_CONDUCT.md

@@ -1,6 +1,6 @@
 # Contributing
 
-That would be awesome if you want to contribute something to BitBLAS!
+That would be awesome if you want to contribute something to TileLang!
 
 - [Contributing](CONTRIBUTING.md#contributing)
   - [Reporting Bugs](CONTRIBUTING.md#reporting-bugs)
@@ -11,7 +11,7 @@ That would be awesome if you want to contribute something to BitBLAS!
 
 ## Reporting Bugs
 
-If you run into any weird behavior while using BitBLAS, feel free to open a new issue in this repository! Please run a **search before opening** a new issue, to make sure that someone else hasn't already reported or solved the bug you've found.
+If you run into any weird behavior while using TileLang, feel free to open a new issue in this repository! Please run a **search before opening** a new issue, to make sure that someone else hasn't already reported or solved the bug you've found.
 
 Any issue you open must include:
 
@@ -25,15 +25,15 @@ Please ask questions in issues.
 
 ## Submitting Pull Requests
 
-All pull requests are super welcomed and greatly appreciated! Issues in need of a solution are marked with a [`♥ help`](https://github.com/ianstormtaylor/BitBLAS/issues?q=is%3Aissue+is%3Aopen+label%3A%22%E2%99%A5+help%22) label if you're looking for somewhere to start.
+All pull requests are super welcomed and greatly appreciated! Issues in need of a solution are marked with a [`♥ help`](https://github.com/ianstormtaylor/TileLang/issues?q=is%3Aissue+is%3Aopen+label%3A%22%E2%99%A5+help%22) label if you're looking for somewhere to start.
 
 Please run `./format.sh` before submitting a pull request to make sure that your code is formatted correctly.
 
 Please include tests and docs with every pull request!
 
 ## Repository Setup
 
-To run the build, you need to have the BitBLAS repository cloned to your computer. After that, you need to `cd` into the directory where you cloned it, and install the dependencies with `python`:
+To run the build, you need to have the TileLang repository cloned to your computer. After that, you need to `cd` into the directory where you cloned it, and install the dependencies with `python`:
 
 ```bash
 python setup.py install

CONTRIBUTING.md

@@ -71,9 +71,9 @@ pip install .  # with -e option if you want to install in editable mode
 
 ### Method 2: Build from Source
 We currently provide three ways to install **tile-lang** from source:
- - [Install from Source (using your own TVM installation)](./docs/Installation.md#install-from-source-with-your-own-tvm-installation)
- - [Install from Source (using the bundled TVM submodule)](./docs/Installation.md#install-from-source-with-our-tvm-submodule)
- - [Install Using the Provided Script](./docs/Installation.md#install-with-provided-script)
+ - [Install from Source (using your own TVM installation)](./docs/get_started/Installation.rst#method-1-install-from-source-using-your-own-tvm-installation)
+ - [Install from Source (using the bundled TVM submodule)](./docs/get_started/Installation.rst#method-2-install-from-source-with-our-tvm-submodule)
+ - [Install Using the Provided Script](./docs/get_started/Installation.rst##method-3-install-using-the-provided-script)
 
 
 ## Quick Start
@@ -192,6 +192,12 @@ In addition to GEMM, we provide a variety of examples to showcase the versatilit
 
 TileLang has now been used in project [BitBLAS](https://github.com/microsoft/BitBLAS).
 
+## Join the Discussion
+
+Welcome to join our Discord community for discussions, support, and collaboration!
+
+[![Join our Discord](https://img.shields.io/badge/Discord-Join%20Us-blue?logo=discord&style=for-the-badge)](https://discord.gg/TUrHyJnKPG)
+
 ## Acknowledgements
 
 We learned a lot from the [TVM](https://github.com/apache/tvm) community and would like to thank them for their contributions. The initial version of this project is mainly contributed by [LeiWang1999](https://github.com/LeiWang1999), [chengyupku](https://github.com/chengyupku) and [nox-410](https://github.com/nox-410). Part of this work was done during the internship at Microsoft Research, under the supervision of Dr. Lingxiao Ma, Dr. Yuqing Xia, Dr. Jilong Xue, and Dr. Fan Yang.

README.md

@@ -1,3 +1,4 @@
+
 Installation Guide
 ==================
 
@@ -65,7 +66,7 @@ After installing the prerequisites, you can clone the TileLang repository and in
 .. code:: bash
 
    git clone --recursive https://github.com/tile-ai/tilelang.git
-   cd TileLang
+   cd tileLang
    pip install .  # Please be patient, this may take some time.
 
 If you want to install TileLang in development mode, you can run the following command:
@@ -76,11 +77,13 @@ If you want to install TileLang in development mode, you can run the following c
 
 We currently provide three methods to install **TileLang**:
 
-1. Install from Source (using your own TVM installation)
-
-2. Install from Source (using the bundled TVM submodule)
+1. `Install from Source (using your own TVM installation)`_
+2. `Install from Source (using the bundled TVM submodule)`_
+3. `Install Using the Provided Script`_
 
-3. Install Using the Provided Script
+.. _Install from Source (using your own TVM installation): #method-1-install-from-source-using-your-own-tvm-installation
+.. _Install from Source (using the bundled TVM submodule): #method-2-install-from-source-using-the-bundled-tvm-submodule
+.. _Install Using the Provided Script: #method-3-install-using-the-provided-script
 
 
 Method 1: Install from Source (Using Your Own TVM Installation)
@@ -93,7 +96,7 @@ If you already have a compatible TVM installation, follow these steps:
    .. code:: bash
 
       git clone --recursive https://github.com/tile-ai/tilelang
-      cd TileLang
+      cd tilelang
 
    **Note**: Use the `--recursive` flag to include necessary submodules.
 
@@ -114,7 +117,7 @@ If you already have a compatible TVM installation, follow these steps:
 
    .. code:: bash
 
-      export PYTHONPATH=/your/path/to/tile-lang/python:$PYTHONPATH
+      export PYTHONPATH=/your/path/to/tilelang/:$PYTHONPATH
       # TVM_IMPORT_PYTHON_PATH is used by 3rd-party frameworks to import TVM
       export TVM_IMPORT_PYTHON_PATH=/your/path/to/tvm/python
 
@@ -128,7 +131,7 @@ If you prefer to use the built-in TVM version, follow these instructions:
    .. code:: bash
 
       git clone --recursive https://github.com/tile-ai/tilelang
-      cd TileLang
+      cd tilelang
 
    **Note**: Ensure the `--recursive` flag is included to fetch submodules.
 
@@ -155,7 +158,7 @@ If you prefer to use the built-in TVM version, follow these instructions:
 
    .. code:: bash
 
-      export PYTHONPATH=/your/path/to/TileLang/python:$PYTHONPATH
+      export PYTHONPATH=/your/path/to/tilelang/:$PYTHONPATH
 
 Method 3: Install Using the Provided Script
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -167,11 +170,11 @@ For a simplified installation, use the provided script:
    .. code:: bash
 
       git clone --recursive https://github.com/tile-ai/tilelang
-      cd TileLang
+      cd tilelang
 
 2. **Run the Installation Script**:
 
    .. code:: bash
 
-      bash install.sh
+      bash install_cuda.sh
       # or bash `install_amd.sh` if you want to enable ROCm runtime

SECURITY.md

@@ -0,0 +1,58 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+import tilelang
+import tilelang.testing
+from tilelang import language as T
+
+
+class BaseCompare(tilelang.testing.CompareBeforeAfter):
+    transform = tilelang.transform.AnnotateDeviceRegions()
+
+
+class TestAnnotateThreadExtent(BaseCompare):
+    """Annotation inserted at the "thread_extent" attribute"""
+
+    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.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)
+        A[i] = 0.0
+
+
+class TestAnnotateDeviceScope(BaseCompare):
+    """Annotation inserted at the "device_scope" attribute"""
+
+    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.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)
+        A[0] = 0.0
+
+
+if __name__ == "__main__":
+    tilelang.testing.main()