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:: check_compiler_abi_compatibility
.. autofunction:: verify_ninja_availability
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`
			`.. autofunction:: check_compiler_abi_compatibility`
			`.. autofunction:: verify_ninja_availability`