OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
56f324d191
pytorch/caffe2/python/operator_test
History
Luke Yeager 014d1fe5c4 Allow test discovery in caffe2/python/ 
Summary:
These are all essentially no-op changes which allow for nose-style (or pytest-style) test discovery.

With this patch, you can use any of these methods to discover and run tests under `caffe2/python`:
```
python -m unittest discover -p '*test*.py' caffe2/python/
python -m nose caffe2/python/
python -m pytest caffe2/python/
```

Future work:

* Get all of the tests to pass
  * Some seem to be testing operations which don't have GPU implementations
  * I get a segfault unless I set `CUDA_VISIBLE_DEVICES=0`
  * Some tests are flaky
* Allow test discovery throughout the whole project (e.g. the `experiments/` dir)
Closes https://github.com/caffe2/caffe2/pull/199

Reviewed By: pietern

Differential Revision: D4704504

Pulled By: Yangqing

fbshipit-source-id: 8f5687ec9c8aa873dfaff30dbf44272bc38a206b
2017-03-14 18:16:41 -07:00
..
activation_ops_test.py Add LeakyReluOp to caffe 2017-02-15 16:00:45 -08:00
atomic_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
checkpoint_test.py snapshot -> checkpoint 2016-12-15 12:01:30 -08:00
conv_test.py Cudnn v6 2017-02-28 17:46:33 -08:00
conv_transpose_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
copy_ops_test.py Gradient for Copy operator 2017-02-16 06:11:27 -08:00
cosine_embedding_criterion_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
counter_ops_test.py AtomicCounter to return previous value on Reset. 2017-02-02 14:59:30 -08:00
cross_entropy_ops_test.py delete redundant comment lines. 2017-02-24 11:04:36 -08:00
dataset_ops_test.py NextScopedBlob with well-defined behavior and respect namescope 2017-02-16 17:16:36 -08:00
duplicate_operands_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
elementwise_op_broadcast_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
elementwise_ops_test.py Sqr op and gradient 2017-03-07 03:03:07 -08:00
emptysample_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
extend_tensor_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
fc_operator_test.py Test for FC operator + fix for docs 2017-01-27 10:44:24 -08:00
filler_ops_test.py add exception for empty shape param 2017-03-10 00:33:59 -08:00
gather_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
gather_ranges_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
given_tensor_fill_op_test.py support fill bool tensors in GivenTensorFill 2017-03-02 20:18:59 -08:00
group_conv_test.py Make all convolution operators allow optional bias term 2016-12-21 15:14:24 -08:00
hsm_test.py Generate huffman tree 2017-01-19 16:14:23 -08:00
index_ops_test.py Change the schema of IndexLoad & IndexFreeze so that state change is captured by the framework 2017-02-14 10:05:12 -08:00
instance_norm_test.py instance norm test fix 2017-02-25 14:31:42 -08:00
margin_ranking_criterion_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
matmul_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
mkl_conv_op_test.py MKL convolution operator 2017-01-23 09:59:30 -08:00
mkl_packed_fc_op_test.py MKL convolution operator 2017-01-23 09:59:30 -08:00
mkl_speed_test.py MKL convolution operator 2017-01-23 09:59:30 -08:00
momentum_sgd_test.py MomemtumSGDUpdate -- version of MomentumSGD with update. 2016-12-15 12:01:29 -08:00
mpi_test.py Setup MPI before test start 2016-12-19 15:59:32 -08:00
one_hot_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
pack_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
partition_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
piecewise_linear_transform_test.py PiecewiseLinearTransformOp transform binary predictions specially 2017-02-15 16:00:44 -08:00
pooling_test.py Unit test for big batch size avg pooling 2017-01-18 19:29:20 -08:00
pow_op_test.py CUDA version of elementwise power + rename to Pow + gradient 2017-03-07 10:20:40 -08:00
python_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
rank_loss_operator_test.py Normalize rank loss gradient to avoid convergence issues when the number of pairs is really large 2016-12-21 17:29:24 -08:00
record_queue_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
recurrent_network_test.py Implement MILSTM in caffe2 2017-03-09 16:32:47 -08:00
reduce_ops_test.py ReduceBack{Sum|Mean}Op CPU & GPU implementation 2017-03-13 16:19:58 -07:00
relu_op_test.py MKLDevice and MKLOperator 2016-12-15 19:59:24 -08:00
reshape_ops_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
segment_ops_test.py Allow test discovery in caffe2/python/ 2017-03-14 18:16:41 -07:00
sequence_ops_test.py Allow use of ReversePackedSegs operator in CUDA context 2017-03-09 15:03:55 -08:00
shape_inference_test.py Add Shape Inference for Reshape Operator 2017-03-13 14:31:27 -07:00
softmax_ops_test.py add soft label functionality to softmax with loss op 2017-02-10 09:01:53 -08:00
sparse_gradient_checker_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
sparse_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
spatial_bn_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
square_root_divide_op_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
stats_ops_test.py Performance counters 2017-02-21 16:31:24 -08:00
string_ops_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
text_file_reader_test.py Add unittest.main() lines to test scripts under python/operator_test 2016-11-29 15:18:37 -08:00
tile_op_test.py Caffe2: Tile operator 2017-02-28 23:17:26 -08:00
unique_uniform_fill_op_test.py UniqueUniformFillOp 2017-02-15 16:00:44 -08:00
utility_ops_test.py Add gradient operator for SumElements 2017-03-07 20:03:07 -08:00