请选择 进入手机版 | 继续访问电脑版
查看: 1357|回复: 0

at_check与torch_check

[复制链接]

664

主题

1232

帖子

6445

积分

xdtech

Rank: 5Rank: 5

积分
6445
发表于 2020-6-20 12:01:02 | 显示全部楼层 |阅读模式
新版pytorch 1.5,

已经抛弃AT_CHECK
换用TORC_CHECK

很多CUDA CPP在pytorch 1.5编译,会遇到“identifier AT_CHECK is undefiened”错误
用TORCH_CHECK替换即可

此外,AT_ERROR仍然保留。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表