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
2f42a4888cacc979bc542b77b43b8c1a7ea4e76e
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: 49ms Template: 2ms
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