Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!7650 modify variable type from int to int32_t 

Merge pull request !7650 from anzhengqi/fixed-variable-type
This commit is contained in:
mindspore-ci-bot 2020-10-23 16:50:27 +08:00 committed by Gitee
parent caab8a1507 516c53f1f0
commit de4da6457e
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
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),

8
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;