提交 7eb0f10b authored 作者: Yann N. Dauphin's avatar Yann N. Dauphin

UsmmCscDense only accepts floats now

上级 482520aa
......@@ -1684,8 +1684,8 @@ class UsmmCscDense(gof.Op):
dtype_out = scalar.upcast(alpha.type.dtype, x_val.type.dtype,
y.type.dtype, z.type.dtype)
if dtype_out in ('complex64', 'complex128'):
raise NotImplementedError('Complex types are not supported in operands')
if dtype_out not in ('float32', 'float64'):
raise NotImplementedError('only float types are supported in operands')
if self.inplace:
assert z.type.dtype == dtype_out
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论