Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!48968 adapter 910b version 

Merge pull request !48968 from jjfeing/ascend_version
This commit is contained in:
i-robot 2023-02-16 08:38:13 +00:00 committed by Gitee
parent 7733480a62 b13ee4619c
commit e24df6f45a
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mindspore/ccsrc/kernel/oplib/op_info_utils.cc
View File

@ -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<std::string> kAscend910Versions = {"Ascend910A", "Ascend910B", "Ascend910PremiumA", "Ascend910ProA",
"Ascend910ProB"};
const std::set<std::string> kAscend920Versions = {"Ascend920A"};
const std::set<std::string> 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 "";
}