Skip to content

GitLab

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

Actually fix builds for pre 0.4

parent 4f8829f9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ ee117aa8
...@@ -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.*so", re.IGNORECASE).match, True))) libaten = list(set(find(torch_dir, re.compile("libaten.*(so)\Z", 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