This matches better how other gpu integration tests are done. Differential Revision: https://reviews.llvm.org/D103099