`CHECK` fail in `tf.linalg.matrix_rank` in TensorFlow
Description
TensorFlow is an open source platform for machine learning. When tf.linalg.matrix_rank receives an empty input a, the GPU kernel gives a CHECK fail that can be used to trigger a denial of service attack. We have patched the issue in GitHub commit c55b476aa0e0bd4ee99d0f3ad18d9d706cd1260a. The fix will be included in TensorFlow 2.10.0. We will also cherrypick this commit on TensorFlow 2.9.1, TensorFlow 2.8.1, and TensorFlow 2.7.2, as these are also affected and still in supported range. There are no known workarounds for this issue.
Affected packages
Versions sourced from the GitHub Security Advisory.
| Package | Affected versions | Patched versions |
|---|---|---|
tensorflowPyPI | < 2.7.2 | 2.7.2 |
tensorflowPyPI | >= 2.8.0, < 2.8.1 | 2.8.1 |
tensorflowPyPI | >= 2.9.0, < 2.9.1 | 2.9.1 |
tensorflow-cpuPyPI | < 2.7.2 | 2.7.2 |
tensorflow-cpuPyPI | >= 2.8.0, < 2.8.1 | 2.8.1 |
tensorflow-cpuPyPI | >= 2.9.0, < 2.9.1 | 2.9.1 |
tensorflow-gpuPyPI | < 2.7.2 | 2.7.2 |
tensorflow-gpuPyPI | >= 2.8.0, < 2.8.1 | 2.8.1 |
tensorflow-gpuPyPI | >= 2.9.0, < 2.9.1 | 2.9.1 |
Affected products
1- Range: < 2.7.2
Patches
1c55b476aa0e0Fix empty batch issue in svd.
2 files changed · +14 −0
tensorflow/core/kernels/linalg/svd_op_gpu.cu.cc+6 −0 modified@@ -395,6 +395,12 @@ class SvdOpGpu : public AsyncOpKernel { OP_REQUIRES_OK_ASYNC(context, context->allocate_output(2, shapeV, &outputV), done); + // If there are zero batches, we are done. + if (shapeRaw.num_elements() == 0) { + done(); + return; + } + if (n == 0 || m == 0) { if (n == m || !compute_uv_ || !full_matrices_) { // S, U, and V are all empty. Nothing to do.
tensorflow/python/kernel_tests/linalg/svd_op_test.py+8 −0 modified@@ -108,6 +108,14 @@ def testExecuteMultipleWithoutError(self): for i in range(0, len(val), 2): self.assertAllEqual(val[i], val[i + 1]) + @test_util.run_in_graph_and_eager_modes(use_gpu=True) + def testEmptyBatches(self): + matrices = constant_op.constant(1.0, shape=[0, 2, 2]) + s, u, v = self.evaluate(linalg_ops.svd(matrices)) + self.assertAllEqual(s, np.zeros([0, 2])) + self.assertAllEqual(u, np.zeros([0, 2, 2])) + self.assertAllEqual(v, np.zeros([0, 2, 2])) + def _GetSvdOpTest(dtype_, shape_, use_static_shape_, compute_uv_, full_matrices_):
Vulnerability mechanics
Generated by null/stub on May 9, 2026. Inputs: CWE entries + fix-commit diffs from this CVE's patches. Citations validated against bundle.
References
5- github.com/advisories/GHSA-9vqj-64pv-w55cghsaADVISORY
- nvd.nist.gov/vuln/detail/CVE-2022-35988ghsaADVISORY
- github.com/tensorflow/tensorflow/commit/c55b476aa0e0bd4ee99d0f3ad18d9d706cd1260aghsax_refsource_MISCWEB
- github.com/tensorflow/tensorflow/releases/tag/v2.10.0ghsaWEB
- github.com/tensorflow/tensorflow/security/advisories/GHSA-9vqj-64pv-w55cghsax_refsource_CONFIRMWEB
News mentions
0No linked articles in our index yet.