未验证 提交 71568672 编写于 作者: S Shijie 提交者: GitHub

fix workspace size (#54884)

上级 7bf37b7d
......@@ -83,7 +83,7 @@ struct MatrixBandPartFunctor {
handle, uplo, n, A, lda, &workspace_size)); \
auto workspace = phi::memory_utils::Alloc( \
dev_ctx.GetPlace(), \
workspace_size, \
workspace_size * sizeof(T), \
phi::Stream(reinterpret_cast<phi::StreamId>(dev_ctx.stream()))); \
T* workspace_ptr = reinterpret_cast<T*>(workspace->ptr()); \
PADDLE_ENFORCE_GPU_SUCCESS(dynload::cusolverDn##C##potrf( \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册