Fix the style of sgl kernel (#10398)
This commit is contained in:
@@ -157,6 +157,7 @@ set(SGL_KERNEL_CUDA_FLAGS
|
||||
"-DCUTLASS_DEBUG_TRACE_LEVEL=0"
|
||||
"--expt-relaxed-constexpr"
|
||||
"--expt-extended-lambda"
|
||||
|
||||
# The following flag leads to the CMAKE_BUILD_PARALLEL_LEVEL breaking,
|
||||
# it triggers OOM with low memory host. Extract the threads number to
|
||||
# option named SGL_KERNEL_COMPILE_THREADS, default value 32.
|
||||
@@ -169,7 +170,8 @@ set(SGL_KERNEL_CUDA_FLAGS
|
||||
"-Xcompiler=-Wno-terminate"
|
||||
"-Xcompiler=-Wfatal-errors"
|
||||
"-Xcompiler=-ftemplate-backtrace-limit=1"
|
||||
"-Xcudafe=--diag_suppress=177" # variable was declared but never referenced
|
||||
"-Xcudafe=--diag_suppress=177" # variable was declared but never referenced
|
||||
"-Xcudafe=--diag_suppress=2361" # invalid narrowing conversion from "char" to "signed char"
|
||||
|
||||
# uncomment to debug
|
||||
# "--ptxas-options=-v"
|
||||
@@ -299,11 +301,12 @@ set(SOURCES
|
||||
|
||||
"csrc/grammar/apply_token_bitmask_inplace_cuda.cu"
|
||||
|
||||
"csrc/mamba/causal_conv1d.cu"
|
||||
|
||||
"csrc/moe/cutlass_moe/w4a8/scaled_mm_entry.cu"
|
||||
"csrc/moe/cutlass_moe/w4a8/w4a8_moe_data.cu"
|
||||
"csrc/moe/cutlass_moe/w4a8/w4a8_grouped_mm_c3x.cu"
|
||||
"csrc/moe/marlin_moe_wna16/ops.cu"
|
||||
"csrc/mamba/causal_conv1d.cu"
|
||||
"csrc/moe/moe_align_kernel.cu"
|
||||
"csrc/moe/moe_fused_gate.cu"
|
||||
"csrc/moe/moe_topk_softmax_kernels.cu"
|
||||
|
||||
Reference in New Issue
Block a user