[xla:cpu] Restore the verbose version of custom call tests
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](https://github.com/tensorflow/tensorflow/pull/65819/checks?check_run_id=23867143761) 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.
[xla:cpu] Restore the verbose version of custom call tests


Reverts 71f2fdb. Applies necessary changes to the new tests.

Rollback reason: The more verbose version provides more context and is self-documenting; thus, it is easier to understand.

Reverts 8a2d31612fbafc7019bb15ad0a5752178cce745c

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/11524 from ROCm:ci_gemm_limit_solutions b8d2a95773ca80d8cb4a2fdac1d2624e4ae2d5c8
