From 3704bf4ee8f7db5d5f4411a707a9d6a93bbf2f58 Mon Sep 17 00:00:00 2001 From: angelayi Date: Mon, 9 Oct 2023 23:40:35 +0000 Subject: [PATCH] [export] Update custom ops docs (#110492) Updating the doc links in the custom ops documentation in export Pull Request resolved: https://github.com/pytorch/pytorch/pull/110492 Approved by: https://github.com/avikchaudhuri --- docs/source/export.rst | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/docs/source/export.rst b/docs/source/export.rst index 723f4a6ca05..465b6e8ae5f 100644 --- a/docs/source/export.rst +++ b/docs/source/export.rst @@ -521,9 +521,13 @@ When tracing, a META implementation (or "meta kernel") is required for all operators. This is used to reason about the input/output shapes for this operator. -Note that the official API for registering custom meta kernels for custom ops is -currently undergoing development. While the final API is being refined, you can -refer to the documentation `here `_. +To register a meta kernel for a C++ Custom Operator, please refer to +`this documentation `__. + +The official API for registering custom meta kernels for custom ops implemented +in python is currently undergoing development. While the final API is being +refined, you can refer to the documentation +`here `_. In the unfortunate case where your model uses an ATen operator that is does not have a meta kernel implementation yet, please file an issue.