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