From ac80f4da5784ac0b745628fe8bf6280708d78e66 Mon Sep 17 00:00:00 2001 From: Chunyuan WU Date: Wed, 9 Jul 2025 16:53:53 +0800 Subject: [PATCH] [CPU] [FP8] set SGLANG_CPU_FP8_CVT_FTZ in CMakeLists.txt (#7885) --- sgl-kernel/csrc/cpu/CMakeLists.txt | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/sgl-kernel/csrc/cpu/CMakeLists.txt b/sgl-kernel/csrc/cpu/CMakeLists.txt index aa77fbad4..421fbedab 100755 --- a/sgl-kernel/csrc/cpu/CMakeLists.txt +++ b/sgl-kernel/csrc/cpu/CMakeLists.txt @@ -58,6 +58,15 @@ endif() file(GLOB SOURCES "${CMAKE_CURRENT_SOURCE_DIR}/*.cpp") +if(NOT DEFINED ENV{SGLANG_CPU_FP8_CVT_FTZ}) + set(ENV{SGLANG_CPU_FP8_CVT_FTZ} "1") +endif() + +if("$ENV{SGLANG_CPU_FP8_CVT_FTZ}" STREQUAL "1") + message(STATUS "Enabling macro: SGLANG_CPU_FP8_CVT_FTZ") + add_compile_definitions(SGLANG_CPU_FP8_CVT_FTZ) +endif() + add_compile_options( -O3 -Wno-unknown-pragmas