Skip to content

[gemini] add static placement policy #4442

@ver217

Description

@ver217

Overview

Current cuda and cpu placement policy cannot set the fraction of sharded params/offloaded params. Moreover, it cannot set optimizer offload and parameter offload independently.

Thus, we can add a static placement policy, through which users can set the fraction of sharded params, offloaded optimizer states and offloaded params.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

Status

Done

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions