VAE error on AMD: torch.AcceleratorError – CUDA error: profiler is not initialized

快速结论：该报错通常出现在 AMD ROCm 环境下使用 ComfyUI 的 VAE 瓦片解码（VAE decode (tiled)）节点时，底层 HIP 报出 VAE error on AMD: torch.AcceleratorError – CUDA error: profiler is not initialized。优先排查 Torch/ROCm 版本组合是否在更新后引入了该问题，并确认是否可通过已知本地补丁绕过。

适用环境：Issue 已确认的环境：Linux Mint 22.3、ROCm 10.1、AMD Radeon RX 7900 XTX、torch 2.14.0+rocm10.1.0、torchvision 0.29.0a0+rocm10.1.0，并使用了 –use-flash-attention；另有 Windows 路径下的 ComfyUI 日志，但未明确完整 Windows 环境版本。

最快修复方案：暂无确认的一步修复方案。Issue 中验证有效的做法是应用评论中给出的 comfy/ops.py 本地补丁，在 AMD HIP 环境且注意力头维度大于 256 时强制使用 SDPBackend.MATH。

注意事项：该补丁属于本地代码修改，不是官方合并修复；升级 ComfyUI 后可能被覆盖或产生冲突。评论者仅报告“That resolved the issue for me”，未覆盖其他显卡、ROCm 版本或工作流，长期稳定性尚未验证。

问题场景

用户在 AMD 显卡上运行 ComfyUI，使用 Krea-2 文本到图像模板，并在工作流中添加或使用“VAE decode (tiled)”节点进行图像生成。更新到较新的 ComfyUI 版本后开始出现该错误；前一天仍可正常使用。用户已尝试禁用自定义节点，问题仍然存在。

报错原文

VAE error on AMD: torch.AcceleratorError - CUDA error: profiler is not initialized
Search for `hipErrorProfilerNotInitialized' in https://rocm.docs.amd.com/projects/HIP/en/latest/index.html for more information.

原因分析

从 Issue 讨论看，问题与 AMD HIP/ROCm 环境下执行 VAE 瓦片解码时的注意力后端选择有关，而不是普通的 NVIDIA CUDA 驱动报错。报错中的 CUDA 字样来自 PyTorch 的错误包装，实际底层提示为 hipErrorProfilerNotInitialized。评论中提供的补丁把 q.shape[-1] > 256 的 HIP 注意力路径改为 SDPBackend.MATH，说明可能原因是当前 SDPA 后端优先级在 AMD + ROCm 组合下触发了不兼容的 profiler/断言路径。该结论仍属于基于补丁有效性的推断，尚未有官方根因说明。

环境排查

确认 PyTorch、torchvision、torchaudio 是否为 ROCm 构建，并记录完整版本，例如评论中的 torch 2.14.0+rocm10.1.0。
确认 ROCm 版本，以及是否与当前 PyTorch 二进制匹配。
确认显卡型号与架构，例如 AMD Radeon RX 7900 XTX。
确认是否启用 –use-flash-attention；评论者虽然使用了该参数，但补丁针对的是 SDPA 后端选择。
确认问题是否只在“VAE decode (tiled)”节点出现，普通 VAE 解码是否正常。
确认是否刚执行过 git pull 更新；评论者明确表示更新后才复发。
禁用自定义节点测试后问题仍存在，可排除多数自定义节点干扰。

解决步骤

先备份 ComfyUI/comfy/ops.py，避免补丁应用失败或后续更新冲突。
在 ComfyUI/comfy/ops.py 中找到 sdpa_kernel(SDPA_BACKEND_PRIORITY, set_priority=True) 所在位置。
按 Issue 评论中的 diff，将后端选择改为：当 q.is_cuda 为真、torch.version.hip 为真且 q.shape[-1] > 256 时，使用 prio = [SDPBackend.MATH]，否则保留原有 SDPA_BACKEND_PRIORITY。
保存文件后重启 ComfyUI，并重新运行 Krea-2 模板中的“VAE decode (tiled)”节点。
如果不想修改源码，可优先尝试关闭或调整与注意力后端相关的启动参数，但 Issue 中未验证该方式有效，不能替代补丁结论。
后续升级 ComfyUI 后需检查该文件是否仍包含补丁，必要时重新评估。

验证方法

重新执行原先会失败的工作流，确认“VAE decode (tiled)”节点不再抛出 VAE error on AMD: torch.AcceleratorError – CUDA error: profiler is not initialized，并能正常输出图像。若仍然报错，记录新的完整 traceback，并确认补丁是否已生效、ComfyUI 是否已重启。