已关闭
[Requirement|需求建议]: mm、bmm、addmm和baddbmm算子支持16In32out场景 #2806
HKFLYE创建于  5月21日关闭于  5月22日
HKFLYE
HKFLYE成员
5月21日 创建

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(设计方案)

likedislike
HKFLYEHKFLYE成员
5月21日 添加了label:requirement
HKFLYEHKFLYE成员
5月21日 关联了pull request:mm和bmm支持16in32out输入
HKFLYEHKFLYE成员
5月21日 删除了关联的pull request:mm和bmm支持16in32out输入
HKFLYEHKFLYE成员
5月21日 关联了pull request:mm和bmm支持16in32out输入
HKFLYEHKFLYE成员
5月21日 将 HuangKun8682 设为负责人
CANN-robotCANN-robot成员
5月22日 关闭了 issue
CANN-robotCANN-robot成员
5月22日 添加了label:resolved
HKFLYEHKFLYE成员
5月30日 修改标题为 “[Requirement|需求建议]: mm、bmm、addmm和baddbmm算子支持16In32out场景”,原标题为“[Requirement|需求建议]: mm和bmm算子支持16In32out场景”
HKFLYEHKFLYE成员
5月30日 关联了pull request:aclnn的mm、bmm、addmm和baddbmm接口支持16in32Out场景