DirectXShaderCompiler中SPIR-V生成错误的isnan()函数问题分析

问题概述

在DirectXShaderCompiler项目中，当使用HLSL编写着色器代码并编译为SPIR-V时，发现了一个关于isnan()函数的生成错误。具体表现为编译器生成的SPIR-V代码中，OpIsNan指令使用了错误的返回类型，导致SPIR-V验证失败。

问题重现

问题可以通过以下简单的计算着色器代码重现：

RWStructuredBuffer<float> Out;
[numthreads(1, 1, 1)]
void main(uint id : SV_DispatchThreadID)
{
    bool bar = isnan(Out[id]);
    Out[id] = (float)bar;
}

当使用dxc编译器以SPIR-V为输出目标编译上述代码时，会生成无效的SPIR-V代码，具体错误是OpIsNan指令使用了uint类型作为返回类型，而SPIR-V规范要求OpIsNan必须返回布尔类型的标量或向量。

技术分析

SPIR-V规范要求

根据SPIR-V规范，OpIsNan指令必须满足以下条件：

1. 结果类型必须是布尔标量或布尔向量
2. 操作数必须是浮点标量或浮点向量

在错误案例中，编译器错误地将OpIsNan的结果类型生成为uint，这直接违反了SPIR-V规范。

问题根源

通过分析可以发现，问题出在编译器处理isnan()函数调用时的类型推导阶段。当isnan()直接作用于缓冲区访问表达式Out[id]时，编译器未能正确推导出返回类型应为布尔值。而当中间加入一个临时变量存储Out[id]的值时，类型推导则能正常工作。

临时解决方案

在问题修复前，开发者可以采用以下临时解决方案：

1. 使用中间变量存储值后再调用isnan()：

float temp = Out[id];
bool bar = isnan(temp);

2. 直接使用SPIR-V内联指令：

[[vk::ext_instruction(156)]]
bool alt_isnan(float v1);

编译器修复

项目维护者已经提交了修复代码，主要修正了类型推导阶段的逻辑错误，确保isnan()函数调用始终生成正确的布尔类型返回值的SPIR-V指令。修复后，编译器将能够正确处理原始代码中的直接缓冲区访问表达式。

开发者建议

对于使用DirectXShaderCompiler生成SPIR-V的开发者，建议：

1. 保持编译器版本更新，及时获取错误修复
2. 在遇到类似问题时，可以考虑添加中间变量作为临时解决方案
3. 关注SPIR-V验证器的错误信息，它们通常能提供有价值的调试线索
4. 对于关键任务代码，建议在重要变更后进行SPIR-V验证

这个问题展示了HLSL到SPIR-V转换过程中类型系统处理的重要性，也提醒开发者在跨平台着色器开发中需要注意不同中间表示规范的具体要求。