pytorch
ec7913af - Cuts test_torch.py runtime in half by marking four tests as slow (#26789)

Commit
6 years ago
Cuts test_torch.py runtime in half by marking four tests as slow (#26789) Summary: - Adds slowTest to four tests On my devfair running test_torch.py takes ~200 seconds with slow tests enabled. Running with the current slowTest annotations takes ~145s. Running with these four additional annotations takes ~64s. test_sum_dim, for example, takes 30s but was not marked as slow. test_det_logdet_slogdet takes 17s on CPU and 22s on CUDA for a total of 39s! test_einsum takes 7s. test_triu_tril takes 5 seconds on CPU and 9s on CUDA for a total of 14s. Several of the current slowTests are faster than this. test_cholesky_solve_batched_many_batches, for example, takes a ~3 seconds on CPU and ~4.5 on CUDA, for a total of 7.5s across both devices. Pull Request resolved: https://github.com/pytorch/pytorch/pull/26789 Differential Revision: D17574282 Pulled By: mruberry fbshipit-source-id: 3e5e505244c09b0ae23bd8c0145828119326719b
Author
Mike Ruberry
Parents
Loading