Commit 1cea1005 authored by Christian Sarofeen's avatar Christian Sarofeen
Browse files

Don't compile cuda code if running clean.

parent 3a08d827
import re import re
import sys
import os import os
import shutil import shutil
import inspect import inspect
...@@ -141,18 +142,20 @@ def CompileCudaFiles(): ...@@ -141,18 +142,20 @@ def CompileCudaFiles():
for object_file in object_files: for object_file in object_files:
extra_link_args.append(object_file) extra_link_args.append(object_file)
if 'clean' not in sys.argv:
print()
print("Arguments used to build CUDA extension:") print()
print("extra_compile_args :", extra_compile_args) print("Arguments used to build CUDA extension:")
print("include_dirs: ", include_dirs) print("extra_compile_args :", extra_compile_args)
print("extra_link_args: ", extra_link_args) print("include_dirs: ", include_dirs)
print("library_dirs: ", library_dirs) print("extra_link_args: ", extra_link_args)
print("libraries: ", main_libraries) print("library_dirs: ", library_dirs)
print() print("libraries: ", main_libraries)
CompileCudaFiles() print()
CompileCudaFiles()
print("Building CUDA extension.")
print("Building CUDA extension.")
cuda_ext = Extension('apex._C', cuda_ext = Extension('apex._C',
[os.path.join('csrc', 'Module.cpp')], [os.path.join('csrc', 'Module.cpp')],
extra_compile_args = extra_compile_args, extra_compile_args = extra_compile_args,
...@@ -161,9 +164,11 @@ cuda_ext = Extension('apex._C', ...@@ -161,9 +164,11 @@ cuda_ext = Extension('apex._C',
library_dirs=library_dirs, library_dirs=library_dirs,
runtime_library_dirs = library_dirs, runtime_library_dirs = library_dirs,
libraries=main_libraries libraries=main_libraries
) )
print("Building module.") if 'clean' not in sys.argv:
print("Building module.")
setup( setup(
name='apex', version='0.1', name='apex', version='0.1',
cmdclass={ cmdclass={
......
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