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

Nvfuser python API import fix (#94036)

Browse Source 
1. Having nvfuser python API import working with both devel and upstream;
2. Add environment variable to allow custom nvfuser code base to be built with upstream pytorch core.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94036
Approved by: https://github.com/malfet, https://github.com/davidberard98
This commit is contained in:
jjsjann123
2023-02-16 20:10:40 +00:00
committed by PyTorch MergeBot
parent 7aaebe00ee
commit 21eb7f70f1
6 changed files with 56 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@ -175,6 +175,9 @@
# NCCL_INCLUDE_DIR
# specify where nccl is installed
#
# NVFUSER_SOURCE_DIR
# specify nvfuser root directory
#
# NVTOOLSEXT_PATH (Windows only)
# specify where nvtoolsext is installed
#