Ensure that indices are on the same device as self
authorSam Gross <sgross@fb.com>
Thu, 29 Nov 2018 01:51:01 +0000 (17:51 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 29 Nov 2018 01:54:32 +0000 (17:54 -0800)
commit533668d7e482ec5a88aaa274027de9cbefad5632
treedf38e9835a98a9c65b5009c8be116dc89aaedd5c
parentda9e49e586016d8905c242f9df1f4aeff2a96853
Ensure that indices are on the same device as self

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

Reviewed By: wat3rBro

Differential Revision: D13242200

Pulled By: colesbury

fbshipit-source-id: 82731cee808681ec612d406342070640eb26e519
aten/src/ATen/native/Indexing.cpp
aten/src/ATen/native/cuda/Loops.cuh
aten/src/ATen/test/basic.cpp