- 13 Sep, 2022 1 commit
-
-
Min Xu authored
-
- 10 Sep, 2022 1 commit
-
-
Min Xu authored
* [minor] [FSDP] add a better for pure fp16 * [minor] [wrap] add a flag to help fsdp pure fp16 wrapping
-
- 07 Sep, 2022 4 commits
-
-
Min Xu authored
-
Min Xu authored
* [feat] add random_sparse_mask api * correct test skip Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
Anupam Bhatnagar authored
-
Min Xu authored
* [fix]: add a context for supporting state_dict from a non-FSDP parent module * formatting Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 26 Aug, 2022 1 commit
-
-
Min Xu authored
* [feat] support sst disabled and dst disabled cases * added tests Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 25 Aug, 2022 1 commit
-
-
Min Xu authored
* update nightly version * update wgit to use numpy for load/store - this is introduced with new nightly torch version, which made torch.save() not producing deterministic bytes - this make tensor<->numpy conversion and then do the save/load to avoid that issues. * fixed tests Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 11 Aug, 2022 1 commit
-
-
Min Xu authored
* added a profiling class * no more type ignore after merging main * fixed a int/round bug * add unit tests * skip if no cuda for a test Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 08 Aug, 2022 2 commits
-
-
Min Xu authored
* update examples and comment * fixed issue with fft/ifft only doing the last dim * fixed a int/round bug; fixed tests * add cuda tests * add atol and rtol * skip cuda test correctly Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
Crutcher Dunnavant authored
-
- 03 Aug, 2022 1 commit
-
-
Riyasat Ohib authored
* [Feat] implements lossy_compress with tests 1. Implements a method lossy_compress that takes in a dense tensor and returns a reconstruction with sst and dst, and optionally with sparsity.
-
- 31 Jul, 2022 1 commit
-
-
Riyasat Ohib authored
[Feat] Implements dense_sst_to_dst and sst_dst_to_dense methods and adds tests 1. Implements the dense_sst_to_dst and sst_dst_to_dense method. 2. Adds tests for perfect reconstruction with all top-k across different dims. 3. Adds tests for the two new methods.
-
- 29 Jul, 2022 1 commit
-
-
Min Xu authored
* [fix]: experimental import fix * [test]: catch issue #1042 in the future * revert trigger for failure * add numpy dep for users Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 28 Jul, 2022 1 commit
-
-
Min Xu authored
Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 27 Jul, 2022 1 commit
-
-
Riyasat Ohib authored
* [Feat] dense to sst implementation 1. Implementation of dense_to_sst function. 2. calculating the threshold for both the cases of top-k-element and top-k-percentage (fraction) 3. assertions to verify that the top_k_elements is smaller than the numel along the same dim 4. top_k_percent to top-k conversion 5. When calculating SST, now the real part of the complex dense_freq is used instead of the magnitudes. * [Feat, Tests] transform method addition, handling of top_k_element None case 1. Addition of a transform method 2. Adds code to handle the dim=None case for top_k_element * [Feat, Refactor] Reorganizations, new assertions and fixes. 1. XOR for validation that both of topk percent and element are not set, or both simultaneously unset. One and only one is set. 3. Distills topk and percent both to topk using unified helper function . 5. Adds a scatter topk values function to scatter values for SST and in future DST. 6. Validation for percentage range, and ensures k is never 0. 7. Uses config validation, adds config validation for top_k_element > 0 if not None.
-
- 26 Jul, 2022 6 commits
-
-
Anupam Bhatnagar authored
-
Min Xu authored
* [fix] handle EMA in the state_dict * better fix
-
Min Xu authored
* [fix]: experimental import fix * Update fairscale/experimental/__init__.py * Update fairscale/experimental/__init__.py Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
Anupam Bhatnagar authored
-
Min Xu authored
* [fix] unclose FD and not load/store metadata many times * one more stat * Update fairscale/experimental/wgit/sha1_store.py * add name to the objects when added * dict key can be int from a state_dict * removed top_level_objects key; it should be added into repo, not sha1_store Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
Min Xu authored
Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 22 Jul, 2022 1 commit
-
-
Min Xu authored
* flip per_tensor's default * fixed original size computation Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 21 Jul, 2022 1 commit
-
-
Min Xu authored
* additional metadata, step 1 * add gzip option to repo::add * add repo:add's return value and some refactoring and todo * added size metadata to sha1_store * added names metadata to sha1_store Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 19 Jul, 2022 1 commit
-
-
Min Xu authored
* formatting change, no logical change * formatting and name change, no logical change * [refactor] sha1_store's path arg - make sha1_store's path arg directly the path, not its parent - this is because sha1_store is not like a .git or a .wgit dir, which is nested inside another "working" dir. It is simply a store, which is using a given dir. - updated repo and tests as well. * remove a test warning due to deprecated API from torch * [refactor] change how dot_wgit_dir_path is used - it should only be assigned in __init__. - we use it in error checking in the rest APIs. * simplify the init a bit * refactor the sanity check * moved some functions, no code change * [feat] added per-tensor add to the repo * enabled gzip compression on add * fix a unit test * add a note * make sha1 store work on general dict * handle general state_dict from a model, not just a module's one-level OrderedDict * formatting Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 18 Jul, 2022 1 commit
-
-
Min Xu authored
Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 15 Jul, 2022 2 commits
-
-
Min Xu authored
Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
Min Xu authored
Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 14 Jul, 2022 2 commits
-
-
Min Xu authored
Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
Min Xu authored
Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 12 Jul, 2022 2 commits
-
-
Min Xu authored
* refactor SHA1_Store - renamed the class - added created_on field and refactored how init is done - wrap long lines * wrapped longer lines * rename json file ref_count.json * make sha1_buf_size an argument * update gitignore * added tmp_dir * added new sha1_store add and tests * update chdir * add debug to test * fixing unit test for 1.8 Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
Min Xu authored
- Thanks to Alexei for spotting this. - Added triggering test case. - enhanced tests with more comments etc. Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 05 Jul, 2022 1 commit
-
-
Riyasat Ohib authored
* [Fix] Restructure for wgit availability as a package * Preliminary implementation of wgit status * [Feat] Addition of wgit status 1. Functionalities to check the status of the repo. 2. Checks if file has been modified, whether changes added or added changes commited. * [test] Addition of tests for weigit status 1. Some minor refactors and docstring changes * [Fix] Changes in repo status test * [test] status test fix 1. made the test status printing order independent * [refactor] Metadata dirs mirroring chkpt paths, changes in wgit status 1. Metadata files are now created within wgit with directory structure mirroring the relative paths of the checkpoint/files they track. 2. Changes in status: 3 statuses now. 3. Changes in tests. 4. Some code refactoring. * [cleanup] minor changes in comments and cleanup
-
- 29 Jun, 2022 2 commits
-
-
Min Xu authored
Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
Crutcher Dunnavant authored
-
- 27 Jun, 2022 1 commit
-
-
Riyasat Ohib authored
-
- 25 Jun, 2022 1 commit
-
-
Min Xu authored
Co-authored-by:Min Xu <min.xu.public@gmail.com>
-
- 24 Jun, 2022 3 commits
-
-
dependabot[bot] authored
Bumps [numpy](https://github.com/numpy/numpy) from 1.21.5 to 1.22.0. - [Release notes](https://github.com/numpy/numpy/releases) - [Changelog](https://github.com/numpy/numpy/blob/main/doc/HOWTO_RELEASE.rst) - [Commits](https://github.com/numpy/numpy/compare/v1.21.5...v1.22.0 ) --- updated-dependencies: - dependency-name: numpy dependency-type: direct:development ... Signed-off-by:
dependabot[bot] <support@github.com> Co-authored-by:
dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
-
Hugo van Kemenade authored
-
Riyasat Ohib authored
weigit: Fixed file tracking with metadata. Changes in sha1_store for better encapsulation. Docstrings. (#1013) * [Feat] Fixed file tracking with metadata. Change in sha1_store for better encapsulation. Tests. 1. Adds metadata creation per added file and independently tracks version of each separate file added. That is, now creates separate metadata files for each file to be tracked. 2. Changes in reference tracking to accomodate the change in 1. 3. Somes changes in SHA1_store for better encapsulation. 4. Modified the tests to reflect above. * [Feat] 1. Added docstrings to the classes. 2. Added a recursively search for the weigit repo upto root. 3. Some refactor of the codes. * [Feat][Refactor] repo and sha1_store add modification and separation. Modification in reference tracking 1. Separation of add functionalities of repo.add and sha1_store.add. 2. Updated the reference tracking. 3. New tests and code refactor * [Fix] Sha1_store fix overlap in first two characters of sha1 hash. 1. Accept multiple sha1 hash's with same two starting characters and create directories accordingly. * [Fix] Minor refactoring and test fix * [Fix] Fix for pygit class initialization in cases when no .gitconfig file is available Co-authored-by:Riyasat Ohib <riohib@devfair0756.h2.fair>
-