Fix torch.linalg.vecdot

Author: asmeurer

array_api_compat/torch/_aliases.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
-from builtins import all as builtin_all
-from builtins import any as builtin_any
+from builtins import (all as builtin_all, any as builtin_any, min as
+                      builtin_min, max as builtin_max)
 from functools import wraps
 from typing import TYPE_CHECKING
 
@@ -712,7 +712,7 @@ def vecdot_linalg(x1: array, x2: array, /, *, axis: int = -1, **kwargs) -> array
     if isdtype(x1.dtype, 'integral') or isdtype(x2.dtype, 'integral'):
         if kwargs:
             raise RuntimeError("vecdot kwargs not supported for integral dtypes")
-        ndim = max(x1.ndim, x2.ndim)
+        ndim = builtin_max(x1.ndim, x2.ndim)
         x1_shape = (1,)*(ndim - x1.ndim) + tuple(x1.shape)
         x2_shape = (1,)*(ndim - x2.ndim) + tuple(x2.shape)
         if x1_shape[axis] != x2_shape[axis]:
@@ -733,4 +733,4 @@ def solve(x1: array, x2: array, /, **kwargs) -> array:
 # torch.trace doesn't support the offset argument and doesn't support stacking
 def trace(x: array, /, *, offset: int = 0, dtype: Optional[Dtype] = None) -> array:
     # Use our wrapped sum to make sure it does upcasting correctly
-    return sum(torch.diagonal(x, offset=offset, dim1=-2, dim2=-1), axis=-1, dtype=dtype)
+    return sum(torch.diagonal(x, offset=offset, dim1=-2, dim2=-1), axis=-1, dtype=dtype)