torch.compiler.assume_constant_result¶
- torch.compiler.assume_constant_result(fn)[source]¶
This function is used to mark a function fn as having a constant result. This allows the compiler to optimize away your function Returns The same function fn
- Parameters
fn – The function to be marked as having a constant result.
Warning
assume_constant_result can if invalid cause safety and soundness issues,
torch.compile()
will not attempt to validate whether the constant assumption is true or not