Add dask-cuda LocalCUDACluster support and VesselFM benchmark rule#128
Draft
Add dask-cuda LocalCUDACluster support and VesselFM benchmark rule#128
Conversation
…eaded scheduler for import scripts (#127) * Wrap dask_setup scripts in if __name__ == "__main__" and force threads scheduler for certain jobs
Co-authored-by: akhanf <11492701+akhanf@users.noreply.github.com>
Copilot
AI
changed the title
[WIP] Add dask-cuda support for GPU cluster setup
Add dask-cuda LocalCUDACluster support and VesselFM benchmark rule
Mar 10, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
VesselFM uses GPUs but was limited to the threaded Dask scheduler. This adds
dask-cudaintegration via a new"cuda"scheduler option and introduces benchmark rules to profile VesselFM performance across different chunk sizes and thread counts.Changes
dask_setup.py"cuda"scheduler branch usingdask_cuda.LocalCUDACluster(threads_per_worker=threads_per_worker)with adask.distributed.Clientvesselfm.pyconfig["dask_scheduler"]instead of hardcoded"threads", enabling GPU-accelerated inferencesnakebids.yml--dask_schedulernow acceptscudaas a valid choicevesselfm_benchmarkconfig block with defaultchunk_sizesandthread_countsfor benchmark sweepsvessels.smkrun_vesselfmgains abenchmark:directive for automatic timingbenchmark_run_vesselfmrule: parameterizeschunk_sizeandthread_countas wildcards, runs each combination 3× viarepeat()all_benchmark_vesselfmaggregate target expanding over all configured chunk/thread combinations# Run benchmark sweep across all chunk_size × thread_count combinations snakemake all_benchmark_vesselfm --cores allWarning
Firewall rules blocked me from connecting to one or more addresses (expand for details)
I tried to connect to the following addresses, but was blocked by firewall rules:
pixi.sh/usr/bin/curl curl -fsSL REDACTED(dns block)If you need me to access, download, or install something from one of these locations, you can either:
💡 You can make Copilot smarter by setting up custom instructions, customizing its development environment and configuring Model Context Protocol (MCP) servers. Learn more Copilot coding agent tips in the docs.