跳到正文
原文
AGI Hunt· cfregly·· 4 小时前AI 评分47

PyTorch 嵌套图断点开关实测:B200 上稳态提速 4.53 倍

AI 导读

开发者 cfregly 在 NVIDIA B200 上用 PyTorch 2.14.1(CUDA 13.0)实测 nestedgraphbreaks=True 开关,配合 torch.compile 在稳态下代码运行快 4.53×。

来源:AGI Hunt · agihunt.info