Update benchmarks/cugraph/standalone/bulk_sampling/cugraph_bulk_sampl…

…ing.py Co-authored-by: Alex Barghi <[email protected]>
rapidsai · Nov 6, 2023 · 25412c0 · 25412c0
1 parent f55925c
commit 25412c0
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/benchmarks/cugraph/standalone/bulk_sampling/cugraph_bulk_sampling.py b/benchmarks/cugraph/standalone/bulk_sampling/cugraph_bulk_sampling.py
@@ -576,7 +576,16 @@ def benchmark_cugraph_bulk_sampling(
             "include_hop_column": False,
         }
     else:
-        sampling_kwargs = {}
+        # FIXME: Update these arguments when CSC mode is fixed in cuGraph-PyG (release 24.02)
+        sampling_kwargs = {
+          "deduplicate_sources": True,
+          "prior_sources_behavior": "exclude",
+          "renumber": True,
+          "compression": "COO",
+          "compress_per_hop": False,
+          "use_legacy_names": False,
+          "include_hop_column": True
+        }
 
     batches_per_partition = 400_000 // batch_size
     execution_time, allocation_counts = sample_graph(