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

Fix MWC unit test (prior to porting it)

Browse Source
This commit is contained in:
Steven Robertson
2014-12-25 13:22:01 -08:00
parent f39563b507
commit 49ccf75c5f
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cuburn/code/mwc.py
View File

@ -102,14 +102,14 @@ def test_mwc(rounds=5000, nblocks=64, blockwidth=512):
print "Trial %d, on CPU: " % trial, print "Trial %d, on CPU: " % trial,
sums = np.zeros(nthreads, dtype=np.uint64) sums = np.zeros(nthreads, dtype=np.uint64)
ctime = time.time() ctime = time.time()
mults = seeds[0].astype(np.uint64) mults = seeds[:,0].astype(np.uint64)
states = seeds[1] states = seeds[:,1]
carries = seeds[2] carries = seeds[:,2]
for i in range(rounds): for i in range(rounds):
step = np.frombuffer((mults * states + carries).data, step = np.frombuffer((mults * states + carries).data,
dtype=np.uint32).reshape((2, nthreads), order='F') dtype=np.uint32).reshape((nthreads, 2))
states[:] = step[0] states[:] = step[:,0]
carries[:] = step[1] carries[:] = step[:,1]
sums += states sums += states
ctime = time.time() - ctime ctime = time.time() - ctime