Skip to content

Commit

Permalink
fix OMP detection
Browse files Browse the repository at this point in the history
  • Loading branch information
david-cortes committed Mar 19, 2023
1 parent d888d80 commit a1ec2a2
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ def add_clang_fp_reassociate(self):
def add_openmp_linkage(self):
arg_omp1 = "-fopenmp"
arg_omp2 = "-fopenmp=libomp"
args_omp2 = ["-fopenmp=libomp", "-lomp"]
args_omp3 = ["-fopenmp=libomp", "-lomp"]
arg_omp4 = "-qopenmp"
arg_omp5 = "-xopenmp"
is_apple = sys.platform[:3].lower() == "dar"
Expand Down Expand Up @@ -231,7 +231,7 @@ def add_openmp_linkage(self):
for e in self.extensions:
e.extra_compile_args += ["-fopenmp=libomp"]
e.extra_link_args += ["-fopenmp"]
elif self.test_supports_compile_arg(arg_omp3, with_omp=True):
elif self.test_supports_compile_arg(args_omp3, with_omp=True):
for e in self.extensions:
e.extra_compile_args += ["-fopenmp=libomp"]
e.extra_link_args += ["-fopenmp", "-lomp"]
Expand Down Expand Up @@ -345,10 +345,9 @@ def test_supports_clang_reassociate(self):
setup(
name = "cmfrec",
packages = ["cmfrec"],
version = '3.5.1-1',
version = '3.5.1-2',
description = 'Collective matrix factorization',
author = 'David Cortes',
author_email = '[email protected]',
url = 'https://github.com/david-cortes/cmfrec',
keywords = ['collaborative filtering', 'collective matrix factorization',
'relational learning'],
Expand Down

0 comments on commit a1ec2a2

Please sign in to comment.