"pcdet/datasets/processor/data_processor.py" did not exist on "760a9d2cb18e1ba8676ad698cf5e8dc9ff6cc99e"
Unverified Commit 4b900e09 authored by Ben Graham's avatar Ben Graham Committed by GitHub
Browse files

Merge pull request #243 from Yannicked/c++17

Support for PyTorch 2.1
parents 8b261bf2 0ceb5da9
...@@ -14,7 +14,7 @@ if torch.cuda.is_available(): ...@@ -14,7 +14,7 @@ if torch.cuda.is_available():
this_dir = os.path.dirname(os.path.realpath(__file__)) this_dir = os.path.dirname(os.path.realpath(__file__))
torch_dir = os.path.dirname(torch.__file__) torch_dir = os.path.dirname(torch.__file__)
conda_include_dir = '/'.join(torch_dir.split('/')[:-4]) + '/include' conda_include_dir = '/'.join(torch_dir.split('/')[:-4]) + '/include'
extra = {'cxx': ['-std=c++14', '-fopenmp','-O3'], 'nvcc': ['-std=c++14', '-Xcompiler', '-fopenmp', '-O3']} extra = {'cxx': ['-std=c++17', '-fopenmp','-O3'], 'nvcc': ['-std=c++17', '-Xcompiler', '-fopenmp', '-O3']}
setup( setup(
name='sparseconvnet', name='sparseconvnet',
......
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