Implement aten::det | feat(torchlib) (#1142)

Author: justinchuby

onnxscript/function_libs/torch_lib/ops/core.py

@@ -2378,12 +2378,6 @@ def aten_dense_dim(self: TensorType) -> int:
     raise NotImplementedError()
 
 
-def aten_det(self: TensorType) -> TensorType:
-    """det(Tensor self) -> Tensor"""
-
-    raise NotImplementedError()
-
-
 @torch_op("aten::detach")
 def aten_detach(self: TensorType) -> TensorType:
     """detach(Tensor(a) self) -> Tensor(a)"""

onnxscript/function_libs/torch_lib/ops/linalg.py

@@ -46,10 +46,11 @@ def aten_linalg_cross(self: TensorType, other: TensorType, dim: int = -1) -> Ten
     raise NotImplementedError()
 
 
-def aten_linalg_det(A: TensorType) -> TensorType:
+@torch_op(("aten::linalg_det", "aten::det"))
+def aten_linalg_det(A: TFloat) -> TFloat:
     """linalg_det(Tensor A) -> Tensor"""
 
-    raise NotImplementedError()
+    return op.Det(A)
 
 
 def aten_linalg_diagonal(

onnxscript/tests/function_libs/torch_lib/ops_test_data.py

@@ -821,6 +821,7 @@ def _where_input_wrangler(
     TorchLibOpInfo("isneginf", core_ops.aten_isneginf),
     TorchLibOpInfo("isposinf", core_ops.aten_isposinf),
     TorchLibOpInfo("lift_fresh_copy", core_ops.aten_lift_fresh_copy),
+    TorchLibOpInfo("linalg.det", linalg_ops.aten_linalg_det),
     TorchLibOpInfo(
         "linalg.vector_norm",
         linalg_ops.aten_linalg_vector_norm,