提交 8a3734fe authored 作者: Frederic's avatar Frederic

Make it work with DebugMode

上级 0ac05faf
......@@ -54,12 +54,14 @@ def multMatVect(v, A, m1, B, m2):
f.input_storage[0].storage[0] = A
f.input_storage[1].storage[0] = v[:3]
f.input_storage[2].storage[0] = m1
r[:3] = f.fn()[0]
f.fn()
r[:3] = f.output_storage[0].storage[0]
f.input_storage[0].storage[0] = B
f.input_storage[1].storage[0] = v[3:]
f.input_storage[2].storage[0] = m2
r[3:] = f.fn()[0]
f.fn()
r[3:] = f.output_storage[0].storage[0]
return r
multMatVect.dot_modulo = None
......@@ -816,12 +818,14 @@ class MRG_RandomStreams(object):
f.input_storage[0].storage[0] = A1p72
f.input_storage[1].storage[0] = v[:3]
f.input_storage[2].storage[0] = M1
r[:3] = f.fn()[0]
f.fn()
r[:3] = f.output_storage[0].storage[0]
f.input_storage[0].storage[0] = A2p72
f.input_storage[1].storage[0] = v[3:]
f.input_storage[2].storage[0] = M2
r[3:] = f.fn()[0]
f.fn()
r[3:] = f.output_storage[0].storage[0]
if inc_rstate:
self.inc_rstate()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论