Added Flexiblas to the list of BLAS/LAPACK libs

NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

https://nvidia.github.io/MinkowskiEngine

Other

2.43k stars 360 forks source link

Added Flexiblas to the list of BLAS/LAPACK libs #420

Closed ccoulombe closed 2 years ago

ccoulombe commented 2 years ago

Flexiblas : https://www.mpi-magdeburg.mpg.de/projects/flexiblas

Can be built with:

TORCH_CUDA_ARCH_LIST="6.0;7.0;7.5;8.0" python setup.py bdist_wheel --blas_include_dirs=$EBROOTFLEXIBLAS/include/flexiblas --blas_library_dirs=$EBROOTFLEXIBLAS/lib --blas=flexiblas --force_cuda --cuda_home=$CUDA_HOME