From 3b9c70571b0a6542e0fa0324cabc06c6e70dc4f8 Mon Sep 17 00:00:00 2001 From: Raul Puri Date: Wed, 6 Jun 2018 07:17:38 -0700 Subject: [PATCH 1/2] libaten search is now libaten match --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 0cb39c8..b69f112 100644 --- a/setup.py +++ b/setup.py @@ -68,7 +68,7 @@ def findcuda(): cuda_headers = find(curdir, lambda file: file.endswith(".cuh"), True) headers = find(curdir, lambda file: file.endswith(".h"), True) -libaten = find(torch_dir, re.compile("libaten", re.IGNORECASE).search, False) +libaten = list(set(find(torch_dir, re.compile("libaten.*(so)\Z", re.IGNORECASE).match, True))) aten_h = find(torch_dir, re.compile("aten.h", re.IGNORECASE).search, False) include_dirs = [os.path.dirname(os.path.dirname(aten_h))] From c6f537cd57fd324c58183c32c4adfa34d8960feb Mon Sep 17 00:00:00 2001 From: Raul Puri Date: Wed, 6 Jun 2018 08:22:28 -0700 Subject: [PATCH 2/2] get libaten path from list --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index b69f112..a034a38 100644 --- a/setup.py +++ b/setup.py @@ -81,7 +81,7 @@ def findcuda(): assert libaten, "Could not find PyTorch's libATen." assert aten_h, "Could not find PyTorch's ATen header." -library_dirs.append(os.path.dirname(libaten)) +library_dirs.append(os.path.dirname(libaten[0])) #create some places to collect important things object_files = []