pytorch/docs/source/cpp_extension.rst

torch.utils.cpp_extension
=========================

.. currentmodule:: torch.utils.cpp_extension
.. autofunction:: CppExtension
.. autofunction:: CUDAExtension
.. autofunction:: BuildExtension
.. autofunction:: load
.. autofunction:: load_inline
.. autofunction:: include_paths
.. autofunction:: get_compiler_abi_compatibility_and_version
.. autofunction:: verify_ninja_availability
.. autofunction:: is_ninja_available
Enable documentation for C++ extensions on the website (#5597) 2018-03-07 13:07:26 +00:00			`torch.utils.cpp_extension`
			`=========================`

			`.. currentmodule:: torch.utils.cpp_extension`
			`.. autofunction:: CppExtension`
			`.. autofunction:: CUDAExtension`
			`.. autofunction:: BuildExtension`
			`.. autofunction:: load`
Inline JIT C++ Extensions (#7059) Adds ability to JIT compile C++ extensions from strings >>> from torch.utils.cpp_extension import load_inline >>> source = ''' at::Tensor sin_add(at::Tensor x, at::Tensor y) { return x.sin() + y.sin(); } ''' >>> module = load_inline(name='inline_extension', cpp_sources=source, functions='sin_add') Fixes #7012 * Inline JIT C++ Extensions * jit_compile_sources -> jit_compile * Split up test into CUDA and non-CUDA parts * Documentation fixes * Implement prologue and epilogue generation * Remove extra newline * Only create the CUDA source file when cuda_sources is passed 2018-04-30 15:48:44 +00:00			`.. autofunction:: load_inline`
Enable documentation for C++ extensions on the website (#5597) 2018-03-07 13:07:26 +00:00			`.. autofunction:: include_paths`
Check clang++/g++ version when compiling CUDA extensions (#63230) Summary: See https://github.com/pytorch/pytorch/issues/55267 Pull Request resolved: https://github.com/pytorch/pytorch/pull/63230 Reviewed By: soulitzer Differential Revision: D34159119 Pulled By: malfet fbshipit-source-id: 6eef7582388bf6a42dcc1d82b6e4b1f40f418dd7 (cherry picked from commit 2056d0a0be7951602de22f8d3b4efc28dd71b6c2) 2022-02-24 07:33:33 +00:00			`.. autofunction:: get_compiler_abi_compatibility_and_version`
Enable documentation for C++ extensions on the website (#5597) 2018-03-07 13:07:26 +00:00			`.. autofunction:: verify_ninja_availability`
correct some cpp extension code usages and documents (#39766) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39766 Test Plan: Imported from OSS Differential Revision: D21967284 Pulled By: glaringlee fbshipit-source-id: 8597916bee247cb5f8c82ed8297119d2f3a72170 2020-06-10 15:28:48 +00:00			`.. autofunction:: is_ninja_available`