add zmq op to travis

.travis.yml

@@ -14,6 +14,7 @@ addons:
     - pandoc
     - libprotobuf-dev
     - protobuf-compiler
+    - libzmqpp-dev
 
 matrix:
   fast_finish: true

tensorpack/user_ops/zmq_recv_op.cc

@@ -71,7 +71,8 @@ class ZMQRecvOp: public AsyncOpKernel {
       TensorShape& shape = tensors[i].shape;
       OP_REQUIRES_OK_ASYNC(ctx, ctx->allocate_output(i, shape, &output), done);
       // reinterpret cast and then memcpy
-      auto ptr = output->bit_casted_shaped<char, 1>({shape.num_elements()}).data();
+      auto ptr = output->bit_casted_shaped<char, 1>(
+          {shape.num_elements() * DataTypeSize(recv_dtype)}).data();
       memcpy(ptr, tensors[i].buf, tensors[i].buf_size);
       ctx->set_output(i, *output);
     }

tests/run-tests.sh

@@ -12,3 +12,5 @@ python -c 'from tensorpack.dataflow.imgaug import transform'
 python -m unittest discover -v
 # python -m tensorpack.models._test
 # segfault for no reason (https://travis-ci.org/ppwwyyxx/tensorpack/jobs/217702985)
+
+python ../tensorpack/user_ops/test-recv-op.py