Description
Tensorflow is an Open Source Machine Learning Framework. The `simplifyBroadcast` function in the MLIR-TFRT infrastructure in TensorFlow is vulnerable to a segfault (hence, denial of service), if called with scalar shapes. If all shapes are scalar, then `maxRank` is 0, so we build an empty `SmallVector`. The fix will be included in TensorFlow 2.8.0. This is the only affected version.
Affected products
| Vendor | Product | Affected | Fixed |
|---|---|---|---|
| tensorflow | tensorflow | >= 2.7.0, < 2.8.0 | — |
Affected packages
| Ecosystem | Package | Affected | Fixed | Record |
|---|---|---|---|---|
| PyPI | tensorflow | >= 2.8.0-rc0, < 2.8.0 | 2.8.0 | GHSA-gwcx-jrx4-92w2 |
| PyPI | tensorflow-cpu | >= 2.8.0-rc0, < 2.8.0 | 2.8.0 | GHSA-gwcx-jrx4-92w2 |
| PyPI | tensorflow-cpu | >= 2.7.0, < 2.8.0 | 2.8.0 | PYSEC-2022-102 |
| PyPI | tensorflow-gpu | >= 2.8.0-rc0, < 2.8.0 | 2.8.0 | GHSA-gwcx-jrx4-92w2 |
| PyPI | tensorflow-gpu | >= 2.7.0, < 2.8.0 | 2.8.0 | PYSEC-2022-157 |
References
- https://github.com/tensorflow/tensorflow/security/advisories/GHSA-gwcx-jrx4-92w2
- https://github.com/tensorflow/tensorflow/commit/35f0fabb4c178253a964d7aabdbb15c6a398b69a
- https://github.com/tensorflow/tensorflow/blob/274df9b02330b790aa8de1cee164b70f72b9b244/tensorflow/compiler/mlir/tfrt/jit/transforms/tf_cpurt_symbolic_shape_optimization.cc#L149-L205