提交 33bd55ef authored 作者: abergeron's avatar abergeron 提交者: GitHub

Merge pull request #21 from lamblin/bool

Fix tests for CAReduceDtype with bool
...@@ -1848,6 +1848,7 @@ class CAReduceDtype(CAReduce): ...@@ -1848,6 +1848,7 @@ class CAReduceDtype(CAReduce):
if dtype is None: if dtype is None:
# If input has a discrete dtype, upcast it to 64 # If input has a discrete dtype, upcast it to 64
return dict( return dict(
bool='int64',
int8='int64', int8='int64',
int16='int64', int16='int64',
int32='int64', int32='int64',
...@@ -1863,6 +1864,7 @@ class CAReduceDtype(CAReduce): ...@@ -1863,6 +1864,7 @@ class CAReduceDtype(CAReduce):
acc_dtype = self.acc_dtype acc_dtype = self.acc_dtype
if acc_dtype is None: if acc_dtype is None:
return dict( return dict(
bool='int64',
int8='int64', int8='int64',
int16='int64', int16='int64',
int32='int64', int32='int64',
......
...@@ -56,7 +56,6 @@ whitelist_flake8 = [ ...@@ -56,7 +56,6 @@ whitelist_flake8 = [
"tensor/tests/test_opt.py", "tensor/tests/test_opt.py",
"tensor/tests/test_basic.py", "tensor/tests/test_basic.py",
"tensor/tests/test_blas.py", "tensor/tests/test_blas.py",
"tensor/tests/test_elemwise.py",
"tensor/tests/test_merge.py", "tensor/tests/test_merge.py",
"tensor/tests/test_gc.py", "tensor/tests/test_gc.py",
"tensor/tests/test_complex.py", "tensor/tests/test_complex.py",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论