xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!69164 fix matmul transpose 

Merge pull request !69164 from 邹文祥/master-fix-gpu-transpose
This commit is contained in:
i-robot 2024-05-16 12:11:09 +00:00 committed by Gitee
parent db5d653e04 3cb9d6b374
commit 905901e482
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/backend/common/graph_kernel/convert_input_and_attr.cc
View File

@ -44,7 +44,8 @@ const std::set<std::string> &GetConvertInputAttrOps() {
prim::kPrimCumSum->name(), prim::kPrimArgmin->name(), prim::kPrimArgmax->name(),
prim::kPrimBiasAdd->name(), prim::kPrimBiasAddGrad->name(), prim::kPrimLayerNorm->name(),
prim::kPrimLayerNormGrad->name(), prim::kPrimLogSoftmax->name(), prim::kPrimLogSoftmaxGrad->name(),
prim::kPrimStridedSlice->name(), prim::kPrimAdamWeightDecay->name(),
prim::kPrimStridedSlice->name(), prim::kPrimAdamWeightDecay->name(), prim::kPrimMatMul->name(),
prim::kPrimBatchMatMul->name(),
};
return convert_input_attr_ops;
}