Speed up test times 🚀

[williambdean]

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

Here are some of the slowest test times:

(6 minutes 19 seconds) tests --ignore=tests/tensor --ignore=tests/scan --ignore=tests/sparse

============================= slowest 50 durations =============================
26.31s call     tests/test_ifelse.py::TestIfelse::test_not_lazy_if_inplace
16.25s call     tests/scalar/test_basic.py::TestUpgradeToFloat::test_unary
8.16s call     tests/scalar/test_basic.py::TestUpgradeToFloat::test_binary
7.98s call     tests/test_rop.py::TestRopLop::test_max
7.66s call     tests/scalar/test_math.py::test_scalarloop_grad_mixed_dtypes[op2-scalar_loop_grads2]
7.61s call     tests/scalar/test_basic.py::TestUpgradeToFloat::test_true_div
6.94s call     tests/link/c/test_cmodule.py::test_cache_race_condition
5.46s call     tests/compile/test_builders.py::TestOpFromGraph::test_L_op_disconnected_output_grad
4.75s call     tests/test_gradient.py::test_undefined_grad_opt
3.91s call     tests/test_gradient.py::test_subgraph_grad
3.64s call     tests/compile/test_compilelock.py::test_locking_multiprocess_spawn
3.64s call     tests/test_rop.py::TestRopLop::test_alloc
3.62s call     tests/test_rop.py::TestRopLop::test_join
3.56s call     tests/scalar/test_math.py::test_scalarloop_grad_mixed_dtypes[op3-scalar_loop_grads3]
3.36s call     tests/compile/test_debugmode.py::test_debugmode_basic
3.02s call     tests/test_gradient.py::test_jacobian_matrix
2.85s call     tests/graph/test_compute_test_value.py::TestComputeTestValue::test_scan
2.79s call     tests/compile/test_misc.py::test_nnet
2.79s call     tests/test_gradient.py::test_known_grads
2.68s call     tests/test_gradient.py::TestHessianVectorProdudoct::test_rosen
2.65s call     tests/compile/function/test_pfunc.py::TestPfunc::test_shared
2.60s call     tests/compile/function/test_pfunc.py::TestAliasingRules::test_input_aliasing_affecting_inplace_operations
2.51s call     tests/test_breakpoint.py::TestPdbBreakpoint::test_infer_shape
2.43s call     tests/test_gradient.py::TestGrad::test_grad_grad_cubic
2.38s call     tests/scalar/test_basic.py::test_grad_abs
2.24s call     tests/compile/test_builders.py::TestOpFromGraph::test_grad_override[OpFromGraph]
2.24s call     tests/compile/test_profiling.py::TestProfiling::test_profiling
2.23s call     tests/compile/test_monitormode.py::test_not_inplace
2.20s call     tests/compile/test_builders.py::TestOpFromGraph::test_compute_test_value
2.16s call     tests/test_gradient.py::TestZeroGrad::test_grad
2.15s call     tests/test_rop.py::TestRopLop::test_elemwise0
2.12s call     tests/graph/test_compute_test_value.py::TestComputeTestValue::test_empty_elemwise
2.09s call     tests/test_rop.py::TestRopLop::test_flatten
2.05s call     tests/compile/function/test_types.py::TestFunction::test_swap_SharedVariable_with_given
2.01s call     tests/compile/function/test_pfunc.py::TestAliasingRules::test_potential_output_aliasing_induced_by_updates
1.95s call     tests/link/c/test_basic.py::test_shared_input_output
1.92s call     tests/compile/test_debugmode.py::TestViewMap::test_aliased_outputs_bad
1.92s call     tests/compile/function/test_pfunc.py::TestPfunc::test_doc
1.88s call     tests/compile/function/test_pfunc.py::TestPfunc::test_shared_mutable
1.88s call     tests/graph/test_compute_test_value.py::TestComputeTestValue::test_scan_err1
1.86s call     tests/test_gradient.py::test_jacobian_vector
1.86s call     tests/compile/test_debugmode.py::TestCheckIsfinite::test_check_isfinite
1.77s call     tests/test_rop.py::TestRopLop::test_elemwise1
1.72s call     tests/test_rop.py::TestRopLop::test_setsubtensor1
1.71s call     tests/compile/test_builders.py::TestOpFromGraph::test_lop_override[OpFromGraph]
1.69s call     tests/test_rop.py::TestRopLop::test_subtensor
1.69s call     tests/test_rop.py::TestRopLop::test_softmax
1.68s call     tests/compile/test_builders.py::TestOpFromGraph::test_rop_multiple_outputs[False]
1.62s call     tests/compile/function/test_function.py::test_function_dump
1.58s call     tests/compile/test_builders.py::TestOpFromGraph::test_shared_grad[cls_ofg1]
=========================== short test summary info ============================

(10 minutes 20 seconds) tests/scan

============================= slowest 50 durations =============================
47.49s call     tests/scan/test_basic.py::TestScan::test_second_derivative_disconnected_cost_with_mit_mot
33.21s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_taps_backwards
30.90s call     tests/scan/test_basic.py::TestScan::test_grad_mitsot
24.51s call     tests/scan/test_basic.py::TestScan::test_R_op[False]
22.66s call     tests/scan/test_basic.py::TestScan::test_R_op_2[False]
22.40s call     tests/scan/test_basic.py::TestScan::test_R_op[True]
20.52s call     tests/scan/test_basic.py::TestExamples::test_grad_multiple_outs_some_disconnected
17.44s call     tests/scan/test_rewriting.py::TestPushOutNonSeqScan::test_pushout_seqs
16.83s call     tests/scan/test_rewriting.py::TestPushOutAddScan::test_machine_translation
16.31s call     tests/scan/test_checkpoints.py::TestScanCheckpoint::test_backward_pass
15.84s call     tests/scan/test_basic.py::TestScan::test_only_nonseq_inputs
15.33s call     tests/scan/test_basic.py::TestScan::test_R_op_2[True]
12.74s call     tests/scan/test_basic.py::TestScan::test_grad_dtype_change
11.62s call     tests/scan/test_basic.py::TestScan::test_grad_of_grad_of_state
9.90s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs
9.31s call     tests/scan/test_basic.py::TestExamples::test_grad_multiple_outs_some_disconnected_2
8.65s call     tests/scan/test_basic.py::TestScan::test_grad_one_output
7.89s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_taps_state
7.67s call     tests/scan/test_basic.py::TestExamples::test_default_value_broadcasted
7.38s call     tests/scan/test_basic.py::TestScan::test_shared_arguments_with_updates
7.37s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until_and_truncate_sequence_taps
7.24s call     tests/scan/test_checkpoints.py::TestScanCheckpoint::test_forward_pass
6.54s call     tests/scan/test_basic.py::test_compute_test_value_grad
6.44s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_taps
5.85s call     tests/scan/test_basic.py::TestScan::test_inner_grad
5.80s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_seqs_different_nsteps
5.77s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_some_uncomputable
5.37s call     tests/scan/test_basic.py::TestScan::test_grad_sitsot
5.22s call     tests/scan/test_rewriting.py::TestSaveMem::test_savemem_opt_0_step
4.80s call     tests/scan/test_rewriting.py::TestSaveMem::test_save_mem_store_steps
4.80s call     tests/scan/test_basic.py::TestScan::test_inner_graph_cloning[default_rng]
4.74s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until_ndim_greater_one
4.56s call     tests/scan/test_basic.py::TestScan::test_grad_duplicate_outputs
3.78s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until
3.44s call     tests/scan/test_rewriting.py::TestScanInplaceOptimizer::test_simple_rnn_2
3.37s call     tests/scan/test_basic.py::TestExamples::test_gibbs_chain
3.31s call     tests/scan/test_basic.py::TestScan::test_non_shaped_input_disconnected_gradient[inside-explicit]
3.27s call     tests/scan/test_views.py::test_foldr_memory_consumption
3.07s call     tests/scan/test_basic.py::TestScan::test_grad_multiple_outs_some_truncate
2.83s call     tests/scan/test_basic.py::TestScan::test_grad_wrt_shared
2.72s call     tests/scan/test_basic.py::TestExamples::test_same
2.70s call     tests/scan/test_basic.py::test_compute_test_values
2.56s call     tests/scan/test_basic.py::TestScan::test_monitor_mode
2.35s call     tests/scan/test_basic.py::test_compute_test_value_grad_cast
2.34s call     tests/scan/test_basic.py::TestGradUntil::test_grad_until_and_truncate
2.26s call     tests/scan/test_basic.py::TestExamples::test_return_steps
2.21s call     tests/scan/test_basic.py::TestScan::test_pickling
2.12s call     tests/scan/test_rewriting.py::TestSaveMem::test_save_mem_reduced_number_of_steps
2.10s call     tests/scan/test_basic.py::TestScan::test_R_op_mitmot[False]
2.00s call     tests/scan/test_rewriting.py::TestSaveMem::test_while_scan_taps
=========================== short test summary info ============================

(4 minutes 38 seconds) tests/sparse

============================= slowest 50 durations =============================
73.57s call     tests/sparse/test_basic.py::TestUsmm::test_basic
14.20s call     tests/sparse/sandbox/test_sp.py::TestSP::test_convolution
9.30s call     tests/sparse/test_basic.py::TestCasting::test_cast
8.84s call     tests/sparse/test_basic.py::TestCasting::test_infer_shape
8.33s call     tests/sparse/sandbox/test_sp.py::TestSP::test_maxpool
5.27s call     tests/sparse/test_basic.py::TestCsmProperties::test_csm_properties_grad
5.27s call     tests/sparse/test_basic.py::TestAddMul::test_AddSD
4.98s call     tests/sparse/test_basic.py::TestStructuredDot::test_dot_sparse_sparse
3.48s call     tests/sparse/test_basic.py::TestCsm::test_csm_grad
2.88s call     tests/sparse/test_basic.py::TestHStack::test_grad
2.67s call     tests/sparse/test_basic.py::TestSpSum::test_grad
2.65s call     tests/sparse/test_basic.py::TestCsm::test_csm_unsorted
2.54s call     tests/sparse/test_basic.py::TestVStack::test_grad
2.49s call     tests/sparse/test_basic.py::TestAddMul::test_AddDS
2.14s call     tests/sparse/test_basic.py::TestDots::test_int32_dtype
1.59s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[any-DenseTensorType-None-None]
1.59s call     tests/sparse/test_basic.py::TestVerifyGradSparse::test_grad_fail
1.47s call     tests/sparse/test_basic.py::TestUsmm::test_infer_shape
1.46s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[var-DenseTensorType-None-None]
1.46s call     tests/sparse/test_basic.py::TestSparseInferShape::test_csm_grad
1.46s call     tests/sparse/test_basic.py::TestMulSV::test_mul_s_v_grad
1.42s call     tests/sparse/test_basic.py::TestStructuredAddSV::test_structured_add_s_v_grad
1.39s call     tests/sparse/test_basic.py::TestTrueDot::test_grad
1.38s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[mean-DenseTensorType-None-None]
1.31s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__divmod__-DenseTensorType]
1.24s call     tests/sparse/test_basic.py::TestConversion::test_format_ndim
1.06s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[argmin-DenseTensorType-None-None]
0.87s call     tests/sparse/test_basic.py::TestConstructSparseFromList::test_adv_sub1_sparse_grad
0.85s call     tests/sparse/test_basic.py::TestDots::test_sparse_sparse
0.73s call     tests/sparse/test_basic.py::TestStructuredDot::test_upcast
0.73s call     tests/sparse/test_basic.py::TestCsm::test_csm_sparser
0.73s call     tests/sparse/test_basic.py::TestStructuredDot::test_structureddot_csc_grad
0.72s call     tests/sparse/test_basic.py::TestTrueDot::test_infer_shape
0.72s call     tests/sparse/test_basic.py::TestStructuredDot::test_structureddot_csr_grad
0.72s call     tests/sparse/test_basic.py::TestSamplingDot::test_op
0.68s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__or__-DenseTensorType]
0.68s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__pow__-DenseTensorType]
0.68s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__xor__-DenseTensorType]
0.68s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__mod__-DenseTensorType]
0.67s call     tests/sparse/test_basic.py::TestClean::test_grad
0.67s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__and__-DenseTensorType]
0.66s call     tests/sparse/test_basic.py::TestCasting::test_grad
0.66s call     tests/sparse/test_var.py::TestSparseVariable::test_binary[__truediv__-DenseTensorType]
0.64s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[deg2rad-DenseTensorType-None-None]
0.64s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[arctan-DenseTensorType-None-None]
0.64s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[__abs__-DenseTensorType-None-None]
0.64s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[rad2deg-DenseTensorType-None-None]
0.64s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[arccosh-DenseTensorType-None-None]
0.63s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[round-DenseTensorType-None-None]
0.63s call     tests/sparse/test_var.py::TestSparseVariable::test_unary[arcsin-DenseTensorType-None-None]
=========================== short test summary info ============================

(19 minutes 10 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 =============================
43.43s call     tests/tensor/test_subtensor.py::TestSubtensor::test_shape_i_const
28.44s call     tests/tensor/test_blockwise.py::TestCholesky::test_grad
13.45s call     tests/tensor/test_subtensor.py::TestSubtensor::test_grad_list
10.35s call     tests/tensor/test_extra_ops.py::test_space_ops[start2-stop2-25-True-0-None-geomspace]
10.28s call     tests/tensor/test_casting.py::TestCasting::test_convert_to_complex
9.36s call     tests/tensor/test_pad.py::test_nd_padding[symmetric-reflect]
7.94s call     tests/tensor/test_pad.py::test_nd_padding[asymmetric-reflect]
7.86s call     tests/tensor/test_fft.py::TestFFT::test_grad_rfft
7.80s call     tests/tensor/test_blockwise.py::TestSolveVector::test_grad
7.78s call     tests/tensor/test_pad.py::test_nd_padding[symmetric-symmetric]
6.92s call     tests/tensor/test_extra_ops.py::test_space_ops[start2-stop2-25-True-0-None-linspace]
6.81s call     tests/tensor/test_extra_ops.py::test_space_ops[1-stop5-30-True-0-None-linspace]
6.72s call     tests/tensor/test_extra_ops.py::test_space_ops[1-stop5-30-True-0-None-geomspace]
6.70s call     tests/tensor/test_nlinalg.py::TestSvd::test_grad[batched=True-(3,3)-compute_uv=True, full_matrices=False, gradient=U]
6.48s call     tests/tensor/test_slinalg.py::test_cholesky_grad_indef
6.40s call     tests/tensor/test_einsum.py::test_general_dot
6.26s call     tests/tensor/test_pad.py::test_reflect_pad[even-symmetrical-3d square]
6.20s call     tests/tensor/test_pad.py::test_symmetric_pad[even-symmetrical-2d square]
6.15s call     tests/tensor/test_blockwise.py::TestMatrixInverse::test_grad
6.04s call     tests/tensor/test_subtensor.py::TestIncSubtensor::test_grad_inc_set
6.01s call     tests/tensor/test_extra_ops.py::test_space_ops[1-10-50-True-0-None-geomspace]
5.98s call     tests/tensor/test_subtensor.py::TestInferShape::test_IncSubtensor
5.89s call     tests/tensor/test_blas_c.py::TestBlasStrides::test_gemm
5.73s call     tests/tensor/random/rewriting/test_basic.py::test_Subtensor_lift[indices2-True-dist_op2-dist_params2-None]
5.68s call     tests/tensor/test_blas_c.py::TestCGemvNoFlags::test_cgemv
5.58s call     tests/tensor/test_nlinalg.py::test_qr_grad[real-shape=(3, 3), gradient_test_case=Q, mode=reduced]
5.52s call     tests/tensor/test_extra_ops.py::TestFillDiagonalOffset::test_gradient[-5]
5.50s call     tests/tensor/test_fft.py::TestFFT::test_rfft_float
5.49s call     tests/tensor/test_extra_ops.py::TestCumOp::test_grad
5.32s call     tests/tensor/test_extra_ops.py::TestRepeat::test_basic[int8-3]
5.31s call     tests/tensor/test_blockwise.py::TestSolveMatrix::test_grad
4.82s call     tests/tensor/test_blas_c.py::TestBlasStridesC::test_gemm
4.82s call     tests/tensor/test_blas_scipy.py::TestBlasStridesScipy::test_gemm
4.81s call     tests/tensor/test_interpolate.py::test_interp
4.80s call     tests/tensor/test_blas_scipy.py::TestBlasStrides::test_gemm
4.79s call     tests/tensor/test_einsum.py::test_einsum_signatures[oij,imj,mjkn,lnk,plk->op-True]
4.75s call     tests/tensor/random/rewriting/test_basic.py::test_Subtensor_lift[indices17-True-dist_op17-dist_params17-None]
4.71s call     tests/tensor/test_sort.py::TestSort::test_grad_negative_axis_4d
4.67s call     tests/tensor/test_pad.py::test_reflect_pad[even-asymmetrical_left-3d square]
4.65s call     tests/tensor/test_pad.py::test_reflect_pad[even-asymmetric_right-3d square]
4.62s call     tests/tensor/test_pad.py::test_symmetric_pad[even-symmetrical-3d square]
4.54s call     tests/tensor/test_nlinalg.py::TestSvd::test_grad[batched=False-(3,3)-compute_uv=True, full_matrices=False, gradient=U]
4.54s call     tests/tensor/test_interpolate.py::test_interp_padded
4.50s call     tests/tensor/test_extra_ops.py::test_space_ops[1-10-50-True-0-None-linspace]
4.48s call     tests/tensor/test_shape.py::TestReshape::test_more_shapes
4.31s call     tests/tensor/test_slinalg.py::test_solve_discrete_lyapunov_gradient[direct-matrix-float]
4.25s call     tests/tensor/test_extra_ops.py::TestBinCount::test_bincountFn[int16]
4.25s call     tests/tensor/test_nlinalg.py::TestEighFloat32::test_grad
4.20s call     tests/tensor/test_extra_ops.py::test_space_ops[1-stop6-30-False--1-None-linspace]
4.19s call     tests/tensor/test_subtensor.py::TestSubtensor::test_ok_list
=========================== short test summary info ============================

(3 minutes 59 seconds) tests/tensor/conv

============================= slowest 50 durations =============================
44.18s call     tests/tensor/conv/test_abstract_conv.py::TestAbstractConvNoOptim::test_all
18.00s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConv3dNoOptim::test_fwd
13.83s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConv3dNoOptim::test_gradweights
12.96s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_fractional_bilinear_upsampling
11.49s call     tests/tensor/conv/test_abstract_conv.py::TestUnsharedConv::test_fwd
9.72s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConv3dNoOptim::test_gradinputs
7.17s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_fractional_bilinear_upsampling_shape
6.65s call     tests/tensor/conv/test_abstract_conv.py::TestUnsharedConv::test_gradinput
4.42s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_kernel_2D
4.06s call     tests/tensor/conv/test_abstract_conv.py::TestUnsharedConv::test_gradweight
4.01s call     tests/tensor/conv/test_abstract_conv.py::TestAssertConvShape::test_basic
3.27s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_kernel_1D
3.24s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConvNoOptim::test_gradweights
3.08s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConvNoOptim::test_fwd
2.95s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_upsampling_1D
2.78s call     tests/tensor/conv/test_abstract_conv.py::TestCausalConv::test_interface
2.65s call     tests/tensor/conv/test_abstract_conv.py::TestConvGradInputsShape::test_check_shape
2.53s call     tests/tensor/conv/test_abstract_conv.py::TestGroupedConvNoOptim::test_gradinputs
1.64s call     tests/tensor/conv/test_abstract_conv.py::TestConv2dGrads::test_conv2d_grad_wrt_inputs
1.61s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_basic
1.60s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv2d
1.32s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_compare_1D_and_2D_upsampling_values
1.31s call     tests/tensor/conv/test_abstract_conv.py::TestConv2dGrads::test_conv2d_grad_wrt_weights
1.26s call     tests/tensor/conv/test_abstract_conv.py::TestAbstractConvNoOptim::test_gradinput_impossible_output_shapes
1.09s call     tests/tensor/conv/test_abstract_conv.py::TestBilinearUpsampling::test_bilinear_upsampling_reshaping
0.94s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv3d
0.91s call     tests/tensor/conv/test_abstract_conv.py::TestConv2dTranspose::test_interface
0.49s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv2d_grad_wrt_weights
0.48s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv3d_grad_wrt_inputs
0.47s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv3d_grad_wrt_weights
0.46s call     tests/tensor/conv/test_abstract_conv.py::TestAssertShape::test_shape_check_conv2d_grad_wrt_inputs
0.31s call     tests/tensor/conv/test_abstract_conv.py::TestAsymmetricPadding::test_gradweight
0.28s call     tests/tensor/conv/test_abstract_conv.py::TestAsymmetricPadding::test_gradinput
0.22s call     tests/tensor/conv/test_abstract_conv.py::TestAsymmetricPadding::test_fwd
0.20s 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.07s call     tests/tensor/conv/test_abstract_conv.py::TestSeparableConv::test_interface2d
0.07s call     tests/tensor/conv/test_abstract_conv.py::TestConvGradInputsShape::test_get_shape
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 ============================

(13 minutes 48 seconds) tests/tensor/rewriting

============================= slowest 50 durations =============================
126.32s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const4
57.99s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_none_slice
51.78s call     tests/tensor/rewriting/test_elemwise.py::TestFusion::test_big_fusion
18.73s call     tests/tensor/rewriting/test_math.py::test_local_log_sum_exp_maximum
15.86s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_prod_of_div
15.17s call     tests/tensor/rewriting/test_basic.py::TestLocalSwitchSink::test_local_mul_switch_sink
14.15s call     tests/tensor/rewriting/test_math.py::TestAlgebraicCanonizer::test_mul_div_cases
13.44s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const3
12.89s call     tests/tensor/rewriting/test_subtensor_lift.py::test_local_subtensor_of_alloc
10.44s call     tests/tensor/rewriting/test_subtensor.py::test_local_IncSubtensor_serialize
10.11s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const2
8.00s call     tests/tensor/rewriting/test_subtensor.py::test_local_subtensor_remove_broadcastable_index
7.46s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const
6.38s call     tests/tensor/rewriting/test_math.py::TestLocalErfc::test_local_log_erfc
6.07s call     tests/tensor/rewriting/test_math.py::test_local_log_add_exp
5.94s call     tests/tensor/rewriting/test_math.py::TestIntDivByOne::test_remove_floor
5.80s call     tests/tensor/rewriting/test_math.py::TestReduceChain::test_local_sum_sum_prod_prod
5.73s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_sum_of_non_scalar_mul
5.19s call     tests/tensor/rewriting/test_math.py::TestSigmoidRewrites::test_exp_over_1_plus_exp
5.16s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_sum_prod_of_scalar_mul
5.07s call     tests/tensor/rewriting/test_special.py::test_softmax_graph
4.95s call     tests/tensor/rewriting/test_math.py::TestLocalReduce::test_local_reduce_broadcast_some_0
4.95s call     tests/tensor/rewriting/test_linalg.py::test_matrix_inverse_rop_lop
4.90s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_sum_of_div
4.72s call     tests/tensor/rewriting/test_basic.py::test_local_useless_slice
4.72s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_none_index
4.62s call     tests/tensor/rewriting/test_math.py::test_merge_abs_bugfix
4.39s call     tests/tensor/rewriting/test_linalg.py::test_det_blockdiag_rewrite
4.17s call     tests/tensor/rewriting/test_basic.py::test_local_flatten_lift[2]
4.16s call     tests/tensor/rewriting/test_math.py::test_local_mul_exp_to_exp_add
4.06s call     tests/tensor/rewriting/test_basic.py::TestLocalSwitchSink::test_local_div_switch_sink
3.38s call     tests/tensor/rewriting/test_math.py::test_log1p
3.16s call     tests/tensor/rewriting/test_basic.py::TestConstantFolding::test_constant_get_stabilized
2.99s call     tests/tensor/rewriting/test_math.py::test_local_subtensor_of_dot
2.87s call     tests/tensor/rewriting/test_math.py::TestFuncInverse::test
2.61s call     tests/tensor/rewriting/test_math.py::test_local_expm1
2.44s call     tests/tensor/rewriting/test_basic.py::TestLocalUselessIncSubtensorAlloc::test_advanced_inc_subtensor
2.43s call     tests/tensor/rewriting/test_math.py::TestLocalSumProd::test_local_sum_prod_alloc
2.43s call     tests/tensor/rewriting/test_blockwise.py::test_blockwise_alloc
2.38s call     tests/tensor/rewriting/test_uncanonicalize.py::TestMinMax::test_optimization_min
2.37s call     tests/tensor/rewriting/test_basic.py::TestLocalElemwiseAlloc::test_remove_alloc_wo_dimshuffle
2.35s call     tests/tensor/rewriting/test_subtensor.py::TestLocalAdvSub1AdvIncSub1::test_assert
2.34s call     tests/tensor/rewriting/test_uncanonicalize.py::test_local_dimshuffle_subtensor
2.25s call     tests/tensor/rewriting/test_special.py::TestLogSoftmaxRewrites::test_local_logsoftmax_grad_rewrite[None]
2.23s call     tests/tensor/rewriting/test_elemwise.py::TestFusion::test_elemwise_fusion[case73]
2.22s call     tests/tensor/rewriting/test_basic.py::test_local_useless_split
2.20s call     tests/tensor/rewriting/test_math.py::TestAlgebraicCanonizer::test_elemwise_multiple_inputs_rewrites
2.18s call     tests/tensor/rewriting/test_subtensor.py::TestLocalSubtensorMerge::test_const6
2.17s call     tests/tensor/rewriting/test_basic.py::TestLocalElemwiseAlloc::test_basic[<lambda>-x_shape3-y_shape3-True]
2.16s call     tests/tensor/rewriting/test_elemwise.py::test_local_useless_composite_outputs
=========================== short test summary info ============================

(15 minutes 20 seconds) tests/tensor/test_math.py

============================= slowest 50 durations =============================
158.69s call     tests/tensor/test_math.py::TestSumProdReduceDtype::test_reduce_custom_dtype
72.75s call     tests/tensor/test_math.py::TestProdWithoutZerosDtype::test_prod_without_zeros_custom_dtype
30.82s call     tests/tensor/test_math.py::TestProdWithoutZerosDtype::test_prod_without_zeros_custom_acc_dtype
27.58s call     tests/tensor/test_math.py::TestMaxAndArgmax::test_grad
16.73s call     tests/tensor/test_math.py::TestComparison::test_isclose
15.67s call     tests/tensor/test_math.py::test_var
15.62s call     tests/tensor/test_math.py::TestSumProdReduceDtype::test_reduce_default_dtype
14.43s call     tests/tensor/test_math.py::TestTensordot::test_basic
13.89s call     tests/tensor/test_math.py::TestPowBroadcast::test_grad
11.56s call     tests/tensor/test_math.py::TestAddBroadcast::test_good
10.51s call     tests/tensor/test_math.py::TestProd::test_verify_grad
9.59s call     tests/tensor/test_math.py::TestMulBroadcast::test_good
9.17s call     tests/tensor/test_math.py::TestArctan2Broadcast::test_grad
8.82s call     tests/tensor/test_math.py::TestSubBroadcast::test_grad
8.12s call     tests/tensor/test_math.py::TestSubBroadcast::test_good
8.02s call     tests/tensor/test_math.py::TestMinMax::test_grad_min
7.86s call     tests/tensor/test_math.py::TestArctan2Broadcast::test_good
7.63s call     tests/tensor/test_math.py::TestModBroadcast::test_good
7.53s call     tests/tensor/test_math.py::TestProd::test_other_grad_tests
7.39s call     tests/tensor/test_math.py::TestCeilIntDivBroadcast::test_good
7.11s call     tests/tensor/test_math.py::TestMaximumBroadcast::test_grad
7.08s call     tests/tensor/test_math.py::TestMinimumBroadcast::test_grad
6.91s call     tests/tensor/test_math.py::TestProd::test_prod_no_zeros_in_input
6.79s call     tests/tensor/test_math.py::TestTrueDivBroadcast::test_grad
6.71s call     tests/tensor/test_math.py::TestTrueDivBroadcast::test_good
6.65s call     tests/tensor/test_math.py::TestModBroadcast::test_grad
6.53s call     tests/tensor/test_math.py::TestMaximumBroadcast::test_good
6.52s call     tests/tensor/test_math.py::TestMinimumBroadcast::test_good
6.51s call     tests/tensor/test_math.py::TestMulBroadcast::test_grad
6.04s call     tests/tensor/test_math.py::TestClip2::test_clip_repeat_verify_grad
5.58s call     tests/tensor/test_math.py::TestProdWithoutZerosDtype::test_prod_without_zeros_default_acc_dtype
5.49s call     tests/tensor/test_math.py::TestClip1::test_good
5.35s call     tests/tensor/test_math.py::TestArgminArgmax::test_uint
5.26s call     tests/tensor/test_math.py::TestMeanDtype::test_mean_default_dtype
4.79s call     tests/tensor/test_math.py::test_cov
4.73s call     tests/tensor/test_math.py::TestPowBroadcast::test_good
4.34s call     tests/tensor/test_math.py::TestOuter::test_grad
4.22s call     tests/tensor/test_math.py::TestInvBroadcast::test_good
4.16s call     tests/tensor/test_math.py::TestAdd::test_complex_all_ops
4.06s call     tests/tensor/test_math.py::TestSqrBroadcast::test_good
4.00s call     tests/tensor/test_math.py::TestAbsBroadcast::test_good
3.84s call     tests/tensor/test_math.py::TestDivimpl::test_impls
3.82s call     tests/tensor/test_math.py::TestDenseDot::test_good
3.51s call     tests/tensor/test_math.py::TestExp2Broadcast::test_grad
3.41s call     tests/tensor/test_math.py::TestClip2::test_clip_repeat_grad
3.25s call     tests/tensor/test_math.py::TestArctanBroadcast::test_good
3.22s call     tests/tensor/test_math.py::TestTanBroadcast::test_good
3.21s call     tests/tensor/test_math.py::TestRad2degBroadcast::test_good
3.20s call     tests/tensor/test_math.py::TestSinhBroadcast::test_good
3.19s call     tests/tensor/test_math.py::TestTanhBroadcast::test_good
=========================== short test summary info ============================

(9 minutes 43 seconds) tests/tensor/test_basic.py tests/tensor/test_inplace.py

============================= slowest 50 durations =============================
64.93s call     tests/tensor/test_basic.py::TestSecondSameRank::test_good
64.80s call     tests/tensor/test_basic.py::TestSecondSameRank::test_bad_runtime
35.53s call     tests/tensor/test_basic.py::TestTriangle::test_tril_triu
19.80s call     tests/tensor/test_basic.py::TestAllocDiag::test_alloc_diag_values
12.35s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_roll
11.17s call     tests/tensor/test_basic.py::TestNdGrid::test_mgrid_numpy_equiv
10.39s call     tests/tensor/test_basic.py::TestSecondBroadcast::test_good
8.08s call     tests/tensor/test_inplace.py::TestAddInplaceBroadcast::test_good
8.07s call     tests/tensor/test_inplace.py::TestSubInplaceBroadcast::test_good
8.07s call     tests/tensor/test_inplace.py::TestMulInplaceBroadcast::test_good
7.02s call     tests/tensor/test_inplace.py::TestModInplaceBroadcast::test_good
6.98s call     tests/tensor/test_inplace.py::TestIntDivInplaceBroadcast::test_good
6.50s call     tests/tensor/test_inplace.py::TestMinimumInplaceBroadcast::test_good
6.49s call     tests/tensor/test_inplace.py::TestMaximumInplaceBroadcast::test_good
5.81s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_stack_hessian
5.44s call     tests/tensor/test_basic.py::TestZerosLikeBroadcast::test_good
4.85s call     tests/tensor/test_basic.py::TestNdGrid::test_mgrid_pytensor_variable_numpy_equiv
4.73s call     tests/tensor/test_basic.py::TestAlloc::test_alloc_constant_folding
4.46s call     tests/tensor/test_inplace.py::TestTrueDivInplaceBroadcast::test_good
4.24s call     tests/tensor/test_inplace.py::TestPowInplaceBroadcast::test_good
4.04s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_broadcastable_flags_many_dims_and_inputs
4.00s call     tests/tensor/test_basic.py::TestAllocBroadcast::test_good
3.97s call     tests/tensor/test_inplace.py::TestSqrInplaceBroadcast::test_good
3.96s call     tests/tensor/test_inplace.py::TestNegInplaceBroadcast::test_good
3.87s call     tests/tensor/test_basic.py::TestSwitchBroadcast::test_grad
3.72s call     tests/tensor/test_inplace.py::TestArctan2InplaceBroadcast::test_good
3.49s call     tests/tensor/test_basic.py::TestInferShape::test_Flatten
3.15s call     tests/tensor/test_inplace.py::TestAbsInplaceBroadcast::test_good
3.12s call     tests/tensor/test_inplace.py::TestTruncInplaceBroadcast::test_good
2.91s call     tests/tensor/test_basic.py::TestAlloc13GradBroadcast::test_grad
2.75s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_join_matrix1[False]
2.66s call     tests/tensor/test_basic.py::TestSwitchBroadcast::test_good
2.50s call     tests/tensor/test_basic.py::TestAllocDimshuffleGradBroadcast::test_grad
2.46s call     tests/tensor/test_basic.py::TestTile::test_tile_grad
2.43s call     tests/tensor/test_basic.py::TestJoinAndSplit::test_broadcastable_flag_assignment_mixed_thisaxes
2.41s call     tests/tensor/test_basic.py::TestOnesLikeBroadcast::test_good
2.36s call     tests/tensor/test_basic.py::TestARange::test_infer_shape[custom]
2.33s call     tests/tensor/test_inplace.py::TestSgnInplaceBroadcast::test_good
2.17s call     tests/tensor/test_inplace.py::TestConjInplaceBroadcast::test_good
2.11s call     tests/tensor/test_basic.py::TestARange::test_grads
2.06s call     tests/tensor/test_basic.py::TestAlloc::test_zeros
1.96s call     tests/tensor/test_inplace.py::test_XOR_inplace
1.96s call     tests/tensor/test_basic.py::TestAllocb1GradBroadcast::test_grad
1.89s call     tests/tensor/test_basic.py::TestAlloc01GradBroadcast::test_grad
1.77s call     tests/tensor/test_basic.py::test_empty
1.60s call     tests/tensor/test_basic.py::TestAllocb4GradBroadcast::test_grad
1.57s call     tests/tensor/test_basic.py::TestTile::test_combined_reps_entries[2]
1.53s call     tests/tensor/test_basic.py::TestExtractDiag::test_grad_3d[0-0-1]
1.52s call     tests/tensor/test_basic.py::TestAllocb3GradBroadcast::test_grad
1.44s call     tests/tensor/test_basic.py::TestTile::test_tile_separate_reps_entries[A_shape6-reps_test6]
=========================== short test summary info ============================

You can find more information on how to contribute here

Automatically generated by GitHub Action
Latest run date: 2025-06-04
Run logs: 15393103893

[ricardoV94]

@wd60622 we need to update, probably because of the new matrix merged recently?

[ricardoV94]

@williambdean this is not including numba / jax CI?

[williambdean]

@williambdean this is not including numba / jax CI?

Would have to check. Is that included here? https://github.com/pymc-devs/pytensor/actions/runs/15284667195

[ricardoV94]

Yes. JAX: https://github.com/pymc-devs/pytensor/actions/runs/15284667195/job/42991990937
Numba: https://github.com/pymc-devs/pytensor/actions/runs/15284667195/job/42991989813