Fix test_sparse_norm_op failure. (#55405) (#55637)
test_sparse_norm_op的case在某些cudnn版本中,sparse和dense的batch norm计算结果存在较大diff。同样的结果可以在pytorch中复现,将数据dump之后用cuda api调用cudnn计算也可以复现。可以确认是cudnn版本导致的单测失败。因此暂时在单测中替换测试case,后续等nv有反馈后再对代码做进一步修改。
Showing
想要评论请 注册 或 登录