frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Clarify how to get extra link flags when building CUDA/C++ extension (#118743)

Browse Source 
Make it a bit more explicit how one parse linker arguments to the build and point to the superclass documentation.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/118743
Approved by: https://github.com/ezyang
This commit is contained in:
Nikolay Bogoychev
2024-02-01 22:35:20 +00:00
committed by PyTorch MergeBot
parent dbba1d4bf5
commit 46ef73505d
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
torch/utils/cpp_extension.py
View File

@ -931,7 +931,8 @@ def CppExtension(name, sources, *args, **kwargs):
bare minimum (but often sufficient) arguments to build a C++ extension.
All arguments are forwarded to the :class:`setuptools.Extension`
constructor.
constructor. Full list arguments can be found at
https://setuptools.pypa.io/en/latest/userguide/ext_modules.html#extension-api-reference
Example:
>>> # xdoctest: +SKIP
@ -944,7 +945,8 @@ def CppExtension(name, sources, *args, **kwargs):
... CppExtension(
... name='extension',
... sources=['extension.cpp'],
... extra_compile_args=['-g']),
... extra_compile_args=['-g'],
... extra_link_flags=['-Wl,--no-as-needed', '-lm'])
... ],
... cmdclass={
... 'build_ext': BuildExtension
@ -979,7 +981,8 @@ def CUDAExtension(name, sources, *args, **kwargs):
library.
All arguments are forwarded to the :class:`setuptools.Extension`
constructor.
constructor. Full list arguments can be found at
https://setuptools.pypa.io/en/latest/userguide/ext_modules.html#extension-api-reference
Example:
>>> # xdoctest: +SKIP
@ -993,7 +996,8 @@ def CUDAExtension(name, sources, *args, **kwargs):
... name='cuda_extension',
... sources=['extension.cpp', 'extension_kernel.cu'],
... extra_compile_args={'cxx': ['-g'],
... 'nvcc': ['-O2']})
... 'nvcc': ['-O2']},
... extra_link_flags=['-Wl,--no-as-needed', '-lcuda'])
... ],
... cmdclass={
... 'build_ext': BuildExtension