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

Assert that curand output dtype is float32.

上级 2869fd17
......@@ -49,6 +49,7 @@ class CURAND_Base(GpuOp):
if self.destructive:
self.destroy_map = {0: [0]}
self.output_type = output_type
assert output_type.dtype == "float32"
def as_destructive(self):
"""Return an destructive version of self"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论