Merge branch 'rocm-jaxlib-v0.6.0-rc' of https://github.com/ROCm/jax into ut-0.6.0-skip-failing

Author: gulsumgudukbay

jax/_src/test_util.py

@@ -34,7 +34,6 @@
 import threading
 from typing import Any, TextIO
 import unittest
-import warnings
 import zlib
 
 from absl.testing import parameterized
@@ -404,7 +403,6 @@ def get_rocm_version():
     major, minor, *_ = version_str.split(".")
     return int(major), int(minor)
   except FileNotFoundError:
-    warnings.warn("ROCm was not installed")
     raise unittest.SkipTest("ROCm was not installed")
 
 def is_device_cuda():