Skip to content

Provide GPU support in released qsimcirq wheel #430

@95-martin-orion

Description

@95-martin-orion

Currently, qsimcirq users who want to run on GPU must compile the code locally, as the wheels are built on Github Actions runners which do not have the CUDA SDK. Compared to the CPU workflow of pip install qsimcirq, this is incovenient; it would be preferable if GPU support was built into the released wheels.

In order to provide this support, we need to investigate the following options (listed from most to least preferable):

  1. Create an image with the CUDA SDK and use it in the release.
    • Minimizes the impact to our release speed
    • Still need to determine if this is possible in Github Actions
  2. Install the CUDA SDK on the Github Actions runner during each release.
    • Definitely possible in Github Actions (an action exists for installing the SDK)
    • May increase release (not CI) runtime by 10+ minutes
  3. Establish a secondary workflow for releasing the GPU wheels.
    • Allows us to use our own devices with the CUDA SDK pre-installed
    • Increases maintenance costs, and a single wheel would be preferable

Metadata

Metadata

Assignees

Labels

area/gpuInvolves CUDA, GPUs, and similar

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions