vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Allow vk add row

Browse source
This commit is contained in:
niansa 2023-06-28 10:19:18 +02:00
parent ed14f0764a
commit e2b721db65
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ggml-vulkan.cpp
View file

@ -366,9 +366,9 @@ void ggml_vk_abmath(kp::Sequence& seq,
seq.record<kp::OpAlgoDispatch>(mgr.algorithm<float, PushConstants>({inA, inB, out}, spirv, {size}, {}, {pushConsts})); seq.record<kp::OpAlgoDispatch>(mgr.algorithm<float, PushConstants>({inA, inB, out}, spirv, {size}, {}, {pushConsts}));
} }
template <typename... Args> template <bool with_row = false, typename... Args>
void ggml_vk_add(Args&&... args) { void ggml_vk_add(Args&&... args) {
return ggml_vk_abmath<'+'>(std::forward<Args>(args)...); return ggml_vk_abmath<'+', with_row>(std::forward<Args>(args)...);
} }
template <bool with_row = false, typename... Args> template <bool with_row = false, typename... Args>