torch.btrifact for tensors with greater than 3 dimensions (#14964)
authorvishwakftw <cs15btech11043@iith.ac.in>
Tue, 12 Mar 2019 08:42:28 +0000 (01:42 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 12 Mar 2019 08:46:07 +0000 (01:46 -0700)
commitf268370b42bc26d4e5d0fd88eb79b44ae162dec5
tree7bfa48ec5100b9a2c155e87d66101c805b9b0e87
parentb161ac9634d11ae3c6a203b22e057cce9b7137d7
torch.btrifact for tensors with greater than 3 dimensions (#14964)

Summary:
Motivation:
- Earlier, `torch.btrifact` could not handle tensors with greater than 3 dimensions. This is because of the check:
>   AT_CHECK(THTensor_(nDimension)(a) == 3, "expected 3D tensor, got size: ", a->sizes());

What is in this PR?:
- Move `btrifact` to ATen
- Remove relation to TH/THC.
- Handle tensors with more than three dimensions
- Tests
- Docs modifications: added a note about the non-pivoting variant.

[blocked due to old magma-cuda binaries]
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14964

Differential Revision: D14405106

Pulled By: soumith

fbshipit-source-id: f051f5d6aaa45f85836a2867176c065733563184
14 files changed:
aten/src/ATen/Declarations.cwrap
aten/src/ATen/native/BatchLinearAlgebra.cpp
aten/src/ATen/native/LegacyDefinitions.cpp
aten/src/ATen/native/LinearAlgebra.cpp
aten/src/ATen/native/LinearAlgebraUtils.h
aten/src/ATen/native/cuda/BatchLinearAlgebra.cu
aten/src/ATen/native/native_functions.yaml
aten/src/TH/generic/THTensorLapack.cpp
aten/src/TH/generic/THTensorLapack.h
aten/src/THC/generic/THCTensorMathBlas.cu
aten/src/THC/generic/THCTensorMathBlas.h
test/test_torch.py
tools/autograd/gen_python_functions.py
torch/_torch_docs.py