10 lines
213 B
Python
10 lines
213 B
Python
import pycuda.driver as cuda
|
|
import pycuda.autoinit
|
|
from pycuda.compiler import SourceModule
|
|
import numpy as np
|
|
|
|
with open('./cuda/kernel.cu', 'r') as file_k:
|
|
code = file_k.read()
|
|
|
|
kernel = SourceModule(code)
|