Skip to content

Commit

Permalink
fix backend name
Browse files Browse the repository at this point in the history
Signed-off-by: João Seródio <[email protected]>
  • Loading branch information
SerodioJ committed Dec 12, 2023
1 parent 7883571 commit 2e4b5da
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dasf/utils/decorators.py
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ def wrapper(*args, **kwargs):
client = get_dask_running_client()
if client is not None: # Runs task according to current client configuration, i.e, Pipeline Executor
func_type = "_lazy"
arch = "gpu" if getattr(client, "backend", None) == "gpu" else "cpu"
arch = "gpu" if getattr(client, "backend", None) == "cupy" else "cpu"
else:
if not is_forced_local(cls) and (is_dask_gpu_supported() or is_dask_supported()):
func_type = "_lazy"
Expand Down

0 comments on commit 2e4b5da

Please sign in to comment.