def __init__(self, src, name='default_program', headers=(), include_names=()): self.ptr = None if isinstance(src, six.binary_type): src = src.decode('UTF-8') if isinstance(name, six.binary_type): name = name.decode('UTF-8') self.src = src self.name = name self.ptr = nvrtc.createProgram(src, name, headers, include_names)
def __init__(self, src, name='default_program', headers=(), include_names=(), name_expressions=None): self.ptr = None if isinstance(src, bytes): src = src.decode('UTF-8') if isinstance(name, bytes): name = name.decode('UTF-8') self.src = src self.name = name self.ptr = nvrtc.createProgram(src, name, headers, include_names) self.name_expressions = name_expressions