From df39cf62a91f9bced50748fb432956b94f63600e Mon Sep 17 00:00:00 2001 From: Zhenglong Li Date: Mon, 22 Feb 2021 16:21:18 +0800 Subject: [PATCH] Fix include issue --- mindspore/ccsrc/minddata/dataset/api/execute.cc | 1 + mindspore/ccsrc/minddata/dataset/include/execute.h | 3 +-- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/minddata/dataset/api/execute.cc b/mindspore/ccsrc/minddata/dataset/api/execute.cc index 93ddb05fbb..6ef9c7d82a 100644 --- a/mindspore/ccsrc/minddata/dataset/api/execute.cc +++ b/mindspore/ccsrc/minddata/dataset/api/execute.cc @@ -17,6 +17,7 @@ #include "minddata/dataset/include/execute.h" #include "minddata/dataset/core/de_tensor.h" #include "minddata/dataset/core/device_tensor.h" +#include "minddata/dataset/core/device_resource.h" #include "minddata/dataset/core/tensor_row.h" #include "minddata/dataset/include/tensor.h" #include "minddata/dataset/include/type_id.h" diff --git a/mindspore/ccsrc/minddata/dataset/include/execute.h b/mindspore/ccsrc/minddata/dataset/include/execute.h index 87afbf9034..e24ef44ecc 100644 --- a/mindspore/ccsrc/minddata/dataset/include/execute.h +++ b/mindspore/ccsrc/minddata/dataset/include/execute.h @@ -22,13 +22,12 @@ #include #include "include/api/context.h" #include "include/api/types.h" -#include "minddata/dataset/core/device_resource.h" #include "minddata/dataset/include/constants.h" #include "minddata/dataset/include/transforms.h" namespace mindspore { namespace dataset { - +class DeviceResource; // class to run tensor operations in eager mode class Execute { public: