尧图建网站 尧图建网站 YAOTU WEB BUILD 免费咨询
ARTICLE DETAIL

资讯详情

深耕网站建设与建站编程的一线实战洞察。

CANN/catlass GEMM/Kernel类模板概述

CANN/catlass GEMM/Kernel类模板概述 Gemm/Kernel Class Template Overview【免费下载链接】catlass本项目是CANN的算子模板库提供NPU上高性能矩阵乘及其相关融合类算子模板样例。项目地址: https://gitcode.com/cann/catlassAPI ListComponentDescriptionbasic_matmulBasic matrix multiplication of the Common templatebasic_matmul_tla_visitorEVG GM workspace kernel entry pointbasic_matmul_tla_ub_visitorEVG UB workspace kernel entry pointAPI BreakdownThe namespace isCatlass::Gemm::Kernel. The class template contains the following core members.TypeNameFunctionstructParamsInput arguments used when invoking the device kernel function through.structArgumentsInput arguments encapsulated prior to device executionStatic functionbool CanImplementArgumentsvalidation interfaceStatic functionsize_t GetWorkspaceSizeCalculates the required workspace size based onArgumentsStatic functionParams ToUnderlyingArgumentsConvertsArgumentsinto kernel arguments (Params)Functionvoid operator()AscendC::AICInputsParamsand executes MMAD computations on the AICFunctionvoid operator()AscendC::AIVInputsParamsand executes calculation on the AIV, such as prologue and epilogue【免费下载链接】catlass本项目是CANN的算子模板库提供NPU上高性能矩阵乘及其相关融合类算子模板样例。项目地址: https://gitcode.com/cann/catlass创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表