pytorch/test
Nick Gibson 4e2ea6e013 [TensorExpr] Remove the Tensor argument from loopnest.reorderAxis (#37873)
Summary:
Remove the requirement for the axes provided to reorderAxis to come from a Tensor. We were using that to determine the relevant loops, but we can alternatively determine it by traversing the parents of each provided For.

resistor does this work for you?
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37873

Differential Revision: D21428016

Pulled By: nickgg

fbshipit-source-id: b16b2f41cb443dfc2c6548b7980731d1e7d89a35
2020-05-06 12:02:15 -07:00
..
backward_compatibility [quant] ConvPackedParams with TorchBind (#35923) 2020-05-05 20:18:36 -07:00
bottleneck_test fix possible pandas import error during tensorboard tests (#29650) 2020-02-04 14:27:46 -08:00
cpp [TensorExpr] Remove the Tensor argument from loopnest.reorderAxis (#37873) 2020-05-06 12:02:15 -07:00
cpp_api_parity Enabled debug symbol in test_cpp_api_parity tests by default. (#36209) 2020-04-08 09:34:32 -07:00
cpp_extensions Back out "Revert D21171334: [pytorch][PR] Change StorageImpl to track byte count rather than element count" (#37893) 2020-05-05 22:43:15 -07:00
custom_operator Support c10::optional<Tensor> in custom C++ autograd function. (#37700) 2020-05-06 01:59:51 -07:00
distributed Enable test_distributed test test_backend_full_group (#37794) 2020-05-05 07:56:57 -07:00
error_messages
expect Prevent unbounded growth of sparse tensor in add operation (#36030) 2020-05-01 12:05:15 -07:00
jit [JIT] rename enable_profiling_mode to enable_profiling_mode_for_profiling_tests (#37825) 2020-05-06 11:30:02 -07:00
mobile Remove callBoxedWorkaround (#36850) 2020-04-24 23:13:31 -07:00
onnx [ONNX] Fix numerical errors in softmax when dim is not last dimension (#37326) 2020-05-04 18:07:38 -07:00
optim
quantization Revert D21409626: [quant][tests] Enable tests to run on all qengine backends 2020-05-06 10:37:41 -07:00
scripts Make test_torch.py pass cuda-memcheck (#29243) 2019-12-14 20:29:58 -08:00
type_hint_tests Fix multiple issues with type annotations (#36358) 2020-04-29 11:16:39 -07:00
HowToWriteTestsUsingFileCheck.md
run_test.py Enable ahead of time compilation for HIPExtensions using ninja (#37800) 2020-05-05 20:53:35 -07:00
simulate_nccl_errors.py
te_utils.py add test_tensorexpr.py (#35776) 2020-04-01 22:05:37 -07:00
test_autograd.py modify select_equals_backward to propage only to a single value (#36316) 2020-05-06 10:50:24 -07:00
test_bundled_inputs.py Add utility for bundling sample inputs with models (#35631) 2020-04-08 13:10:36 -07:00
test_complex.py Temporary methods for real and imag values of complex tensors (#35879) 2020-04-05 07:22:02 -07:00
test_cpp_api_parity.py Refactor C++ API parity test mechanism and turn it on in CI again (#35190) 2020-04-03 11:20:36 -07:00
test_cpp_extensions_aot.py Fix lint (#37922) 2020-05-06 09:29:34 -07:00
test_cpp_extensions_jit.py Don't use NonVariableTypeMode in custom ops (#37355) 2020-04-28 20:11:31 -07:00
test_cuda_primary_ctx.py Move pytorch distributed tests to separate folder for contbuild. (#30445) 2020-01-22 21:16:59 -08:00
test_cuda.py Disable tests failing on test2 in ROCm CI (#37427) 2020-04-29 09:56:28 -07:00
test_dataloader.py Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
test_determination.py run the simple executor for jit tests by default, add profiling jobs … (#37017) 2020-04-28 19:16:52 -07:00
test_dispatch.py Switch to pybind11 style registration function API. (#36258) 2020-04-16 10:44:21 -07:00
test_distributions.py [takeover] BTRS algorithm for fast/efficient binomial sampling (#36858) 2020-04-22 15:53:41 -07:00
test_docs_coverage.py Revert "Revert D21337640: [pytorch][PR] Split up documentation into subpages and clean up some warnings" (#37778) 2020-05-04 14:32:35 -07:00
test_expecttest.py Move pytorch distributed tests to separate folder for contbuild. (#30445) 2020-01-22 21:16:59 -08:00
test_function_schema.py Move pytorch distributed tests to separate folder for contbuild. (#30445) 2020-01-22 21:16:59 -08:00
test_indexing.py Changes warnings generated in cpp to show point of Python origination (#36052) 2020-04-25 21:18:58 -07:00
test_jit_cuda_fuser.py New CUDA Fuser: Unrolling support, interface refactor (#36435) 2020-04-16 09:20:24 -07:00
test_jit_disabled.py Move pytorch distributed tests to separate folder for contbuild. (#30445) 2020-01-22 21:16:59 -08:00
test_jit_fuser_legacy.py switching to a simple/full executor 2019-11-11 13:41:35 -08:00
test_jit_fuser_profiling.py run the simple executor for jit tests by default, add profiling jobs … (#37017) 2020-04-28 19:16:52 -07:00
test_jit_fuser_te.py [JIT] rename enable_profiling_mode to enable_profiling_mode_for_profiling_tests (#37825) 2020-05-06 11:30:02 -07:00
test_jit_fuser.py [JIT] rename enable_profiling_mode to enable_profiling_mode_for_profiling_tests (#37825) 2020-05-06 11:30:02 -07:00
test_jit_legacy.py Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
test_jit_profiling.py run the simple executor for jit tests by default, add profiling jobs … (#37017) 2020-04-28 19:16:52 -07:00
test_jit_py3.py [jit] fix named tuples as attributes (#37251) 2020-04-24 17:48:44 -07:00
test_jit_simple.py Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
test_jit_string.py [PyTorch][TorchScript] Add support for join on List of strings in TorchScript (#32847) 2020-01-31 18:20:38 -08:00
test_jit.py [JIT] rename enable_profiling_mode to enable_profiling_mode_for_profiling_tests (#37825) 2020-05-06 11:30:02 -07:00
test_logging.py Move pytorch distributed tests to separate folder for contbuild. (#30445) 2020-01-22 21:16:59 -08:00
test_mkldnn.py Use unittest assertWarns instead (#36411) 2020-04-13 15:56:42 -07:00
test_mobile_optimizer.py Propagate module lints for mobile scripted module. (#37046) 2020-04-27 10:20:12 -07:00
test_multiprocessing_spawn.py Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
test_multiprocessing.py Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
test_namedtensor.py Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
test_namedtuple_return_api.py Added cummin 2020-01-17 10:51:58 -08:00
test_nn.py [ROCm] Enable bfloat16 for ops in BERT model (#37634) 2020-05-05 21:24:56 -07:00
test_numba_integration.py Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
test_optim.py Add SWA to PyTorch mainline (#35032) 2020-04-27 07:42:19 -07:00
test_overrides.py Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
test_quantization.py [wip] Make quantization modules work with DataParallel (#37032) 2020-05-05 13:06:43 -07:00
test_serialization.py Remove python-2 or python<3.5 checks from unit tests (#37252) 2020-04-24 17:42:04 -07:00
test_show_pickle.py Add torch.utils.show_pickle for showing pickle contents in saved models (#35168) 2020-04-03 15:11:20 -07:00
test_sparse.py Fix for tests on ROCm (#37616) 2020-05-04 07:16:54 -07:00
test_tensorboard.py Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
test_tensorexpr.py correctly set and restore states in te tests (#37210) 2020-04-24 20:16:51 -07:00
test_throughput_benchmark.py ThroughputBenchmark: integration with Autograd Profiler (#36282) 2020-04-13 08:53:40 -07:00
test_torch.py Reduction should not coalesce_dimensions when splitting for 32bit indexing (#37788) 2020-05-05 23:44:00 -07:00
test_type_hints.py Fixes type annotations for named tensors #27846 (#36890) 2020-04-28 06:51:22 -07:00
test_type_info.py Move pytorch distributed tests to separate folder for contbuild. (#30445) 2020-01-22 21:16:59 -08:00
test_type_promotion.py Extend some of the basic ops to kHalf (#37121) 2020-04-29 10:49:16 -07:00
test_utils.py Remove (most) Python 2 support from Python code (#35615) 2020-04-22 09:23:14 -07:00
test_xnnpack_integration.py Enable relu fusion with prepacked linear/conv. (#35705) 2020-04-03 15:38:45 -07:00