Skip to content
Snippets Groups Projects
  1. Aug 22, 2017
  2. Aug 20, 2017
  3. Aug 16, 2017
  4. Aug 15, 2017
  5. Aug 13, 2017
    • Nicolas Vasilache's avatar
      [WIP] C++ topi contributions (#312) · f08de2b6
      Nicolas Vasilache authored
      * [WIP] C++ topi contributions
      
      Summary:
      This diff implements C++ topi contributions for:
        - relu with parametrix threshold
        - pad with generic padBefore / padAfter specification
        - matmult with transposes
        - conv2d_nchw, conv2d_hwcn with runtime constant padding and strides
        - depthwise_conv2d_nchw with runtime constant padding and strides
        - group_conv2d_ngchw with runtime constant padding and strides
        - broadcast_to a broadcastable shape
        - broadcast_bop where bop is an usual binary op (+ - * / %)
      
      Convolution padding is implemented using the pad operation.
      To avoid extra memory consumption, it is generally recommended to inline the padding with the autoinliner.
      Unfortunately in its current form the elemwise checks are too restrictive to allow inlining.
      So this diff also proposes an extension to LHS injective (i.e. no reduction axis in the current IR design)
      
      Test Plan:
      Tested in C++ testsuite in a separate repository, I am looking for suggestions to quickly spin up some tests for tvm.
      
      Reviewers: tqchen
      
      Subscribers:
      
      Tasks:
      
      Tags:
      
      Blame Revision:
      
      * Review + Lint + GSG C++
      f08de2b6
    • Tianqi Chen's avatar
    • Tianqi Chen's avatar
  6. Aug 12, 2017
  7. Aug 11, 2017
  8. Aug 10, 2017
  9. Aug 09, 2017
  10. Aug 08, 2017
  11. Aug 07, 2017
  12. Aug 05, 2017
  13. Aug 04, 2017
  14. Aug 03, 2017
  15. Aug 01, 2017
  16. Jul 31, 2017
  17. Jul 28, 2017
  18. Jul 26, 2017
  19. Jul 25, 2017
  20. Jul 24, 2017
  21. Jul 20, 2017
  22. Jul 19, 2017
  23. Jul 18, 2017
    • Jian Weng's avatar
      [API] Prefetch schedule supported (#258) · 01cbc61a
      Jian Weng authored
      * prefetch interface added
      
      * prefetch python comments modified. prefetch info data structure maintained.
      
      * start injecting prefetches. first step (domain touch) implemented.
      
      * domain touch tested.
      
      * Prefetch ir_mutator and ir_visitor dispatch registered.
      
      * modify domain touched from passing a func_ref to passing a tensor
      
      * modify domain touched from passing a func_ref to passing a tensor
      
      * modify Tensor copy to Tensor ref
      
      * temp commit for rebase
      
      * debug info removed, typo fixed, ready to rebase
      
      * prefetch flatten test add!
      
      * roll back builtin functions to side effect functions
      
      * lint error fixed!
      
      * add cache line size to storage flatten argument
      
      * forgot modifications add
      
      * change code style to dmlc-like; get rid of can_prove, use manually compute instead
      
      * python lint error fixed
      
      * modify instrinsic name to pass tests
      
      * [TEST] get rid of str(), replace them by accessing attributes
      
      * change map to list comprehension
      
      * redundant numpy import removed
      01cbc61a
  24. Jul 17, 2017
  25. Jul 16, 2017
Loading