From a876d617ae7f81d360a75bc94d91eb1fb1ba550b Mon Sep 17 00:00:00 2001 From: Mark Saroufim Date: Sun, 23 Jun 2024 19:20:52 -0700 Subject: [PATCH] Add link to new custom op tutorial --- torchao/csrc/README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torchao/csrc/README.md b/torchao/csrc/README.md index b08a0a06ba..b6fced45b4 100644 --- a/torchao/csrc/README.md +++ b/torchao/csrc/README.md @@ -6,6 +6,8 @@ This folder is an example of how to integrate your own custom kernels into ao su The goal is that you can focus on just writing your custom CUDA or C++ kernel and we can package it up so it's available via `torchao.ops.your_custom_kernel`. +To learn more about custom ops in PyTorch you can refer to the [PyTorch Custom Operators Landing Page](https://pytorch.org/tutorials/advanced/custom_ops_landing_page.html) + ## How to add your own kernel in ao