• ver217's avatar
    [nvme] CPUAdam and HybridAdam support NVMe offload (#1360) · c415240d
    ver217 authored
    * impl nvme optimizer
    
    * update cpu adam
    
    * add unit test
    
    * update hybrid adam
    
    * update docstr
    
    * add TODOs
    
    * update CI
    
    * fix CI
    
    * fix CI
    
    * fix CI path
    
    * fix CI path
    
    * fix CI path
    
    * fix install tensornvme
    
    * fix CI
    
    * fix CI path
    
    * fix CI env variables
    
    * test CI
    
    * test CI
    
    * fix CI
    
    * fix nvme optim __del__
    
    * fix adam __del__
    
    * fix nvme optim
    
    * fix CI env variables
    
    * fix nvme optim import
    
    * test CI
    
    * test CI
    
    * fix CI
    c415240d
test_nvme.py 1.88 KB