Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
pcmoritz committed Dec 1, 2018
1 parent 4836342 commit f89d5df
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/pyarrow/tensorflow/plasma_op.cc
Original file line number Diff line number Diff line change
Expand Up @@ -189,7 +189,7 @@ class TensorToPlasmaOp : public tf::AsyncOpKernel {
// async memcpy. Under the hood it performs cuMemHostRegister(), see:
// http://docs.nvidia.com/cuda/cuda-driver-api/group__CUDA__MEM.html#group__CUDA__MEM_1gf0a9fe11544326dabd743b7aa6b54223
CHECK(stream_executor->HostMemoryRegister(static_cast<void*>(data),
static_cast<tf::uint64>(total_bytes)));
static_cast<tf::uint64>(total_bytes)));

{
tf::mutex_lock l(d2h_stream_mu);
Expand Down

0 comments on commit f89d5df

Please sign in to comment.