PR #11981: [ROCM] fixing pjrt_c_api_gpu_test
PR #11981: [ROCM] fixing pjrt_c_api_gpu_test

Imported from GitHub PR https://github.com/openxla/xla/pull/11981


Fixing platform name CUDA/ROCM

@xla-rotation: could you please have a look?
Copybara import of the project:

--
d114eb0deb691b5cf6907880f26d241a2813f26d by Pavel Emeliyanenko <pavel.emeliyanenko@amd.com>:

fixing the test

Merging this change closes #11981

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/11981 from ROCm:ci_pjrt_c_api_gpu_test_fix d114eb0deb691b5cf6907880f26d241a2813f26d
