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

Make Blas version optional in HloAlgorithmDenylist #67152

Closed copybara-service[bot] closed 1 week ago

copybara-service[bot] commented 1 week ago

Make Blas version optional in HloAlgorithmDenylist

Most cuDNN algorithms nowadays don't call cuBlas kernels anymore, so the Blas version is often not relevant when adding a new entry to the denylist.

So this change makes it possible to leave the blas version field empty and it will just match any blas version.

It also does various cleanups:

The original behaviour stays intact, so it shouldn't break any user of the denylist feature.