run the new tests under test/nn (#3955)

* run the new tests

* add torch_pin

* run test only for dropout

* update skip

* make yapf happy

* Delete .torch_pin

Co-authored-by: JackCaoG <[email protected]>

Author: kshitij12345

test/pytorch_test_base.py

@@ -268,6 +268,10 @@
         'test_upsamplingBicubic2d_correctness_xla',  # FIXME! Got dtypes torch.float32 and torch.float64
         'test_conv3d_same_padding_backward_xla',  # XLA tensors do not have storage,
         'test_CTCLoss_no_batch_dim_xla',  # Value out of range
+    },
+
+    # test/nn/test_dropout.py
+    'TestDropoutNNDeviceTypeXLA': {
         'test_Dropout2d_xla',  # Started to pass
     },
 

test/run_tests.sh

@@ -94,6 +94,7 @@ function run_op_tests {
   run_test python3 "$CDIR/../../test/test_indexing.py" "$@" -v TestIndexingXLA
   run_test python3 "$CDIR/../../test/test_indexing.py" "$@" -v NumpyTestsXLA
   run_dynamic python3 "$CDIR/../../test/test_nn.py" "$@" -v TestNNDeviceTypeXLA
+  run_dynamic python3 "$CDIR/../../test/nn/test_dropout.py" "$@" -v TestDropoutNNDeviceTypeXLA
   run_dynamic python3 "$CDIR/../../test/test_type_promotion.py" "$@" -v TestTypePromotionXLA
   run_dynamic python3 "$CDIR/test_operations.py" "$@" --verbosity=$VERBOSITY
   run_opbyop python3 "$CDIR/test_operations.py" "$@" --verbosity=$VERBOSITY