clab / clab/dynet

Per device memory allocation

Open
#1,480 0 comments 0 reactions 0 assignees View on GitHub
enhancement
Dominant language
C++
Stars
3.4k
Forks
701
PR merge metrics
No merged PRs in 30d

Description

Hi,

I have a multi-device network with custom nodes in the CG. As of now my nodes use a memory allocator I made, but I want to move the one of dynet (i.e. use aux_memory). They are based on dynamic programming so they need a lot of memory, which is not alway the case of the rest of the network that runs on GPU.

Dynet provides a command line option for memory allocation but it cannot be specified per device: `--dynet-mem FOR,BACK,PARAM,SCRATCH`.
Is there any reason for that?

Best,
Caio

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by tracing how the `--dynet-mem FOR,BACK,PARAM,SCRATCH` option is handled and how `aux_memory` is used for custom nodes. Determine whether memory settings can be scoped to individual devices and clarify the expected behavior for multi-device networks. Done should mean a defined, tested per-device allocation mechanism, but the issue does not name relevant files or tests.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.