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

No one assigned

    Labels

    area/gpuInvolves CUDA, GPUs, and similar

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions