Browse Source

!7650 modify variable type from int to int32_t

Merge pull request !7650 from anzhengqi/fixed-variable-type
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
de4da6457e
2 changed files with 5 additions and 5 deletions
  1. +1
    -1
      mindspore/ccsrc/minddata/dataset/engine/datasetops/device_queue_op.cc
  2. +4
    -4
      mindspore/ccsrc/minddata/dataset/engine/datasetops/device_queue_op.h

+ 1
- 1
mindspore/ccsrc/minddata/dataset/engine/datasetops/device_queue_op.cc View File

@@ -33,7 +33,7 @@
namespace mindspore {
namespace dataset {
DeviceQueueOp::DeviceQueueOp(std::string channel_name, DeviceType device_type, int32_t device_id, int32_t prefetch_size,
bool send_epoch_end, int total_batch)
bool send_epoch_end, int32_t total_batch)
: PipelineOp(1),
channel_name_(channel_name),
device_type_(device_type),


+ 4
- 4
mindspore/ccsrc/minddata/dataset/engine/datasetops/device_queue_op.h View File

@@ -86,7 +86,7 @@ class DeviceQueueOp : public PipelineOp {
return *this;
}

Builder &SetTotalBatch(int total_batch) {
Builder &SetTotalBatch(int32_t total_batch) {
builder_total_batch_ = total_batch;
return *this;
}
@@ -107,13 +107,13 @@ class DeviceQueueOp : public PipelineOp {
DeviceType builder_device_type_;
std::string builder_channel_name_;
bool builder_send_epoch_end_;
int builder_total_batch_;
int32_t builder_total_batch_;
};

// Name: constructor
// Description
DeviceQueueOp(std::string channel_name, DeviceType device_type, int32_t device_id, int32_t prefetch_size,
bool send_epoch_end, int total_batch);
bool send_epoch_end, int32_t total_batch);

// Name: destructor
// Description
@@ -189,7 +189,7 @@ class DeviceQueueOp : public PipelineOp {
const int32_t prefetch_size_;
const bool send_epoch_end_;
bool stop_send_;
int total_batch_;
int32_t total_batch_;

#ifdef ENABLE_TDTQUE
std::shared_ptr<TdtPlugin> tdtInstancePtr;


Loading…
Cancel
Save