get_build_directory

paddle.utils.cpp_extension. get_build_directory ( ) [源代码]

此接口返回编译自定义 OP 时生成动态链接库所在的 build 目录路径。此目录可以通过 export PADDLE_EXTENSION_DIR=XXX 来设置。若未设定,则默认使用 ~/.cache/paddle_extension 作为 build 目录。

返回

编译自定义 OP 的 build 目录路径。

代码示例

from paddle.utils.cpp_extension import get_build_directory

build_dir = get_build_directory()
print(build_dir)