423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-20 06:10:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

pytorch 2.1, drop mac builds

Browse Source 
Summary: Build updates for PyTorch 2.1

Reviewed By: MichaelRamamonjisoa

Differential Revision: D50345762

fbshipit-source-id: 89bf4edf1c21566aa86a3abca9b4df7c4d1d17a2
This commit is contained in:
Jeremy Reizenstein
2023-10-17 06:04:06 -07:00
committed by Facebook GitHub Bot
parent eaf0709d6a
commit 28f914bf3b
6 changed files with 67 additions and 266 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@@ -49,7 +49,7 @@ def get_extensions():
source_cuda = glob.glob(os.path.join(extensions_dir, "**", "*.cu"), recursive=True)
extension = CppExtension
extra_compile_args = {"cxx": ["-std=c++14"]}
extra_compile_args = {"cxx": ["-std=c++17"]}
define_macros = []
include_dirs = [extensions_dir]
@@ -74,7 +74,7 @@ def get_extensions():
"-D__CUDA_NO_HALF2_OPERATORS__",
]
if os.name != "nt":
nvcc_args.append("-std=c++14")
nvcc_args.append("-std=c++17")
if cub_home is None:
prefix = os.environ.get("CONDA_PREFIX", None)
if prefix is not None and os.path.isdir(prefix + "/include/cub"):