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

10 lines
208 B
C++
Raw 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: 1.25.2 Page: 56ms Template: 1ms
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