提交 dcac8718 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #4046 from AnishShah/issue4035

tip for mixed dtypes
...@@ -1346,6 +1346,16 @@ class ProfileStats(object): ...@@ -1346,6 +1346,16 @@ class ProfileStats(object):
" generator supported on the GPU.", file = file) " generator supported on the GPU.", file = file)
break break
# tip 6
for a in self.apply_time:
node = a
if (isinstance(node.op, T.Dot) and
len(set(i.dtype for i in node.inputs)) != 1):
print(" - You have a dot operation that has different dtype "
" for inputs (%s). Make sure that the inputs have same "
" dtype." % [i.type for i in node.inputs], file = file)
printed_tip = True
if not printed_tip: if not printed_tip:
print(" Sorry, no tip for today.", file = file) print(" Sorry, no tip for today.", file = file)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论