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

enable GpuGer to be inserted with "blas.ldflags="

上级 5907d548
......@@ -514,6 +514,8 @@ def local_gpu_ger(node):
tensor.blas_c.CGer(destructive=False): gpu_ger_no_inplace,
tensor.blas.Ger(destructive=True): gpu_ger_no_inplace,
tensor.blas.Ger(destructive=False): gpu_ger_no_inplace,
tensor.blas_scipy.ScipyGer(destructive=True): gpu_ger_no_inplace,
tensor.blas_scipy.ScipyGer(destructive=False): gpu_ger_no_inplace,
}
if node.op == gpu_from_host:
host_input = node.inputs[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论