pytorch
390fe749 - Migrate `torch.lstsq` to ATen (#59400)

Commit

3 years ago

Migrate `torch.lstsq` to ATen (#59400) Summary: Closes https://github.com/pytorch/pytorch/issues/24726, closes https://github.com/pytorch/pytorch/issues/44011 This builds on the port from https://github.com/pytorch/pytorch/issues/44011. I've rebased on master and addressed mruberry's comments. There were also some unnecessary copies of `B` taking place that I've cleaned up. This function is already deprecated, but since it's the last lapack routine in TH, it's still worth porting. Pull Request resolved: https://github.com/pytorch/pytorch/pull/59400 Reviewed By: mruberry Differential Revision: D28922060 Pulled By: ngimel fbshipit-source-id: cfd7ec8b50d2ab886f0e04a2a557e4e410ee8184

Author

peterbell10

Committer

facebook-github-bot

Parents

da972afd

Files16

BUILD.bazel
aten/src
- ATen
  - LegacyTHFunctionsCPU.cpp
  - LegacyTHFunctionsCPU.h
  - native
    - BatchLinearAlgebra.cpp
    - native_functions.yaml
- TH
  - CMakeLists.txt
  - TH.h
  - THLapack.cpp
  - THLapack.h
  - THTensor.h
  - THTensorLapack.cpp
  - generic
    - THLapack.cpp
    - THLapack.h
    - THTensorLapack.cpp
    - THTensorLapack.h
tools
- build_variables.bzl

pytorch 390fe749 - Migrate `torch.lstsq` to ATen (#59400)

pytorch
390fe749 - Migrate `torch.lstsq` to ATen (#59400)