get_build_directory

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

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

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

代码示例

  1. from paddle.utils.cpp_extension import get_build_directory
  2. build_dir = get_build_directory()
  3. print(build_dir)