Crashes due to overflow and `CHECK`-fail in ops with large tensor shapes

Summary

TensorFlow is an open source platform for machine learning. In affected versions TensorFlow allows tensor to have a large number of dimensions and each dimension can be as large as desired. However, the total number of elements in a tensor must fit within an `int64_t`. If an overflow occurs, `MultiplyWithoutOverflow` would return a negative result. In the majority of TensorFlow codebase this then results in a `CHECK`-failure. Newer constructs exist which return a `Status` instead of crashing the binary. This is similar to CVE-2021-29584. The fix will be included in TensorFlow 2.7.0. We will also cherrypick this commit on TensorFlow 2.6.1, TensorFlow 2.5.2, and TensorFlow 2.4.4, as these are also affected and still in supported range.

Timeline

1. Nov 5, 2021, 7:55 PM UTCCVE published
2. Aug 4, 2024, 3:08 AM UTCRecord last updated

Affected versions

• tensorflow tensorflow

  From 2.6.0 · fixed in 2.6.1; From 2.5.0 · fixed in 2.5.2; fixed in 2.4.4

• tensorflowPyPI

  fixed in 2.4.4; From 2.5.0 · fixed in 2.5.2; From 2.6.0 · fixed in 2.6.1; From 2.7.0rc0 · fixed in 2.7.0

• tensorflowPyPI

  From 2.6.0 · fixed in 2.6.1

• tensorflowPyPI

  From 2.5.0 · fixed in 2.5.2

• tensorflowPyPI

  fixed in 2.4.4

• tensorflow-cpuPyPI

  From 2.6.0 · fixed in 2.6.1

• tensorflow-cpuPyPI

  From 2.5.0 · fixed in 2.5.2

• tensorflow-cpuPyPI

  fixed in 2.4.4

• tensorflow-gpuPyPI

  From 2.6.0 · fixed in 2.6.1

• tensorflow-gpuPyPI

  From 2.5.0 · fixed in 2.5.2

• tensorflow-gpuPyPI

  fixed in 2.4.4

• tensorflow-cpuPyPI

  fixed in 2.4.4; From 2.5.0 · fixed in 2.5.2; From 2.6.0 · fixed in 2.6.1; From 2.7.0rc0 · fixed in 2.7.0

• tensorflow-gpuPyPI

  fixed in 2.4.4; From 2.5.0 · fixed in 2.5.2; From 2.6.0 · fixed in 2.6.1; From 2.7.0rc0 · fixed in 2.7.0

References 6