Merge pull request #520 from rapidsai/branch-0.18

[gpuCI] Auto-merge branch-0.18 to branch-0.19 [skip ci]
rapidsai · Feb 9, 2021 · 0751c4b · 0751c4b
2 parents 1a553c4 + f9ce83a
commit 0751c4b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dask_cuda/benchmarks/local_cupy.py b/dask_cuda/benchmarks/local_cupy.py
@@ -75,7 +75,7 @@ async def _run(client, args):
         res = client.compute(func(*func_args))
         await client.gather(res)
         if args.type == "gpu":
-            await client.run(xp.cuda.Device().synchronize)
+            await client.run(lambda xp: xp.cuda.Device().synchronize(), xp)
         took = clock() - t1
 
     return {