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
2fa2a50bf950797cb59d48908d205c655ec02654
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: 77ms 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