- 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.
-