Skip to content

Commit aacaecb

Browse files
author
Pavel Sukhachev
committed
Improved darwin support
1 parent bd59115 commit aacaecb

File tree

4 files changed

+9
-7
lines changed

4 files changed

+9
-7
lines changed

Compiler/Compilers/Compiler.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,13 +52,15 @@ def setupPlatform(forPython):
5252
libraryDirs = [os.path.join(bindir, "libs"), bindir]
5353
libraries = ["python%s%s" % sys.version_info[:2]]
5454

55-
elif sys.platform == "linux":
56-
oext, libext, soext = ".o", ".a", ".so"
55+
elif sys.platform in ("linux", "darwin"):
56+
oext, libext, soext = ".o", ".a", (".so" if sys.platform == "linux" else ".dylib")
5757
linkext, debugext = [], []
5858

5959
if forPython:
6060
libraryDirs = [config["LIBDIR"]]
61-
libraries = ["python%s.%sm" % sys.version_info[:2]]
61+
62+
(major, minor), mext = sys.version_info[:2], "m" if sys.platform == "linux" else ""
63+
libraries = ["python%s.%s%s" % (major, minor, mext)]
6264

6365
else:
6466
raise NotImplementedError(sys.platform)

Intel/ThirdParty/finddnnl.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ def findDNNL():
99
libnames += ["/usr/local/lib/%s" % libname for libname in libnames]
1010

1111
elif sys.platform == "darwin":
12-
libnames = ["libdnnl.%s.dylib" % v for v in versions]
12+
libnames = ["/usr/local/lib/libdnnl.%s.dylib" % v for v in versions]
1313

1414
elif sys.platform == "win32":
1515
libpaths = [
@@ -32,6 +32,6 @@ def findDNNL():
3232
pass
3333

3434
else:
35-
return clib
35+
return libname, clib
3636

3737
raise OSError("dnnl library not found (searched for following version(s): %s)" % versions)

Intel/ThirdParty/libdnnl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from PuzzleLib.Intel.ThirdParty.finddnnl import findDNNL
33

44

5-
_libdnnl = findDNNL()
5+
_, _libdnnl = findDNNL()
66

77

88
class dnnlError(Exception):

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,7 @@ def installIntelPackage(src, dst, options):
239239
from PuzzleLib.Intel.ThirdParty.finddnnl import findDNNL
240240

241241
print("\nChecking dnnl installation ...")
242-
lib = findDNNL()
242+
lib, _ = findDNNL();
243243

244244
if os.path.commonpath([dst, lib]) == dst:
245245
data = {pathToPackageName(dst): [lib]}

0 commit comments

Comments
 (0)