message: Fix unit test for matrix square root: Don't try to take the matrix square root of a matrix for which the square root may not exist.
diff: @@ -102,13 +102,13 @@ class SquareRootOpTest(test.TestCase):
  self.evaluate(gen_linalg_ops.matrix_square_root(tensor))
  def testConcurrentExecutesWithoutError(self):
  - self.skipTest("Triggers assert in matrix_sqrt_quasi_triangular_diagonal")
  -
  with test_util.use_gpu():
  matrix1 = random_ops.random_normal([5, 5], seed=42)
  matrix2 = random_ops.random_normal([5, 5], seed=42)
  - sqrt1 = gen_linalg_ops.matrix_square_root(matrix1)
  - sqrt2 = gen_linalg_ops.matrix_square_root(matrix2)
  + square1 = math_ops.matmul(matrix1, matrix1)
  + square2 = math_ops.matmul(matrix2, matrix2)
  + sqrt1 = gen_linalg_ops.matrix_square_root(square1)
  + sqrt2 = gen_linalg_ops.matrix_square_root(square2)
  all_ops = [sqrt1, sqrt2]
  sqrt = self.evaluate(all_ops)
  self.assertAllEqual(sqrt[0], sqrt[1])
  