- 07 Sep, 2022 2 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>
-
- 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>
-
- 08 Aug, 2022 1 commit
-
-
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>
-
- 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.
-
- 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.
-