[GO] add two cgo api, test=develop (#29659)

PaddlePaddle · Dec 17, 2020 · 7684b91 · 7684b91
1 parent af8ded7
commit 7684b91
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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))
 }