fix gpu build (apache#69)

reminisce · Jun 7, 2017 · 39d12bc · 39d12bc
1 parent a7c8fe8
commit 39d12bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/executor/attach_op_execs_pass.cc b/src/executor/attach_op_execs_pass.cc
@@ -44,7 +44,7 @@ class ForwardOpExecutor : public OpExecutor {
       GetDefaultBlobs<gpu>(aux_array_, &aux_data_, &temp_aux_, op_ctx);
       GetDefaultBlobs<gpu>(out_array, &out_data_, &temp_out_, op_ctx);
       op_->Forward(op_ctx, in_data_, req, out_data_, aux_data_);
-      CastNonDefaultStorage<gpu>(out_array_, temp_out_, op_ctx);
+      CastNonDefaultStorage<gpu>(out_array, temp_out_, op_ctx);
 #else
       LOG(FATAL) << MXNET_GPU_NOT_ENABLED_ERROR;
 #endif