Skip to content
Snippets Groups Projects
  1. Nov 22, 2022
  2. Nov 06, 2022
  3. Oct 17, 2022
  4. Jun 13, 2022
    • Kaushik Kulkarni's avatar
      Introduce attributes to store information about reduction dimensions (#224) · a0fa84a5
      Kaushik Kulkarni authored
      * Rename EinsumAxisDescriptor subclasses.
      
      In order to avoid coming off as something that is more generally applicable to
      any pt.Array node.
      
      * adds helpers to get all reduction induction variables
      
      * implements ReductionDescriptor
      
      * defines IndexLambda.var_to_reduction_descr
      
      * defines Einsum.redn_axis_to_redn_descr
      
      * test reduction_axis
      
      * Propagates loopy iname tags for ReductionAxis
      
      * tests loopy tag propagation for ReductionAxis
      
      * introduces ReductionAxis/EinsumIndex specific exception types
      
      * prefer immutables.Map over pyrsistent.PMap
      a0fa84a5
  5. Jun 21, 2021
  6. May 20, 2021
  7. Jan 31, 2021
  8. Jan 09, 2021
  9. Dec 15, 2020
  10. Nov 28, 2020
  11. Nov 24, 2020
  12. Oct 29, 2020
  13. Oct 14, 2020
  14. Jul 22, 2020
  15. Jun 23, 2020
  16. May 25, 2020
  17. May 24, 2020
  18. May 21, 2020
Loading