search

pytorch / tutorials

PyTorch tutorials.
https://pytorch.org/tutorials/
BSD 3-Clause "New" or "Revised" License
8.22k stars 4.07k forks source link

[docs] Mention inline CUDA extensions in tutorial #2491

Open vadimkantorov opened 6 years ago

vadimkantorov commented 6 years ago

It would be nice to mention load_inline (https://github.com/pytorch/pytorch/pull/7059) in the tutorial https://pytorch.org/tutorials/advanced/cpp_extension.html

A practical use-case for an example may be a bilateral filter from https://github.com/NVIDIA/FastPhotoStyle/blob/master/smooth_filter.py (currentlly uses CuPY and pynvrtc).

cc @jlin27 @mruberry

vadimkantorov commented 3 years ago

At least load_inline now appears in docs

vadimkantorov commented 2 years ago

Although, caching of compiled binaries was recently still broken: https://github.com/pytorch/pytorch/issues/68066