提交 45bd014f authored 作者: Frederic's avatar Frederic 提交者: Yann N. Dauphin

Make it work with DebugMode

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