Enable detectron on AMD GPU
authorSandeep Kumar <sandeep.kumar@amd.com>
Tue, 12 Mar 2019 23:22:41 +0000 (16:22 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 12 Mar 2019 23:29:42 +0000 (16:29 -0700)
commit6248266d911f180b1af43f48022efcf0ef29d733
tree794b8ab0314416336a6723d377ecb35055012ee2
parent1cfb50334f408c182e24e0cd786045a3d30ba457
Enable detectron on AMD GPU

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

Differential Revision: D14429234

Pulled By: bddppq

fbshipit-source-id: 5cb8750bd9db0ff8a179977d2bfbb180265cce81
15 files changed:
modules/CMakeLists.txt
modules/detectron/CMakeLists.txt
modules/detectron/batch_permutation_op.cu
modules/detectron/group_spatial_softmax_op.cu
modules/detectron/ps_roi_pool_op.cu
modules/detectron/roi_pool_f_op.cu
modules/detectron/sample_as_op.cu
modules/detectron/select_smooth_l1_loss_op.cu
modules/detectron/sigmoid_cross_entropy_loss_op.cu
modules/detectron/sigmoid_focal_loss_op.cu
modules/detectron/smooth_l1_loss_op.cu
modules/detectron/softmax_focal_loss_op.cu
modules/detectron/spatial_narrow_as_op.cu
modules/detectron/upsample_nearest_op.cu
tools/amd_build/build_amd.py