From b13ee4619ca9e8bd3c0795603e0cfbed734415ec Mon Sep 17 00:00:00 2001 From: jjfeing Date: Thu, 16 Feb 2023 11:03:22 +0800 Subject: [PATCH] adapter ascend 910b --- mindspore/ccsrc/kernel/oplib/op_info_utils.cc | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/mindspore/ccsrc/kernel/oplib/op_info_utils.cc b/mindspore/ccsrc/kernel/oplib/op_info_utils.cc index 8cae1cee018..21d56146bf5 100644 --- a/mindspore/ccsrc/kernel/oplib/op_info_utils.cc +++ b/mindspore/ccsrc/kernel/oplib/op_info_utils.cc @@ -1,5 +1,5 @@ /** - * Copyright 2022 Huawei Technologies Co., Ltd + * Copyright 2022-2023 Huawei Technologies Co., Ltd * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -34,16 +34,16 @@ namespace { std::string GetAscendSuffixPath(const std::string &version) { const std::set kAscend910Versions = {"Ascend910A", "Ascend910B", "Ascend910PremiumA", "Ascend910ProA", "Ascend910ProB"}; - const std::set kAscend920Versions = {"Ascend920A"}; + const std::set kAscend910BVersions = {"Ascend910B1", "Ascend910B2", "Ascend910B3", "Ascend910B4"}; const std::string kAscend910SuffixPath = "/opp/built-in/op_impl/ai_core/tbe/config/ascend910/aic-ascend910-ops-info.json"; - const std::string kAscend920SuffixPath = - "/opp/built-in/op_impl/ai_core/tbe/config/ascend920/aic-ascend920-ops-info.json"; + const std::string kAscend910BSuffixPath = + "/opp/built-in/op_impl/ai_core/tbe/config/ascend910b/aic-ascend910b-ops-info.json"; if (kAscend910Versions.count(version) != 0) { return kAscend910SuffixPath; } - if (kAscend920Versions.count(version) != 0) { - return kAscend920SuffixPath; + if (kAscend910BVersions.count(version) != 0) { + return kAscend910BSuffixPath; } return ""; }