提交 ab798a28 编写于 作者: X Xin Pan

clarify the fraction_of_gpu_memory flag

test=develop
上级 d770b9bd
...@@ -46,6 +46,7 @@ struct RWLock { ...@@ -46,6 +46,7 @@ struct RWLock {
private: private:
pthread_rwlock_t lock_; pthread_rwlock_t lock_;
}; };
// TODO(paddle-dev): Support RWLock for WIN32 for correctness.
#else #else
// https://stackoverflow.com/questions/7125250/making-pthread-rwlock-wrlock-recursive // https://stackoverflow.com/questions/7125250/making-pthread-rwlock-wrlock-recursive
// In windows, rw_lock seems like a hack. Use empty object and do nothing. // In windows, rw_lock seems like a hack. Use empty object and do nothing.
......
...@@ -20,8 +20,11 @@ limitations under the License. */ ...@@ -20,8 +20,11 @@ limitations under the License. */
#include "paddle/fluid/platform/enforce.h" #include "paddle/fluid/platform/enforce.h"
DEFINE_double(fraction_of_gpu_memory_to_use, 0.92, DEFINE_double(fraction_of_gpu_memory_to_use, 0.92,
"Default use 92% of GPU memory for PaddlePaddle," "Allocate a trunk of gpu memory that is this fraction of the "
"reserve the rest for page tables, etc"); "total gpu memory size. Future memory usage will be allocated "
"from the trunk. If the trunk doesn't have enough gpu memory, "
"additional trunks of the same size will be requested from gpu "
"until the gpu has no memory left for another trunk.");
namespace paddle { namespace paddle {
namespace platform { namespace platform {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册