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

Adjust softjoin and vec to be easier to format

Browse Source
This commit is contained in:
Steven Robertson 2010-09-02 15:05:24 -04:00
parent 0a3f9551ad
commit bf79dc7fa0
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
cuburnlib/ptx.py
View File

@ -73,7 +73,7 @@ from collections import namedtuple
def _softjoin(args, sep): def _softjoin(args, sep):
"""Intersperses 'sep' between 'args' without coercing to string.""" """Intersperses 'sep' between 'args' without coercing to string."""
return reduce(lambda l, x: l + [x, sep], args, [])[:-1] return [(arg, sep) for arg in args[:-1]] + args[-1:]
BlockCtx = namedtuple('BlockCtx', 'locals code injectors') BlockCtx = namedtuple('BlockCtx', 'locals code injectors')
PTXStmt = namedtuple('PTXStmt', 'prefix op vars semi indent') PTXStmt = namedtuple('PTXStmt', 'prefix op vars semi indent')
@ -406,7 +406,12 @@ class Mem(object):
>>> op.ld.global.v2.u32(vec(reg1, reg2), addr(areg)) >>> op.ld.global.v2.u32(vec(reg1, reg2), addr(areg))
""" """
return ['{', _softjoin(args, ', '), '}'] assert len(args) >= 2, "vector loads make no sense with < 2 args"
joined = _softjoin(args, ', ')
# This makes a list like [('{', ('arg1', ', ')), ('argf', '}')], which
# when compacted comes out like ['{arg1,', 'argf}'], which lets the
# formatter work properly. There's no semantic value to this.
return [('{', joined[0])] + joined[1:-1] + [(joined[-1], '}')]
@staticmethod @staticmethod
def addr(areg, aoffset=''): def addr(areg, aoffset=''):
@ -565,7 +570,7 @@ class PTXEntryPoint(PTXFragment):
# List of (type, name) pairs for entry params, e.g. [('u32', 'thing')] # List of (type, name) pairs for entry params, e.g. [('u32', 'thing')]
entry_params = [] entry_params = []
def entry(self, ctx): def entry(self):
""" """
PTX DSL function that comprises the body of the PTX statement. PTX DSL function that comprises the body of the PTX statement.