print out multiline output

This commit is contained in:
Andrei Nechaev
2019-07-12 17:00:56 -04:00
parent b88614e6e5
commit 98566ce171
3 changed files with 33 additions and 16 deletions
+13 -7
View File
@@ -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