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 comment to DSL

Browse Source
This commit is contained in:
Steven Robertson
2010-09-01 23:34:24 -04:00
parent 32f68ea1d5
commit 4a80d946db
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cuburnlib/ptx.py
View File

@ -453,6 +453,13 @@ class _LabelFactory(object):
self.block.inject(name, Label(name))
self.block.code(prefix='%s:' % name, semi=False)
class Comment(object):
"""Add a single-line comment to the PTX output."""
def __init__(self, block):
self.block = block
def __call__(self, comment):
self.block.code(op=['// ', comment], semi=False)
class PTXFragment(object):
"""
An object containing PTX DSL functions.
@ -627,6 +634,7 @@ class _PTXStdLib(PTXFragment):
addr=Mem.addr,
vec=Mem.vec,
label=_LabelFactory(self.block),
comment=Comment(self.block),
get_gtid=self._get_gtid)
class PTXModule(object):