mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!5902 GPU add support device 

Merge pull request !5902 from VectorSL/format
This commit is contained in:
mindspore-ci-bot 2020-09-09 10:12:15 +08:00 committed by Gitee
parent 963c5c28cf 197b00331e
commit 31844bff2e
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/ccsrc/runtime/device/gpu/kernel_info_setter.cc
View File

@ -27,6 +27,7 @@
#include "utils/ms_utils.h"
#include "backend/kernel_compiler/oplib/oplib.h"
#include "backend/kernel_compiler/oplib/opinfo.h"
#include "runtime/device/gpu/cuda_common.h"
namespace mindspore {
namespace device {
@ -236,6 +237,12 @@ void UpdateKernelFormatInfo(const CNodePtr &kernel_node, const std::vector<TypeI
} // namespace
void FormatTransformChecker::CheckSupportFormatTransform(const std::shared_ptr<session::KernelGraph> &kernel_graph) {
// TensorCore can be used only in Volta or newer devices.
const int marjor_sm = GET_MAJOR_SM;
if (marjor_sm < RECOMMEND_SM) {
format_transform_ = false;
return;
}
auto kernels = kernel_graph->execution_order();
size_t conv_cnt = 0;
size_t bn_cnt = 0;