Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Core] Add MultiprocessingGPUExecutor #4539

Merged
merged 10 commits into from
May 14, 2024

Commits on May 1, 2024

  1. [Core] Add MultiprocessingGPUExecutor

    This introduces a python multiprocessing-based executor that can be used as an alternative to Ray for single-node inferencing.
    
    With the changes in this PR, Ray will continue to be used for parallel workers if it's installed, otherwise vanilla python multiprocessing is used. It can also be overridden with --no-worker-use-ray.
    
    The existing distributed tests have been updated to run with/without Ray.
    
    Worker processes are shut down when the LLMEngine is garbage collected.
    
    Co-authored-by: SAHIL SUNEJA <suneja@us.ibm.com>
    njhill and sahilsuneja1 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    eb7e5d4 View commit details
    Browse the repository at this point in the history

Commits on May 2, 2024

  1. Configuration menu
    Copy the full SHA
    079480f View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    3ef4da3 View commit details
    Browse the repository at this point in the history

Commits on May 4, 2024

  1. Merge remote-tracking branch 'refs/remotes/origin/main' into multipro…

    …c-gpu-executor
    
    # Conflicts:
    #	tests/distributed/test_basic_distributed_correctness.py
    njhill committed May 4, 2024
    Configuration menu
    Copy the full SHA
    e7d06d9 View commit details
    Browse the repository at this point in the history

Commits on May 11, 2024

  1. Configuration menu
    Copy the full SHA
    5ac9525 View commit details
    Browse the repository at this point in the history

Commits on May 12, 2024

  1. Configuration menu
    Copy the full SHA
    f728fdd View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    9161e5e View commit details
    Browse the repository at this point in the history

Commits on May 13, 2024

  1. Configuration menu
    Copy the full SHA
    3dc25c5 View commit details
    Browse the repository at this point in the history

Commits on May 14, 2024

  1. Configuration menu
    Copy the full SHA
    4c6e4ed View commit details
    Browse the repository at this point in the history
  2. Merge remote-tracking branch 'refs/remotes/origin/main' into multipro…

    …c-gpu-executor
    
    # Conflicts:
    #	.buildkite/test-pipeline.yaml
    njhill committed May 14, 2024
    Configuration menu
    Copy the full SHA
    1431de6 View commit details
    Browse the repository at this point in the history