diff --git a/torch/compiler/__init__.py b/torch/compiler/__init__.py index 038e25d8fd3..9aa06211b00 100644 --- a/torch/compiler/__init__.py +++ b/torch/compiler/__init__.py @@ -209,7 +209,7 @@ def list_backends(exclude_tags=("debug", "experimental")) -> list[str]: def assume_constant_result(fn): """ This function is used to mark a function `fn` as having a constant result. - This allows the compiler to optimize away your function + This allows the compiler to optimize away your function. Returns The same function `fn` Args: @@ -227,8 +227,8 @@ def assume_constant_result(fn): def disable(fn=None, recursive=True): """ - This function provides a decorator to disable compilation on a function - It also provides the option of recursively disabling called functions + This function provides a decorator to disable compilation on a function. + It also provides the option of recursively disabling called functions. Args: fn (optional): The function to disable