diff --git a/src/math/snrm2/op_host/snrm2.cpp b/src/math/snrm2/op_host/snrm2.cpp index 137848b445de66a092ac0e515c3efd806fe90256..1192a9903c13dccf371a3092be460c2819734aa9 100644 --- a/src/math/snrm2/op_host/snrm2.cpp +++ b/src/math/snrm2/op_host/snrm2.cpp @@ -135,7 +135,8 @@ public: this->SetInferShape(ge::InferShape).SetInferDataType(ge::InferDataType); this->AICore() .SetTiling(optiling::TilingFunc) - .AddConfig("ascend910b"); + .AddConfig("ascend910b") + .AddConfig("ascend310b"); } }; OP_ADD(Snrm2);