diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml
index 667250ce86e..87cd7cb7165 100644
--- a/aten/src/ATen/native/native_functions.yaml
+++ b/aten/src/ATen/native/native_functions.yaml
@@ -3,6 +3,9 @@
 - func: checkpoint(Tensor self) -> Tensor
   variants: method
 
+- func: try_checkpoint(Tensor self) -> Tensor
+  variants: method
+
 - func: is_checkpoint(Tensor self) -> bool
   variants: method