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

tensorcore can only used in Volta or newer devices

This commit is contained in:
VectorSL 2020-09-08 17:00:45 +08:00
parent 2a9c458870
commit 197b00331e
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;