Port `gather` to structured kernel (#63312)
authorKushashwa Ravi Shrimali <kushashwaravishrimali@gmail.com>
Thu, 2 Sep 2021 08:08:53 +0000 (01:08 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 2 Sep 2021 08:36:21 +0000 (01:36 -0700)
commit76e187aa08556ce90e84b17e836784ffbb6905e0
tree88b56ad783f58add02740fa5e29ce85eae275ad5
parentee8a6c1d141ae49e23323bdd485fb4b390541f69
Port `gather` to structured kernel (#63312)

Summary:
Will add a description once this is ready for review.

cc: ysiraichi ezyang

Pull Request resolved: https://github.com/pytorch/pytorch/pull/63312

Reviewed By: iramazanli

Differential Revision: D30597447

Pulled By: ezyang

fbshipit-source-id: d36e59835c2f4b38e286032dd2a1111a7e16b7e5
aten/src/ATen/native/ScatterGatherChecks.h
aten/src/ATen/native/TensorAdvancedIndexing.cpp
aten/src/ATen/native/TensorAdvancedIndexing.h
aten/src/ATen/native/cpu/ScatterGatherKernel.cpp
aten/src/ATen/native/cuda/ScatterGatherKernel.cu
aten/src/ATen/native/native_functions.yaml
test/test_torch.py