PR #11352: [ROCm] Triton in XLA for ROCm - changed ir_emitter_triton_test 
PR #11352: [ROCm] Triton in XLA for ROCm - changed ir_emitter_triton_test 

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

Modified the test case to pass on rocm and cuda.
There will be at least one more XLA related PR - to switch on Triton usage for ROCm, but only after PR for adding Triton build files for ROCm on openxla/triton.
Copybara import of the project:

--
dd80a6f44295642efd3ae6af6ffd3e2a3302d36e by Zoran Jovanovic <zjovanov@amd.com>:

[ROCm] Triton in XLA for ROCm - changed ir_emitter_triton_test to pass on rocm and cuda

Merging this change closes #11352

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/11352 from ROCm:rocm_triton_backend_5 dd80a6f44295642efd3ae6af6ffd3e2a3302d36e
