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

[modifier refactor] Add constant pruning tests #1752

Merged
merged 14 commits into from
Oct 27, 2023

Commits on Oct 23, 2023

  1. Initial commit

    rahul-tuli committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    782267b View commit details
    Browse the repository at this point in the history
  2. Add end to end tests

    rahul-tuli committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    780ec6d View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    41877aa View commit details
    Browse the repository at this point in the history
  4. Move imports inside the test fuctions so

    that torch isn't imported unless running the tests
    rahul-tuli committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    c86daa8 View commit details
    Browse the repository at this point in the history
  5. Configuration menu
    Copy the full SHA
    84149e8 View commit details
    Browse the repository at this point in the history
  6. [Bugfix] .dict() method on Recipe (#1753)

    * Bugfix .dict() method on Recipe
    
    * Remove extraneous local test, [faulty commit]
    rahul-tuli committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    fe5151a View commit details
    Browse the repository at this point in the history
  7. [modifier refactor] Add serialization tests (#1755)

    * Add serialization tests
    
    * Clean up
    
    * Keep original stage and group names
    Clean up _get_yaml_dict
    
    * fix comment
    
    * Typo
    rahul-tuli committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    599b816 View commit details
    Browse the repository at this point in the history
  8. [Unit Tests][Modifier Refactor] (#1756)

    * Move valid recipes to a helper file
    Add tests for session.py
    
    * Increase test coverage of src/sparseml/core/session.py
    to 100%
    Run Style
    Add logs to .gitignore
    
    * Increase coverage of tests/sparseml/core/test_state.py
    to 100%
    
    * add tests for lifecycle/event.py
    
    * Increase code coverage of lifecycle/event to
    100%
    
    * increase lifecycle/session.py code coverage to 93%
    
    * Address review comments from @Satrat
    rahul-tuli committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    4865641 View commit details
    Browse the repository at this point in the history
  9. Address review comments on 1752 (#1772)

    Update makefile to only ignore *pytorch.py files in modifier dir
    Fix order in test
    Add regex to makefile
    Add helper function to determine if torch tests should be run
    Check masks
    Make transformers import optional in sparsegpt.py
    rahul-tuli committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    f01d5ec View commit details
    Browse the repository at this point in the history
  10. Fix merge conflict

    rahul-tuli committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    585b7e8 View commit details
    Browse the repository at this point in the history
  11. Configuration menu
    Copy the full SHA
    1b86ceb View commit details
    Browse the repository at this point in the history
  12. [Bug][ConstantPruningModifier] Fix mask de register bug (#1773)

    * Fix mask de-register logic
    
    * forgot to remove commented out line
    rahul-tuli committed Oct 23, 2023
    Configuration menu
    Copy the full SHA
    32469e3 View commit details
    Browse the repository at this point in the history
  13. Configuration menu
    Copy the full SHA
    2c7f050 View commit details
    Browse the repository at this point in the history

Commits on Oct 26, 2023

  1. Fix session reset (#1790)

    rahul-tuli authored Oct 26, 2023
    Configuration menu
    Copy the full SHA
    40111cf View commit details
    Browse the repository at this point in the history