mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-07-05 15:55:14 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

Populate arch by default; add --keep

Browse Source
This commit is contained in:
Steven Robertson
2017-04-24 16:39:15 -07:00
parent 582221dd0f
commit 9892acbc7f
3 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cuburn/render.py
View File

@ -218,9 +218,9 @@ class Renderer(object):
_modrefs = {}
@classmethod
def compile(cls, gnm, arch=None):
def compile(cls, gnm, arch=None, keep=False):
packer, lib = iter.mkiterlib(gnm)
cubin = util.compile('iter', assemble_code(lib), arch=arch)
cubin = util.compile('iter', assemble_code(lib), arch=arch, keep=keep)
return packer, lib, cubin
def load(self, cubin):
@ -232,8 +232,8 @@ class Renderer(object):
self._modrefs[cubin] = mod
return mod
def __init__(self, gnm, gprof):
self.packer, self.lib, self.cubin = self.compile(gnm)
def __init__(self, gnm, gprof, keep=False, arch=None):
self.packer, self.lib, self.cubin = self.compile(gnm, keep=keep, arch=arch)
self.mod = self.load(self.cubin)
self.filts = filters.create(gprof)
self.out = output.get_output_for_profile(gprof)