提交 d16b5cfa authored 作者: Frederic's avatar Frederic

a few pep8

上级 97b7f9bc
...@@ -875,8 +875,8 @@ class test_fusion(unittest.TestCase): ...@@ -875,8 +875,8 @@ class test_fusion(unittest.TestCase):
fyv * fzv, 'float32'), # 2 fyv * fzv, 'float32'), # 2
(fx * fy + fz, (fx, fy, fz), (fxv, fyv, fzv), 1, fxv * (fx * fy + fz, (fx, fy, fz), (fxv, fyv, fzv), 1, fxv *
fyv + fzv, 'float32'), # 3 fyv + fzv, 'float32'), # 3
(fw + fx + fy + fz, (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1, fwv + (fw + fx + fy + fz, (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1,
fxv + fyv + fzv, 'float32'), fwv + fxv + fyv + fzv, 'float32'),
((fw + fx) + (fy + fz), (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1, ((fw + fx) + (fy + fz), (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1,
fwv + fxv + fyv + fzv, 'float32'), # 5 fwv + fxv + fyv + fzv, 'float32'), # 5
(((fw + fx) + fy) + fz, (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1, (((fw + fx) + fy) + fz, (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1,
...@@ -886,22 +886,22 @@ class test_fusion(unittest.TestCase): ...@@ -886,22 +886,22 @@ class test_fusion(unittest.TestCase):
((fw + (fx + fy) + fz), (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1, ((fw + (fx + fy) + fz), (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1,
fwv + fxv + fyv + fzv, 'float32'), fwv + fxv + fyv + fzv, 'float32'),
(fw + (fx + (fy + fz)), (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1, (fw + (fx + (fy + fz)), (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1,
fwv + fxv + fyv +fzv, 'float32'), fwv + fxv + fyv + fzv, 'float32'),
((fw+fx)+(fy+fz), (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1, ((fw+fx)+(fy+fz), (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1,
fwv+fxv+fyv+fzv, 'float32'), # 10 fwv+fxv+fyv+fzv, 'float32'), # 10
(fw*fx*fy*fz, (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1, fwv* (fw*fx*fy*fz, (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1,
fxv*fyv*fzv, 'float32'), fwv * fxv * fyv * fzv, 'float32'),
(fw+fx*fy*fz, (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1, fwv+ (fw+fx*fy*fz, (fw, fx, fy, fz), (fwv, fxv, fyv, fzv), 1,
fxv*fyv*fzv, 'float32'), fwv + fxv * fyv * fzv, 'float32'),
(fx+fy*fz*fx, (fx, fy, fz), (fxv, fyv, fzv), 1, (fx+fy*fz*fx, (fx, fy, fz), (fxv, fyv, fzv), 1,
fxv+fyv*fzv*fxv, 'float32'), fxv + fyv * fzv * fxv, 'float32'),
(fx*fy+fz+fy, (fx, fy, fz), (fxv, fyv, fzv), 1, (fx*fy+fz+fy, (fx, fy, fz), (fxv, fyv, fzv), 1,
fxv*fyv+fzv+fyv, 'float32'), fxv * fyv + fzv + fyv, 'float32'),
(fx*fy*fz*fw+fx+fy+fz+fw, (fw, fx, fy, fz), (fwv, fxv, (fx*fy*fz*fw+fx+fy+fz+fw, (fw, fx, fy, fz), (fwv, fxv,
fyv, fzv), 1, fxv*fyv*fzv*fwv+fxv+fyv+fzv+fwv, 'float32'), # 15 fyv, fzv), 1, fxv*fyv*fzv*fwv+fxv+fyv+fzv+fwv, 'float32'), # 15
#test with constant #test with constant
((fw+fx)+(fy+fz)+ 2.,(fw,fx,fy,fz),(fwv,fxv,fyv,fzv) ((fw+fx)+(fy+fz)+ 2.,(fw,fx,fy,fz),(fwv,fxv,fyv,fzv),
,1,fwv+fxv+fyv+fzv+2,'float32'), 1,fwv+fxv+fyv+fzv+2,'float32'),
(((fw+fx)+2.+fy)+fz,(fw,fx,fy,fz),(fwv,fxv,fyv,fzv), (((fw+fx)+2.+fy)+fz,(fw,fx,fy,fz),(fwv,fxv,fyv,fzv),
1,fwv+fxv+fyv+fzv+2,'float32'), 1,fwv+fxv+fyv+fzv+2,'float32'),
((fw+(fx+2.+fy))+fz,(fw,fx,fy,fz),(fwv,fxv,fyv,fzv), ((fw+(fx+2.+fy))+fz,(fw,fx,fy,fz),(fwv,fxv,fyv,fzv),
...@@ -1027,10 +1027,12 @@ class test_fusion(unittest.TestCase): ...@@ -1027,10 +1027,12 @@ class test_fusion(unittest.TestCase):
fail2 = [] fail2 = []
fail3 = [] fail3 = []
fail4 = [] fail4 = []
for id, [g, sym_inputs, val_inputs, nb_elemwise, answer, out_dtype] in enumerate(cases): for id, [g, sym_inputs, val_inputs,
nb_elemwise, answer, out_dtype] in enumerate(cases):
if isinstance(out_dtype, dict): if isinstance(out_dtype, dict):
out_dtype = out_dtype[config.cast_policy] out_dtype = out_dtype[config.cast_policy]
if gpu and (out_dtype!='float32' or any(i.dtype != 'float32' for i in g.owner.inputs)): if (gpu and (out_dtype != 'float32' or
any(i.dtype != 'float32' for i in g.owner.inputs))):
print "Skip test %d as the gpu code currently supports only float32" % id print "Skip test %d as the gpu code currently supports only float32" % id
continue continue
print "new cases", id print "new cases", id
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论