已关闭
[Requirement|需求建议]: mm、bmm、addmm和baddbmm算子支持16In32out场景 #2806
HKFLYE创建于 5月21日关闭于 5月22日
5月21日 添加了label:requirement
5月21日 关联了pull request:mm和bmm支持16in32out输入
5月21日 删除了关联的pull request:mm和bmm支持16in32out输入
5月21日 关联了pull request:mm和bmm支持16in32out输入
5月21日 将 HuangKun8682 设为负责人
5月22日 关闭了 issue
5月22日 添加了label:resolved
5月30日 修改标题为 “[Requirement|需求建议]: mm、bmm、addmm和baddbmm算子支持16In32out场景”,原标题为“[Requirement|需求建议]: mm和bmm算子支持16In32out场景”
5月30日 修改标题为 “[Requirement|需求建议]: mm、bmm、addmm和baddbmm算子支持16In32out场景”,原标题为“[Requirement|需求建议]: mm和bmm算子支持16In32out场景”
5月30日 关联了pull request:aclnn的mm、bmm、addmm和baddbmm接口支持16in32Out场景
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
Backgroud(背景信息)
torch功能一致性
参考:
https://docs.pytorch.org/docs/2.12/generated/torch.mm.html
https://docs.pytorch.org/docs/2.12/generated/torch.bmm.html
支持用户out_dtype参数输入为torch.float32时,kernel内部高精度计算
Origin(信息来源)
matmul算子功能一致性要求
Benefit / Necessity (价值/作用)
Design(设计方案)