Nikita Shulga 2025-01-22 19:56:31 +00:00 committed by PyTorch MergeBot
parent 3917053f63
commit 95ff9f0340

View file

@ -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