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:

(3 minutes 2 seconds) --doctest-modules pytensor --ignore=pytensor/misc/check_duplicate_key.py --ignore=pytensor/link
============================= slowest 50 durations =============================
48.48s call     pytensor/tensor/pad.py::pytensor.tensor.pad.pad
14.24s call     pytensor/gradient.py::pytensor.gradient.hessian_vector_product
10.07s call     pytensor/tensor/basic.py::pytensor.tensor.basic.tril
8.51s call     pytensor/tensor/math.py::pytensor.tensor.math.isclose
7.37s call     pytensor/tensor/einsum.py::pytensor.tensor.einsum._general_dot
3.65s call     pytensor/gradient.py::pytensor.gradient.verify_grad
2.95s call     pytensor/tensor/einsum.py::pytensor.tensor.einsum._delta
2.34s call     pytensor/tensor/basic.py::pytensor.tensor.basic._nd_grid
1.75s call     pytensor/tensor/basic.py::pytensor.tensor.basic.triu
1.21s call     pytensor/gradient.py::pytensor.gradient.grad_scale
1.17s call     pytensor/tensor/basic.py::pytensor.tensor.basic.stacklists
1.16s call     pytensor/tensor/einsum.py::pytensor.tensor.einsum._iota
1.13s call     pytensor/tensor/variable.py::pytensor.tensor.variable._tensor_py_operators.inc
1.07s call     pytensor/raise_op.py::pytensor.raise_op.Assert
0.76s call     pytensor/tensor/basic.py::pytensor.tensor.basic.Split
0.69s call     pytensor/gradient.py::pytensor.gradient.grad_clip
0.64s call     pytensor/tensor/basic.py::pytensor.tensor.basic.stack
0.62s call     pytensor/tensor/utils.py::pytensor.tensor.utils.shape_of_variables
0.61s call     pytensor/tensor/io.py::pytensor.tensor.io.load
0.61s call     pytensor/tensor/subtensor.py::pytensor.tensor.subtensor.slice_at_axis
0.60s call     pytensor/graph/basic.py::pytensor.graph.basic.Variable.eval
0.60s call     pytensor/tensor/extra_ops.py::pytensor.tensor.extra_ops.searchsorted
0.57s call     pytensor/tensor/subtensor.py::pytensor.tensor.subtensor.flip
0.56s call     pytensor/tensor/variable.py::pytensor.tensor.variable._tensor_py_operators.set
0.53s call     pytensor/sparse/basic.py::pytensor.sparse.basic.block_diag
0.03s call     pytensor/tensor/extra_ops.py::pytensor.tensor.extra_ops.Unique
0.02s call     pytensor/tensor/slinalg.py::pytensor.tensor.slinalg.cholesky

0.005s hidden.  Use -vv to show these durations.)
=========================== short test summary info ============================
(7 minutes 38 seconds) tests --ignore=tests/tensor --ignore=tests/scan --ignore=tests/sparse
============================= slowest 50 durations =============================
30.31s call     tests/test_ifelse.py::TestIfelse::test_not_lazy_if_inplace
23.44s call     tests/scalar/test_basic.py::TestUpgradeToFloat::test_unary
11.47s call     tests/scalar/test_basic.py::TestUpgradeToFloat::test_binary
10.61s call     tests/scalar/test_basic.py::TestUpgradeToFloat::test_true_div
8.72s call     tests/scalar/test_math.py::test_scalarloop_grad_mixed_dtypes[op2-scalar_loop_grads2]
8.38s call     tests/test_gradient.py::test_undefined_grad_opt
7.71s call     tests/test_rop.py::TestRopLop::test_max
7.57s call     tests/link/c/test_cmodule.py::test_cache_race_condition
7.46s call     tests/compile/test_compilelock.py::test_locking_multiprocess_spawn
6.66s call     tests/compile/test_builders.py::TestOpFromGraph::test_L_op_disconnected_output_grad
4.92s call     tests/graph/test_compute_test_value.py::TestComputeTestValue::test_scan
4.84s call     tests/test_gradient.py::test_subgraph_grad
4.31s call     tests/compile/test_debugmode.py::test_debugmode_basic
4.24s call     tests/compile/test_misc.py::test_nnet
4.08s 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.56s call     tests/test_gradient.py::test_known_grads
3.49s call     tests/test_gradient.py::TestHessianVectorProdudoct::test_rosen
3.37s call     tests/test_rop.py::TestRopLop::test_join
3.24s call     tests/compile/function/test_pfunc.py::TestAliasingRules::test_input_aliasing_affecting_inplace_operations
3.19s call     tests/test_breakpoint.py::TestPdbBreakpoint::test_infer_shape
3.10s call     tests/compile/function/test_pfunc.py::TestPfunc::test_shared
2.94s call     tests/scalar/test_basic.py::test_grad_abs
2.87s call     tests/compile/test_builders.py::TestOpFromGraph::test_compute_test_value
2.87s call     tests/test_gradient.py::TestGrad::test_grad_grad_cubic
2.83s call     tests/compile/test_builders.py::TestOpFromGraph::test_grad_override[OpFromGraph]
2.82s call     tests/link/c/test_op.py::test_ExternalCOp_c_code_cache_version
2.78s call     tests/compile/test_profiling.py::TestProfiling::test_profiling
2.77s call     tests/compile/test_monitormode.py::test_not_inplace
2.60s call     tests/test_rop.py::TestRopLop::test_elemwise0
2.56s call     tests/compile/function/test_types.py::TestFunction::test_swap_SharedVariable_with_given
2.56s call     tests/test_gradient.py::TestZeroGrad::test_grad
2.52s call     tests/compile/test_debugmode.py::TestViewMap::test_aliased_outputs_bad
2.52s call     tests/graph/test_compute_test_value.py::TestComputeTestValue::test_empty_elemwise
2.52s call     tests/compile/test_builders.py::TestOpFromGraph::test_nested[OpFromGraph]
2.50s call     tests/link/c/test_basic.py::test_shared_input_output
2.48s call     tests/graph/test_compute_test_value.py::TestComputeTestValue::test_scan_err1
2.45s call     tests/compile/function/test_pfunc.py::TestAliasingRules::test_potential_output_aliasing_induced_by_updates
2.41s call     tests/compile/function/test_pfunc.py::TestPfunc::test_shared_mutable
2.40s call     tests/compile/test_debugmode.py::TestCheckIsfinite::test_check_isfinite
2.33s call     tests/compile/function/test_pfunc.py::TestPfunc::test_doc
2.19s call     tests/test_rop.py::TestRopLop::test_flatten
2.13s call     tests/compile/test_builders.py::TestOpFromGraph::test_lop_override[OpFromGraph]
2.03s call     tests/test_rop.py::TestRopLop::test_elemwise1
1.96s call     tests/test_rop.py::TestRopLop::test_unbroadcast
1.96s call     tests/test_rop.py::TestRopLop::test_setsubtensor1
1.90s call     tests/compile/test_builders.py::TestOpFromGraph::test_shared_grad[cls_ofg1]
1.87s call     tests/compile/test_builders.py::TestOpFromGraph::test_grad[OpFromGraph]
1.87s call     tests/test_gradient.py::TestGrad::test_grad_duplicate_input
1.86s call     tests/compile/function/test_pfunc.py::TestPfunc::test_givens
=========================== short test summary info ============================
(12 minutes 19 seconds) tests/scan
============================= slowest 50 durations =============================
60.32s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_taps_backwards
36.66s call     tests/scan/test_basic.py::TestScan::test_grad_mitsot
28.65s call     tests/scan/test_checkpoints.py::TestScanCheckpoint::test_backward_pass
26.90s call     tests/scan/test_basic.py::TestScan::test_R_op
26.83s call     tests/scan/test_basic.py::TestExamples::test_grad_multiple_outs_some_disconnected
24.23s call     tests/scan/test_rewriting.py::TestPushOutNonSeqScan::test_pushout_seqs
23.33s call     tests/scan/test_basic.py::TestScan::test_only_nonseq_inputs
20.28s call     tests/scan/test_rewriting.py::TestPushOutAddScan::test_machine_translation
18.84s call     tests/scan/test_basic.py::TestScan::test_grad_dtype_change
18.68s call     tests/scan/test_basic.py::TestScan::test_grad_of_grad_of_state
18.16s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs
17.92s call     tests/scan/test_basic.py::TestScan::test_R_op_2
14.96s call     tests/scan/test_basic.py::TestScan::test_grad_one_output
14.26s call     tests/scan/test_basic.py::TestScan::test_shared_arguments_with_updates
13.29s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_taps_state
12.31s call     tests/scan/test_basic.py::TestExamples::test_default_value_broadcasted
10.29s call     tests/scan/test_checkpoints.py::TestScanCheckpoint::test_forward_pass
9.61s call     tests/scan/test_basic.py::test_compute_test_value_grad
9.02s call     tests/scan/test_basic.py::TestExamples::test_grad_multiple_outs_some_disconnected_2
8.86s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_some_uncomputable
8.80s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_taps
8.36s call     tests/scan/test_basic.py::TestScan::test_grad_duplicate_outputs
8.25s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_seqs_different_nsteps
8.24s call     tests/scan/test_rewriting.py::TestSaveMem::test_savemem_opt_0_step
8.15s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until
8.14s call     tests/scan/test_basic.py::TestScan::test_grad_sitsot
7.97s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until_and_truncate_sequence_taps
7.92s call     tests/scan/test_basic.py::TestScan::test_inner_graph_cloning[default_rng]
7.32s call     tests/scan/test_basic.py::TestScan::test_inner_grad
6.04s call     tests/scan/test_basic.py::TestScan::test_shared_updates
5.64s call     tests/scan/test_basic.py::TestScan::test_grad_wrt_shared
5.54s call     tests/scan/test_rewriting.py::TestSaveMem::test_save_mem_store_steps
5.52s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until_ndim_greater_one
5.38s call     tests/scan/test_rewriting.py::TestSaveMem::test_save_mem_reduced_number_of_steps
5.03s call     tests/scan/test_rewriting.py::TestScanInplaceOptimizer::test_simple_rnn_2
4.91s call     tests/scan/test_basic.py::TestExamples::test_return_steps
4.75s call     tests/scan/test_views.py::test_foldr_memory_consumption
4.69s call     tests/scan/test_basic.py::test_compute_test_value_grad_cast
4.59s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until_and_truncate
4.46s call     tests/scan/test_basic.py::TestExamples::test_gibbs_chain
4.35s call     tests/scan/test_basic.py::TestScan::test_pickling
3.95s call     tests/scan/test_basic.py::TestExamples::test_same
3.92s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_some_truncate
3.61s call     tests/scan/test_basic.py::TestScan::test_backwards
3.43s call     tests/scan/test_basic.py::TestScan::test_only_shared_no_input_no_output
3.39s call     tests/scan/test_rewriting.py::TestSaveMem::test_save_mem
3.29s call     tests/scan/test_basic.py::TestExamples::test_past_future_taps_shared
3.15s call     tests/scan/test_basic.py::test_compute_test_values
3.08s call     tests/scan/test_rewriting.py::TestPushOutAddScan::test_sum_dot
3.04s call     tests/scan/test_basic.py::TestScan::test_using_negative_taps_sequence
=========================== short test summary info ============================
(5 minutes 6 seconds) tests/sparse
============================= slowest 50 durations =============================
76.97s call     tests/sparse/test_basic.py::TestUsmm::test_basic
19.55s call     tests/sparse/sandbox/test_sp.py::TestSP::test_convolution
9.90s call     tests/sparse/sandbox/test_sp.py::TestSP::test_maxpool
9.23s call     tests/sparse/test_basic.py::TestCasting::test_cast
8.83s call     tests/sparse/test_basic.py::TestCasting::test_infer_shape
6.49s call     tests/sparse/test_basic.py::TestAddMul::test_AddSD
6.39s call     tests/sparse/test_basic.py::TestCsmProperties::test_csm_properties_grad
6.20s call     tests/sparse/test_basic.py::TestStructuredDot::test_dot_sparse_sparse
4.26s call     tests/sparse/test_basic.py::TestCsm::test_csm_grad
3.39s call     tests/sparse/test_basic.py::TestHStack::test_grad
3.09s call     tests/sparse/test_basic.py::TestSpSum::test_grad
3.09s call     tests/sparse/test_basic.py::TestCsm::test_csm_unsorted
2.94s call     tests/sparse/test_basic.py::TestAddMul::test_AddDS
2.04s call     tests/sparse/test_basic.py::TestVerifyGradSparse::test_grad_fail
1.96s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[any-DenseTensorType-None-None]
1.89s call     tests/sparse/test_basic.py::TestSparseInferShape::test_csm_grad
1.84s call     tests/sparse/test_basic.py::TestUsmm::test_infer_shape
1.76s call     tests/sparse/test_basic.py::TestMulSV::test_mul_s_v_grad
1.74s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[mean-DenseTensorType-None-None]
1.73s call     tests/sparse/test_basic.py::TestStructuredAddSV::test_structured_add_s_v_grad
1.66s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[var-DenseTensorType-None-None]
1.58s call     tests/sparse/test_basic.py::TestConversion::test_format_ndim
1.54s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__divmod__-DenseTensorType]
1.50s call     tests/sparse/test_basic.py::TestVStack::test_grad
1.47s call     tests/sparse/test_basic.py::TestTrueDot::test_grad
1.29s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[argmin-DenseTensorType-None-None]
1.24s call     tests/sparse/test_basic.py::TestDots::test_int32_dtype
1.17s call     tests/sparse/test_basic.py::TestConstructSparseFromList::test_adv_sub1_sparse_grad
0.90s call     tests/sparse/test_basic.py::TestCsm::test_csm_sparser
0.87s call     tests/sparse/test_basic.py::TestStructuredDot::test_structureddot_csc_grad
0.87s call     tests/sparse/test_basic.py::TestStructuredDot::test_structureddot_csr_grad
0.85s call     tests/sparse/test_basic.py::TestSamplingDot::test_op
0.80s call     tests/sparse/test_basic.py::TestDots::test_sparse_sparse
0.78s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__mod__-DenseTensorType]
0.78s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__and__-DenseTensorType]
0.78s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__or__-DenseTensorType]
0.77s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[deg2rad-DenseTensorType-None-None]
0.77s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__truediv__-DenseTensorType]
0.77s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__xor__-DenseTensorType]
0.75s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__pow__-DenseTensorType]
0.75s call     tests/sparse/test_basic.py::TestTrueDot::test_infer_shape
0.74s call     tests/sparse/sandbox/test_sp.py::TestSP::test_multilayer_conv
0.74s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[rad2deg-DenseTensorType-None-None]
0.73s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[__abs__-DenseTensorType-None-None]
0.73s call     tests/sparse/test_basic.py::TestCasting::test_grad
0.73s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[log2-DenseTensorType-None-None]
0.73s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[sqrt-DenseTensorType-None-None]
0.73s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[std-DenseTensorType-None-None]
0.73s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[round-DenseTensorType-None-None]
0.73s call     tests/sparse/test_basic.py::TestSquareDiagonal::test_infer_shape
=========================== short test summary info ============================
(20 minutes 9 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 =============================
45.82s call     tests/tensor/test_subtensor.py::TestSubtensor::test_shape_i_const
33.92s call     tests/tensor/test_blockwise.py::TestCholesky::test_grad
12.58s call     tests/tensor/test_subtensor.py::TestSubtensor::test_grad_list
11.80s call     tests/tensor/test_extra_ops.py::test_space_ops[start2-stop2-25-True-0-None-geomspace]
11.50s call     tests/tensor/test_casting.py::TestCasting::test_convert_to_complex
8.57s call     tests/tensor/test_fft.py::TestFFT::test_grad_rfft
8.28s call     tests/tensor/test_subtensor.py::TestIncSubtensor::test_grad_inc_set
8.10s call     tests/tensor/test_nlinalg.py::TestSvd::test_grad[batched=True-(3,3)-compute_uv=True, full_matrices=False, gradient=U]
8.07s call     tests/tensor/test_extra_ops.py::test_space_ops[start2-stop2-25-True-0-None-linspace]
8.07s call     tests/tensor/test_subtensor.py::TestInferShape::test_IncSubtensor
7.93s call     tests/tensor/test_extra_ops.py::test_space_ops[1-stop5-30-True-0-None-geomspace]
7.89s call     tests/tensor/test_extra_ops.py::test_space_ops[1-stop5-30-True-0-None-linspace]
7.86s call     tests/tensor/test_pad.py::test_nd_padding[symmetric-symmetric]
7.77s call     tests/tensor/test_einsum.py::test_einsum_signatures[oij,imj,mjkn,lnk,plk->op-True]
7.71s call     tests/tensor/test_fft.py::TestFFT::test_rfft_float
7.47s call     tests/tensor/test_extra_ops.py::TestCumOp::test_grad
7.35s call     tests/tensor/test_extra_ops.py::test_space_ops[1-10-50-True-0-None-geomspace]
7.34s call     tests/tensor/test_blockwise.py::TestMatrixInverse::test_grad
7.31s call     tests/tensor/test_einsum.py::test_einsum_signatures[oij,imj,mjkn,lnk,plk->op-False]
7.26s call     tests/tensor/test_slinalg.py::test_cholesky_grad_indef
7.08s call     tests/tensor/random/rewriting/test_basic.py::test_Subtensor_lift[indices2-True-dist_op2-dist_params2-None]
7.01s call     tests/tensor/test_blas_c.py::TestBlasStrides::test_gemm
7.00s call     tests/tensor/test_extra_ops.py::TestFillDiagonalOffset::test_gradient[-5]
6.89s call     tests/tensor/test_einsum.py::test_general_dot
6.70s call     tests/tensor/test_extra_ops.py::TestRepeat::test_basic[int8-3]
6.70s call     tests/tensor/test_blockwise.py::TestSolveVector::test_grad
6.21s call     tests/tensor/test_blockwise.py::TestSolveMatrix::test_grad
5.93s call     tests/tensor/test_interpolate.py::test_interp
5.91s call     tests/tensor/random/rewriting/test_basic.py::test_Subtensor_lift[indices17-True-dist_op17-dist_params17-None]
5.76s call     tests/tensor/test_blas_c.py::TestBlasStridesC::test_gemm
5.71s call     tests/tensor/test_pad.py::test_reflect_pad[even-symmetrical-3d square]
5.70s call     tests/tensor/test_blas_scipy.py::TestBlasStridesScipy::test_gemm
5.69s call     tests/tensor/test_blas_scipy.py::TestBlasStrides::test_gemm
5.65s call     tests/tensor/test_extra_ops.py::TestBinCount::test_bincountFn[int8]
5.58s call     tests/tensor/test_pad.py::test_nd_padding[symmetric-reflect]
5.53s call     tests/tensor/test_interpolate.py::test_interp_padded
5.48s call     tests/tensor/test_extra_ops.py::TestBinCount::test_bincountFn[int16]
5.36s call     tests/tensor/test_pad.py::test_reflect_pad[even-asymmetric_right-3d square]
5.34s call     tests/tensor/test_nlinalg.py::TestSvd::test_grad[batched=False-(3,3)-compute_uv=True, full_matrices=False, gradient=U]
5.33s call     tests/tensor/test_pad.py::test_symmetric_pad[even-symmetrical-2d square]
5.32s call     tests/tensor/test_pad.py::test_nd_padding[asymmetric-reflect]
5.28s call     tests/tensor/test_blas_c.py::TestCGemvNoFlags::test_cgemv
5.24s call     tests/tensor/test_pad.py::test_reflect_pad[even-asymmetrical_left-3d square]
5.19s call     tests/tensor/test_nlinalg.py::TestEighFloat32::test_grad
5.02s call     tests/tensor/test_extra_ops.py::TestBinCount::test_bincountFn[int32]
5.02s call     tests/tensor/test_extra_ops.py::test_space_ops[1-10-50-True-0-None-linspace]
4.97s call     tests/tensor/test_shape.py::TestReshape::test_more_shapes
4.92s call     tests/tensor/test_slinalg.py::test_solve_discrete_lyapunov_gradient[direct-matrix-float]
4.84s call     tests/tensor/test_extra_ops.py::test_space_ops[start4-stop4-25-False--1-None-linspace]
4.82s call     tests/tensor/test_extra_ops.py::test_space_ops[1-stop6-30-False--1-None-linspace]
=========================== short test summary info ============================
(4 minutes 29 seconds) tests/tensor/conv
============================= slowest 50 durations =============================
52.02s call     tests/tensor/conv/test_abstract_conv.py::TestAbstractConvNoOptim::test_all
22.44s 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.63s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConv3dNoOptim::test_gradweights
13.03s call     tests/tensor/conv/test_abstract_conv.py::TestUnsharedConv::test_fwd
10.19s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConv3dNoOptim::test_gradinputs
7.57s call     tests/tensor/conv/test_abstract_conv.py::TestUnsharedConv::test_gradinput
7.06s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_fractional_bilinear_upsampling_shape
4.98s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_kernel_2D
4.86s call     tests/tensor/conv/test_abstract_conv.py::TestAssertConvShape::test_basic
4.78s 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.74s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConvNoOptim::test_gradweights
3.65s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_kernel_1D
3.45s call     tests/tensor/conv/test_abstract_conv.py::TestCausalConv::test_interface
3.12s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_upsampling_1D
2.96s 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.20s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_basic
1.94s call     tests/tensor/conv/test_abstract_conv.py::TestConv2dGrads::test_conv2d_grad_wrt_inputs
1.81s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv2d
1.52s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_compare_1D_and_2D_upsampling_values
1.40s call     tests/tensor/conv/test_abstract_conv.py::TestAbstractConvNoOptim::test_gradinput_impossible_output_shapes
1.39s call     tests/tensor/conv/test_abstract_conv.py::TestConv2dGrads::test_conv2d_grad_wrt_weights
1.30s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_upsampling_reshaping
1.27s 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_weights
0.64s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv3d_grad_wrt_inputs
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.35s call     tests/tensor/conv/test_abstract_conv.py::TestAsymmetricPadding::test_gradweight
0.32s call     tests/tensor/conv/test_abstract_conv.py::TestAsymmetricPadding::test_gradinput
0.25s call     tests/tensor/conv/test_abstract_conv.py::TestAsymmetricPadding::test_fwd
0.23s 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.08s 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 37 seconds) tests/tensor/rewriting
============================= slowest 50 durations =============================
159.57s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const4
101.60s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_none_slice
57.35s call     tests/tensor/rewriting/test_elemwise.py::TestFusion::test_big_fusion
23.09s call     tests/tensor/rewriting/test_math.py::test_local_log_sum_exp_maximum
19.33s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_prod_of_div
19.05s call     tests/tensor/rewriting/test_basic.py::TestLocalSwitchSink::test_local_mul_switch_sink
16.75s call     tests/tensor/rewriting/test_math.py::TestAlgebraicCanonizer::test_mul_div_cases
16.54s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const3
15.77s call     tests/tensor/rewriting/test_subtensor.py::test_local_subtensor_of_alloc
14.05s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const
13.29s call     tests/tensor/rewriting/test_subtensor.py::test_local_IncSubtensor_serialize
12.56s call     tests/tensor/rewriting/test_subtensor.py::test_local_subtensor_remove_broadcastable_index
10.45s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const2
8.79s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_none_index
8.26s call     tests/tensor/rewriting/test_math.py::TestLocalErfc::test_local_log_erfc
7.65s call     tests/tensor/rewriting/test_math.py::TestIntDivByOne::test_remove_floor
7.62s call     tests/tensor/rewriting/test_math.py::test_local_log_add_exp
6.99s call     tests/tensor/rewriting/test_math.py::TestReduceChain::test_local_sum_sum_prod_prod
6.89s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_sum_of_non_scalar_mul
6.66s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_sum_prod_of_scalar_mul
6.65s call     tests/tensor/rewriting/test_math.py::TestSigmoidRewrites::test_exp_over_1_plus_exp
6.46s call     tests/tensor/rewriting/test_basic.py::test_local_flatten_lift[2]
6.22s call     tests/tensor/rewriting/test_special.py::test_softmax_graph
6.09s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_scalar5
6.08s call     tests/tensor/rewriting/test_math.py::TestLocalReduce::test_local_reduce_broadcast_some_0
5.92s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_sum_of_div
5.42s call     tests/tensor/rewriting/test_math.py::test_merge_abs_bugfix
5.15s call     tests/tensor/rewriting/test_basic.py::test_local_useless_slice
5.11s call     tests/tensor/rewriting/test_basic.py::TestLocalSwitchSink::test_local_div_switch_sink
4.88s call     tests/tensor/rewriting/test_linalg.py::test_det_blockdiag_rewrite
4.71s call     tests/tensor/rewriting/test_math.py::test_local_mul_exp_to_exp_add
4.40s call     tests/tensor/rewriting/test_linalg.py::test_rop_lop
4.17s call     tests/tensor/rewriting/test_math.py::test_log1p
4.15s call     tests/tensor/rewriting/test_basic.py::TestConstantFolding::test_constant_get_stabilized
4.13s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_scalar6
4.06s call     tests/tensor/rewriting/test_math.py::test_local_subtensor_of_dot
3.95s call     tests/tensor/rewriting/test_blockwise.py::test_blockwise_alloc
3.75s call     tests/tensor/rewriting/test_uncanonicalize.py::TestMinMax::test_optimization_min
3.61s call     tests/tensor/rewriting/test_basic.py::TestLocalUselessIncSubtensorAlloc::test_advanced_inc_subtensor
3.37s call     tests/tensor/rewriting/test_uncanonicalize.py::test_local_dimshuffle_subtensor
3.34s call     tests/tensor/rewriting/test_math.py::TestFuncInverse::test
3.31s call     tests/tensor/rewriting/test_basic.py::TestLocalUselessIncSubtensorAlloc::test_incsubtensor
3.25s call     tests/tensor/rewriting/test_subtensor.py::TestLocalAdvSub1AdvIncSub1::test_assert
3.21s call     tests/tensor/rewriting/test_math.py::test_local_expm1
3.05s call     tests/tensor/rewriting/test_math.py::TestSigmoidRewrites::test_grad_log1msigm
3.03s call     tests/tensor/rewriting/test_basic.py::test_local_useless_split
3.03s call     tests/tensor/rewriting/test_basic.py::TestLocalElemwiseAlloc::test_remove_alloc_wo_dimshuffle
2.99s call     tests/tensor/rewriting/test_special.py::TestLogSoftmaxRewrites::test_local_logsoftmax_grad_rewrite[None]
2.95s 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
=========================== short test summary info ============================
(19 minutes 13 seconds) tests/tensor/test_math.py
============================= slowest 50 durations =============================
200.03s call     tests/tensor/test_math.py::TestSumProdReduceDtype::test_reduce_custom_dtype
91.22s call     tests/tensor/test_math.py::TestProdWithoutZerosDtype::test_prod_without_zeros_custom_dtype
38.20s call     tests/tensor/test_math.py::TestProdWithoutZerosDtype::test_prod_without_zeros_custom_acc_dtype
33.57s call     tests/tensor/test_math.py::TestMaxAndArgmax::test_grad
21.09s call     tests/tensor/test_math.py::TestComparison::test_isclose
19.92s call     tests/tensor/test_math.py::test_var
19.82s call     tests/tensor/test_math.py::TestSumProdReduceDtype::test_reduce_default_dtype
19.61s call     tests/tensor/test_math.py::TestTensordot::test_basic
17.92s call     tests/tensor/test_math.py::TestPowBroadcast::test_grad
13.56s call     tests/tensor/test_math.py::TestAddBroadcast::test_good
13.37s call     tests/tensor/test_math.py::TestProd::test_verify_grad
11.49s call     tests/tensor/test_math.py::TestMinMax::test_uint
11.34s call     tests/tensor/test_math.py::TestMulBroadcast::test_good
10.92s call     tests/tensor/test_math.py::TestSubBroadcast::test_grad
10.79s call     tests/tensor/test_math.py::TestArctan2Broadcast::test_grad
10.39s call     tests/tensor/test_math.py::TestMinMax::test_grad_min
9.85s call     tests/tensor/test_math.py::TestSubBroadcast::test_good
9.71s call     tests/tensor/test_math.py::TestProd::test_other_grad_tests
9.62s call     tests/tensor/test_math.py::TestModBroadcast::test_good
9.35s call     tests/tensor/test_math.py::TestCeilIntDivBroadcast::test_good
8.86s call     tests/tensor/test_math.py::TestProd::test_prod_no_zeros_in_input
8.54s call     tests/tensor/test_math.py::TestArctan2Broadcast::test_good
8.45s call     tests/tensor/test_math.py::TestTrueDivBroadcast::test_grad
8.42s call     tests/tensor/test_math.py::TestTrueDivBroadcast::test_good
8.29s call     tests/tensor/test_math.py::TestMaximumBroadcast::test_grad
8.22s call     tests/tensor/test_math.py::TestModBroadcast::test_grad
8.21s call     tests/tensor/test_math.py::TestMinimumBroadcast::test_grad
8.10s call     tests/tensor/test_math.py::TestClip2::test_clip_repeat_verify_grad
8.06s call     tests/tensor/test_math.py::TestMulBroadcast::test_grad
7.91s call     tests/tensor/test_math.py::TestMaximumBroadcast::test_good
7.79s call     tests/tensor/test_math.py::TestMinimumBroadcast::test_good
7.50s call     tests/tensor/test_math.py::TestArgminArgmax::test_uint
7.17s call     tests/tensor/test_math.py::TestProdWithoutZerosDtype::test_prod_without_zeros_default_acc_dtype
7.05s call     tests/tensor/test_math.py::TestClip1::test_good
6.68s call     tests/tensor/test_math.py::TestMeanDtype::test_mean_default_dtype
6.41s call     tests/tensor/test_math.py::TestNegBroadcast::test_good
6.31s call     tests/tensor/test_math.py::test_cov
6.18s call     tests/tensor/test_math.py::TestPowBroadcast::test_good
5.61s call     tests/tensor/test_math.py::TestOuter::test_grad
5.58s call     tests/tensor/test_math.py::TestInvBroadcast::test_good
5.33s call     tests/tensor/test_math.py::TestDivimpl::test_impls
5.22s call     tests/tensor/test_math.py::TestSqrBroadcast::test_good
5.10s call     tests/tensor/test_math.py::TestAbsBroadcast::test_good
5.10s call     tests/tensor/test_math.py::TestAdd::test_complex_all_ops
4.76s call     tests/tensor/test_math.py::TestDenseDot::test_good
4.75s call     tests/tensor/test_math.py::TestTensordot::test_weird_valid_axes
4.62s call     tests/tensor/test_math.py::TestExp2Broadcast::test_grad
4.23s call     tests/tensor/test_math.py::TestExp2Broadcast::test_good
4.20s call     tests/tensor/test_math.py::TestExpBroadcast::test_good
4.15s call     tests/tensor/test_math.py::TestTruncBroadcast::test_good
=========================== short test summary info ============================
(11 minutes 58 seconds) tests/tensor/test_basic.py tests/tensor/test_inplace.py
============================= slowest 50 durations =============================
75.46s call     tests/tensor/test_basic.py::TestSecondSameRank::test_bad_runtime
74.90s call     tests/tensor/test_basic.py::TestSecondSameRank::test_good
49.53s call     tests/tensor/test_basic.py::TestTriangle::test_tril_triu
41.32s call     tests/tensor/test_basic.py::test_tile
24.74s call     tests/tensor/test_basic.py::TestAllocDiag::test_alloc_diag_values
14.07s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_roll
13.03s call     tests/tensor/test_basic.py::TestSecondBroadcast::test_good
10.30s call     tests/tensor/test_inplace.py::TestMulInplaceBroadcast::test_good
10.14s call     tests/tensor/test_basic.py::TestNdGrid::test_mgrid_numpy_equiv
10.10s call     tests/tensor/test_inplace.py::TestAddInplaceBroadcast::test_good
9.90s call     tests/tensor/test_inplace.py::TestSubInplaceBroadcast::test_good
8.78s call     tests/tensor/test_inplace.py::TestModInplaceBroadcast::test_good
8.70s call     tests/tensor/test_inplace.py::TestIntDivInplaceBroadcast::test_good
8.10s call     tests/tensor/test_basic.py::test_tile_grad
8.05s call     tests/tensor/test_inplace.py::TestMinimumInplaceBroadcast::test_good
7.98s call     tests/tensor/test_inplace.py::TestMaximumInplaceBroadcast::test_good
7.41s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_stack_hessian
6.60s call     tests/tensor/test_basic.py::TestAlloc::test_alloc_constant_folding
6.35s call     tests/tensor/test_basic.py::TestZerosLikeBroadcast::test_good
5.79s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_broadcastable_flags_many_dims_and_inputs
5.55s call     tests/tensor/test_inplace.py::TestTrueDivInplaceBroadcast::test_good
5.48s call     tests/tensor/test_basic.py::TestAllocBroadcast::test_good
5.35s call     tests/tensor/test_inplace.py::TestPowInplaceBroadcast::test_good
5.00s call     tests/tensor/test_inplace.py::TestSqrInplaceBroadcast::test_good
4.95s call     tests/tensor/test_inplace.py::TestNegInplaceBroadcast::test_good
4.86s call     tests/tensor/test_basic.py::TestARange::test_infer_shape[custom]
4.77s call     tests/tensor/test_basic.py::TestSwitchBroadcast::test_grad
4.67s call     tests/tensor/test_inplace.py::TestArctan2InplaceBroadcast::test_good
4.62s call     tests/tensor/test_basic.py::TestInferShape::test_Flatten
4.07s call     tests/tensor/test_inplace.py::TestAbsInplaceBroadcast::test_good
4.05s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_join_matrix1[False]
4.04s call     tests/tensor/test_inplace.py::TestTruncInplaceBroadcast::test_good
3.67s call     tests/tensor/test_basic.py::TestAlloc13GradBroadcast::test_grad
3.18s call     tests/tensor/test_basic.py::TestAllocDimshuffleGradBroadcast::test_grad
3.01s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_broadcastable_flag_assignment_mixed_thisaxes
2.97s call     tests/tensor/test_inplace.py::TestSgnInplaceBroadcast::test_good
2.83s call     tests/tensor/test_inplace.py::TestConjInplaceBroadcast::test_good
2.76s call     tests/tensor/test_basic.py::TestSwitchBroadcast::test_good
2.76s call     tests/tensor/test_basic.py::TestNdGrid::test_mgrid_pytensor_variable_numpy_equiv
2.62s call     tests/tensor/test_basic.py::TestExtractDiag::test_grad_2d[0-0-1]
2.59s call     tests/tensor/test_basic.py::test_empty
2.56s call     tests/tensor/test_inplace.py::test_XOR_inplace
2.53s call     tests/tensor/test_basic.py::TestInferShape::test_Split
2.50s call     tests/tensor/test_basic.py::TestAlloc01GradBroadcast::test_grad
2.42s call     tests/tensor/test_basic.py::TestAllocb1GradBroadcast::test_grad
2.09s call     tests/tensor/test_basic.py::TestAlloc::test_ones
2.03s call     tests/tensor/test_basic.py::TestExtractDiag::test_grad_2d[1-0-1]
1.91s call     tests/tensor/test_basic.py::TestAllocb4GradBroadcast::test_grad
1.86s call     tests/tensor/test_basic.py::TestAllocb3GradBroadcast::test_grad
1.75s call     tests/tensor/test_basic.py::TestMakeVector::test_make_vector[int32-inputs1]
=========================== short test summary info ============================
(12 minutes 7 seconds) tests/tensor/test_blas.py tests/tensor/test_elemwise.py tests/tensor/test_math_scipy.py
============================= slowest 50 durations =============================
173.63s call     tests/tensor/test_elemwise.py::TestCAReduce::test_c
47.53s call     tests/tensor/test_blas.py::TestGemmNoFlags::test_gemm
26.27s call     tests/tensor/test_blas.py::TestBatchedDot::test_grad
25.31s call     tests/tensor/test_math_scipy.py::TestGammaIncBroadcast::test_grad
22.81s call     tests/tensor/test_blas.py::test_dot22scalar
16.84s call     tests/tensor/test_elemwise.py::TestCAReduce::test_perform
11.97s call     tests/tensor/test_math_scipy.py::TestGammaInccBroadcast::test_grad
11.08s call     tests/tensor/test_math_scipy.py::TestBetaincBroadcast::test_grad
10.09s call     tests/tensor/test_elemwise.py::TestCAReduce::test_c_noopt
8.77s call     tests/tensor/test_math_scipy.py::test_gammainc_ddk_tabulated_values
8.53s call     tests/tensor/test_blas.py::TestBatchedDot::test_good
8.12s call     tests/tensor/test_blas.py::test_dot22scalar_cast
7.84s call     tests/tensor/test_blas.py::test_local_dot22_to_dot22scalar
7.54s call     tests/tensor/test_elemwise.py::TestBroadcast::test_c
7.47s call     tests/tensor/test_elemwise.py::TestCAReduce::test_perform_nan
7.25s call     tests/tensor/test_math_scipy.py::TestHyp2F1Broadcast::test_grad
6.34s call     tests/tensor/test_blas.py::test_dot_w_self
6.30s call     tests/tensor/test_math_scipy.py::TestBetaIncGrad::test_stan_grad_partial
5.97s call     tests/tensor/test_elemwise.py::TestBroadcast::test_c_inplace
5.73s call     tests/tensor/test_blas.py::TestBlasStrides::test_gemm
5.66s call     tests/tensor/test_elemwise.py::TestCAReduce::test_infer_shape
5.50s call     tests/tensor/test_elemwise.py::TestElemwise::test_infer_shape
5.20s call     tests/tensor/test_math_scipy.py::TestHyp2F1Grad::test_hyp2f1_grad_stan_cases
4.91s 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.48s call     tests/tensor/test_math_scipy.py::TestErfcxBroadcast::test_grad
4.37s call     tests/tensor/test_math_scipy.py::TestGammalnBroadcast::test_good
4.31s call     tests/tensor/test_math_scipy.py::TestGammalnInplaceBroadcast::test_good
3.97s call     tests/tensor/test_math_scipy.py::TestErfBroadcast::test_good
3.82s call     tests/tensor/test_blas.py::TestGemm::test_non_contiguous
3.50s call     tests/tensor/test_blas.py::test_gemm_nested
3.42s call     tests/tensor/test_blas.py::test_gemm_opt_vector_stuff
3.31s call     tests/tensor/test_blas.py::test_upcasting_scalar_nogemm
3.11s call     tests/tensor/test_math_scipy.py::TestErfBroadcast::test_grad
3.09s call     tests/tensor/test_math_scipy.py::TestErfcBroadcast::test_grad
3.08s call     tests/tensor/test_elemwise.py::TestDimShuffle::test_c_or_py
3.05s call     tests/tensor/test_elemwise.py::TestCAReduce::test_c_nan
3.04s call     tests/tensor/test_math_scipy.py::TestSigmoidInplaceBroadcast::test_grad
3.01s call     tests/tensor/test_blas.py::test_gemm_unrolled
2.97s call     tests/tensor/test_math_scipy.py::TestSoftplusInplaceBroadcast::test_grad
2.91s call     tests/tensor/test_math_scipy.py::test_kve
2.59s call     tests/tensor/test_math_scipy.py::TestBetaIncGrad::test_boik_robison_cox
2.49s call     tests/tensor/test_blas.py::TestBlasStrides::test_dot22scalar
2.41s call     tests/tensor/test_blas.py::TestGer::test_c64_7_1
2.36s call     tests/tensor/test_blas.py::test_gemm_with_vector
2.29s call     tests/tensor/test_math_scipy.py::TestHyp2F1Grad::test_unused_grad_loop_opt[wrt3]
2.19s call     tests/tensor/test_math_scipy.py::TestHyp2F1Grad::test_unused_grad_loop_opt[wrt5]
2.19s call     tests/tensor/test_math_scipy.py::TestHyp2F1Grad::test_unused_grad_loop_opt[wrt4]
2.18s call     tests/tensor/test_math_scipy.py::TestLog1mexpBroadcast::test_good
=========================== short test summary info ============================

You can find more information on how to contribute here

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

@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