mirror of
https://github.com/pfloos/quack
synced 2024-12-22 12:23:42 +01:00
update make_ninja file for cuda objects
This commit is contained in:
parent
3009ffe8f7
commit
a10e4c8c23
@ -3,13 +3,13 @@ import os
|
||||
import sys
|
||||
import subprocess
|
||||
|
||||
|
||||
|
||||
DEBUG=False
|
||||
try:
|
||||
DEBUG = sys.argv[1] == "debug"
|
||||
except:
|
||||
pass
|
||||
import argparse
|
||||
parser = argparse.ArgumentParser(description='This script generate the compilation files for QuAcK.')
|
||||
parser.add_argument('-d', '--debug', action='store_true', help='Debug mode. Default is false.')
|
||||
parser.add_argument('-u', '--use-gpu', action='store_true', help='Use GPU. Default is false.')
|
||||
args = parser.parse_args()
|
||||
DEBUG = args.debug
|
||||
USE_GPU = args.use_gpu
|
||||
|
||||
|
||||
if "QUACK_ROOT" not in os.environ:
|
||||
@ -120,6 +120,7 @@ IDIR=$QUACK_ROOT/include
|
||||
LDIR=$QUACK_ROOT/lib
|
||||
BDIR=$QUACK_ROOT/bin
|
||||
SDIR=$QUACK_ROOT/src
|
||||
CUDA_DIR=$QUACK_ROOT/src/cuda/build
|
||||
|
||||
LIBXC_VERSION=5.0.0
|
||||
|
||||
@ -248,6 +249,9 @@ rule build_lib
|
||||
sources = [ "$SDIR/{0}/{1}".format(exe_dir,x) for x in os.listdir(exe_dir) ]
|
||||
sources = filter(lambda x: x.endswith(".f") or x.endswith(".f90"), sources)
|
||||
sources = " ".join(sources)
|
||||
if USE_GPU:
|
||||
f.write("build $BDIR/{0}: build_exe $CUDA_DIR/cuda.a {1} {2}\n".format(exe_dir,libs,sources))
|
||||
else:
|
||||
f.write("build $BDIR/{0}: build_exe {1} {2}\n".format(exe_dir,libs,sources))
|
||||
f.write(" dir = {0} \n".format(exe_dir) )
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user