Safran/PVD
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix: change hardcoded compiler path

Browse source
This commit is contained in:
Laurent FAINSIN 2023-04-06 09:31:16 +02:00
parent 0ff2e1d30f
commit 38da17d7d9
6 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
README.md
View file

@ -19,8 +19,8 @@ mamba env create --file env.yml
Install PyTorchEMD by Install PyTorchEMD by
```bash ```bash
cd metrics/PyTorchEMD cd metrics/PyTorchEMD
module load gcc module load gcc/11.2.0
module load mpfr module load mpfr/4.0.2
conda activate PVD conda activate PVD
python setup.py install python setup.py install
cp build/**/emd_cuda.cpython-*-x86_64-linux-gnu.so . cp build/**/emd_cuda.cpython-*-x86_64-linux-gnu.so .

2
metrics/ChamferDistancePytorch/chamfer2D/setup.py
View file

@ -10,7 +10,7 @@ setup(
]), ]),
], ],
extra_cuda_cflags=['--compiler-bindir=/usr/bin/gcc-8'], extra_cuda_cflags=['--compiler-bindir=/softs/gcc/11.2.0/bin/gcc'],
cmdclass={ cmdclass={
'build_ext': BuildExtension 'build_ext': BuildExtension
}) })

2
metrics/ChamferDistancePytorch/chamfer3D/dist_chamfer_3D.py
View file

@ -15,7 +15,7 @@ if not chamfer_found:
"/".join(os.path.abspath(__file__).split('/')[:-1] + ["chamfer3D.cu"]), "/".join(os.path.abspath(__file__).split('/')[:-1] + ["chamfer3D.cu"]),
], ],
extra_cuda_cflags=['--compiler-bindir=/usr/bin/gcc-8'],) extra_cuda_cflags=['--compiler-bindir=/softs/gcc/11.2.0/bin/gcc'],)
print("Loaded JIT 3D CUDA chamfer distance") print("Loaded JIT 3D CUDA chamfer distance")
else: else:

2
metrics/ChamferDistancePytorch/chamfer3D/setup.py
View file

@ -10,7 +10,7 @@ setup(
]), ]),
], ],
extra_cuda_cflags=['--compiler-bindir=/usr/bin/gcc-8'], extra_cuda_cflags=['--compiler-bindir=/softs/gcc/11.2.0/bin/gcc'],
cmdclass={ cmdclass={
'build_ext': BuildExtension 'build_ext': BuildExtension
}) })

2
metrics/ChamferDistancePytorch/chamfer5D/setup.py
View file

@ -10,7 +10,7 @@ setup(
]), ]),
], ],
extra_cuda_cflags=['--compiler-bindir=/usr/bin/gcc-8'], extra_cuda_cflags=['--compiler-bindir=/softs/gcc/11.2.0/bin/gcc'],
cmdclass={ cmdclass={
'build_ext': BuildExtension 'build_ext': BuildExtension
}) })

2
modules/functional/backend.py
View file

@ -5,7 +5,7 @@ from torch.utils.cpp_extension import load
_src_path = os.path.dirname(os.path.abspath(__file__)) _src_path = os.path.dirname(os.path.abspath(__file__))
_backend = load(name='_pvcnn_backend', _backend = load(name='_pvcnn_backend',
extra_cflags=['-O3', '-std=c++17'], extra_cflags=['-O3', '-std=c++17'],
extra_cuda_cflags=['--compiler-bindir=/usr/bin/gcc-8'], extra_cuda_cflags=['--compiler-bindir=/softs/gcc/11.2.0/bin/gcc'],
sources=[os.path.join(_src_path,'src', f) for f in [ sources=[os.path.join(_src_path,'src', f) for f in [
'ball_query/ball_query.cpp', 'ball_query/ball_query.cpp',
'ball_query/ball_query.cu', 'ball_query/ball_query.cu',