[ROCm] Add float8_e4m3fnuz test

[Ruturaj4]

trafficstars

[google-cla[bot]]

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

[Ruturaj4]

@jakevdp can you review this please

[jakevdp]

Can you say more about why you want to enable this test? It's only testing dot_general with preferred_element_type – we do have other coverage for dot_general without this argument.

[Ruturaj4]

@jakevdp Thanks for your comments. I revised the patch. Currently only enabled for rocm.