mirror of
https://github.com/andreinechaev/nvcc4jupyter.git
synced 2026-06-14 11:10:49 +05:30
print out multiline output
This commit is contained in:
@@ -4,8 +4,9 @@ import tempfile
|
||||
import uuid
|
||||
|
||||
from IPython.core.magic import Magics, cell_magic, magics_class
|
||||
from common import helper
|
||||
|
||||
compiler = '/usr/local/cuda/bin/nvcc -Wno-deprecated-gpu-targets'
|
||||
compiler = '/usr/local/cuda/bin/nvcc'
|
||||
ext = '.cu'
|
||||
|
||||
|
||||
@@ -14,21 +15,26 @@ class NVCCPlugin(Magics):
|
||||
|
||||
def __init__(self, shell):
|
||||
super(NVCCPlugin, self).__init__(shell)
|
||||
from common import helper
|
||||
|
||||
self.argparser = helper.get_argparser()
|
||||
|
||||
@staticmethod
|
||||
def compile(file_path):
|
||||
subprocess.check_output([compiler, file_path + ext, "-o", file_path + ".out"], stderr=subprocess.STDOUT)
|
||||
subprocess.check_output(
|
||||
[compiler, file_path + ext, "-o", file_path + ".out", '-Wno-deprecated-gpu-targets'], stderr=subprocess.STDOUT)
|
||||
|
||||
def run(self, file_path, timeit=False):
|
||||
if timeit:
|
||||
stmt = f"subprocess.check_output(['{file_path}.out'], stderr=subprocess.STDOUT)"
|
||||
output = self.shell.run_cell_magic(magic_name="timeit", line="-q -o import subprocess", cell=stmt)
|
||||
output = self.shell.run_cell_magic(
|
||||
magic_name="timeit", line="-q -o import subprocess", cell=stmt)
|
||||
else:
|
||||
output = subprocess.check_output([file_path + ".out"], stderr=subprocess.STDOUT)
|
||||
output = subprocess.check_output(
|
||||
[file_path + ".out"], stderr=subprocess.STDOUT)
|
||||
output = output.decode('utf8')
|
||||
return output
|
||||
|
||||
helper.print_out(output)
|
||||
return None
|
||||
|
||||
@cell_magic
|
||||
def cu(self, line, cell):
|
||||
@@ -46,6 +52,6 @@ class NVCCPlugin(Magics):
|
||||
self.compile(file_path)
|
||||
output = self.run(file_path, timeit=args.timeit)
|
||||
except subprocess.CalledProcessError as e:
|
||||
print(e.output.decode("utf8"))
|
||||
helper.print_out(e.output.decode("utf8"))
|
||||
output = None
|
||||
return output
|
||||
|
||||
Reference in New Issue
Block a user