Files
nvcc/nvcc_plugin.py
T

9 lines
227 B
Python

from IPython.core.interactiveshell import InteractiveShell
from v1.v1 import NVCCPlugin as NVCC_V1
def load_ipython_extension(shell: InteractiveShell):
nvcc_plugin = NVCC_V1(shell)
shell.register_magics(nvcc_plugin)