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

[Text Generation][V2] LinearRouter to accept SPLIT/JOIN #1434

Draft
wants to merge 50 commits into
base: feature/damian/no_kv_cache
Choose a base branch
from

Commits on Oct 26, 2023

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

Commits on Oct 31, 2023

  1. [Pipeline Refactor] Additional functionality, engine operator, linear…

    … router and image classification pipeline/operators/example (#1325)
    
    * initial functionality and working example with image classification
    
    * remove testing image
    
    * update args
    
    * initial functionality and working example with image classification
    
    * remove testing image
    
    * pr comments
    
    * defines schemas for operators and test
    
    * add image classification test, PR comments
    
    * fix input/output handling in pipeline and operator base classes to be more generic; remove context
    
    * add additional operator input message
    
    * typo fix
    dsikka committed Oct 31, 2023
    Configuration menu
    Copy the full SHA
    224e116 View commit details
    Browse the repository at this point in the history

Commits on Nov 1, 2023

  1. [v2] EngineOperator updates to make continuous batching easier (#1371)

    * [v2] EngineOperator updates to make continuous batching easier
    
    * test fixes
    bfineran committed Nov 1, 2023
    Configuration menu
    Copy the full SHA
    58b0758 View commit details
    Browse the repository at this point in the history

Commits on Nov 3, 2023

  1. [Pipeline Refactor] Update routes, text generation initial functional…

    …ity (#1348)
    
    * initial functionality and working example with image classification
    
    * remove testing image
    
    * rebase fixes
    
    * initial functionality and working example with image classification
    
    * text gen
    
    * updates func
    
    * prompt inference, initial functionality
    
    * remove image; update state docstring
    
    * Fix typo
    
    * add todo for split/join
    
    * remove context, clean-up args, remove prefill_preprocess_operaator
    
    * fix docstrings
    dsikka committed Nov 3, 2023
    Configuration menu
    Copy the full SHA
    e1ff108 View commit details
    Browse the repository at this point in the history
  2. [Pipeline Refactor] Additional Operators, Route update and completed …

    …generation functionality (#1356)
    
    * initial functionality and working example with image classification
    
    * remove testing image
    
    * rebase fixes
    
    * initial functionality and working example with image classification
    
    * text gen
    
    * updates func
    
    * prompt inference, initial functionality
    
    * remove image; update state docstring
    
    * Fix typo
    
    * add todo for split/join
    
    * remove context, clean-up args, remove prefill_preprocess_operaator
    
    * fix docstrings
    
    * initial functionality and working example with image classification
    
    * updates func
    
    * prompt inference, initial functionality
    
    * finish generation operators and update routes
    
    * further breakdown operators
    
    * add operators
    
    * fix can_operate condition
    
    * update can_operate to not rely on the inference_state
    
    * rebase + update
    
    * fix condition
    
    * fix capacity settting again
    
    * typo fixes
    dsikka committed Nov 3, 2023
    Configuration menu
    Copy the full SHA
    59457b7 View commit details
    Browse the repository at this point in the history
  3. add split/join functionality

    dsikka committed Nov 3, 2023
    Configuration menu
    Copy the full SHA
    f18d5f3 View commit details
    Browse the repository at this point in the history

Commits on Nov 7, 2023

  1. update router to include split/join in parent class, refactor pipelin…

    …e code to remove repeat code, update map function
    dsikka committed Nov 7, 2023
    Configuration menu
    Copy the full SHA
    2c4d231 View commit details
    Browse the repository at this point in the history
  2. process multiple generations

    dsikka committed Nov 7, 2023
    Configuration menu
    Copy the full SHA
    672ca20 View commit details
    Browse the repository at this point in the history

Commits on Nov 8, 2023

  1. initial commit

    dbogunowicz committed Nov 8, 2023
    Configuration menu
    Copy the full SHA
    304eb35 View commit details
    Browse the repository at this point in the history
  2. fix error

    dbogunowicz committed Nov 8, 2023
    Configuration menu
    Copy the full SHA
    71515ac View commit details
    Browse the repository at this point in the history

Commits on Nov 9, 2023

  1. Merge remote-tracking branch 'origin/features/v2/run_multiple' into f…

    …eature/damian/v2/factor_out_transformation_utils
    dbogunowicz committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    6f1b175 View commit details
    Browse the repository at this point in the history
  2. [Pipeline Refactor] Split/Join Functionality for multiple prompts (#1384

    )
    
    * add split/join functionality
    
    * update router to include split/join in parent class, refactor pipeline code to remove repeat code, update map function
    
    * process multiple generations
    
    * move map to base class
    dsikka committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    041174b View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    a508342 View commit details
    Browse the repository at this point in the history
  4. additional changes

    dsikka committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    cbb0e86 View commit details
    Browse the repository at this point in the history
  5. unit testing completion

    dsikka committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    2541581 View commit details
    Browse the repository at this point in the history
  6. remove debug

    dsikka committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    8c8989d View commit details
    Browse the repository at this point in the history
  7. fix

    dsikka committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    f8d75e3 View commit details
    Browse the repository at this point in the history
  8. add todo

    dsikka committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    fd1e466 View commit details
    Browse the repository at this point in the history
  9. more clean-up

    dsikka committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    64c0552 View commit details
    Browse the repository at this point in the history
  10. fix test

    dsikka committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    913665a View commit details
    Browse the repository at this point in the history
  11. add docstrings/comments

    dsikka committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    e15521f View commit details
    Browse the repository at this point in the history
  12. break out tests to individual unit test files; add conftest and make …

    …scope of fixtures module to help with speed
    dsikka committed Nov 9, 2023
    Configuration menu
    Copy the full SHA
    379481e View commit details
    Browse the repository at this point in the history

Commits on Nov 10, 2023

  1. Merge remote-tracking branch 'origin/features/v2/unit_testing' into f…

    …eature/damian/v2/factor_out_transformation_utils
    dbogunowicz committed Nov 10, 2023
    Configuration menu
    Copy the full SHA
    a90a20a View commit details
    Browse the repository at this point in the history
  2. [Pipeline Refactor] Unit Testing for Text Generation Operators (#1392)

    * unit testing for text generation operators
    
    * additional changes
    
    * unit testing completion
    
    * remove debug
    
    * fix
    
    * add todo
    
    * more clean-up
    
    * fix test
    
    * add docstrings/comments
    
    * break out tests to individual unit test files; add conftest and make scope of fixtures module to help with speed
    
    * fix name
    dsikka committed Nov 10, 2023
    Configuration menu
    Copy the full SHA
    0a50d1d View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    c0c4240 View commit details
    Browse the repository at this point in the history

Commits on Nov 13, 2023

  1. Configuration menu
    Copy the full SHA
    4f248dd View commit details
    Browse the repository at this point in the history
  2. [Continuous Batching] Queue Implementation to support batching groupi…

    …ng and prioritization (#1373)
    
    * [Continuous Batching] Queue Implementation to support batching grouping and prioritization
    
    * has_key method
    
    * thread safety
    
    * add blocking option for pop_batch
    
    * update docstring
    
    * allow mutex to be shared across continuous batching objects
    
    * revert last commit
    bfineran committed Nov 13, 2023
    Configuration menu
    Copy the full SHA
    20980a7 View commit details
    Browse the repository at this point in the history
  3. [Continuous Batching] Executor thread for running continuous batching (

    …#1374)
    
    * [Continuous Batching] Executor thread for running continuous batching
    
    * quality
    
    * ensure that executor stops when main thread does - clean up test hack
    bfineran committed Nov 13, 2023
    Configuration menu
    Copy the full SHA
    d81012d View commit details
    Browse the repository at this point in the history
  4. [ContinuousBatching] ContinuousBatchingScheduler Implementation (#1375)

    * [ContinuousBatching] ContinuousBatchingScheduler Implementation
    
    * cleanup unnecessary stop condition
    bfineran committed Nov 13, 2023
    Configuration menu
    Copy the full SHA
    5c48505 View commit details
    Browse the repository at this point in the history
  5. [continuous batching] singleton pattern for scheduler (#1391)

    * [continuous batching] singleton pattern for scheduler
    
    * catch from review
    bfineran committed Nov 13, 2023
    Configuration menu
    Copy the full SHA
    e1b7f37 View commit details
    Browse the repository at this point in the history

Commits on Nov 14, 2023

  1. Configuration menu
    Copy the full SHA
    98f7a6d View commit details
    Browse the repository at this point in the history
  2. [Pipeline Refactor][Text-Generation] Create a helper function for cre…

    …ating engine_inputs (#1364)
    
    * rebasing off my initial commit
    
    * cleanups
    
    * unit testing for text generation operators
    
    * additional changes
    
    * unit testing completion
    
    * remove debug
    
    * fix
    
    * add todo
    
    * more clean-up
    
    * fix test
    
    * add docstrings/comments
    
    * break out tests to individual unit test files; add conftest and make scope of fixtures module to help with speed
    
    * Delete tests/deepsparse/v2/unit/text_generation/test_msic.py
    
    ---------
    
    Co-authored-by: Dipika Sikka <dipikasikka1@gmail.com>
    dbogunowicz and dsikka committed Nov 14, 2023
    Configuration menu
    Copy the full SHA
    bbd534d View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    d1683b4 View commit details
    Browse the repository at this point in the history

Commits on Nov 17, 2023

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

Commits on Nov 20, 2023

  1. Configuration menu
    Copy the full SHA
    fa96efb View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    e41ddf8 View commit details
    Browse the repository at this point in the history
  3. [Pipeline Refactor][Text-Generation] Refactor transformers helpers …

    …functions (#1394)
    
    * add split/join functionality
    
    * update router to include split/join in parent class, refactor pipeline code to remove repeat code, update map function
    
    * process multiple generations
    
    * initial commit
    
    * fix error
    
    * unit testing for text generation operators
    
    * additional changes
    
    * unit testing completion
    
    * remove debug
    
    * fix
    
    * add todo
    
    * more clean-up
    
    * fix test
    
    * add docstrings/comments
    
    * break out tests to individual unit test files; add conftest and make scope of fixtures module to help with speed
    
    * Delete tests/deepsparse/v2/unit/text_generation/test_msic.py
    
    * pipeline runs, but incorrectly
    
    * Revert "pipeline runs, but incorrectly"
    
    This reverts commit 51c4ee6.
    
    * PR review comments
    
    ---------
    
    Co-authored-by: Dipika Sikka <dipikasikka1@gmail.com>
    dbogunowicz and dsikka committed Nov 20, 2023
    Configuration menu
    Copy the full SHA
    b80a417 View commit details
    Browse the repository at this point in the history
  4. [Text Generation][V2] End-to-end tests (#1402)

    * initial commit
    
    * initial commit
    
    * its working now
    
    * beautification
    
    * thank you Dipika <3
    
    * ready to review
    dbogunowicz committed Nov 20, 2023
    Configuration menu
    Copy the full SHA
    1b9238a View commit details
    Browse the repository at this point in the history

Commits on Nov 21, 2023

  1. Configuration menu
    Copy the full SHA
    89f11e5 View commit details
    Browse the repository at this point in the history
  2. integration tests pass

    dbogunowicz committed Nov 21, 2023
    Configuration menu
    Copy the full SHA
    9b441f5 View commit details
    Browse the repository at this point in the history
  3. [Pipeline Refactor][Text Generation][Continuous Batching] Integration (

    …#1409)
    
    * update split/join
    
    * use map
    
    * update
    
    * run end-to-end
    
    * clean-up
    
    * fix bug with batch size, introduce SplitRoute dataclass
    
    * update tests to use new inputs/outputs
    
    * use the normal scheduler for internal kv_cache
    
    * add pipeline inpuits
    
    * clean-up
    
    * change engine type, update docstrings, update override function to be more generic
    
    * move subgraph functionality to its own function; clean-up cont batching in text gen pipeline
    
    * update linear pathway to also use subgraph execution
    
    * rebase fix
    
    * fix tests
    dsikka committed Nov 21, 2023
    Configuration menu
    Copy the full SHA
    c858b1f View commit details
    Browse the repository at this point in the history
  4. [Pipeline Refactor] Operator Registry (#1420)

    * initial registry functionality
    
    * use sparsezoo mixin
    dsikka committed Nov 21, 2023
    Configuration menu
    Copy the full SHA
    bb3ff41 View commit details
    Browse the repository at this point in the history

Commits on Nov 22, 2023

  1. Configuration menu
    Copy the full SHA
    19434e7 View commit details
    Browse the repository at this point in the history
  2. fix tricky rebase

    dbogunowicz committed Nov 22, 2023
    Configuration menu
    Copy the full SHA
    90de2b3 View commit details
    Browse the repository at this point in the history
  3. one more cleanup

    dbogunowicz committed Nov 22, 2023
    Configuration menu
    Copy the full SHA
    66ca295 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    dcded1d View commit details
    Browse the repository at this point in the history
  5. Configuration menu
    Copy the full SHA
    127aa00 View commit details
    Browse the repository at this point in the history

Commits on Nov 27, 2023

  1. ready for review

    dbogunowicz committed Nov 27, 2023
    Configuration menu
    Copy the full SHA
    af57698 View commit details
    Browse the repository at this point in the history

Commits on Nov 28, 2023

  1. cleanup

    dbogunowicz committed Nov 28, 2023
    Configuration menu
    Copy the full SHA
    4397c80 View commit details
    Browse the repository at this point in the history
  2. initial commit

    dbogunowicz committed Nov 28, 2023
    Configuration menu
    Copy the full SHA
    6f1214e View commit details
    Browse the repository at this point in the history