提交 31844bff 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!5902 GPU add support device

Merge pull request !5902 from VectorSL/format
......@@ -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;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册