diff --git a/nxbench/benchmarks/benchmark.py b/nxbench/benchmarks/benchmark.py index 370b50c..50aae0f 100644 --- a/nxbench/benchmarks/benchmark.py +++ b/nxbench/benchmarks/benchmark.py @@ -181,6 +181,8 @@ def prepare_benchmark( except ImportError: logger.exception("nx-parallel backend not available") return None + nx.config.backends.parallel.active = True + nx.config.backends.parallel.n_jobs = num_thread return nxp.ParallelGraph(original_graph) if "cugraph" in backend and is_nx_cugraph_available():