diff options
| author | Brandon Amos <bamos@cs.cmu.edu> | 2017-03-26 20:06:34 +0300 |
|---|---|---|
| committer | Brandon Amos <bamos@cs.cmu.edu> | 2017-03-26 20:06:34 +0300 |
| commit | 69723acd5014062758897e09f9c1ebfe9158f650 (patch) | |
| tree | 3fdfcdd73b780b791695ecc5aad786f268e05c34 | |
| parent | dad316fb3617b8d1299a585ad804ab13b48ffb50 (diff) | |
Update btrisolve argument order.
| -rw-r--r-- | lib/TH/generic/THTensorLapack.c | 2 | ||||
| -rw-r--r-- | lib/TH/generic/THTensorLapack.h | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/lib/TH/generic/THTensorLapack.c b/lib/TH/generic/THTensorLapack.c index 9e3309c..4f88e23 100644 --- a/lib/TH/generic/THTensorLapack.c +++ b/lib/TH/generic/THTensorLapack.c @@ -1010,7 +1010,7 @@ void THTensor_(btrifact)(THTensor *ra_, THIntTensor *rpivots_, THIntTensor *rinf } } -void THTensor_(btrisolve)(THTensor *rb_, THTensor *atf, THTensor *b, THIntTensor *pivots) +void THTensor_(btrisolve)(THTensor *rb_, THTensor *b, THTensor *atf, THIntTensor *pivots) { THArgCheck(THTensor_(nDimension)(atf) == 3, 1, "expected 3D tensor, got %dD", THTensor_(nDimension)(atf)); diff --git a/lib/TH/generic/THTensorLapack.h b/lib/TH/generic/THTensorLapack.h index 26da0fd..29db83b 100644 --- a/lib/TH/generic/THTensorLapack.h +++ b/lib/TH/generic/THTensorLapack.h @@ -20,6 +20,6 @@ TH_API void THTensor_(ormqr)(THTensor *ra_, THTensor *a, THTensor *tau, THTensor TH_API void THTensor_(pstrf)(THTensor *ra_, THIntTensor *rpiv_, THTensor*a, const char* uplo, real tol); TH_API void THTensor_(btrifact)(THTensor *ra_, THIntTensor *rpivots_, THIntTensor *rinfo_, THTensor *a); -TH_API void THTensor_(btrisolve)(THTensor *rb_, THTensor *a, THTensor *b, THIntTensor *pivots); +TH_API void THTensor_(btrisolve)(THTensor *rb_, THTensor *b, THTensor *atf, THIntTensor *pivots); #endif |