pytorch
41bdfe39 - [ONNX] Fix cuda test case (#63597) (#64378)

Commit
4 years ago
[ONNX] Fix cuda test case (#63597) (#64378) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/64378 * skip script test for unsupported autocast. * Fix test case by adding missed `autocast` and `model.cuda()`. Test Plan: Imported from OSS Reviewed By: jansel Differential Revision: D30919600 Pulled By: malfet fbshipit-source-id: 3231fc672d97de487d6e4460626df0ba25f212ce Co-authored-by: BowenBao <bowbao@microsoft.com>
Author
Parents
Loading