diff options
| author | Kawrakow <iwankawrakow@gmail.com> | 2025-07-03 15:36:52 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-07-03 15:36:52 +0200 |
| commit | 9534461c01e132672821e53ce6e5e560dc58e829 (patch) | |
| tree | 18d4487dece1ae25d0a44e02cc44260f94a91697 /ggml/src/vulkan-shaders/vulkan-shaders-gen.cpp | |
| parent | db8dee50516d1437f94017de196f87dcf54be12d (diff) | |
Vulkan: fused rms norm (#577)
Co-authored-by: Iwan Kawrakow <iwan.kawrakow@gmail.com>
Diffstat (limited to 'ggml/src/vulkan-shaders/vulkan-shaders-gen.cpp')
| -rw-r--r-- | ggml/src/vulkan-shaders/vulkan-shaders-gen.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ggml/src/vulkan-shaders/vulkan-shaders-gen.cpp b/ggml/src/vulkan-shaders/vulkan-shaders-gen.cpp index 0f244dea..d622f1bd 100644 --- a/ggml/src/vulkan-shaders/vulkan-shaders-gen.cpp +++ b/ggml/src/vulkan-shaders/vulkan-shaders-gen.cpp @@ -498,6 +498,7 @@ void process_shaders() { string_to_spv("norm_f32", "norm.comp", merge_maps(base_dict, {{"A_TYPE", "float"}, {"D_TYPE", "float"}})); string_to_spv("group_norm_f32", "group_norm.comp", merge_maps(base_dict, {{"A_TYPE", "float"}, {"D_TYPE", "float"}})); string_to_spv("rms_norm_f32", "rms_norm.comp", merge_maps(base_dict, {{"A_TYPE", "float"}, {"D_TYPE", "float"}})); + string_to_spv("fused_rms_norm_f32", "fused_rms_norm.comp", merge_maps(base_dict, {{"A_TYPE", "float"}, {"B_TYPE", "float"}, {"D_TYPE", "float"}})); string_to_spv("rms_norm_back_f32", "rms_norm_back.comp", merge_maps(base_dict, {{"A_TYPE", "float"}, {"B_TYPE", "float"}, {"D_TYPE", "float"}})); string_to_spv("cpy_f32_f32", "copy.comp", {{"A_TYPE", "float"}, {"D_TYPE", "float"}}); |