torch.compiler.keep_tensor_guards_unsafe

torch.compiler.keep_tensor_guards_unsafe(guard_entries, keep_parameters=False) [source]

A common function to keep tensor guards on all tensors. This is unsafe to use by default. But if you don’t expect any changes in the model code, you can just keep the tensor guards.

>> opt_mod = torch.compile( >> mod, >> options={“guard_filter_fn”: torch.compiler.keep_tensor_guards}, >> )