compile SageAttention faing error C2872: “std” for latest torch nightly

[NOFOX]

ENV: Win11 ,VS2022, Torch:
print(torch.version)
2.7.0.dev20250302+cu128
print(torchvision.version)
0.22.0.dev20250302+cu128
print(torch.cuda.is_available())
True
print(torch.cuda.get_device_name(0))
NVIDIA GeForce RTX 5080
print(torch.cuda.get_device_capability(0))
(12, 0)

compile SageAttention faing error C2872: “std”
C:/tools/AI/ComfyUI-aki-v1.4/python/lib/site-packages/torch/include\torch/csrc/dynamo/compiled_autograd.h(1076): error C2872: “std”: 不明确的符号
C:/Program Files/Microsoft Visual Studio/2022/Community/VC/Tools/MSVC/14.43.34808/include\valarray(20): note: 可能是“std”
C:/tools/AI/ComfyUI-aki-v1.4/python/lib/site-packages/torch/include\torch/csrc/dynamo/compiled_autograd.h(1076): note: 或 “std”
C:/tools/AI/ComfyUI-aki-v1.4/python/lib/site-packages/torch/include\torch/csrc/dynamo/compiled_autograd.h(1076): note: 模板实例化上下文(最早的实例化上下文)为
C:/tools/AI/ComfyUI-aki-v1.4/python/lib/site-packages/torch/include\torch/csrc/dynamo/compiled_autograd.h(1121): note: 查看对正在编译的 类 模板 实例化“torch::dynamo::autograd::IValuePacker<__int64>”的引用
C:/tools/AI/ComfyUI-aki-v1.4/python/lib/site-packages/torch/include\torch/csrc/dynamo/compiled_autograd.h(1059): note: 在编译 类 模板 成员函数“c10::TypePtr torch::dynamo::autograd::IValuePacker<__int64>::packed_type(void)”时
C:/tools/AI/ComfyUI-aki-v1.4/python/lib/site-packages/torch/include\torch/csrc/dynamo/compiled_autograd.h(1121): note: 请参阅 "torch::dynamo::autograd::IValuePacker::packed_type" 中对 "torch::dynamo::autograd::IValuePacker<__int64>::packed_type" 的第一个引用
ninja: build stopped: subcommand failed.

cc @peterjc123 @mszhanyi @skyline75489 @nbcsm @iremyux @Blackhex @chauhang @penguinwu

[malfet]

Interesting, looks like yet another manifestation of pytorch/audio#3877

[NOFOX]

Interesting, looks like yet another manifestation of pytorch/audio#3877

Yes. I tried use the latest audio build success breach :36815ef914216a61307477ccff3ee6fb0d7fda20
also has same issue facing cause by pytorch code:

[SirJackYang]

试试这个链接里边的torch2.6以及torchvision2.6，支持NVIDIA GeForce RTX 5080
https://huggingface.co/w-e-w/torch-2.6.0-cu128.nv

[cyyever]

Add #include <type_traits> to torch/csrc/dynamo/compiled_autograd.h

[NOFOX]

Add #include <type_traits> to torch/csrc/dynamo/compiled_autograd.h
After add this header file Issue still there ：

And I don't thing it will working. It not about missing header.it related the std naming pollution:

[NOFOX]

试试这个链接里边的torch2.6以及torchvision2.6，支持NVIDIA GeForce RTX 5080 https://huggingface.co/w-e-w/torch-2.6.0-cu128.nv
谢谢，这个包一早就试过了，不过因为一些其它原因需要切抱到2.7。

[cyyever]

Notice that Windows should go to the upper branch. I have no idea why the else branch was triggered, a quick fix is to remove that branch.

我建议你直接修改C:/tools/AI/ComfyUI-aki-v1.4/python/lib/site-packages/torch/include\torch/csrc/dynamo/compiled_autograd.h，把那些有问题的行删掉。

[malfet]

Add #include <type_traits> to torch/csrc/dynamo/compiled_autograd.h

I've tried that already, didn't really help...

[cyyever]

@malfet The problematic lines are wrapped inside not _WIN32 branch, no idea why they were compiled by MSVC. Is it possible that the macro isn't correct?