https://ontrack-internal.amd.com/browse/SWDEV-560689
PYTORCH_TEST_WITH_ROCM=1 python test/test_transformers.py TestTransformersCUDA.test_transformerencoder_fastpath_use_torchscript_False_enable_nested_tensor_False_use_autocast_False_d_model_256_cuda
Observed output:
pastedImage_10_14_2025__21_40_14_394.png
Expected output: All testcases should pass