1. 09 4月, 2022 1 次提交
  2. 30 3月, 2022 1 次提交
    • F
      Add new APIs for GPU memory monitoring (max_memory_allocated,... · afe02e9d
      From00 提交于
      Add new APIs for GPU memory monitoring (max_memory_allocated, max_memory_reserved, memory_allocated, memory_reserved) (#38657)
      
      * Add new API memory_reserved
      
      * Add memory_allocated, max_memory_reserved and max_memory_allocater
      
      * Fix CI error
      
      * Fix CI error
      
      * Enhance UT
      
      * Add FLAGS_memory_stats_opt
      
      * Add STATS macro functions
      
      * Add StatAllocator
      
      * Fix CI errors
      
      * Add UT
      
      * Fix CI errors
      afe02e9d
  3. 20 2月, 2022 1 次提交
  4. 08 2月, 2022 1 次提交
    • F
      Support allocate CUDA managed memory (#39075) · 42910361
      From00 提交于
      * Rough implementation for experiment
      
      * Support allocate cuda managed memory
      
      * Fix CI error
      
      * Modify UT
      
      * Check whether support memory oversubscription
      
      * Fix ROCM Compile error
      
      * Fix ROCM Compile error
      
      * Fix UT cuda_managed_memory_test
      
      * Set UT timeout to 40
      
      * Add UT OOMExceptionTest
      
      * Set UT timeout to 50
      42910361
  5. 06 2月, 2022 1 次提交
  6. 28 1月, 2022 1 次提交
  7. 17 12月, 2021 1 次提交
    • F
      Get base pointer from Allocation (#37978) · 431a2d6a
      From00 提交于
      * Get GPU BasePtr from CUDA allocation
      
      * Fix compile error for ROCm
      
      * Add BasePtr function for IPUPlace in naive_best_fit_allocator.cc
      
      * Add alignment for BuddyAllocator
      
      * Set address alignment of BuddyAllocator to 32 bytes
      
      * Fix CI error
      
      * Remove code for naive_best_fit strategy
      431a2d6a
  8. 03 12月, 2021 1 次提交