• HELSON's avatar
    [zero] add constant placement policy (#1705) · 1468e4bc
    HELSON authored
    * fixes memory leak when paramter is in fp16 in ZeroDDP init.
    * bans chunk releasement in CUDA. Only when a chunk is about to offload, it is allowed to release.
    * adds a constant placement policy. With it, users can allocate a reserved caching memory space for parameters.
    1468e4bc
test_zeroddp_state_dict.py 4.4 KB