Enable tests working on ROCm 2.1 dual gfx906
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17473
Reviewed By: bddppq
Differential Revision: D14210243
Pulled By: ezyang
fbshipit-source-id: 519032a1e73c13ecb260ea93102dc8efb645e070
