Skip to content

Commit 91540a8

Browse files
authored
Add new Python and C++/CUDA Custom Op tutorials (#2888)
* Add new Python and C++/CUDA Custom Op tutorials I want to land this before PyTorch 2.4 (so we can link to these in PyTorch's nightly documentation) and then have a follow-up PR for 2.4 that actually runs the scripts (so that they can generate outputs). pytorch/pytorch#127443 to remind myself of the above. NB: These two tutorials replace all of the existing custom ops (and cpp extensions) tutorials: - advanced/cpp_extension - advanced/torch_script_custom_ops - advanced/torch_script_custom_classes - advanced/dispatcher In a follow-up PR we will add warnings to all of those tutorials stating that they are deprecated but we will preserve the text just in case people still need them (e.g. if they are not using PyTorch 2.4). Test Plan: - I tested these locally. * Fix cpp custom ops tutorial per review * update * update index.rst * Update * update * update
1 parent 63f987d commit 91540a8

File tree

6 files changed

+771
-3
lines changed

6 files changed

+771
-3
lines changed

Diff for: .jenkins/validate_tutorials_built.py

+1
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
"intermediate_source/fx_conv_bn_fuser",
3030
"intermediate_source/_torch_export_nightly_tutorial", # does not work on release
3131
"advanced_source/super_resolution_with_onnxruntime",
32+
"advanced_source/python_custom_ops", # https://github.com/pytorch/pytorch/issues/127443
3233
"advanced_source/ddp_pipeline", # requires 4 gpus
3334
"advanced_source/usb_semisup_learn", # fails with CUDA OOM error, should try on a different worker
3435
"prototype_source/fx_graph_mode_ptq_dynamic",

0 commit comments

Comments
 (0)