review.tizen.org Git - platform/upstream/pytorch.git/commit

projects / platform / upstream / pytorch.git / commit

author	Vishwak Srinivasan <cs15btech11043@iith.ac.in>
	Wed, 20 Mar 2019 18:06:56 +0000 (11:06 -0700)
committer	Facebook Github Bot <facebook-github-bot@users.noreply.github.com>
	Wed, 20 Mar 2019 18:11:32 +0000 (11:11 -0700)
commit	a519217ee72bdef12b5d133b7e941d92d6ccd3cf
tree	665982141e548653738f47418ca8922e70f9d41e	tree \| snapshot
parent	e312801453d1121f929b956c2862dfa58d6b3ac3	commit \| diff

Add batched version of trtrs (#18025)

Summary:
- Remove single batch TH/THC implementations
- Remove `_batch_trtrs_lower` from `multivariate_normal`
- Add tests for batched behavior
- Modify trtrs_backward to accommodate for batched case
- Modify docs

In a future PR, this will be renamed to `triangular_solve`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18025

Differential Revision: D14523004

Pulled By: ifedan

fbshipit-source-id: 11c6a967d107f969b60e5a5c73ce6bb8099ebbe1

19 files changed:

aten/src/ATen/Declarations.cwrap		diff \| blob \| history
aten/src/ATen/native/BatchLinearAlgebra.cpp		diff \| blob \| history
aten/src/ATen/native/LegacyDefinitions.cpp		diff \| blob \| history
aten/src/ATen/native/cuda/BatchLinearAlgebra.cu		diff \| blob \| history
aten/src/ATen/native/native_functions.yaml		diff \| blob \| history
aten/src/TH/generic/THLapack.cpp		diff \| blob \| history
aten/src/TH/generic/THLapack.h		diff \| blob \| history
aten/src/TH/generic/THTensorLapack.cpp		diff \| blob \| history
aten/src/TH/generic/THTensorLapack.h		diff \| blob \| history
aten/src/THC/generic/THCTensorMathMagma.cu		diff \| blob \| history
aten/src/THC/generic/THCTensorMathMagma.h		diff \| blob \| history
test/test_autograd.py		diff \| blob \| history
test/test_cuda.py		diff \| blob \| history
test/test_torch.py		diff \| blob \| history
tools/autograd/templates/Functions.cpp		diff \| blob \| history
torch/_torch_docs.py		diff \| blob \| history
torch/distributions/kl.py		diff \| blob \| history
torch/distributions/lowrank_multivariate_normal.py		diff \| blob \| history
torch/distributions/multivariate_normal.py		diff \| blob \| history

Domain: Machine Learning / ML Framework;

RSS Atom