Add support for HIP to DispatchStub. (#14413)
authorEdward Yang <ezyang@fb.com>
Wed, 28 Nov 2018 19:05:36 +0000 (11:05 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 28 Nov 2018 19:07:45 +0000 (11:07 -0800)
commit7c4aef9dfc71c0c79eb0c442a111127793da4611
treeb19635d444b3f3c9bb881583acf65d973624709b
parent7749804099b8a64aea4bf91e298a20976f9b10ad
Add support for HIP to DispatchStub. (#14413)

Summary:
I feel a bit bad writing this patch, because there isn't really
any reason not to use the normal dispatch mechanism for CUDA
and HIP here (so we have *yet another dispatcher*), but I don't
really want to sign up to rewrite DispatchStub to deduplicate the
dispatcher right now.

Need to natively add support for HIP here, as I don't want to
have to HIPify files which are not in a CUDA directory.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14413

Differential Revision: D13220358

Pulled By: ezyang

fbshipit-source-id: cc61218322589a1dc2ab8eb9d5ddd3c616f6b712
aten/src/ATen/native/DispatchStub.h