Skip to content

Commit 423dcf3

Browse files
committed
added time for the rbm on the gpu.
1 parent a8b4541 commit 423dcf3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: code/test.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def speed():
5555
expected_times_32=numpy.asarray([ 13.29699826, 32.42813158, 68.03559947, 105.54640913, 107.00527334,
5656
242.41721797, 490.40798998, 528.88854146])
5757
expected_times_gpu=numpy.asarray([ 3.07663488, 7.55523491, 27.05203891, 9.58915591, 24.13007045,
58-
24.77524018, 145.93923092, float('NaN')])
58+
24.77524018, 145.93923092, 349.31434679])
5959

6060
def time_test(m,l,idx,f,**kwargs):
6161
if not to_exec[idx]:

0 commit comments

Comments
 (0)