add sampling_scaling_penalties kernel (#2846)
This commit is contained in:
@@ -32,6 +32,7 @@ add_library(_kernels SHARED
|
||||
src/sgl-kernel/csrc/trt_reduce_kernel.cu
|
||||
src/sgl-kernel/csrc/moe_align_kernel.cu
|
||||
src/sgl-kernel/csrc/int8_gemm_kernel.cu
|
||||
src/sgl-kernel/csrc/sampling_scaling_penalties.cu
|
||||
src/sgl-kernel/csrc/sgl_kernel_ops.cu
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user