1. 25 2月, 2019 3 次提交
    • Z
      update with develop. test=develop · 9261cf39
      Zhen Wang 提交于
      9261cf39
    • Z
      add set_attr for IrOpNode. test=develop · 0bf809c9
      Zhen Wang 提交于
      0bf809c9
    • J
      [MKL-DNN] MKL-DNN specific Tensor modification (#15429) · dec9cf53
      Jacek Czaja 提交于
      * - Implemented draft of primitive desc keeping in Tensor
      
      test=develop
      
      - TransposeMKLDNNHandler::AcquireSrcMemory was reimplemented
      
      - Added nchw and nc formats setting for sake of compatiblity
      
      Fixed unit tests
      
      - Worakaround to problem with 5D data in conv
      
      - Added 3D and 1D MKL-DNN formats for name handles for tensor
      
      test=develop
      
      - Fix to UTs
      
      test=develop
      
      - Conv fp32 op was updated
      
      Cosmetic fixes
      
      test=develop
      
      - tensor mkldnn cosmetics
      
      test=develop
      
      - Moved most of mkl-dnn specific code from Tensor to mkl-dnn utils
      
      * - Lint fixes
      
      test=develop
      
      * - setting prim dec in Tensor , sets also layout to kMKLDNN
      
      test=develop
      
      * - Moved creation of prim desc totally out of Tensor
      
      test=develop
      
      * - Cosmetic fixes adter review
      
      test=develop
      dec9cf53
  2. 24 2月, 2019 1 次提交
  3. 22 2月, 2019 4 次提交
  4. 21 2月, 2019 6 次提交
    • T
      disable dam temporarily (#15860) · e3dd6970
      Tao Luo 提交于
      test=develop
      e3dd6970
    • D
      test=develop · 35a90e06
      Dun Liang 提交于
      35a90e06
    • D
      test=develop · c9080f51
      Dun Liang 提交于
      c9080f51
    • D
      test=develop · 1c7bb0e4
      Dun Liang 提交于
      1c7bb0e4
    • X
      add per kernel config and remove const_cast. · 5eb87506
      Xin Pan 提交于
      test=develop
      5eb87506
    • D
      Profiler refine and add CUDA runtime api tracer (#15301) · a83e4704
      Dun 提交于
      * refine profiler && add runtime tracer
      
      * test=develop
      
      * test=develop
      
      * test=develop
      
      * test=develop
      
      * test=develop
      
      * test=develop
      
      * test=develop
      
      * test=develop
      
      * fix bug && test=develop
      
      * add thread id map && test=develop
      
      * test=develop
      
      * testing
      
      * bug fix
      
      * remove cuda event && refine code && test=develop
      
      * test=develop
      
      * test=develop
      
      * test=develop
      
      * fix windows temp file && test=develop
      
      * test=develop
      
      * fix windows bug && test=develop
      
      * fix start up issue && test=develop
      
      * code polish &&  test=develop
      
      * remove unused code && test=develop
      
      * add some cupti cbid && test=develop
      
      * add FLAGS_multiple_of_cupti_buffer_size && test=develop
      
      * fix compile error && test=develop
      
      * add keyword && test=develop
      
      * fix && test=develop
      
      * code polish && test=develop
      a83e4704
  5. 20 2月, 2019 2 次提交
  6. 19 2月, 2019 3 次提交
  7. 14 2月, 2019 1 次提交
  8. 11 2月, 2019 2 次提交
  9. 03 2月, 2019 1 次提交
  10. 02 2月, 2019 2 次提交
  11. 31 1月, 2019 2 次提交
  12. 28 1月, 2019 2 次提交
  13. 24 1月, 2019 2 次提交
    • Y
      Add the CUDA kernel for beam_search op (#15020) · 3008fa12
      Yiqun Liu 提交于
      * Refine the beam_search op and test.
      
      * A basic CUDA implementation of beam_search for small batch_size.
      
      * Implement CUDA kernel for beam_search_op.
      
      * Use multiple CUDA threads in the same block to select the top beam.
      
      * Update the python api of beam_search op.
      
      * Enable extend function in CPU kernel of beam_search op.
      
      * Unify the CUDA codes.
      test=develop
      
      * Unify the CPU kernel of beam_search op.
      
      * Ensure the seletced items of beam_search_op's CPU kernel sorted by scores.
      
      * Update the description of beam_search in API.spec.
      
      * Enable the use of CUDA kernel in beam_search op.
      
      * Exclude the beam_search's CUDA unittest when there is no CUDA gpu, and delete some debuging statements.
      test=develop
      
      * Follow comments.
      test=develop
      
      * Call the CPU kernel for beam_search op when batch_size > 4.
      test=develop
      
      * Remove the except of is_empty op in PrepareData.
      test=develop
      3008fa12
    • S
      lazy_allocator · 51227bd4
      sneaxiy 提交于
      test=develop
      51227bd4
  14. 23 1月, 2019 1 次提交
  15. 16 1月, 2019 1 次提交
  16. 14 1月, 2019 2 次提交
  17. 11 1月, 2019 4 次提交
  18. 10 1月, 2019 1 次提交
    • X
      Conv int8 residual (#15145) · 8f17c714
      xiaolil1 提交于
      * Enable basic MKL-DNN INT8 Conv OP
      test=develop
      
      * Modify test case
      test=develop
      
      * Clean unittest code
      test=develop
      
      * Fix test
      test=develop
      
      * Modify test
      test=develop
      
      * Enable MKL-DNN INT8 Conv with Relu Fusion OP
      test=develop
      
      * Enable INT8 Conv with residual fusion OP
      test=develop
      
      * Modify code.
      test=develop
      
      * Modify basic INT8 Conv
      test=develop
      
      * Modify Conv.
      test=develop
      
      * fix style
      test=develop
      
      * Fix style
      test=develop
      
      * Fix test
      test=develop
      
      * Modify code.
      test=develop
      
      * Fix test
      test=develop
      8f17c714