Commit cc5e9f7c authored by Lei Wang's avatar Lei Wang Committed by LeiWang1999
Browse files

[Release] Bump Version to 0.1.5 (#551)

* Update VERSION to 0.1.5

* Add DEBUG_MODE support in setup.py and update CMake build type; enhance pypi.Dockerfile with git installation
parent c9e503be
...@@ -2,7 +2,7 @@ FROM nvidia/cuda:12.1.0-devel-ubuntu18.04 ...@@ -2,7 +2,7 @@ FROM nvidia/cuda:12.1.0-devel-ubuntu18.04
RUN set -eux; \ RUN set -eux; \
apt-get update; \ apt-get update; \
apt-get install -y wget curl libtinfo-dev zlib1g-dev libssl-dev build-essential libedit-dev libxml2-dev; \ apt-get install -y wget curl libtinfo-dev zlib1g-dev libssl-dev build-essential libedit-dev libxml2-dev git; \
curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh; \ curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh; \
bash Miniconda3-latest-Linux-x86_64.sh -b -p /miniconda3; \ bash Miniconda3-latest-Linux-x86_64.sh -b -p /miniconda3; \
rm Miniconda3-latest-Linux-x86_64.sh rm Miniconda3-latest-Linux-x86_64.sh
......
...@@ -37,6 +37,8 @@ ROOT_DIR = os.path.dirname(__file__) ...@@ -37,6 +37,8 @@ ROOT_DIR = os.path.dirname(__file__)
USE_LLVM = os.environ.get("USE_LLVM", "False").lower() == "true" USE_LLVM = os.environ.get("USE_LLVM", "False").lower() == "true"
# Add ROCM control environment variable # Add ROCM control environment variable
USE_ROCM = os.environ.get("USE_ROCM", "False").lower() == "true" USE_ROCM = os.environ.get("USE_ROCM", "False").lower() == "true"
# Build with Debug mode
DEBUG_MODE = os.environ.get("DEBUG_MODE", "False").lower() == "true"
def load_module_from_path(module_name, path): def load_module_from_path(module_name, path):
...@@ -158,7 +160,7 @@ def get_tilelang_version(with_cuda=True, with_system_info=True, with_commit_id=F ...@@ -158,7 +160,7 @@ def get_tilelang_version(with_cuda=True, with_system_info=True, with_commit_id=F
stderr=subprocess.DEVNULL, stderr=subprocess.DEVNULL,
encoding='utf-8').strip() encoding='utf-8').strip()
except subprocess.SubprocessError as error: except subprocess.SubprocessError as error:
raise RuntimeError("Failed to get git commit id") from error logger.warning(f"Ignore commit id because failed to get git commit id: {str(error)}")
if commit_id: if commit_id:
version += f"+{commit_id}" version += f"+{commit_id}"
...@@ -635,8 +637,8 @@ class CMakeBuild(build_ext): ...@@ -635,8 +637,8 @@ class CMakeBuild(build_ext):
# -DCMAKE_LIBRARY_OUTPUT_DIRECTORY sets where built libraries go # -DCMAKE_LIBRARY_OUTPUT_DIRECTORY sets where built libraries go
# -DPYTHON_EXECUTABLE ensures that the correct Python is used # -DPYTHON_EXECUTABLE ensures that the correct Python is used
cmake_args = [ cmake_args = [
f"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={extdir}", f"-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={extdir}", f"-DPYTHON_EXECUTABLE={sys.executable}",
f"-DPYTHON_EXECUTABLE={sys.executable}", f"-DCMAKE_BUILD_TYPE={'Debug' if DEBUG_MODE else 'Release'}"
] ]
# Create the temporary build directory (if it doesn't exist). # Create the temporary build directory (if it doesn't exist).
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment