Logo
Explore Help
Register Sign In
20231088/vllm
1
0
Fork 0
You've already forked vllm
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
vllm/csrc/core/math.hpp

10 lines
208 B
C++
Raw Permalink Normal View History

[Kernel] Update `cutlass_scaled_mm` to support 2d group (blockwise) scaling (#11868)
2025-01-30 21:33:00 -05:00
#pragma once
[Kernel]: Cutlass 2:4 Sparsity + FP8/Int8 Quant Support (#10995) Co-authored-by: Faraz Shahsavan <faraz.shahsavan@gmail.com> Co-authored-by: ilmarkov <markovilya197@gmail.com> Co-authored-by: Rahul Tuli <rahul@neuralmagic.com> Co-authored-by: rshaw@neuralmagic.com <rshaw@neuralmagic.com>
2024-12-18 09:57:16 -05:00
#include <climits>
#include <iostream>
[Kernel] Update `cutlass_scaled_mm` to support 2d group (blockwise) scaling (#11868)
2025-01-30 21:33:00 -05:00
inline constexpr uint32_t next_pow_2(uint32_t const num) {
[Kernel]: Cutlass 2:4 Sparsity + FP8/Int8 Quant Support (#10995) Co-authored-by: Faraz Shahsavan <faraz.shahsavan@gmail.com> Co-authored-by: ilmarkov <markovilya197@gmail.com> Co-authored-by: Rahul Tuli <rahul@neuralmagic.com> Co-authored-by: rshaw@neuralmagic.com <rshaw@neuralmagic.com>
2024-12-18 09:57:16 -05:00
if (num <= 1) return num;
return 1 << (CHAR_BIT * sizeof(num) - __builtin_clz(num - 1));
[Kernel] Update `cutlass_scaled_mm` to support 2d group (blockwise) scaling (#11868)
2025-01-30 21:33:00 -05:00
}
Reference in New Issue Copy Permalink
Powered by Gitea Version: 23.0.0 Page: 164ms Template: 4ms
English
Bahasa Indonesia Deutsch English Español Français Gaeilge Italiano Latviešu Magyar nyelv Nederlands Polski Português de Portugal Português do Brasil Suomi Svenska Türkçe Čeština Ελληνικά Български Русский Українська فارسی മലയാളം 日本語 简体中文 繁體中文(台灣) 繁體中文(香港) 한국어
Licenses API