Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Speed up test times 🚀 #1124

Open
wd60622 opened this issue Dec 13, 2024 · 0 comments
Open

Speed up test times 🚀 #1124

wd60622 opened this issue Dec 13, 2024 · 0 comments

Comments

@wd60622
Copy link
Contributor

wd60622 commented Dec 13, 2024

If you are motivated to help speed up some tests, we would appreciate it!

Here are some of the slowest test times:

(2 minutes 58 seconds) --doctest-modules pytensor --ignore=pytensor/misc/check_duplicate_key.py --ignore=pytensor/link
============================= slowest 50 durations =============================
46.78s call     pytensor/tensor/pad.py::pytensor.tensor.pad.pad
13.95s call     pytensor/gradient.py::pytensor.gradient.hessian_vector_product
9.87s call     pytensor/tensor/basic.py::pytensor.tensor.basic.tril
8.34s call     pytensor/tensor/math.py::pytensor.tensor.math.isclose
7.22s call     pytensor/tensor/einsum.py::pytensor.tensor.einsum._general_dot
3.53s call     pytensor/gradient.py::pytensor.gradient.verify_grad
2.82s call     pytensor/tensor/einsum.py::pytensor.tensor.einsum._delta
2.28s call     pytensor/tensor/basic.py::pytensor.tensor.basic._nd_grid
1.71s call     pytensor/tensor/basic.py::pytensor.tensor.basic.triu
1.17s call     pytensor/gradient.py::pytensor.gradient.grad_scale
1.15s call     pytensor/tensor/basic.py::pytensor.tensor.basic.stacklists
1.14s call     pytensor/tensor/einsum.py::pytensor.tensor.einsum._iota
1.08s call     pytensor/tensor/variable.py::pytensor.tensor.variable._tensor_py_operators.inc
1.06s call     pytensor/raise_op.py::pytensor.raise_op.Assert
0.75s call     pytensor/tensor/basic.py::pytensor.tensor.basic.Split
0.67s call     pytensor/gradient.py::pytensor.gradient.grad_clip
0.62s call     pytensor/tensor/basic.py::pytensor.tensor.basic.stack
0.61s call     pytensor/tensor/io.py::pytensor.tensor.io.load
0.59s call     pytensor/tensor/utils.py::pytensor.tensor.utils.shape_of_variables
0.59s call     pytensor/graph/basic.py::pytensor.graph.basic.Variable.eval
0.58s call     pytensor/tensor/subtensor.py::pytensor.tensor.subtensor.slice_at_axis
0.58s call     pytensor/tensor/extra_ops.py::pytensor.tensor.extra_ops.searchsorted
0.55s call     pytensor/tensor/subtensor.py::pytensor.tensor.subtensor.flip
0.54s call     pytensor/tensor/variable.py::pytensor.tensor.variable._tensor_py_operators.set
0.52s call     pytensor/sparse/basic.py::pytensor.sparse.basic.block_diag
0.03s call     pytensor/tensor/extra_ops.py::pytensor.tensor.extra_ops.Unique
0.01s call     pytensor/tensor/slinalg.py::pytensor.tensor.slinalg.cholesky

0.005s hidden.  Use -vv to show these durations.)
=========================== short test summary info ============================
(7 minutes 41 seconds) tests --ignore=tests/tensor --ignore=tests/scan --ignore=tests/sparse
============================= slowest 50 durations =============================
30.33s call     tests/test_ifelse.py::TestIfelse::test_not_lazy_if_inplace
23.42s call     tests/scalar/test_basic.py::TestUpgradeToFloat::test_unary
11.64s call     tests/scalar/test_basic.py::TestUpgradeToFloat::test_binary
10.40s call     tests/scalar/test_basic.py::TestUpgradeToFloat::test_true_div
8.64s call     tests/scalar/test_math.py::test_scalarloop_grad_mixed_dtypes[op2-scalar_loop_grads2]
8.37s call     tests/test_gradient.py::test_undefined_grad_opt
7.73s call     tests/test_rop.py::TestRopLop::test_max
7.70s call     tests/link/c/test_cmodule.py::test_cache_race_condition
7.39s call     tests/compile/test_compilelock.py::test_locking_multiprocess_spawn
6.67s call     tests/compile/test_builders.py::TestOpFromGraph::test_L_op_disconnected_output_grad
4.98s call     tests/graph/test_compute_test_value.py::TestComputeTestValue::test_scan
4.81s call     tests/test_gradient.py::test_subgraph_grad
4.28s call     tests/compile/test_debugmode.py::test_debugmode_basic
4.20s call     tests/compile/test_misc.py::test_nnet
4.03s call     tests/scalar/test_math.py::test_scalarloop_grad_mixed_dtypes[op3-scalar_loop_grads3]
3.74s call     tests/test_gradient.py::test_jacobian_matrix
3.60s call     tests/test_gradient.py::test_known_grads
3.48s call     tests/test_gradient.py::TestHessianVectorProdudoct::test_rosen
3.25s call     tests/compile/function/test_pfunc.py::TestAliasingRules::test_input_aliasing_affecting_inplace_operations
3.21s call     tests/test_rop.py::TestRopLop::test_join
3.19s call     tests/compile/function/test_pfunc.py::TestPfunc::test_shared
3.19s call     tests/test_breakpoint.py::TestPdbBreakpoint::test_infer_shape
3.02s call     tests/compile/function/test_function.py::test_function_dump
2.94s call     tests/scalar/test_basic.py::test_grad_abs
2.89s call     tests/test_gradient.py::TestGrad::test_grad_grad_cubic
2.87s call     tests/compile/test_builders.py::TestOpFromGraph::test_compute_test_value
2.83s call     tests/link/c/test_op.py::test_ExternalCOp_c_code_cache_version
2.82s call     tests/compile/test_builders.py::TestOpFromGraph::test_grad_override[OpFromGraph]
2.78s call     tests/compile/test_profiling.py::TestProfiling::test_profiling
2.77s call     tests/compile/test_monitormode.py::test_not_inplace
2.57s call     tests/compile/function/test_types.py::TestFunction::test_swap_SharedVariable_with_given
2.55s call     tests/compile/test_debugmode.py::TestViewMap::test_aliased_outputs_bad
2.54s call     tests/test_rop.py::TestRopLop::test_elemwise0
2.54s call     tests/graph/test_compute_test_value.py::TestComputeTestValue::test_empty_elemwise
2.54s call     tests/test_gradient.py::TestZeroGrad::test_grad
2.52s call     tests/compile/test_builders.py::TestOpFromGraph::test_nested[OpFromGraph]
2.52s call     tests/link/c/test_basic.py::test_shared_input_output
2.48s call     tests/compile/function/test_pfunc.py::TestAliasingRules::test_potential_output_aliasing_induced_by_updates
2.46s call     tests/graph/test_compute_test_value.py::TestComputeTestValue::test_scan_err1
2.43s call     tests/compile/function/test_pfunc.py::TestPfunc::test_shared_mutable
2.37s call     tests/compile/test_debugmode.py::TestCheckIsfinite::test_check_isfinite
2.36s call     tests/compile/function/test_pfunc.py::TestPfunc::test_doc
2.16s call     tests/test_rop.py::TestRopLop::test_flatten
2.13s call     tests/compile/test_builders.py::TestOpFromGraph::test_lop_override[OpFromGraph]
2.00s call     tests/test_rop.py::TestRopLop::test_elemwise1
1.93s call     tests/test_rop.py::TestRopLop::test_unbroadcast
1.92s call     tests/test_rop.py::TestRopLop::test_setsubtensor1
1.91s call     tests/compile/test_builders.py::TestOpFromGraph::test_shared_grad[cls_ofg1]
1.89s call     tests/test_gradient.py::TestGrad::test_grad_duplicate_input
1.86s call     tests/compile/test_builders.py::TestOpFromGraph::test_grad[OpFromGraph]
=========================== short test summary info ============================
(12 minutes 4 seconds) tests/scan
============================= slowest 50 durations =============================
56.98s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_taps_backwards
35.76s call     tests/scan/test_basic.py::TestScan::test_grad_mitsot
28.58s call     tests/scan/test_checkpoints.py::TestScanCheckpoint::test_backward_pass
25.84s call     tests/scan/test_basic.py::TestScan::test_R_op
25.29s call     tests/scan/test_basic.py::TestExamples::test_grad_multiple_outs_some_disconnected
23.61s call     tests/scan/test_rewriting.py::TestPushOutNonSeqScan::test_pushout_seqs
22.51s call     tests/scan/test_basic.py::TestScan::test_only_nonseq_inputs
20.55s call     tests/scan/test_rewriting.py::TestPushOutAddScan::test_machine_translation
17.90s call     tests/scan/test_basic.py::TestScan::test_grad_of_grad_of_state
17.78s call     tests/scan/test_basic.py::TestScan::test_grad_dtype_change
17.42s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs
17.35s call     tests/scan/test_basic.py::TestScan::test_R_op_2
14.50s call     tests/scan/test_basic.py::TestScan::test_grad_one_output
13.94s call     tests/scan/test_basic.py::TestScan::test_shared_arguments_with_updates
12.69s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_taps_state
12.23s call     tests/scan/test_basic.py::TestExamples::test_default_value_broadcasted
9.99s call     tests/scan/test_checkpoints.py::TestScanCheckpoint::test_forward_pass
9.24s call     tests/scan/test_basic.py::test_compute_test_value_grad
8.68s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_some_uncomputable
8.50s call     tests/scan/test_basic.py::TestExamples::test_grad_multiple_outs_some_disconnected_2
8.34s call     tests/scan/test_basic.py::TestScan::test_inner_graph_cloning[default_rng]
8.25s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_taps
8.14s call     tests/scan/test_rewriting.py::TestSaveMem::test_savemem_opt_0_step
7.90s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until
7.82s call     tests/scan/test_basic.py::TestScan::test_grad_sitsot
7.77s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_seqs_different_nsteps
7.76s call     tests/scan/test_basic.py::TestScan::test_grad_duplicate_outputs
7.60s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until_and_truncate_sequence_taps
7.02s call     tests/scan/test_basic.py::TestScan::test_inner_grad
5.70s call     tests/scan/test_basic.py::TestScan::test_shared_updates
5.49s call     tests/scan/test_rewriting.py::TestSaveMem::test_save_mem_store_steps
5.38s call     tests/scan/test_rewriting.py::TestSaveMem::test_save_mem_reduced_number_of_steps
5.33s call     tests/scan/test_basic.py::TestScan::test_grad_wrt_shared
5.31s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until_ndim_greater_one
5.14s call     tests/scan/test_rewriting.py::TestScanInplaceOptimizer::test_simple_rnn_2
4.99s call     tests/scan/test_basic.py::TestExamples::test_return_steps
4.75s call     tests/scan/test_views.py::test_foldr_memory_consumption
4.41s call     tests/scan/test_basic.py::TestExamples::test_gibbs_chain
4.41s call     tests/scan/test_basic.py::test_compute_test_value_grad_cast
4.39s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until_and_truncate
4.31s call     tests/scan/test_basic.py::TestScan::test_pickling
3.89s call     tests/scan/test_basic.py::TestExamples::test_same
3.83s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_some_truncate
3.47s call     tests/scan/test_basic.py::TestScan::test_backwards
3.40s call     tests/scan/test_rewriting.py::TestSaveMem::test_save_mem
3.32s call     tests/scan/test_basic.py::TestExamples::test_past_future_taps_shared
3.30s call     tests/scan/test_basic.py::TestScan::test_only_shared_no_input_no_output
3.13s call     tests/scan/test_basic.py::test_compute_test_values
3.06s call     tests/scan/test_rewriting.py::TestPushOutAddScan::test_sum_dot
3.01s call     tests/scan/test_basic.py::TestExamples::test_infer_shape_remove_stuff
=========================== short test summary info ============================
(5 minutes 22 seconds) tests/sparse
============================= slowest 50 durations =============================
84.11s call     tests/sparse/test_basic.py::TestUsmm::test_basic
20.64s call     tests/sparse/sandbox/test_sp.py::TestSP::test_convolution
10.76s call     tests/sparse/test_basic.py::TestCasting::test_cast
10.40s call     tests/sparse/sandbox/test_sp.py::TestSP::test_maxpool
10.12s call     tests/sparse/test_basic.py::TestCasting::test_infer_shape
6.70s call     tests/sparse/test_basic.py::TestAddMul::test_AddSD
6.67s call     tests/sparse/test_basic.py::TestCsmProperties::test_csm_properties_grad
6.35s call     tests/sparse/test_basic.py::TestStructuredDot::test_dot_sparse_sparse
4.39s call     tests/sparse/test_basic.py::TestCsm::test_csm_grad
3.57s call     tests/sparse/test_basic.py::TestHStack::test_grad
3.26s call     tests/sparse/test_basic.py::TestSpSum::test_grad
3.20s call     tests/sparse/test_basic.py::TestCsm::test_csm_unsorted
3.02s call     tests/sparse/test_basic.py::TestAddMul::test_AddDS
2.12s call     tests/sparse/test_basic.py::TestVerifyGradSparse::test_grad_fail
2.06s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[any-DenseTensorType-None-None]
2.00s call     tests/sparse/test_basic.py::TestSparseInferShape::test_csm_grad
1.89s call     tests/sparse/test_basic.py::TestUsmm::test_infer_shape
1.84s call     tests/sparse/test_basic.py::TestMulSV::test_mul_s_v_grad
1.83s call     tests/sparse/test_basic.py::TestStructuredAddSV::test_structured_add_s_v_grad
1.82s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[mean-DenseTensorType-None-None]
1.77s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[var-DenseTensorType-None-None]
1.67s call     tests/sparse/test_basic.py::TestConversion::test_format_ndim
1.64s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__divmod__-DenseTensorType]
1.58s call     tests/sparse/test_basic.py::TestVStack::test_grad
1.54s call     tests/sparse/test_basic.py::TestTrueDot::test_grad
1.37s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[argmin-DenseTensorType-None-None]
1.29s call     tests/sparse/test_basic.py::TestDots::test_int32_dtype
1.23s call     tests/sparse/test_basic.py::TestConstructSparseFromList::test_adv_sub1_sparse_grad
0.96s call     tests/sparse/test_basic.py::TestCsm::test_csm_sparser
0.92s call     tests/sparse/test_basic.py::TestDots::test_sparse_sparse
0.91s call     tests/sparse/test_basic.py::TestStructuredDot::test_structureddot_csc_grad
0.89s call     tests/sparse/test_basic.py::TestStructuredDot::test_structureddot_csr_grad
0.88s call     tests/sparse/test_basic.py::TestSamplingDot::test_op
0.85s call     tests/sparse/test_basic.py::TestTrueDot::test_infer_shape
0.83s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__and__-DenseTensorType]
0.82s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__truediv__-DenseTensorType]
0.82s call     tests/sparse/test_basic.py::TestStructuredDot::test_upcast
0.82s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__or__-DenseTensorType]
0.82s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__xor__-DenseTensorType]
0.82s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__mod__-DenseTensorType]
0.82s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[deg2rad-DenseTensorType-None-None]
0.80s call     tests/sparse/sandbox/test_sp.py::TestSP::test_multilayer_conv
0.80s call     tests/sparse/test_basic.py::TestCasting::test_grad
0.79s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[std-DenseTensorType-None-None]
0.79s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__pow__-DenseTensorType]
0.78s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[rad2deg-DenseTensorType-None-None]
0.78s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[round-DenseTensorType-None-None]
0.78s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[log1p-DenseTensorType-None-None]
0.77s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[__abs__-DenseTensorType-None-None]
0.77s call     tests/sparse/test_basic.py::TestSquareDiagonal::test_infer_shape
=========================== short test summary info ============================
(19 minutes 47 seconds) tests/tensor --ignore=tests/tensor/conv --ignore=tests/tensor/rewriting --ignore=tests/tensor/test_math.py --ignore=tests/tensor/test_basic.py --ignore=tests/tensor/test_blas.py --ignore=tests/tensor/test_math_scipy.py --ignore=tests/tensor/test_inplace.py --ignore=tests/tensor/test_elemwise.py
============================= slowest 50 durations =============================
46.69s call     tests/tensor/test_subtensor.py::TestSubtensor::test_shape_i_const
34.34s call     tests/tensor/test_blockwise.py::TestCholesky::test_grad
13.10s call     tests/tensor/test_subtensor.py::TestSubtensor::test_grad_list
11.96s call     tests/tensor/test_extra_ops.py::test_space_ops[start2-stop2-25-True-0-None-geomspace]
11.61s call     tests/tensor/test_casting.py::TestCasting::test_convert_to_complex
8.66s call     tests/tensor/test_fft.py::TestFFT::test_grad_rfft
8.40s call     tests/tensor/test_subtensor.py::TestIncSubtensor::test_grad_inc_set
8.24s call     tests/tensor/test_nlinalg.py::TestSvd::test_grad[batched=True-(3,3)-compute_uv=True, full_matrices=False, gradient=U]
8.09s call     tests/tensor/test_extra_ops.py::test_space_ops[start2-stop2-25-True-0-None-linspace]
8.04s call     tests/tensor/test_subtensor.py::TestInferShape::test_IncSubtensor
7.96s call     tests/tensor/test_extra_ops.py::test_space_ops[1-stop5-30-True-0-None-geomspace]
7.96s call     tests/tensor/test_extra_ops.py::test_space_ops[1-stop5-30-True-0-None-linspace]
7.84s call     tests/tensor/test_fft.py::TestFFT::test_rfft_float
7.79s call     tests/tensor/test_einsum.py::test_einsum_signatures[oij,imj,mjkn,lnk,plk->op-True]
7.60s call     tests/tensor/test_extra_ops.py::TestCumOp::test_grad
7.48s call     tests/tensor/test_pad.py::test_nd_padding[symmetric-symmetric]
7.44s call     tests/tensor/test_slinalg.py::test_cholesky_grad_indef
7.41s call     tests/tensor/test_extra_ops.py::test_space_ops[1-10-50-True-0-None-geomspace]
7.41s call     tests/tensor/test_blockwise.py::TestMatrixInverse::test_grad
7.37s call     tests/tensor/test_einsum.py::test_einsum_signatures[oij,imj,mjkn,lnk,plk->op-False]
7.11s call     tests/tensor/random/rewriting/test_basic.py::test_Subtensor_lift[indices2-True-dist_op2-dist_params2-None]
7.07s call     tests/tensor/test_blas_c.py::TestBlasStrides::test_gemm
7.04s call     tests/tensor/test_extra_ops.py::TestFillDiagonalOffset::test_gradient[-5]
6.92s call     tests/tensor/test_einsum.py::test_general_dot
6.81s call     tests/tensor/test_blockwise.py::TestSolveVector::test_grad
6.73s call     tests/tensor/test_extra_ops.py::TestRepeat::test_basic[int8-3]
6.41s call     tests/tensor/test_pad.py::test_nd_padding[asymmetric-symmetric]
6.25s call     tests/tensor/test_blockwise.py::TestSolveMatrix::test_grad
6.07s call     tests/tensor/random/rewriting/test_basic.py::test_Subtensor_lift[indices17-True-dist_op17-dist_params17-None]
5.92s call     tests/tensor/test_interpolate.py::test_interp
5.79s call     tests/tensor/test_blas_c.py::TestBlasStridesC::test_gemm
5.75s call     tests/tensor/test_blas_scipy.py::TestBlasStrides::test_gemm
5.75s call     tests/tensor/test_blas_scipy.py::TestBlasStridesScipy::test_gemm
5.75s call     tests/tensor/test_extra_ops.py::TestBinCount::test_bincountFn[int8]
5.68s call     tests/tensor/test_pad.py::test_nd_padding[symmetric-reflect]
5.67s call     tests/tensor/test_pad.py::test_reflect_pad[even-symmetrical-3d square]
5.53s call     tests/tensor/test_interpolate.py::test_interp_padded
5.53s call     tests/tensor/test_extra_ops.py::TestBinCount::test_bincountFn[int16]
5.46s call     tests/tensor/test_nlinalg.py::TestSvd::test_grad[batched=False-(3,3)-compute_uv=True, full_matrices=False, gradient=U]
5.32s call     tests/tensor/test_pad.py::test_symmetric_pad[even-symmetrical-2d square]
5.32s call     tests/tensor/test_pad.py::test_reflect_pad[even-asymmetric_right-3d square]
5.28s call     tests/tensor/test_blas_c.py::TestCGemvNoFlags::test_cgemv
5.25s call     tests/tensor/test_pad.py::test_reflect_pad[even-asymmetrical_left-3d square]
5.23s call     tests/tensor/test_nlinalg.py::TestEighFloat32::test_grad
5.16s call     tests/tensor/test_pad.py::test_nd_padding[asymmetric-reflect]
5.06s call     tests/tensor/test_extra_ops.py::TestBinCount::test_bincountFn[int32]
5.06s call     tests/tensor/test_extra_ops.py::test_space_ops[1-10-50-True-0-None-linspace]
5.00s call     tests/tensor/test_shape.py::TestReshape::test_more_shapes
4.95s call     tests/tensor/test_slinalg.py::test_solve_discrete_lyapunov_gradient[direct-matrix-float]
4.90s call     tests/tensor/test_extra_ops.py::test_space_ops[start4-stop4-25-False--1-None-linspace]
=========================== short test summary info ============================
(4 minutes 46 seconds) tests/tensor/conv
============================= slowest 50 durations =============================
52.79s call     tests/tensor/conv/test_abstract_conv.py::TestAbstractConvNoOptim::test_all
22.70s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_fractional_bilinear_upsampling
21.80s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConv3dNoOptim::test_fwd
16.66s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConv3dNoOptim::test_gradweights
12.79s call     tests/tensor/conv/test_abstract_conv.py::TestUnsharedConv::test_fwd
10.24s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConv3dNoOptim::test_gradinputs
7.61s call     tests/tensor/conv/test_abstract_conv.py::TestAssertConvShape::test_basic
7.46s call     tests/tensor/conv/test_abstract_conv.py::TestUnsharedConv::test_gradinput
7.05s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_fractional_bilinear_upsampling_shape
5.03s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_kernel_2D
4.86s call     tests/tensor/conv/test_abstract_conv.py::TestUnsharedConv::test_gradweight
3.75s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConvNoOptim::test_fwd
3.75s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConvNoOptim::test_gradweights
3.67s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_kernel_1D
3.58s call     tests/tensor/conv/test_abstract_conv.py::TestCausalConv::test_interface
3.19s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_upsampling_1D
2.99s call     tests/tensor/conv/test_abstract_conv.py::TestConvGradInputsShape::test_check_shape
2.95s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConvNoOptim::test_gradinputs
2.21s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_basic
1.96s call     tests/tensor/conv/test_abstract_conv.py::TestConv2dGrads::test_conv2d_grad_wrt_inputs
1.86s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv2d
1.55s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_compare_1D_and_2D_upsampling_values
1.42s call     tests/tensor/conv/test_abstract_conv.py::TestConv2dGrads::test_conv2d_grad_wrt_weights
1.40s call     tests/tensor/conv/test_abstract_conv.py::TestAbstractConvNoOptim::test_gradinput_impossible_output_shapes
1.33s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_upsampling_reshaping
1.28s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv3d
1.23s call     tests/tensor/conv/test_abstract_conv.py::TestConv2dTranspose::test_interface
0.64s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv3d_grad_wrt_inputs
0.64s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv3d_grad_wrt_weights
0.62s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv2d_grad_wrt_inputs
0.62s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv2d_grad_wrt_weights
0.36s call     tests/tensor/conv/test_abstract_conv.py::TestAsymmetricPadding::test_gradweight
0.33s call     tests/tensor/conv/test_abstract_conv.py::TestAsymmetricPadding::test_gradinput
0.26s call     tests/tensor/conv/test_abstract_conv.py::TestAsymmetricPadding::test_fwd
0.24s call     tests/tensor/conv/test_abstract_conv.py::TestAbstractConvNoOptim::test_gradinput_arbitrary_output_shapes
0.09s call     tests/tensor/conv/test_abstract_conv.py::TestSeparableConv::test_interface3d
0.09s call     tests/tensor/conv/test_abstract_conv.py::TestConvGradInputsShape::test_get_shape
0.08s call     tests/tensor/conv/test_abstract_conv.py::TestSeparableConv::test_interface2d
0.04s call     tests/tensor/conv/test_abstract_conv.py::test_constant_shapes
0.03s call     tests/tensor/conv/test_abstract_conv.py::TestConvTypes::test_constant_input
0.01s call     tests/tensor/conv/test_abstract_conv.py::TestConvTypes::test_grad_types

0.005s hidden.  Use -vv to show these durations.)
=========================== short test summary info ============================
(17 minutes 18 seconds) tests/tensor/rewriting
============================= slowest 50 durations =============================
154.61s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const4
98.10s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_none_slice
56.60s call     tests/tensor/rewriting/test_elemwise.py::TestFusion::test_big_fusion
22.27s call     tests/tensor/rewriting/test_math.py::test_local_log_sum_exp_maximum
19.08s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_prod_of_div
18.56s call     tests/tensor/rewriting/test_basic.py::TestLocalSwitchSink::test_local_mul_switch_sink
16.54s call     tests/tensor/rewriting/test_math.py::TestAlgebraicCanonizer::test_mul_div_cases
16.01s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const3
15.46s call     tests/tensor/rewriting/test_subtensor.py::test_local_subtensor_of_alloc
13.76s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const
13.11s call     tests/tensor/rewriting/test_subtensor.py::test_local_IncSubtensor_serialize
12.11s call     tests/tensor/rewriting/test_subtensor.py::test_local_subtensor_remove_broadcastable_index
10.10s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const2
8.52s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_none_index
7.92s call     tests/tensor/rewriting/test_math.py::TestLocalErfc::test_local_log_erfc
7.62s call     tests/tensor/rewriting/test_math.py::test_local_log_add_exp
7.60s call     tests/tensor/rewriting/test_math.py::TestIntDivByOne::test_remove_floor
6.88s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_sum_of_non_scalar_mul
6.82s call     tests/tensor/rewriting/test_math.py::TestReduceChain::test_local_sum_sum_prod_prod
6.50s call     tests/tensor/rewriting/test_math.py::TestSigmoidRewrites::test_exp_over_1_plus_exp
6.43s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_sum_prod_of_scalar_mul
6.18s call     tests/tensor/rewriting/test_basic.py::test_local_flatten_lift[2]
6.06s call     tests/tensor/rewriting/test_special.py::test_softmax_graph
5.98s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_sum_of_div
5.91s call     tests/tensor/rewriting/test_math.py::TestLocalReduce::test_local_reduce_broadcast_some_0
5.88s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_scalar5
5.72s call     tests/tensor/rewriting/test_basic.py::test_local_useless_slice
5.33s call     tests/tensor/rewriting/test_math.py::test_merge_abs_bugfix
4.98s call     tests/tensor/rewriting/test_basic.py::TestLocalSwitchSink::test_local_div_switch_sink
4.67s call     tests/tensor/rewriting/test_linalg.py::test_det_blockdiag_rewrite
4.58s call     tests/tensor/rewriting/test_math.py::test_local_mul_exp_to_exp_add
4.36s call     tests/tensor/rewriting/test_linalg.py::test_rop_lop
4.20s call     tests/tensor/rewriting/test_math.py::test_log1p
4.08s call     tests/tensor/rewriting/test_math.py::test_local_subtensor_of_dot
4.01s call     tests/tensor/rewriting/test_basic.py::TestConstantFolding::test_constant_get_stabilized
3.85s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_scalar6
3.79s call     tests/tensor/rewriting/test_blockwise.py::test_blockwise_alloc
3.65s call     tests/tensor/rewriting/test_uncanonicalize.py::TestMinMax::test_optimization_min
3.55s call     tests/tensor/rewriting/test_basic.py::TestLocalUselessIncSubtensorAlloc::test_advanced_inc_subtensor
3.37s call     tests/tensor/rewriting/test_math.py::TestFuncInverse::test
3.34s call     tests/tensor/rewriting/test_uncanonicalize.py::test_local_dimshuffle_subtensor
3.27s call     tests/tensor/rewriting/test_basic.py::TestLocalUselessIncSubtensorAlloc::test_incsubtensor
3.18s call     tests/tensor/rewriting/test_subtensor.py::TestLocalAdvSub1AdvIncSub1::test_assert
3.10s call     tests/tensor/rewriting/test_math.py::test_local_expm1
2.99s call     tests/tensor/rewriting/test_basic.py::test_local_useless_split
2.91s call     tests/tensor/rewriting/test_basic.py::TestLocalElemwiseAlloc::test_remove_alloc_wo_dimshuffle
2.89s call     tests/tensor/rewriting/test_special.py::TestLogSoftmaxRewrites::test_local_logsoftmax_grad_rewrite[None]
2.88s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const6
2.88s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_sum_prod_alloc
2.85s call     tests/tensor/rewriting/test_math.py::TestSigmoidRewrites::test_grad_log1msigm
=========================== short test summary info ============================
(19 minutes 10 seconds) tests/tensor/test_math.py
============================= slowest 50 durations =============================
193.32s call     tests/tensor/test_math.py::TestSumProdReduceDtype::test_reduce_custom_dtype
90.95s call     tests/tensor/test_math.py::TestProdWithoutZerosDtype::test_prod_without_zeros_custom_dtype
38.66s call     tests/tensor/test_math.py::TestProdWithoutZerosDtype::test_prod_without_zeros_custom_acc_dtype
33.32s call     tests/tensor/test_math.py::TestMaxAndArgmax::test_grad
20.69s call     tests/tensor/test_math.py::TestComparison::test_isclose
19.31s call     tests/tensor/test_math.py::test_var
18.95s call     tests/tensor/test_math.py::TestSumProdReduceDtype::test_reduce_default_dtype
18.66s call     tests/tensor/test_math.py::TestTensordot::test_basic
17.93s call     tests/tensor/test_math.py::TestAddBroadcast::test_good
16.85s call     tests/tensor/test_math.py::TestPowBroadcast::test_grad
12.89s call     tests/tensor/test_math.py::TestProd::test_verify_grad
11.23s call     tests/tensor/test_math.py::TestMulBroadcast::test_good
10.95s call     tests/tensor/test_math.py::TestMinMax::test_uint
10.63s call     tests/tensor/test_math.py::TestArctan2Broadcast::test_grad
10.59s call     tests/tensor/test_math.py::TestSubBroadcast::test_grad
9.97s call     tests/tensor/test_math.py::TestMinMax::test_grad_min
9.53s call     tests/tensor/test_math.py::TestSubBroadcast::test_good
9.27s call     tests/tensor/test_math.py::TestModBroadcast::test_good
9.23s call     tests/tensor/test_math.py::TestCeilIntDivBroadcast::test_good
9.16s call     tests/tensor/test_math.py::TestProd::test_other_grad_tests
8.61s call     tests/tensor/test_math.py::TestProd::test_prod_no_zeros_in_input
8.40s call     tests/tensor/test_math.py::TestArctan2Broadcast::test_good
8.29s call     tests/tensor/test_math.py::TestMinimumBroadcast::test_grad
8.22s call     tests/tensor/test_math.py::TestTrueDivBroadcast::test_grad
8.20s call     tests/tensor/test_math.py::TestMaximumBroadcast::test_grad
8.16s call     tests/tensor/test_math.py::TestTrueDivBroadcast::test_good
7.99s call     tests/tensor/test_math.py::TestMinimumBroadcast::test_good
7.93s call     tests/tensor/test_math.py::TestModBroadcast::test_grad
7.92s call     tests/tensor/test_math.py::TestMaximumBroadcast::test_good
7.88s call     tests/tensor/test_math.py::TestClip2::test_clip_repeat_verify_grad
7.81s call     tests/tensor/test_math.py::TestMulBroadcast::test_grad
7.33s call     tests/tensor/test_math.py::TestArgminArgmax::test_uint
7.02s call     tests/tensor/test_math.py::TestProdWithoutZerosDtype::test_prod_without_zeros_default_acc_dtype
6.62s call     tests/tensor/test_math.py::TestClip1::test_good
6.61s call     tests/tensor/test_math.py::TestMeanDtype::test_mean_default_dtype
6.14s call     tests/tensor/test_math.py::test_cov
6.07s call     tests/tensor/test_math.py::TestNegBroadcast::test_good
5.86s call     tests/tensor/test_math.py::TestPowBroadcast::test_good
5.55s call     tests/tensor/test_math.py::TestInvBroadcast::test_good
5.50s call     tests/tensor/test_math.py::TestOuter::test_grad
5.14s call     tests/tensor/test_math.py::TestSqrBroadcast::test_good
5.03s call     tests/tensor/test_math.py::TestDivimpl::test_impls
4.91s call     tests/tensor/test_math.py::TestAbsBroadcast::test_good
4.84s call     tests/tensor/test_math.py::TestDenseDot::test_good
4.83s call     tests/tensor/test_math.py::TestAdd::test_complex_all_ops
4.51s call     tests/tensor/test_math.py::TestTensordot::test_weird_valid_axes
4.39s call     tests/tensor/test_math.py::TestExp2Broadcast::test_grad
4.06s call     tests/tensor/test_math.py::TestTruncBroadcast::test_good
4.03s call     tests/tensor/test_math.py::TestSinBroadcast::test_good
4.02s call     tests/tensor/test_math.py::TestExpm1Broadcast::test_good
=========================== short test summary info ============================
(12 minutes 30 seconds) tests/tensor/test_basic.py tests/tensor/test_inplace.py
============================= slowest 50 durations =============================
78.57s call     tests/tensor/test_basic.py::TestSecondSameRank::test_bad_runtime
78.23s call     tests/tensor/test_basic.py::TestSecondSameRank::test_good
50.38s call     tests/tensor/test_basic.py::TestTriangle::test_tril_triu
43.43s call     tests/tensor/test_basic.py::test_tile
25.79s call     tests/tensor/test_basic.py::TestAllocDiag::test_alloc_diag_values
14.68s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_roll
13.54s call     tests/tensor/test_basic.py::TestSecondBroadcast::test_good
10.88s call     tests/tensor/test_basic.py::TestNdGrid::test_mgrid_numpy_equiv
10.17s call     tests/tensor/test_inplace.py::TestAddInplaceBroadcast::test_good
10.08s call     tests/tensor/test_inplace.py::TestSubInplaceBroadcast::test_good
10.01s call     tests/tensor/test_inplace.py::TestMulInplaceBroadcast::test_good
9.10s call     tests/tensor/test_inplace.py::TestIntDivInplaceBroadcast::test_good
8.94s call     tests/tensor/test_inplace.py::TestModInplaceBroadcast::test_good
8.77s call     tests/tensor/test_basic.py::test_tile_grad
8.25s call     tests/tensor/test_inplace.py::TestMaximumInplaceBroadcast::test_good
8.24s call     tests/tensor/test_inplace.py::TestMinimumInplaceBroadcast::test_good
7.53s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_stack_hessian
6.77s call     tests/tensor/test_basic.py::TestAlloc::test_alloc_constant_folding
6.56s call     tests/tensor/test_basic.py::TestZerosLikeBroadcast::test_good
6.00s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_broadcastable_flags_many_dims_and_inputs
5.77s call     tests/tensor/test_basic.py::TestAllocBroadcast::test_good
5.68s call     tests/tensor/test_inplace.py::TestTrueDivInplaceBroadcast::test_good
5.61s call     tests/tensor/test_inplace.py::TestPowInplaceBroadcast::test_good
5.29s call     tests/tensor/test_basic.py::TestARange::test_infer_shape[custom]
5.14s call     tests/tensor/test_inplace.py::TestSqrInplaceBroadcast::test_good
5.13s call     tests/tensor/test_inplace.py::TestNegInplaceBroadcast::test_good
5.07s call     tests/tensor/test_basic.py::TestSwitchBroadcast::test_grad
4.80s call     tests/tensor/test_inplace.py::TestArctan2InplaceBroadcast::test_good
4.78s call     tests/tensor/test_basic.py::TestInferShape::test_Flatten
4.19s call     tests/tensor/test_inplace.py::TestAbsInplaceBroadcast::test_good
4.12s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_join_matrix1[False]
4.10s call     tests/tensor/test_inplace.py::TestTruncInplaceBroadcast::test_good
3.88s call     tests/tensor/test_basic.py::TestAlloc13GradBroadcast::test_grad
3.41s call     tests/tensor/test_basic.py::TestAllocDimshuffleGradBroadcast::test_grad
3.11s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_broadcastable_flag_assignment_mixed_thisaxes
3.10s call     tests/tensor/test_basic.py::TestSwitchBroadcast::test_good
3.09s call     tests/tensor/test_inplace.py::TestSgnInplaceBroadcast::test_good
2.88s call     tests/tensor/test_basic.py::TestNdGrid::test_mgrid_pytensor_variable_numpy_equiv
2.84s call     tests/tensor/test_inplace.py::TestConjInplaceBroadcast::test_good
2.76s call     tests/tensor/test_basic.py::TestExtractDiag::test_grad_2d[0-0-1]
2.63s call     tests/tensor/test_inplace.py::test_XOR_inplace
2.62s call     tests/tensor/test_basic.py::TestInferShape::test_Split
2.61s call     tests/tensor/test_basic.py::TestAlloc01GradBroadcast::test_grad
2.56s call     tests/tensor/test_basic.py::test_empty
2.54s call     tests/tensor/test_basic.py::TestAllocb1GradBroadcast::test_grad
2.19s call     tests/tensor/test_basic.py::TestAlloc::test_ones
2.12s call     tests/tensor/test_basic.py::TestExtractDiag::test_grad_2d[1-0-1]
1.99s call     tests/tensor/test_basic.py::TestAllocb4GradBroadcast::test_grad
1.90s call     tests/tensor/test_basic.py::TestAllocb3GradBroadcast::test_grad
1.80s call     tests/tensor/test_basic.py::TestMakeVector::test_make_vector[int32-inputs1]
=========================== short test summary info ============================
(12 minutes 16 seconds) tests/tensor/test_blas.py tests/tensor/test_elemwise.py tests/tensor/test_math_scipy.py
============================= slowest 50 durations =============================
176.67s call     tests/tensor/test_elemwise.py::TestCAReduce::test_c
46.78s call     tests/tensor/test_blas.py::TestGemmNoFlags::test_gemm
26.11s call     tests/tensor/test_blas.py::TestBatchedDot::test_grad
25.48s call     tests/tensor/test_math_scipy.py::TestGammaIncBroadcast::test_grad
23.38s call     tests/tensor/test_blas.py::test_dot22scalar
16.88s call     tests/tensor/test_elemwise.py::TestCAReduce::test_perform
11.96s call     tests/tensor/test_math_scipy.py::TestGammaInccBroadcast::test_grad
10.92s call     tests/tensor/test_math_scipy.py::TestBetaincBroadcast::test_grad
10.08s call     tests/tensor/test_elemwise.py::TestCAReduce::test_c_noopt
8.70s call     tests/tensor/test_math_scipy.py::test_gammainc_ddk_tabulated_values
8.66s call     tests/tensor/test_blas.py::TestBatchedDot::test_good
8.43s call     tests/tensor/test_blas.py::test_dot22scalar_cast
8.05s call     tests/tensor/test_blas.py::test_local_dot22_to_dot22scalar
7.55s call     tests/tensor/test_elemwise.py::TestCAReduce::test_perform_nan
7.50s call     tests/tensor/test_elemwise.py::TestBroadcast::test_c
7.24s call     tests/tensor/test_math_scipy.py::TestHyp2F1Broadcast::test_grad
6.55s call     tests/tensor/test_blas.py::test_dot_w_self
6.25s call     tests/tensor/test_math_scipy.py::TestBetaIncGrad::test_stan_grad_partial
5.95s call     tests/tensor/test_elemwise.py::TestBroadcast::test_c_inplace
5.69s call     tests/tensor/test_blas.py::TestBlasStrides::test_gemm
5.58s call     tests/tensor/test_elemwise.py::TestCAReduce::test_infer_shape
5.43s call     tests/tensor/test_elemwise.py::TestElemwise::test_infer_shape
5.17s call     tests/tensor/test_math_scipy.py::TestHyp2F1Grad::test_hyp2f1_grad_stan_cases
5.07s call     tests/tensor/test_blas.py::TestGemm::test_factorised_scalar
4.79s call     tests/tensor/test_math_scipy.py::TestBetaincInplaceBroadcast::test_grad
4.73s call     tests/tensor/test_blas.py::test_gemm_opt0
4.61s call     tests/tensor/test_math_scipy.py::TestErfcxBroadcast::test_grad
4.45s call     tests/tensor/test_math_scipy.py::TestGammalnInplaceBroadcast::test_good
4.45s call     tests/tensor/test_math_scipy.py::TestGammalnBroadcast::test_good
3.91s call     tests/tensor/test_math_scipy.py::TestErfBroadcast::test_good
3.77s call     tests/tensor/test_blas.py::TestGemm::test_non_contiguous
3.50s call     tests/tensor/test_blas.py::test_gemm_nested
3.43s call     tests/tensor/test_blas.py::test_gemm_opt_vector_stuff
3.21s call     tests/tensor/test_blas.py::test_upcasting_scalar_nogemm
3.18s call     tests/tensor/test_math_scipy.py::TestErfcBroadcast::test_grad
3.09s call     tests/tensor/test_math_scipy.py::TestErfBroadcast::test_grad
3.08s call     tests/tensor/test_elemwise.py::TestCAReduce::test_c_nan
3.07s call     tests/tensor/test_math_scipy.py::TestSigmoidInplaceBroadcast::test_grad
3.05s call     tests/tensor/test_blas.py::test_gemm_unrolled
3.04s call     tests/tensor/test_elemwise.py::TestDimShuffle::test_c_or_py
3.00s call     tests/tensor/test_math_scipy.py::TestSoftplusInplaceBroadcast::test_grad
2.94s call     tests/tensor/test_math_scipy.py::test_kve
2.92s call     tests/tensor/test_blas.py::TestGemm::test_basic_4
2.61s call     tests/tensor/test_math_scipy.py::TestBetaIncGrad::test_boik_robison_cox
2.47s call     tests/tensor/test_blas.py::TestBlasStrides::test_dot22scalar
2.45s call     tests/tensor/test_blas.py::TestGer::test_c64_7_1
2.38s call     tests/tensor/test_blas.py::test_gemm_with_vector
2.27s call     tests/tensor/test_math_scipy.py::TestHyp2F1Grad::test_unused_grad_loop_opt[wrt3]
2.25s call     tests/tensor/test_math_scipy.py::TestErfcxBroadcast::test_good
2.21s call     tests/tensor/test_math_scipy.py::TestHyp2F1Grad::test_unused_grad_loop_opt[wrt5]
=========================== short test summary info ============================

You can find more information on how to contribute here

Automatically generated by GitHub Action
Latest run date: 2025-01-23

@ricardoV94 ricardoV94 pinned this issue Dec 26, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant