cat_shape_check: Fixes dimension in the error message for CUDA cat shape check and...
authorPalwisha Akhtar <msoyturk20@ku.edu.tr>
Thu, 9 Sep 2021 19:49:03 +0000 (12:49 -0700)
committerFacebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
Thu, 9 Sep 2021 19:51:11 +0000 (12:51 -0700)
commitd43fb75a2196742c806a2e938d10cd2fba1cc168
tree34478c81061ae188ff5b1754e74f9943a06f7e4e
parent2c243ed112b4061e8e287be9c10fc22a463053d9
cat_shape_check: Fixes dimension in the error message for CUDA cat shape check and removes unnecessary offending index information (#64556)

Summary:
Fixes: https://github.com/pytorch/pytorch/issues/64207

Thank you, SsnL for providing the reproducing script.

cc ngimel

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

Reviewed By: albanD

Differential Revision: D30843859

Pulled By: ngimel

fbshipit-source-id: 457ebe80eaef793d9f5d35ee962b6697e5de1907
aten/src/ATen/native/TensorShape.cpp
aten/src/ATen/native/TensorShape.h [new file with mode: 0644]
aten/src/ATen/native/cuda/Shape.cu