Skip to content

add max nbor size from 1024 to 4096#227

Merged
amcadmus merged 1 commit intodeepmodeling:develfrom
denghuilu:devel-submit
May 27, 2020
Merged

add max nbor size from 1024 to 4096#227
amcadmus merged 1 commit intodeepmodeling:develfrom
denghuilu:devel-submit

Conversation

@denghuilu
Copy link
Member

1、use TensorFlow API allocates temp memory instead of using cudaMalloc.
2、due to the shared memory limit, max nbor size is limited to 4096 currently.

@amcadmus amcadmus merged commit c7f8473 into deepmodeling:devel May 27, 2020
njzjz pushed a commit to njzjz/deepmd-kit that referenced this pull request Sep 21, 2023
fix bug: transfer back the tf checkpoints
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants