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)