notkshitij/nvcc
1
0
Fork 0
mirror of https://github.com/andreinechaev/nvcc4jupyter.git synced 2026-06-16 12:10:49 +05:30
Code Issues Packages Projects Releases Wiki Activity

Add option to give nvcc extra arguments

Browse Source
This commit is contained in:
Cosmin Ștefan Ciocan
2024-01-23 22:53:47 +00:00
parent 5cd225851b
commit 881c67f5f1
2 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nvcc4jupyter/parsers.py
+1
View File
@@ -19,6 +19,7 @@ def get_parser_cuda() -> argparse.ArgumentParser:
parser.add_argument("-t", "--timeit", action="store_true") parser.add_argument("-t", "--timeit", action="store_true")
parser.add_argument("-p", "--profile", action="store_true") parser.add_argument("-p", "--profile", action="store_true")
parser.add_argument("-a", "--profiler-args", type=str, default="") parser.add_argument("-a", "--profiler-args", type=str, default="")
parser.add_argument("-c", "--compiler-args", type=str, default="")
return parser return parser
nvcc4jupyter/plugin.py
+14 -13
View File
@@ -87,7 +87,10 @@ class NVCCPlugin(Magics):
shutil.rmtree(group_dirpath) shutil.rmtree(group_dirpath)
def _compile( def _compile(
self, group_name: str, executable_fname: str = DEFAULT_EXEC_FNAME self,
group_name: str,
executable_fname: str = DEFAULT_EXEC_FNAME,
compiler_args: str = "",
) -> str: ) -> str:
""" """
Compiles all source files in a given group together with all source Compiles all source files in a given group together with all source
@@ -97,6 +100,7 @@ class NVCCPlugin(Magics):
group_name: The name of the source file group to be compiled. group_name: The name of the source file group to be compiled.
executable_fname: The output executable file name. Defaults to executable_fname: The output executable file name. Defaults to
"cuda_exec.out". "cuda_exec.out".
compiler_args: The optional "nvcc" compiler arguments.
Raises: Raises:
RuntimeError: If the group does not exist or if does not have any RuntimeError: If the group does not exist or if does not have any
@@ -121,18 +125,12 @@ class NVCCPlugin(Magics):
executable_fpath = os.path.join(group_dirpath, executable_fname) executable_fpath = os.path.join(group_dirpath, executable_fname)
args = [ args = ["nvcc"]
"nvcc", args.extend(compiler_args.split())
"-I" + shared_dirpath + "," + group_dirpath, args.append("-I" + shared_dirpath + "," + group_dirpath)
]
args.extend(source_files) args.extend(source_files)
args.extend( args.extend(["-o", executable_fpath, "-Wno-deprecated-gpu-targets"])
[
"-o",
executable_fpath,
"-Wno-deprecated-gpu-targets",
]
)
subprocess.check_output(args, stderr=subprocess.STDOUT) subprocess.check_output(args, stderr=subprocess.STDOUT)
return executable_fpath return executable_fpath
@@ -188,7 +186,10 @@ class NVCCPlugin(Magics):
self, group_name: str, args: argparse.Namespace self, group_name: str, args: argparse.Namespace
) -> str: ) -> str:
try: try:
exec_fpath = self._compile(group_name) exec_fpath = self._compile(
group_name=group_name,
compiler_args=args.compiler_args,
)
output = self._run( output = self._run(
exec_fpath=exec_fpath, exec_fpath=exec_fpath,
timeit=args.timeit, timeit=args.timeit,