pytorch
12dde7f5 - cdist performance improvement for euclidean distance (#25799)

Commit
5 years ago
cdist performance improvement for euclidean distance (#25799) Summary: jacobrgardner https://github.com/pytorch/pytorch/issues/15253#issuecomment-491467128 preposed a way to speedup euclidean distance calculation. This PR is implementation of this solution for normal and batch version. Also simonepri provided performance metrics https://github.com/pytorch/pytorch/issues/15253#issuecomment-502363581 ![image](https://user-images.githubusercontent.com/12058312/64460756-44a24580-d0c9-11e9-9f7f-a5942f4c832d.png) Current implementation has speedup comparing to jacobrgardner approach ![image](https://user-images.githubusercontent.com/12058312/64461495-5553bb00-d0cb-11e9-87e6-302b8cc7e12b.png) Pull Request resolved: https://github.com/pytorch/pytorch/pull/25799 Differential Revision: D17964982 Pulled By: ifedan fbshipit-source-id: bf7bd0dbfca51fd39e667da55139347480f30a2f
Author
Parents
Loading