Skip to content

Add optional GPU execution via CuPy#142

Open
01-1 wants to merge 1 commit into
AIcrowd:mainfrom
01-1:codex/gpu-execution
Open

Add optional GPU execution via CuPy#142
01-1 wants to merge 1 commit into
AIcrowd:mainfrom
01-1:codex/gpu-execution

Conversation

@01-1

@01-1 01-1 commented Jun 24, 2026

Copy link
Copy Markdown

Summary

  • add opt-in CuPy execution for selected backend NumPy calls while preserving CPU ndarray outputs and FLOP accounting
  • expose configure_gpu, gpu_status, gpu_enabled, and gpu_available at the top level
  • add a gpu optional dependency extra and README documentation for local GPU execution
  • add focused tests for public controls, CPU array contract preservation, fake-CuPy offload, and CUDA runtime fallback

Testing

  • UV_CACHE_DIR=/tmp/flopscope-uv-cache uv run --python 3.10 pytest -q tests/test_gpu.py

@01-1 01-1 force-pushed the codex/gpu-execution branch from bca1a43 to 7b96161 Compare June 24, 2026 20:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant