feat: vulkan matmul#6579

Open

Cat-myq wants to merge 1 commit intoTencent:masterfrom

Cat-myq:feat/matmul-vulkan

Contributor

Cat-myq commented Mar 6, 2026

功能说明

实现 matmul 算子的 vulkan 版本：

vulkan 版本：基于 ncnn 的 vulkan 框架，仿照 gemm 算子的 vulkan 实现方式，实现 matmul 矩阵乘法的 vulkan 内核;

修改文件列表

src/layer/vulkan/matmul_vulkan.cpp # 核心：matmul vulkan 实现
src/layer/vulkan/matmul_vulkan.h # 头文件：声明 vulkan 相关函数/类
src/layer/vulkan/shader/matmul.comp
src/layer/vulkan/shader/matmul_sg.comp
src/layer/vulkan/shader/matmul_cm.comp

测试情况

环境：Windows 11-NVIDIA RTX 3060；Android mali-g78；Android Areno-730
测试用例：
- Windows测试不同 shape下（1D、3D、4D、batch、transB）CPU vs Vulkan结果对比
- Android 无崩溃现象


          feat: vulkan matmul

88e0927

github-actions bot added the vulkan label

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels