"website/git@developer.sourcefind.cn:OpenDAS/pytorch3d.git" did not exist on "3c9f06581ac24194c13f4e3edb5f55301fe78802"
Commit c999e986 authored by rusty1s's avatar rusty1s
Browse files

cleanup

parent 1bd23ac8
import os import os
import os.path as osp import os.path as osp
import sys
import glob import glob
from setuptools import setup, find_packages from setuptools import setup, find_packages
...@@ -22,30 +21,14 @@ def get_extensions(): ...@@ -22,30 +21,14 @@ def get_extensions():
define_macros = [] define_macros = []
extra_compile_args = {'cxx': [], 'nvcc': []} extra_compile_args = {'cxx': [], 'nvcc': []}
# flags = os.getenv('EXTRA_COMPILE_ARGS', '')
# extra_compile_args['cxx'] += [] if flags == '' else flags.split(' ')
# extra_compile_args['nvcc'] += [] if flags == '' else flags.split(' ')
libraries = []
# Windows users: Make sure that your VS path is included, i.e.:
# extra_compile_args['cxx'] += ['-I{VISUAL_STUDIO_DIR}\\include']
# extra_compile_args['nvcc'] += ['-I{VISUAL_STUDIO_DIR}\\include']
if WITH_CUDA: if WITH_CUDA:
Extension = CUDAExtension Extension = CUDAExtension
define_macros += [('WITH_CUDA', None)] define_macros += [('WITH_CUDA', None)]
# extra_compile_args['cxx'] += ['-O0']
nvcc_flags = os.getenv('NVCC_FLAGS', '') nvcc_flags = os.getenv('NVCC_FLAGS', '')
nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ') nvcc_flags = [] if nvcc_flags == '' else nvcc_flags.split(' ')
nvcc_flags += ['-arch=sm_35', '--expt-relaxed-constexpr'] nvcc_flags += ['-arch=sm_35', '--expt-relaxed-constexpr']
extra_compile_args['nvcc'] += nvcc_flags extra_compile_args['nvcc'] += nvcc_flags
if sys.platform == 'win32':
# extra_compile_args['cxx'] += ['/MP']
# libraries = ['ATen', '_C']
pass
extensions_dir = osp.join(osp.dirname(osp.abspath(__file__)), 'csrc') extensions_dir = osp.join(osp.dirname(osp.abspath(__file__)), 'csrc')
main_files = glob.glob(osp.join(extensions_dir, '*.cpp')) main_files = glob.glob(osp.join(extensions_dir, '*.cpp'))
extensions = [] extensions = []
...@@ -62,7 +45,6 @@ def get_extensions(): ...@@ -62,7 +45,6 @@ def get_extensions():
include_dirs=[extensions_dir], include_dirs=[extensions_dir],
define_macros=define_macros, define_macros=define_macros,
extra_compile_args=extra_compile_args, extra_compile_args=extra_compile_args,
libraries=libraries,
) )
extensions += [extension] extensions += [extension]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment