| import cupy | |
| #@cupy.memoize(for_each_device=True) | |
| def cupy_launch(strFunction, strKernel): | |
| # return cupy.cuda.compile_with_cache(strKernel).get_function(strFunction) | |
| return cupy.RawKernel(strKernel, strFunction) | |
| # end |
| import cupy | |
| #@cupy.memoize(for_each_device=True) | |
| def cupy_launch(strFunction, strKernel): | |
| # return cupy.cuda.compile_with_cache(strKernel).get_function(strFunction) | |
| return cupy.RawKernel(strKernel, strFunction) | |
| # end |