1. 14 5月, 2020 1 次提交
  2. 11 9月, 2019 1 次提交
    • H
      Replace TemporaryAllocator by CUDADeviceContextAllocator (#18989) · 12542320
      Huihuang Zheng 提交于
      TemporaryAllocator is a singleton used for allocating memory for Cudnn. Since it is a singleton, we can delete it for better performance in memory.
      
      We replace TemporaryAllocator by CUDADeviceContextAllocator and CUDADeviceContextAllocation, which uses stream callback to delete the memory allocated for the stream to avoid singleton.
      
      Also added data_feed_proto to operator to fix CI in CPU compilation
      12542320
  3. 30 8月, 2019 1 次提交
  4. 14 8月, 2019 1 次提交
  5. 25 5月, 2019 1 次提交
  6. 26 3月, 2019 1 次提交
  7. 12 11月, 2018 1 次提交
    • Y
      Fix gather & stack op (#14355) · bd294378
      Yibing Liu 提交于
      * Add int type support for stack_op
      
      * Improve gather op to support index with shape N x 1
      
      test=develop
      
      * Fix stack_op kernel's registry
      
      test=develop
      bd294378
  8. 12 2月, 2018 1 次提交
  9. 10 2月, 2018 2 次提交
  10. 26 12月, 2017 1 次提交
  11. 12 12月, 2017 1 次提交
    • Q
      Refine device context (#6433) · 61ec0b95
      QI JUN 提交于
      There are mainly following fixes:
      
      - take `DeviceContext` as the template parameter of math functors and OpKernel instead of `Place`
      - remove `eigen_device` interface in base class  `DeviceContext`
      - remove `GetEigenDevice` interface in `ExecutionContext` and base class `DeviceContext`
      - remove unused `platform::EigenDeviceConverter`
      - rename `REGISTER_OP_GPU_KERNEL` to `REGISTER_OP_CUDA_KERNEL`
      - rename `USE_GPU_ONLY_OP` to `USE_CUDA_ONLY_OP`
      61ec0b95
  12. 04 10月, 2017 1 次提交
  13. 03 10月, 2017 1 次提交
  14. 29 9月, 2017 2 次提交