PR #10066: [ROCM] rocm tracer fixes for jax profiler
PR #10066: [ROCM] rocm tracer fixes for jax profiler

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

1. Don't subscribe to HIP_API activity events
2. Use array for per_device_collector
3. Fix device_id indexing
Copybara import of the project:

--
50d23258e024e75d640a241775a4a6dbfbfb5d51 by Rahul Batra <rahbatra@amd.com>:

[ROCM] rocm tracer fixes for jax profiler

Merging this change closes #10066

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/10066 from ROCm:ci_profiler_fixes 50d23258e024e75d640a241775a4a6dbfbfb5d51
