Skip to content

Instantly share code, notes, and snippets.

View c200chromebook's full-sized avatar

c200chromebook

View GitHub Profile
import numba.cuda as cuda
import numba as nb
import numpy as np
from numba.core import sigutils
from numba.cuda.compiler import _Kernel
from numba.cuda.codegen import CUDACodeLibrary
from numba.cuda.compiler import Dispatcher
import pickle
@cuda.jit(device=True, debug=True)
def expensive(x):
return math.pow(x, 1.001)
@cuda.jit(device=True, debug=True)
def cheap(x):
return math.sqrt(x)
@c200chromebook
c200chromebook / this one works
Created February 21, 2020 15:16
working one
import numpy as np
import numba
from numba import cuda
import math
@cuda.jit(argtypes=[numba.int32[:]], debug=True)
import numpy as np
import numba
from numba import cuda
import math
@cuda.jit(device=True)
def expensive(x):
return math.pow(x, 1.001)