at_check与torch_check

665 主题	1234 帖子	6568 积分

Rank: 5 Rank: 5

发表于 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仍然保留。