提交 0833ce24 编写于 作者: C chengtbf 提交者: Will Zhang

fix bug of print kernel to print useless content (#495)

上级 65dd29f7
......@@ -11,10 +11,13 @@ void PrintBlobImpl(PersistentOutStream& out_stream, const Blob* blob) {
const T* dptr = blob->dptr<T>();
for (int64_t i = 0; i < blob->shape().At(0); ++i) {
if (blob->has_data_id()) {
for (size_t j = 0; j != JobDesc::Singleton()->SizeOfOneDataId(); ++j) {
if (*(blob->data_id(i) + j) == '\0') { break; }
out_stream.Write(blob->data_id(i) + j, 1);
size_t data_id_size = 0;
for (; data_id_size != JobDesc::Singleton()->SizeOfOneDataId();
++data_id_size) {
if (*(blob->data_id(i) + data_id_size) == '\0') { break; }
}
if (data_id_size == 0) { continue; }
out_stream.Write(blob->data_id(i), data_id_size);
out_stream.Write(",", 1);
}
for (int64_t j = 0; j < blob->shape().Count(1); ++j) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册