Fix throw abort

[hjchen2]

No description provided.

[github-actions]

Speed stats:

GPU Name: GeForce GTX 1080 

PyTorch resnet50 time: 139.8ms (= 6988.4ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 127.8ms (= 6388.3ms / 50, input_shape=[16, 3, 224, 224], backward is enabled)
Relative speed: 1.09 (= 139.8ms / 127.8ms)

PyTorch resnet50 time: 85.0ms (= 4252.4ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 74.2ms (= 3712.4ms / 50, input_shape=[8, 3, 224, 224], backward is enabled)
Relative speed: 1.15 (= 85.0ms / 74.2ms)

PyTorch resnet50 time: 57.7ms (= 2884.7ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 48.0ms (= 2399.1ms / 50, input_shape=[4, 3, 224, 224], backward is enabled)
Relative speed: 1.20 (= 57.7ms / 48.0ms)

PyTorch resnet50 time: 47.9ms (= 2392.9ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 41.8ms (= 2087.9ms / 50, input_shape=[2, 3, 224, 224], backward is enabled)
Relative speed: 1.15 (= 47.9ms / 41.8ms)

PyTorch resnet50 time: 41.0ms (= 2049.0ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
OneFlow resnet50 time: 41.1ms (= 2056.5ms / 50, input_shape=[1, 3, 224, 224], backward is enabled)
Relative speed: 1.00 (= 41.0ms / 41.1ms)