Unverified Commit 92941ce7 authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

bump sgl-kernel 0.0.5.post4 (#4768)

parent 2bb0e7cf
...@@ -86,6 +86,7 @@ option(SGL_KERNEL_ENABLE_SM100A "Enable SM100A" OFF) ...@@ -86,6 +86,7 @@ option(SGL_KERNEL_ENABLE_SM100A "Enable SM100A" OFF)
option(SGL_KERNEL_ENABLE_SM90A "Enable SM90A" OFF) option(SGL_KERNEL_ENABLE_SM90A "Enable SM90A" OFF)
option(SGL_KERNEL_ENABLE_BF16 "Enable BF16" ON) option(SGL_KERNEL_ENABLE_BF16 "Enable BF16" ON)
option(SGL_KERNEL_ENABLE_FP8 "Enable FP8" ON) option(SGL_KERNEL_ENABLE_FP8 "Enable FP8" ON)
option(SGL_KERNEL_ENABLE_FP4 "Enable FP4" OFF)
if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8" OR SGL_KERNEL_ENABLE_SM100A) if ("${CUDA_VERSION}" VERSION_GREATER_EQUAL "12.8" OR SGL_KERNEL_ENABLE_SM100A)
list(APPEND SGL_KERNEL_CUDA_FLAGS list(APPEND SGL_KERNEL_CUDA_FLAGS
...@@ -118,6 +119,12 @@ if (SGL_KERNEL_ENABLE_FP8) ...@@ -118,6 +119,12 @@ if (SGL_KERNEL_ENABLE_FP8)
) )
endif() endif()
if (SGL_KERNEL_ENABLE_FP4)
list(APPEND SGL_KERNEL_CUDA_FLAGS
"-DENABLE_NVFP4=1"
)
endif()
string(REPLACE "-D__CUDA_NO_HALF_OPERATORS__" "" CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS}") string(REPLACE "-D__CUDA_NO_HALF_OPERATORS__" "" CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS}")
string(REPLACE "-D__CUDA_NO_HALF_CONVERSIONS__" "" CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS}") string(REPLACE "-D__CUDA_NO_HALF_CONVERSIONS__" "" CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS}")
string(REPLACE "-D__CUDA_NO_BFLOAT16_CONVERSIONS__" "" CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS}") string(REPLACE "-D__CUDA_NO_BFLOAT16_CONVERSIONS__" "" CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS}")
......
...@@ -45,7 +45,8 @@ format: check-deps ## Format all source files ...@@ -45,7 +45,8 @@ format: check-deps ## Format all source files
@pre-commit run --all-files @pre-commit run --all-files
FILES_TO_UPDATE = python/sgl_kernel/version.py \ FILES_TO_UPDATE = python/sgl_kernel/version.py \
pyproject.toml pyproject.toml \
pyproject_rocm.toml
update: ## Update version numbers across project files. Usage: make update <new_version> update: ## Update version numbers across project files. Usage: make update <new_version>
@if [ -z "$(filter-out $@,$(MAKECMDGOALS))" ]; then \ @if [ -z "$(filter-out $@,$(MAKECMDGOALS))" ]; then \
......
...@@ -8,7 +8,7 @@ build-backend = "scikit_build_core.build" ...@@ -8,7 +8,7 @@ build-backend = "scikit_build_core.build"
[project] [project]
name = "sgl-kernel" name = "sgl-kernel"
version = "0.0.5.post3" version = "0.0.5.post4"
description = "Kernel Library for SGLang" description = "Kernel Library for SGLang"
readme = "README.md" readme = "README.md"
requires-python = ">=3.9" requires-python = ">=3.9"
......
...@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta" ...@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
[project] [project]
name = "sgl-kernel" name = "sgl-kernel"
version = "0.0.5.post3" version = "0.0.5.post4"
description = "Kernel Library for SGLang" description = "Kernel Library for SGLang"
readme = "README.md" readme = "README.md"
requires-python = ">=3.9" requires-python = ">=3.9"
......
__version__ = "0.0.5.post3" __version__ = "0.0.5.post4"
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