|
|
|
@@ -30,7 +30,7 @@ std::shared_ptr<TdtPlugin> TdtPlugin::GetInstance() { |
|
|
|
} |
|
|
|
|
|
|
|
TdtStatus TdtPlugin::hostPush(TensorRow ts_row, bool is_wait, std::string channel_name, bool profiling, int32_t &time) { |
|
|
|
MS_LOG(INFO) << "TDT channel name is " << channel_name << "."; |
|
|
|
MS_LOG(DEBUG) << "TDT channel name is " << channel_name << "."; |
|
|
|
std::vector<DataItem> items; |
|
|
|
double start_time; |
|
|
|
auto ret = translate(ts_row, items); |
|
|
|
@@ -121,8 +121,8 @@ TdtStatus TdtPlugin::translate(const TensorRow &ts_row, std::vector<DataItem> &i |
|
|
|
data_item.dataLen_ = ts->SizeInBytes(); |
|
|
|
data_item.dataPtr_ = std::shared_ptr<void>(reinterpret_cast<uchar *>(&(*ts->begin<uint8_t>())), [](void *elem) {}); |
|
|
|
items.emplace_back(data_item); |
|
|
|
MS_LOG(INFO) << "TDT data type is " << datatype << ", data shape is " << dataShapes << ", data length is " |
|
|
|
<< ts->Size() << "."; |
|
|
|
MS_LOG(DEBUG) << "TDT data type is " << datatype << ", data shape is " << dataShapes << ", data length is " |
|
|
|
<< ts->Size() << "."; |
|
|
|
} |
|
|
|
return SUCCESS; |
|
|
|
} |
|
|
|
|