Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
@arcadiaphy
arcadiaphy committed May 16, 2019
1 parent 889fc6e commit 808104d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/operator/tensor/la_op.cu
View file
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ NNVM_REGISTER_OP(_linalg_inverse)
.set_attr<FCompute>("FCompute<gpu>", LaOpForward<gpu, 2, 2, 1, 1, inverse>);

NNVM_REGISTER_OP(_backward_linalg_inverse)
.set_attr<FCompute>("FCompute<gpu>", LaOpBackward<gpu, 2, 2, 3, 1, inverse_backward>);
.set_attr<FCompute>("FCompute<gpu>", LaOpBackward<gpu, 2, 2, 2, 1, inverse_backward>);

#if MXNET_USE_CUSOLVER == 1

Expand Down

0 comments on commit 808104d

Please sign in to comment.