未验证 提交 7684b918 编写于 作者: Q Qi Li 提交者: GitHub

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

上级 af8ded77
...@@ -94,6 +94,10 @@ func (config *AnalysisConfig) MemoryPoolInitSizeMb() int { ...@@ -94,6 +94,10 @@ func (config *AnalysisConfig) MemoryPoolInitSizeMb() int {
return int(C.PD_MemoryPoolInitSizeMb(config.c)) return int(C.PD_MemoryPoolInitSizeMb(config.c))
} }
func (config *AnalysisConfig) FractionOfGpuMemoryForPool() float32 {
return float32(C.PD_FractionOfGpuMemoryForPool(config.c))
}
func (config *AnalysisConfig) EnableCudnn() { func (config *AnalysisConfig) EnableCudnn() {
C.PD_EnableCUDNN(config.c) C.PD_EnableCUDNN(config.c)
} }
...@@ -142,6 +146,10 @@ func (config *AnalysisConfig) EnableMkldnn() { ...@@ -142,6 +146,10 @@ func (config *AnalysisConfig) EnableMkldnn() {
C.PD_EnableMKLDNN(config.c) C.PD_EnableMKLDNN(config.c)
} }
func (config *AnalysisConfig) MkldnnEnabled() bool {
return ConvertCBooleanToGo(C.PD_MkldnnEnabled(config.c))
}
func (config *AnalysisConfig) SetCpuMathLibraryNumThreads(n int) { func (config *AnalysisConfig) SetCpuMathLibraryNumThreads(n int) {
C.PD_SetCpuMathLibraryNumThreads(config.c, C.int(n)) C.PD_SetCpuMathLibraryNumThreads(config.c, C.int(n))
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册