Add depreciated parameters

rdspring1 · Apr 27, 2024 · 5db4698 · 5db4698
1 parent 1222e26
commit 5db4698
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/thunder/torch/__init__.py b/thunder/torch/__init__.py
@@ -3296,10 +3296,17 @@ def cross_entropy(
     a: TensorProxy,
     target: TensorProxy,
     weight: None | TensorProxy = None,
+    size_average: None | Any = None,
     ignore_index: int = None,
+    reduce: None | Any = None,
     reduction: str = "mean",
     label_smoothing: float = 0.0,
 ) -> TensorProxy:
+    utils.check(
+        size_average is None and reduce is None,
+        lambda: f"Deprecated size_average={size_average} and reduce={reduce} is not supported!",
+    )
+
     # Resolve ignore_index if it is not specified by user.
     if ignore_index is None:
         ignore_index = -1