def __init__(self, name, retain_file=False, gen_flex=False):
self.num_kernels = 0
self.module = None
self.functions = dict()
self.arg_descs = dict()
self.cache = dict()
self.compiled = False
self.retain_file = retain_file
self.buffer = StringIO()
# Open file and add header
if self.retain_file:
self.f = tempfile.NamedTemporaryFile(mode='w', suffix='.c', prefix=name, delete=False)
self.filename = self.f.name
self.buffer.write(_includes_template)
if gen_flex:
self.buffer.write(_flex_includes_template)
评论列表
文章目录