diff options
author | Wenyong Huang <weyo.huang@gmail.com> | 2018-08-01 17:57:11 +0300 |
---|---|---|
committer | Wenyong Huang <weyo.huang@gmail.com> | 2018-08-04 06:19:50 +0300 |
commit | 3e19457912a5310c3f7d566b668a23f1b9f92c3c (patch) | |
tree | 1d663c4471fe890a354595b563525396b08fd20a /src/graph/node_operators_unary.h | |
parent | 78cc62e1beddc95b96902467f1e9fccf9db590a3 (diff) |
avoid cudaMalloc in CopyRows
Diffstat (limited to 'src/graph/node_operators_unary.h')
-rw-r--r-- | src/graph/node_operators_unary.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/graph/node_operators_unary.h b/src/graph/node_operators_unary.h index cbeded24..b51c8a75 100644 --- a/src/graph/node_operators_unary.h +++ b/src/graph/node_operators_unary.h @@ -657,7 +657,7 @@ struct RowsNodeOp : public UnaryNodeOp { NodeOps forwardOps() { // @TODO: solve this with a tensor! - return {NodeOp(CopyRows(val_, child(0)->val(), indices_))}; + return {NodeOp(CopyRows(val_, child(0)->val(), indices_, graph()->allocator()))}; } NodeOps backwardOps() { |