• 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
cuda_managed_allocator.cc 3.2 KB