Metal着色器编译入门遇阻：如何创建可动态链接的MTLLibrary？

要生成包含未定义extern函数、可在运行时动态绑定的Metal库，核心是调整编译参数以允许库中存在未解析符号，以下是具体实现步骤：

1. 编译生成AIR中间文件 #

先用xcrun metal的-c参数将Metal代码编译为AIR（Apple Intermediate Representation）文件，这一步仅做编译不链接，不会因未定义符号报错：

xcrun metal -c shader.metal -o shader.air

2. 生成“部分绑定”的Metal库 #

直接用xcrun metallib编译AIR文件会触发未定义符号错误，需添加--allow-undeclared-identifiers参数，允许库中保留未解析的符号引用：

xcrun metallib --allow-undeclared-identifiers shader.air -o shader.metallib

生成的shader.metallib将保留对add函数的引用，且不会触发链接错误。

运行时绑定extern函数的两种方式 #

方式一：通过MTLDynamicLibrary加载动态库 #

1. 编写实现add函数的Metal代码（示例文件impl.metal）：

uint add(uint a, uint b) {
    return a + b;
}

2. 将其编译为Metal动态库，需使用-dynamic-library参数：

xcrun metal -dynamic-library impl.metal -o impl.dylib

3. 运行时加载动态库并绑定到计算管道：

// 加载主业务库
id<MTLLibrary> shaderLibrary = [device newLibraryWithURL:[NSURL fileURLWithPath:@"shader.metallib"] error:nil];
id<MTLFunction> kernelFunc = [shaderLibrary newFunctionWithName:@"kernel_func"];

// 加载包含符号实现的动态库
id<MTLDynamicLibrary> dynamicLib = [device newDynamicLibraryWithURL:[NSURL fileURLWithPath:@"impl.dylib"] error:nil];

// 配置管道描述符，预加载动态库完成符号解析
MTLComputePipelineDescriptor *pipelineDesc = [[MTLComputePipelineDescriptor alloc] init];
pipelineDesc.computeFunction = kernelFunc;
pipelineDesc.preloadedLibraries = @[dynamicLib];

// 创建计算管道
NSError *error = nil;
id<MTLComputePipelineState> pipeline = [device newComputePipelineStateWithDescriptor:pipelineDesc error:&error];

方式二：通过MTLLinkedFunctions绑定单个函数 #

这种方式无需生成完整动态库，可直接绑定指定函数，灵活性更高：

1. 编译实现add函数的Metal代码为普通Metal库：

xcrun metallib impl.metal -o impl.metallib

2. 运行时提取目标函数，通过MTLLinkedFunctions完成符号绑定：

// 加载主业务库与符号实现库
id<MTLLibrary> shaderLibrary = [device newLibraryWithURL:[NSURL fileURLWithPath:@"shader.metallib"] error:nil];
id<MTLLibrary> implLibrary = [device newLibraryWithURL:[NSURL fileURLWithPath:@"impl.metallib"] error:nil];

id<MTLFunction> kernelFunc = [shaderLibrary newFunctionWithName:@"kernel_func"];
id<MTLFunction> addFunc = [implLibrary newFunctionWithName:@"add"];

// 创建LinkedFunctions并绑定符号映射
MTLLinkedFunctions *linkedFuncs = [[MTLLinkedFunctions alloc] init];
[linkedFuncs setFunction:addFunc forName:@"add"];

// 配置管道描述符
MTLComputePipelineDescriptor *pipelineDesc = [[MTLComputePipelineDescriptor alloc] init];
pipelineDesc.computeFunction = kernelFunc;
pipelineDesc.linkedFunctions = linkedFuncs;

// 创建计算管道
NSError *error = nil;
id<MTLComputePipelineState> pipeline = [device newComputePipelineStateWithDescriptor:pipelineDesc error:&error];

关于WWDC示例的差异说明 #

WWDC2021提到的extern技术，核心是将符号解析延迟到运行时，允许编译阶段保留未定义符号；而早期的Dynamic Library示例通常直接生成包含完整符号的动态库，通过install name机制在运行时自动加载，无需额外声明extern——二者是动态绑定的不同实现路径，前者更适合按需绑定符号，后者侧重完整动态库的加载。

内容的提问来源于stack exchange，提问作者Joseph Johnston