Set driver default

pytorch · Oct 16, 2024 · 2c99db1 · 2c99db1
1 parent f8bac9a
commit 2c99db1
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/experimental/torch_xla2/torch_xla2/ops/jaten.py b/experimental/torch_xla2/torch_xla2/ops/jaten.py
@@ -2413,10 +2413,7 @@ def _aten_linalg_eigh(A, UPLO='L'):
 
 
 @op(torch.ops.aten.linalg_lstsq)
-def _aten_linalg_lstsq(A, B, rcond=None, driver=None):
-  if not driver:
-      driver = 'gelsy'  # Set default driver
-
+def _aten_linalg_lstsq(A, B, rcond=None, driver='gelsy'):
   input_dtype = A.dtype
 
   m = A.shape[-2]