diff options
| author | Brandon Amos <bamos@cs.cmu.edu> | 2017-06-13 20:45:40 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-06-22 19:44:51 +0300 |
| commit | 3a322076d8c3f5e52b6777691f801a92ea880b7c (patch) | |
| tree | eb1228efbf76878caea42f3d4619ec48503be253 | |
| parent | 3864f3782a74715e35d98176f9014a74524a868a (diff) | |
btrifact: Make pivoting optional.
| -rw-r--r-- | lib/TH/generic/THTensorLapack.c | 5 | ||||
| -rw-r--r-- | lib/TH/generic/THTensorLapack.h | 2 |
2 files changed, 5 insertions, 2 deletions
diff --git a/lib/TH/generic/THTensorLapack.c b/lib/TH/generic/THTensorLapack.c index d0196c9..d4e52f6 100644 --- a/lib/TH/generic/THTensorLapack.c +++ b/lib/TH/generic/THTensorLapack.c @@ -938,9 +938,12 @@ void THTensor_(ormqr)(THTensor *ra_, THTensor *a, THTensor *tau, THTensor *c, co THTensor_(free)(work); } -void THTensor_(btrifact)(THTensor *ra_, THIntTensor *rpivots_, THIntTensor *rinfo_, THTensor *a) +void THTensor_(btrifact)(THTensor *ra_, THIntTensor *rpivots_, THIntTensor *rinfo_, int pivot, THTensor *a) { THArgCheck(THTensor_(nDimension)(a) == 3, 1, "expected 3D tensor, got %dD", THTensor_(nDimension)(a)); + if (!pivot) { + THError("btrifact without pivoting is not implemented on the CPU"); + } if (ra_ != a) { THTensor_(resizeAs)(ra_, a); diff --git a/lib/TH/generic/THTensorLapack.h b/lib/TH/generic/THTensorLapack.h index 29db83b..8785943 100644 --- a/lib/TH/generic/THTensorLapack.h +++ b/lib/TH/generic/THTensorLapack.h @@ -19,7 +19,7 @@ TH_API void THTensor_(orgqr)(THTensor *ra_, THTensor *a, THTensor *tau); TH_API void THTensor_(ormqr)(THTensor *ra_, THTensor *a, THTensor *tau, THTensor *c, const char *side, const char *trans); 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_(btrifact)(THTensor *ra_, THIntTensor *rpivots_, THIntTensor *rinfo_, int pivot, THTensor *a); TH_API void THTensor_(btrisolve)(THTensor *rb_, THTensor *b, THTensor *atf, THIntTensor *pivots); #endif |