Fix the failed unit test on A100 (#1720)
Summary:
Print detailed test info in unit testing.
Skip tests failed because of CPU OOM.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1720
Reviewed By: FindHao
Differential Revision: D46564352
Pulled By: xuzhao9
fbshipit-source-id: 5f0a5617acb29ac03470b7a3bdd76985529cad20
