PR #12224: [GPU] Fix handling of flags in the cuDNN FMHA test.
PR #12224: [GPU] Fix handling of flags in the cuDNN FMHA test.

Imported from GitHub PR https://github.com/openxla/xla/pull/12224

The test got broken by https://github.com/openxla/xla/commit/8799ff0f4b2aa3437a9963f7c9968ee361a62fd7, this commit fixes it.
Copybara import of the project:

--
80528497321ee6020126b15035050f4c1a0beea9 by Ilia Sergachev <isergachev@nvidia.com>:

[GPU] Fix handling of flags in the cuDNN FMHA test.

Merging this change closes #12224

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/12224 from openxla:fix_fmha_test 80528497321ee6020126b15035050f4c1a0beea9
