# tf.test.is_built_with_rocm

<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.14.0/tensorflow/python/platform/test.py#L113-L131) |

Returns whether TensorFlow was built with ROCm (GPU) support.

#### View aliases


**Compat aliases for migration**

See
[Migration guide](https://www.tensorflow.org/guide/migrate) for
more details.

[`tf.compat.v1.test.is_built_with_rocm`](https://www.tensorflow.org/api_docs/python/tf/test/is_built_with_rocm)

<br />

    tf.test.is_built_with_rocm()

This method should only be used in tests written with [`tf.test.TestCase`](https://www.tensorflow.org/versions/r2.14/api_docs/python/tf/test/TestCase). A
typical usage is to skip tests that should only run with ROCm (GPU).

    class MyTest(tf.test.TestCase):

      def test_add_on_gpu(self):
        if not tf.test.is_built_with_rocm():
          self.skipTest("test is only applicable on GPU")

        with tf.device("GPU:0"):
          self.assertEqual(tf.math.add(1.0, 2.0), 3.0)

TensorFlow official binary is NOT built with ROCm.