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

Add performance tuning (maxnregs) to entries

Browse Source
This commit is contained in:
Steven Robertson
2010-09-10 12:52:47 -04:00
parent 086e4e4fb4
commit 7e0d36af7d
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cuburnlib/ptx.py
View File

@ -605,6 +605,7 @@ class PTXEntryPoint(PTXFragment):
entry_name = ""
# List of (type, name) pairs for entry params, e.g. [('u32', 'thing')]
entry_params = []
maxnreg = None
def entry(self):
"""
@ -817,9 +818,11 @@ class PTXModule(object):
# This is kind of hackish compared to everything else
params = [Reg('.param.' + str(type), name)
for (type, name) in ent.entry_params]
_block.code(op='.entry %s ' % ent.entry_name, semi=False,
_block.code(op='.entry %s' % ent.entry_name, semi=False,
vars=['(', ', '.join(['%s %s' % (r.type, r.name)
for r in params]), ')'])
if ent.maxnreg:
_block.code(op='.maxnreg %d' % ent.maxnreg, semi=False)
with Block(_block):
[_block.inject(r.name, r) for r in params]
for dep in insts: