Skip to content

GitLab

Unverified Commit 4f8829f9 authored by Christian Sarofeen's avatar Christian Sarofeen Committed by GitHub
Browse files

Fix pre 0.4 builds. 

Fix regex expression to search for libaten
parent 70d2e3b7
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ 4f8829f9
...@@ -106,7 +106,7 @@ cuda_files = find(curdir, lambda file: file.endswith(".cu"), True) ...@@ -106,7 +106,7 @@ cuda_files = find(curdir, lambda file: file.endswith(".cu"), True)
cuda_headers = find(curdir, lambda file: file.endswith(".cuh"), True) cuda_headers = find(curdir, lambda file: file.endswith(".cuh"), True)
headers = find(curdir, lambda file: file.endswith(".h"), True) headers = find(curdir, lambda file: file.endswith(".h"), True)
libaten = list(set(find(torch_dir, re.compile("libaten", re.IGNORECASE).search, True))) libaten = list(set(find(torch_dir, re.compile("libaten.*so", re.IGNORECASE).match, True)))
libaten_names = [os.path.splitext(os.path.basename(entry))[0] for entry in libaten] libaten_names = [os.path.splitext(os.path.basename(entry))[0] for entry in libaten]
for i, entry in enumerate(libaten_names): for i, entry in enumerate(libaten_names):
if entry[:3]=='lib': if entry[:3]=='lib':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment