multi files working version + refactoring

v1/v1.py

@@ -0,0 +1,51 @@
+import os
+import subprocess
+import tempfile
+import uuid
+
+from IPython.core.magic import Magics, cell_magic, magics_class
+
+compiler = '/usr/local/cuda/bin/nvcc'
+ext = '.cu'
+
+
+@magics_class
+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)
+
+    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)
+        else:
+            output = subprocess.check_output([file_path + ".out"], stderr=subprocess.STDOUT)
+            output = output.decode('utf8')
+        return output
+
+    @cell_magic
+    def cu(self, line, cell):
+        try:
+            args = self.argparser.parse_args(line.split())
+        except SystemExit as e:
+            self.argparser.print_help()
+            return
+
+        with tempfile.TemporaryDirectory() as tmp_dir:
+            file_path = os.path.join(tmp_dir, str(uuid.uuid4()))
+            with open(file_path + ext, "w") as f:
+                f.write(cell)
+            try:
+                self.compile(file_path)
+                output = self.run(file_path, timeit=args.timeit)
+            except subprocess.CalledProcessError as e:
+                print(e.output.decode("utf8"))
+                output = None
+        return output