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),仅供参考