编译TensorFlow 2.20.0（ROCM环境+ -march=native）时遭遇FFT模块重载歧义错误

问题背景 #

我正在从官方仓库的v2.20.0分支源码编译TensorFlow 2.20.0，环境是ROCM，并且使用了-march=native编译选项。具体配置如下：

• 环境变量：

  export HERMETIC_PYTHON_VERSION=3.13
  export ROCM_PATH=/opt/rocm/
  export PATH="$PATH:/opt/rocm/bin"
  

• 工具链版本：

  • Python 3.13
  • Clang 18.1.8
  • Bazel 7.4.1（通过bazelisk）

• 编译命令：

  bazel build //tensorflow/tools/pip_package:wheel --repo_env=USE_PYWRAP_RULES=1 --repo_env=WHEEL_NAME=tensorflow_cpu --copt=-w --config=nonccl -j 16
  

• configure配置文件：

  build --action_env PYTHON_BIN_PATH="/usr/bin/python3.13"
  build --action_env PYTHON_LIB_PATH="/usr/lib/python3.13/site-packages"
  build --python_path="/usr/bin/python3.13"
  build --config=rocm
  build --action_env ROCM_PATH="/opt/rocm/"
  build --action_env CLANG_COMPILER_PATH="/usr/lib/llvm18/bin/clang-18"
  build --repo_env=CC=/usr/lib/llvm18/bin/clang-18
  build --repo_env=BAZEL_COMPILER=/usr/lib/llvm18/bin/clang-18
  build:opt --copt=-march=native
  build:opt --host_copt=-march=native
  test --test_size_filters=small,medium
  test --test_env=LD_LIBRARY_PATH
  test:v1 --test_tag_filters=-benchmark-test,-no_oss,-oss_excluded,-no_gpu,-oss_serial
  test:v1 --build_tag_filters=-benchmark-test,-no_oss,-oss_excluded,-no_gpu
  test:v2 --test_tag_filters=-benchmark-test,-no_oss,-oss_excluded,-no_gpu,-oss_serial,-v1only
  test:v2 --build_tag_filters=-benchmark-test,-no_oss,-oss_excluded,-no_gpu,-v1only
  

编译到FFT模块时持续失败，核心报错是函数重载歧义：

call of overloaded ‘raw(ptrdiff_t)’ is ambiguous
if (src == &dst.raw(it.oofs(0))) return; // in-place
bazel-out/k8-opt/bin/external/ducc/_virtual_includes/fft/ducc/src/ducc0/fft/fftnd_impl.h:486:22: note: there are 2 candidates
external/ducc/src/ducc0/infra/mav.h:118:35: note: candidate 1: ‘const T& ducc0::detail_mav::cmembuf::raw(I) const [with I = long int; T = double]’
118 | template const T &raw(I i) const

解决方案建议 #

看起来这是TensorFlow依赖的ducc库在Clang 18下遇到的模板函数推导歧义问题，新版本编译器对模板参数的检查更严格，加上-march=native的指令集优化可能加剧了类型匹配的冲突。这里给你几个实用的解决方向：

• 降级Clang版本到17
  TensorFlow 2.20.0的官方兼容文档里，推荐的Clang版本大概率还没到18，新版本编译器的严格检查可能和ducc的旧代码不兼容。你可以切换到Clang 17，重新配置环境变量和configure文件，再尝试编译。

• 手动修改ducc源码消除歧义
  如果不想降级编译器，直接修改报错的代码是最直接的方式：找到报错文件bazel-out/k8-opt/bin/external/ducc/_virtual_includes/fft/ducc/src/ducc0/fft/fftnd_impl.h的第486行，把dst.raw(it.oofs(0))改成dst.raw(static_cast<long int>(it.oofs(0)))——通过显式类型转换，告诉编译器该用哪个重载版本，就能消除歧义。修改后重新执行编译命令即可。

• 替换-march=native为具体架构参数
  -march=native会启用当前CPU的所有指令集，可能导致ptrdiff_t等类型的宽度变化，触发模板推导问题。你可以换成对应AMD CPU的具体架构参数，比如-march=znver4（根据你的CPU型号调整），更新configure里的-march=native后重新编译。

• 尝试降级Python版本
  Python 3.13是比较新的版本，可能和TensorFlow 2.20.0的某些依赖存在隐性兼容性问题。你可以尝试切换到Python 3.11或3.12，调整环境变量和configure配置后再试。

备注：内容来源于stack exchange，提问作者Eugene S