diff --git a/go/paddle/config.go b/go/paddle/config.go index dcdb64008be775d2fe83069b1b05eb3458159c4c..89f7d7e63ff2a858f058ad22ea424b29f66a4477 100644 --- a/go/paddle/config.go +++ b/go/paddle/config.go @@ -94,6 +94,10 @@ func (config *AnalysisConfig) MemoryPoolInitSizeMb() int { return int(C.PD_MemoryPoolInitSizeMb(config.c)) } +func (config *AnalysisConfig) FractionOfGpuMemoryForPool() float32 { + return float32(C.PD_FractionOfGpuMemoryForPool(config.c)) +} + func (config *AnalysisConfig) EnableCudnn() { C.PD_EnableCUDNN(config.c) } @@ -142,6 +146,10 @@ func (config *AnalysisConfig) EnableMkldnn() { C.PD_EnableMKLDNN(config.c) } +func (config *AnalysisConfig) MkldnnEnabled() bool { + return ConvertCBooleanToGo(C.PD_MkldnnEnabled(config.c)) +} + func (config *AnalysisConfig) SetCpuMathLibraryNumThreads(n int) { C.PD_SetCpuMathLibraryNumThreads(config.c, C.int(n)) }