File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
paddle/fluid/operators/detail Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 13
13
// limitations under the License.
14
14
15
15
#include " paddle/fluid/operators/detail/variable_response.h"
16
- #include < string.h>
16
+
17
+ #include < string>
18
+ #include < utility>
19
+ #include < vector>
20
+
17
21
#include " paddle/fluid/operators/detail/send_recv.pb.h"
18
22
#include " paddle/fluid/operators/detail/sendrecvop_utils.h"
19
23
@@ -151,7 +155,7 @@ bool VariableResponse::CopySelectRowsTensorData(
151
155
auto * tensor = slr->mutable_value ();
152
156
tensor->Resize (dims);
153
157
PADDLE_ENFORCE_EQ (
154
- tensor->numel (),
158
+ static_cast < size_t >( tensor->numel () ),
155
159
length / framework::SizeOfType (
156
160
paddle::operators::detail::ToTypeIndex (meta_.data_type ())));
157
161
void * tensor_data = tensor->mutable_data (
You can’t perform that action at this time.
0 commit comments