你遇到的是C++静态成员变量未定义导致的链接错误，错误信息里的undefined symbol: _ZN3xla3gpu17DeviceAssignTable17deviceAssignTableE直接指向你的DeviceAssignTable类里的静态成员deviceAssignTable——你只在头文件里声明了它，但没有在.cpp文件中定义它，链接器找不到这个变量的实际内存分配，所以抛出了这个错误。

问题根源 #

在C++中，类的静态成员变量（比如你的static DeviceAssignTable* deviceAssignTable;）属于类本身，而非类的实例。你在头文件里只是声明了这个变量，但必须在类外部的.cpp文件中定义它，否则链接阶段会找不到对应的符号。

解决方案步骤 #

1. 创建对应的.cpp文件并定义静态成员

在tensorflow/compiler/xla/service/gpu目录下创建ndpx_device_table.cc文件，内容如下：

#include "tensorflow/compiler/xla/service/gpu/ndpx_device_table.h"

// 定义静态成员变量，这是解决链接错误的关键！
namespace xla {
namespace gpu {
DeviceAssignTable* DeviceAssignTable::deviceAssignTable = nullptr;

// 实现类的成员函数
void DeviceAssignTable::AssignDevice(int hlo_id, Device device) {
  device_table_[hlo_id] = device;
}

Device DeviceAssignTable::FindAssignInfo(int hlo_id) {
  auto it = device_table_.find(hlo_id);
  if (it != device_table_.end()) {
    return it->second;
  }
  // 可根据业务需求返回默认设备，示例返回GPU
  return Device::GPU;
}
} // namespace gpu
} // namespace xla

2. 检查Bazel BUILD文件配置

确保你的BUILD文件里的cc_library包含了这个新的.cpp文件。比如在tensorflow/compiler/xla/service/gpu/BUILD中，新增一个单独的库或者把文件加入到已有库中：

# 新增单独的设备表库
cc_library(
    name = "ndpx_device_table",
    srcs = ["ndpx_device_table.cc"],
    hdrs = ["ndpx_device_table.h"],
    visibility = ["//tensorflow/compiler/xla/service:friends"],
    deps = [
        "//tensorflow/compiler/xla/service:hlo",
        # 按需添加其他XLA基础依赖
    ],
)

# 确保使用这个单例的优化Pass依赖该库
cc_library(
    name = "your_custom_optimization_pass",
    srcs = ["your_custom_pass.cc"],
    hdrs = ["your_custom_pass.h"],
    deps = [
        ":ndpx_device_table",
        # 其他必要依赖
    ],
)

3. 清理缓存并重新构建

为避免Bazel缓存导致的问题，先清理缓存再重新构建：

bazel clean --expunge
bazel build //tensorflow/python:pywrap_tensorflow_internal

为什么之前修改其他Pass没遇到？ #

其他优化Pass可能没有使用静态成员变量，或者已经在对应.cpp文件中正确定义了静态成员。而你这次的单例模式依赖静态指针，必须显式定义才能让链接器找到它的符号。

内容的提问来源于stack exchange，提问作者Jueon Park