未验证 提交 f8b3e576 编写于 作者: 0 0x45f 提交者: GitHub

Fix Warpctc error when using muti-gpu (#41389)

上级 5d6d14bc
......@@ -203,7 +203,7 @@ class WarpCTCFunctor {
void init(const Context& dev_ctx, const size_t blank) {
warpctc_version_ = phi::dynload::get_warpctc_version();
if (dev_ctx.GetPlace() == phi::GPUPlace()) {
if (paddle::platform::is_gpu_place(dev_ctx.GetPlace())) {
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
options_.loc = CTC_GPU;
options_.stream =
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册