notkshitij/nvcc
1
0
Fork 0
mirror of https://github.com/andreinechaev/nvcc4jupyter.git synced 2026-06-13 18:50:47 +05:30
Code Issues Packages Projects Releases Wiki Activity

path to nvcc

Browse Source
This commit is contained in:
Andrei Nechaev
2018-05-04 11:01:45 -04:00
parent 208e67cf07
commit b302ae215a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nvcc_plugin.py
+2 -1
View File
@@ -6,6 +6,7 @@ import tempfile
import subprocess import subprocess
import IPython.core.magic as ipym import IPython.core.magic as ipym
compiler = '/usr/local/cuda/bin/nvcc'
ext = '.cu' ext = '.cu'
@@ -25,7 +26,7 @@ class NVCCPlugin(ipym.Magics):
@staticmethod @staticmethod
def compile(file_path): def compile(file_path):
subprocess.check_output(["nvcc", file_path + ext, "-o", file_path + ".out"], stderr=subprocess.STDOUT) subprocess.check_output([compiler, file_path + ext, "-o", file_path + ".out"], stderr=subprocess.STDOUT)
def run(self, file_path, timeit=False): def run(self, file_path, timeit=False):
if timeit: if timeit: