search

tensorflow / tensorflow

An Open Source Machine Learning Framework for Everyone
https://tensorflow.org
Apache License 2.0
182.91k stars 73.92k forks source link

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

Closed copybara-service[bot] closed 1 week ago

copybara-service[bot] commented 1 week ago

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