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

Make store_per_thread reuse gtid in multiple calls when possible

Browse Source
This commit is contained in:
Steven Robertson 2010-09-10 18:45:32 -04:00
parent 943e92b80c
commit e71a8422e5
2 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cuburn/device_code.py
View File

@ -132,8 +132,8 @@ class IterThread(PTXEntryPoint):
label('all_cps_done')
# TODO this is for testing, move it to a debug statement
std.store_per_thread(g_num_rounds, num_rounds)
std.store_per_thread(g_num_writes, num_writes)
std.store_per_thread(g_num_rounds, num_rounds,
g_num_writes, num_writes)
@instmethod
def upload_cp_stream(self, ctx, cp_stream, num_cps):

22
cuburn/ptx.py
View File

@ -689,18 +689,22 @@ class _PTXStdLib(PTXFragment):
op.mad.lo.u32(dst, cta, ncta, tid)
@ptx_func
def store_per_thread(self, base, val):
def store_per_thread(self, *args):
"""Store b32 at `base+gtid*4`. Super-common debug pattern."""
with block("Per-thread store of %s" % str(val)):
with block("Per-thread storing values"):
reg.u32('spt_base spt_offset')
op.mov.u32(spt_base, base)
self.get_gtid(spt_offset)
op.mad.lo.u32(spt_base, spt_offset, 4, spt_base)
if isinstance(val, float):
# Turn a constant float into the big-endian PTX binary float
# representation, 0fXXXXXXXX (where XX is hex byte)
val = '0f%x%x%x%x' % reversed(map(ord, struct.pack('f', val)))
op.st.b32(addr(spt_base), val)
op.mul.lo.u32(spt_offset, spt_offset, 4)
for i in range(0, len(args), 2):
base, val = args[i], args[i+1]
op.mov.u32(spt_base, base)
op.add.u32(spt_base, spt_base, spt_offset)
if isinstance(val, float):
# Turn a constant float into the big-endian PTX binary f32
# representation, 0fXXXXXXXX (where XX is hex byte)
val = '0f%x%x%x%x' % reversed(map(ord,
struct.pack('f', val)))
op.st.b32(addr(spt_base), val)
@ptx_func
def set_is_first_thread(self, p_dst):