-
Notifications
You must be signed in to change notification settings - Fork 4
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
13d86b8
commit 5206ee6
Showing
1 changed file
with
15 additions
and
15 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -48,11 +48,11 @@ | |
import distutils.sysconfig | ||
import distutils.ccompiler | ||
|
||
sources_list = ['sklearn_freiburg_extension/computation/interface/nearestNeighbors_PythonInterface.cpp', 'sklearn_freiburg_extension/computation/nearestNeighbors.cpp', | ||
'sklearn_freiburg_extension/computation/inverseIndex.cpp', 'sklearn_freiburg_extension/computation/inverseIndexStorageUnorderedMap.cpp'] | ||
depends_list = ['sklearn_freiburg_extension/computation/nearestNeighbors.h', 'sklearn_freiburg_extension/computation/inverseIndex.h', 'sklearn_freiburg_extension/computation/kSizeSortedMap.h', | ||
'sklearn_freiburg_extension/computation/typeDefinitions.h', 'sklearn_freiburg_extension/computation/parsePythonToCpp.h', 'sklearn_freiburg_extension/computation/sparseMatrix.h', | ||
'sklearn_freiburg_extension/computation/inverseIndexStorage.h', 'sklearn_freiburg_extension/computation/inverseIndexStorageUnorderedMap.h','sklearn_freiburg_extension/computation/hash.h'] | ||
sources_list = ['sparse_neighbors_search/computation/interface/nearestNeighbors_PythonInterface.cpp', 'sparse_neighbors_search/computation/nearestNeighbors.cpp', | ||
'sparse_neighbors_search/computation/inverseIndex.cpp', 'sparse_neighbors_search/computation/inverseIndexStorageUnorderedMap.cpp'] | ||
depends_list = ['sparse_neighbors_search/computation/nearestNeighbors.h', 'sparse_neighbors_search/computation/inverseIndex.h', 'sparse_neighbors_search/computation/kSizeSortedMap.h', | ||
'sparse_neighbors_search/computation/typeDefinitions.h', 'sparse_neighbors_search/computation/parsePythonToCpp.h', 'sparse_neighbors_search/computation/sparseMatrix.h', | ||
'sparse_neighbors_search/computation/inverseIndexStorage.h', 'sparse_neighbors_search/computation/inverseIndexStorageUnorderedMap.h','sparse_neighbors_search/computation/hash.h'] | ||
openmp = True | ||
if "--openmp" in sys.argv: | ||
module1 = Extension('_nearestNeighbors', sources = sources_list, depends = depends_list, | ||
|
@@ -231,7 +231,7 @@ def build_extensions(self): | |
|
||
if (locate_cuda() == None or no_cuda): | ||
print("No Cuda found or no cuda forced. Installation without GPU support.") | ||
setup (name = 'sklearn_freiburg_extension', | ||
setup (name = 'sparse_neighbors_search', | ||
author = 'Joachim Wolff', | ||
author_email = '[email protected]', | ||
url='https://github.com/joachimwolff/minHashNearestNeighbors', | ||
|
@@ -244,18 +244,18 @@ def build_extensions(self): | |
"scikit-learn >= 0.16.0",], | ||
ext_modules = [module1], | ||
# cmdclass={'build_ext': custom_build_ext_gcc}, | ||
packages=['sklearn_freiburg_extension', | ||
'sklearn_freiburg_extension.neighbors', | ||
'sklearn_freiburg_extension.cluster', | ||
packages=['sparse_neighbors_search', | ||
'sparse_neighbors_search.neighbors', | ||
'sparse_neighbors_search.cluster', | ||
# 'bioinf.computation', | ||
], | ||
platforms = "Linux, Mac OS X", | ||
version = '0.1' | ||
) | ||
else: | ||
print ("CUDA found on system. Installing MinHash with CUDA-Support.") | ||
sources_list.extend(['sklearn_freiburg_extension/computation/kernel.cu', 'sklearn_freiburg_extension/computation/inverseIndexCuda.cu', 'sklearn_freiburg_extension/computation/nearestNeighborsCuda.cu']) | ||
depends_list.extend(['sklearn_freiburg_extension/computation/typeDefinitionsCuda.h', 'sklearn_freiburg_extension/computation/kernel.h', 'sklearn_freiburg_extension/computation/inverseIndexCuda.h', 'sklearn_freiburg_extension/computation/nearestNeighborsCuda.h', ]) | ||
sources_list.extend(['sparse_neighbors_search/computation/kernel.cu', 'sparse_neighbors_search/computation/inverseIndexCuda.cu', 'sparse_neighbors_search/computation/nearestNeighborsCuda.cu']) | ||
depends_list.extend(['sparse_neighbors_search/computation/typeDefinitionsCuda.h', 'sparse_neighbors_search/computation/kernel.h', 'sparse_neighbors_search/computation/inverseIndexCuda.h', 'sparse_neighbors_search/computation/nearestNeighborsCuda.h', ]) | ||
# Extension('_nearestNeighbors', sources = sources_list, depends = depends_list, | ||
# define_macros=[('OPENMP', None)], extra_link_args = ["-lm", "-lrt","-lgomp"], | ||
# extra_compile_args=["-fopenmp", "-O3", "-std=c++11"]) | ||
|
@@ -298,7 +298,7 @@ def build_extensions(self): | |
platforms = "Linux, Mac OS X" | ||
) | ||
|
||
setup(name='sklearn_freiburg_extension', | ||
setup(name='sparse_neighbors_search', | ||
# random metadata. there's more you can supploy | ||
author='Joachim Wolff', | ||
ext_modules = [ext], | ||
|
@@ -318,9 +318,9 @@ def build_extensions(self): | |
"scipy >= 0.14.0", | ||
"scikit-learn >= 0.16.0",], | ||
# ext_modules = [module1], | ||
packages=['sklearn_freiburg_extension', | ||
'sklearn_freiburg_extension.neighbors', | ||
'sklearn_freiburg_extension.cluster', | ||
packages=['sparse_neighbors_search', | ||
'sparse_neighbors_search.neighbors', | ||
'sparse_neighbors_search.cluster', | ||
], | ||
platforms = "Linux, Mac OS X", | ||
version = '0.1') |