Driver handling in svdvals function in torch_frontend

ivy/functional/backends/torch/linear_algebra.py

@@ -414,9 +414,11 @@ def svd(
         return results(D)
 
 
-@with_unsupported_dtypes({"2.1.0 and below": ("float16", "bfloat16")}, backend_version)
-def svdvals(x: torch.Tensor, /, *, out: Optional[torch.Tensor] = None) -> torch.Tensor:
-    return torch.linalg.svdvals(x, out=out)
+@with_unsupported_dtypes({"2.0.1 and below": ("float16", "bfloat16")}, backend_version)
+def svdvals(
+    x: torch.Tensor, /, *, driver: Optional[str], out: Optional[torch.Tensor] = None
+) -> torch.Tensor:
+    return torch.linalg.svdvals(x, driver=driver, out=out)
 
 
 svdvals.support_native_out = True

ivy/functional/frontends/torch/linalg.py

@@ -314,8 +314,10 @@ def svd(A, /, *, full_matrices=True, driver=None, out=None):
     {"2.1.0 and below": ("float32", "float64", "complex32", "complex64")}, "torch"
 )
 def svdvals(A, *, driver=None, out=None):
-    # TODO: add handling for driver
-    return ivy.svdvals(A, out=out)
+    if driver in ["gesvd", "gesvdj", "gesvda", None]:
+        return ivy.svdvals(A, driver=driver, out=out)
+    else:
+        raise ValueError("Unsupported SVD driver")
 
 
 @to_ivy_arrays_and_back

ivy/functional/ivy/linear_algebra.py

@@ -2263,7 +2263,11 @@ def svd(
 @handle_array_function
 @handle_device
 def svdvals(
-    x: Union[ivy.Array, ivy.NativeArray], /, *, out: Optional[ivy.Array] = None
+    x: Union[ivy.Array, ivy.NativeArray],
+    /,
+    *,
+    driver: Optional[str],
+    out: Optional[ivy.Array] = None,
 ) -> ivy.Array:
     """
     Return the singular values of a matrix (or a stack of matrices) ``x``.
@@ -2273,6 +2277,9 @@ def svdvals(
     x
         input array having shape ``(..., M, N)`` and whose innermost two dimensions form
         ``MxN`` matrices.
+    driver
+        optional output array,name of the cuSOLVER method to be used. This keyword argument only works on CUDA inputs.
+        Available options are: None, gesvd, gesvdj, and gesvda.Default: None.
     out
         optional output array, for writing the result to. It must have a shape that the
         inputs broadcast to.

ivy_tests/test_ivy/test_frontends/test_torch/test_linalg.py

@@ -1156,10 +1156,12 @@ def test_torch_svd(
 @handle_frontend_test(
     fn_tree="torch.linalg.svdvals",
     dtype_and_x=_get_dtype_and_matrix(batch=True),
+    driver=st.sampled_from([None, "gesvd", "gesvdj", "gesvda"]),
 )
 def test_torch_svdvals(
     *,
     dtype_and_x,
+    driver,
     on_device,
     fn_tree,
     frontend,
@@ -1174,6 +1176,7 @@ def test_torch_svdvals(
         test_flags=test_flags,
         fn_tree=fn_tree,
         on_device=on_device,
+        driver=driver,
         A=x[0],
     )