Small fixes for pdist (#16210)
authorErik Brinkman <erk@fb.com>
Thu, 24 Jan 2019 03:37:29 +0000 (19:37 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 24 Jan 2019 03:40:16 +0000 (19:40 -0800)
commit3cba115abb750b74ecc246f76e0592ca3ca751a9
tree86d6960328fc182c9609284b12f30123dc287fc5
parent0a3932acb2f091b803cc6fe520673d90fc9cebf3
Small fixes for pdist (#16210)

Summary:
pdist was recently patched to remove buggy batch support and fix issues
with large tensors. This fixed missed a few spots, and didn't handle a
few recommendations that this commit addresses.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16210

Differential Revision: D13791914

Pulled By: gchanan

fbshipit-source-id: 0595841be1b298f7268fd4c02a6628acfec918f2
aten/src/ATen/native/cpu/DistanceOpsKernel.cpp
aten/src/ATen/native/cuda/DistanceKernel.cu