Port torch.linspace to ATen and parallelize it on CPU.
authorGregory Chanan <gchanan@fb.com>
Tue, 18 Dec 2018 22:56:43 +0000 (14:56 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 18 Dec 2018 23:01:49 +0000 (15:01 -0800)
commit2469f7e02e5458fdbf9eba12d86eec5178b42584
tree3b6e9768fd24d4f28b18e30cd65d11b02690d6b4
parent3118124cd6132588c4bf6614cbff1a3abb977bd9
Port torch.linspace to ATen and parallelize it on CPU.

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15320

Reviewed By: ezyang

Differential Revision: D13498995

Pulled By: gchanan

fbshipit-source-id: fba655d51d978fffaa53a5e4cae4a99ebfb0eddc
aten/src/ATen/Declarations.cwrap
aten/src/ATen/native/RangeFactories.cpp [new file with mode: 0644]
aten/src/ATen/native/TensorFactories.cpp
aten/src/ATen/native/cuda/RangeFactories.cu [new file with mode: 0644]
aten/src/ATen/native/native_functions.yaml
aten/src/TH/generic/THTensorMath.h
aten/src/TH/generic/THTensorMoreMath.cpp
aten/src/THC/generic/THCTensorMath.cu
aten/src/THC/generic/THCTensorMath.h
test/test_torch.py