issues
search
pytorch
/
tensordict
TensorDict is a pytorch dedicated tensor container.
MIT License
803
stars
65
forks
source link
[BugFix] fix _expand_to_match_shape for single bool tensor
#902
Closed
vmoens
closed
1 month ago
github-actions[bot]
commented
1 month ago
$\color{#D29922}\textsf{\Large\⚠\kern{0.2cm}\normalsize Warning}$ Result of CPU Benchmark Tests
Total Benchmarks: 144. Improved: $\large\color{#35bf28}26$. Worsened: $\large\color{#d91a1a}13$.
Expand to view detailed results
| Name | Max | Mean | Ops | Ops on Repo `HEAD` | Change | | ------------------------------------------ | --------- | --------- | --------------- | ------------------ | ----------------------------------- | | test_plain_set_nested | 45.5250μs | 21.9617μs | 45.5338 KOps/s | 44.0560 KOps/s | $\color{#35bf28}+3.35\\%$ | | test_plain_set_stack_nested | 63.4780μs | 22.1794μs | 45.0869 KOps/s | 43.9914 KOps/s | $\color{#35bf28}+2.49\\%$ | | test_plain_set_nested_inplace | 89.7970μs | 23.9012μs | 41.8389 KOps/s | 40.3770 KOps/s | $\color{#35bf28}+3.62\\%$ | | test_plain_set_stack_nested_inplace | 76.5330μs | 24.1039μs | 41.4871 KOps/s | 41.0286 KOps/s | $\color{#35bf28}+1.12\\%$ | | test_items | 24.0540μs | 2.5795μs | 387.6664 KOps/s | 357.7499 KOps/s | $\textbf{\color{#35bf28}+8.36\\%}$ | | test_items_nested | 0.7392ms | 0.3611ms | 2.7695 KOps/s | 2.7502 KOps/s | $\color{#35bf28}+0.70\\%$ | | test_items_nested_locked | 2.1839ms | 0.3632ms | 2.7534 KOps/s | 2.7660 KOps/s | $\color{#d91a1a}-0.45\\%$ | | test_items_nested_leaf | 0.3255ms | 89.1020μs | 11.2231 KOps/s | 11.6143 KOps/s | $\color{#d91a1a}-3.37\\%$ | | test_items_stack_nested | 0.5639ms | 0.3645ms | 2.7434 KOps/s | 2.7288 KOps/s | $\color{#35bf28}+0.54\\%$ | | test_items_stack_nested_leaf | 0.1630ms | 85.4703μs | 11.7000 KOps/s | 11.5941 KOps/s | $\color{#35bf28}+0.91\\%$ | | test_items_stack_nested_locked | 0.7707ms | 0.3696ms | 2.7054 KOps/s | 2.7677 KOps/s | $\color{#d91a1a}-2.25\\%$ | | test_keys | 47.6290μs | 3.8536μs | 259.4956 KOps/s | 244.1452 KOps/s | $\textbf{\color{#35bf28}+6.29\\%}$ | | test_keys_nested | 0.3137ms | 0.1430ms | 6.9935 KOps/s | 6.8945 KOps/s | $\color{#35bf28}+1.44\\%$ | | test_keys_nested_locked | 1.9063ms | 0.1537ms | 6.5081 KOps/s | 6.6518 KOps/s | $\color{#d91a1a}-2.16\\%$ | | test_keys_nested_leaf | 0.2186ms | 0.1237ms | 8.0815 KOps/s | 8.0074 KOps/s | $\color{#35bf28}+0.93\\%$ | | test_keys_stack_nested | 0.2408ms | 0.1447ms | 6.9100 KOps/s | 6.8815 KOps/s | $\color{#35bf28}+0.42\\%$ | | test_keys_stack_nested_leaf | 0.2161ms | 0.1238ms | 8.0749 KOps/s | 8.0556 KOps/s | $\color{#35bf28}+0.24\\%$ | | test_keys_stack_nested_locked | 0.2599ms | 0.1493ms | 6.6962 KOps/s | 6.6383 KOps/s | $\color{#35bf28}+0.87\\%$ | | test_values | 10.3143μs | 1.1639μs | 859.1836 KOps/s | 861.2822 KOps/s | $\color{#d91a1a}-0.24\\%$ | | test_values_nested | 91.4700μs | 48.9993μs | 20.4085 KOps/s | 20.0089 KOps/s | $\color{#35bf28}+2.00\\%$ | | test_values_nested_locked | 94.2850μs | 48.7434μs | 20.5156 KOps/s | 19.9248 KOps/s | $\color{#35bf28}+2.97\\%$ | | test_values_nested_leaf | 96.3670μs | 44.7104μs | 22.3662 KOps/s | 22.3119 KOps/s | $\color{#35bf28}+0.24\\%$ | | test_values_stack_nested | 0.1002ms | 50.6312μs | 19.7507 KOps/s | 19.9275 KOps/s | $\color{#d91a1a}-0.89\\%$ | | test_values_stack_nested_leaf | 96.4800μs | 44.2706μs | 22.5884 KOps/s | 22.3222 KOps/s | $\color{#35bf28}+1.19\\%$ | | test_values_stack_nested_locked | 0.1024ms | 51.1807μs | 19.5386 KOps/s | 20.0695 KOps/s | $\color{#d91a1a}-2.65\\%$ | | test_membership | 2.8383μs | 0.7063μs | 1.4159 MOps/s | 1.1023 MOps/s | $\textbf{\color{#35bf28}+28.45\\%}$ | | test_membership_nested | 27.9420μs | 2.6925μs | 371.3982 KOps/s | 372.0381 KOps/s | $\color{#d91a1a}-0.17\\%$ | | test_membership_nested_leaf | 29.2140μs | 2.7090μs | 369.1374 KOps/s | 326.7643 KOps/s | $\textbf{\color{#35bf28}+12.97\\%}$ | | test_membership_stacked_nested | 28.4840μs | 2.6800μs | 373.1388 KOps/s | 369.4733 KOps/s | $\color{#35bf28}+0.99\\%$ | | test_membership_stacked_nested_leaf | 30.5770μs | 2.7093μs | 369.0963 KOps/s | 369.4705 KOps/s | $\color{#d91a1a}-0.10\\%$ | | test_membership_nested_last | 41.9380μs | 3.9699μs | 251.8933 KOps/s | 250.1299 KOps/s | $\color{#35bf28}+0.71\\%$ | | test_membership_nested_leaf_last | 24.2750μs | 4.0002μs | 249.9870 KOps/s | 250.9570 KOps/s | $\color{#d91a1a}-0.39\\%$ | | test_membership_stacked_nested_last | 58.9900μs | 12.9737μs | 77.0792 KOps/s | 252.3239 KOps/s | $\textbf{\color{#d91a1a}-69.45\\%}$ | | test_membership_stacked_nested_leaf_last | 43.1510μs | 13.1713μs | 75.9226 KOps/s | 248.8684 KOps/s | $\textbf{\color{#d91a1a}-69.49\\%}$ | | test_nested_getleaf | 57.0260μs | 11.0861μs | 90.2031 KOps/s | 95.1283 KOps/s | $\textbf{\color{#d91a1a}-5.18\\%}$ | | test_nested_get | 51.8170μs | 10.6157μs | 94.1997 KOps/s | 99.3060 KOps/s | $\textbf{\color{#d91a1a}-5.14\\%}$ | | test_stacked_getleaf | 50.0640μs | 10.9301μs | 91.4908 KOps/s | 94.0257 KOps/s | $\color{#d91a1a}-2.70\\%$ | | test_stacked_get | 33.6420μs | 10.4294μs | 95.8824 KOps/s | 99.0112 KOps/s | $\color{#d91a1a}-3.16\\%$ | | test_nested_getitemleaf | 53.8700μs | 11.3772μs | 87.8949 KOps/s | 90.0638 KOps/s | $\color{#d91a1a}-2.41\\%$ | | test_nested_getitem | 60.3820μs | 10.5287μs | 94.9789 KOps/s | 97.7026 KOps/s | $\color{#d91a1a}-2.79\\%$ | | test_stacked_getitemleaf | 40.4650μs | 11.2858μs | 88.6073 KOps/s | 89.9988 KOps/s | $\color{#d91a1a}-1.55\\%$ | | test_stacked_getitem | 56.8350μs | 10.5348μs | 94.9237 KOps/s | 97.4208 KOps/s | $\color{#d91a1a}-2.56\\%$ | | test_lock_nested | 3.0446ms | 0.5262ms | 1.9006 KOps/s | 1.6648 KOps/s | $\textbf{\color{#35bf28}+14.16\\%}$ | | test_lock_stack_nested | 0.6106ms | 0.4677ms | 2.1381 KOps/s | 2.0306 KOps/s | $\textbf{\color{#35bf28}+5.29\\%}$ | | test_unlock_nested | 0.9455ms | 0.4449ms | 2.2478 KOps/s | 1.9542 KOps/s | $\textbf{\color{#35bf28}+15.03\\%}$ | | test_unlock_stack_nested | 0.5492ms | 0.3819ms | 2.6186 KOps/s | 2.4484 KOps/s | $\textbf{\color{#35bf28}+6.95\\%}$ | | test_flatten_speed | 0.2396ms | 0.1078ms | 9.2736 KOps/s | 9.3572 KOps/s | $\color{#d91a1a}-0.89\\%$ | | test_unflatten_speed | 0.6077ms | 0.4529ms | 2.2078 KOps/s | 2.2503 KOps/s | $\color{#d91a1a}-1.89\\%$ | | test_common_ops | 5.3983ms | 1.1482ms | 870.9199 Ops/s | 850.5103 Ops/s | $\color{#35bf28}+2.40\\%$ | | test_creation | 26.3890μs | 2.5970μs | 385.0580 KOps/s | 399.5637 KOps/s | $\color{#d91a1a}-3.63\\%$ | | test_creation_empty | 65.9130μs | 18.9621μs | 52.7368 KOps/s | 48.3839 KOps/s | $\textbf{\color{#35bf28}+9.00\\%}$ | | test_creation_nested_1 | 50.9250μs | 22.1614μs | 45.1235 KOps/s | 42.3411 KOps/s | $\textbf{\color{#35bf28}+6.57\\%}$ | | test_creation_nested_2 | 90.2990μs | 26.4394μs | 37.8224 KOps/s | 35.5785 KOps/s | $\textbf{\color{#35bf28}+6.31\\%}$ | | test_clone | 87.3230μs | 17.8868μs | 55.9072 KOps/s | 55.9399 KOps/s | $\color{#d91a1a}-0.06\\%$ | | test_getitem[int] | 0.9763ms | 13.3332μs | 75.0008 KOps/s | 76.8919 KOps/s | $\color{#d91a1a}-2.46\\%$ | | test_getitem[slice_int] | 0.1541ms | 34.0167μs | 29.3973 KOps/s | 30.5000 KOps/s | $\color{#d91a1a}-3.62\\%$ | | test_getitem[range] | 0.4018ms | 57.1003μs | 17.5131 KOps/s | 17.5687 KOps/s | $\color{#d91a1a}-0.32\\%$ | | test_getitem[tuple] | 0.1322ms | 27.2979μs | 36.6328 KOps/s | 36.5675 KOps/s | $\color{#35bf28}+0.18\\%$ | | test_getitem[list] | 0.2565ms | 52.0509μs | 19.2120 KOps/s | 19.3084 KOps/s | $\color{#d91a1a}-0.50\\%$ | | test_setitem_dim[int] | 57.7170μs | 32.7733μs | 30.5126 KOps/s | 26.6784 KOps/s | $\textbf{\color{#35bf28}+14.37\\%}$ | | test_setitem_dim[slice_int] | 0.1685ms | 70.7185μs | 14.1406 KOps/s | 13.1765 KOps/s | $\textbf{\color{#35bf28}+7.32\\%}$ | | test_setitem_dim[range] | 0.1389ms | 92.2327μs | 10.8421 KOps/s | 10.1312 KOps/s | $\textbf{\color{#35bf28}+7.02\\%}$ | | test_setitem_dim[tuple] | 0.1077ms | 58.9015μs | 16.9775 KOps/s | 15.6625 KOps/s | $\textbf{\color{#35bf28}+8.40\\%}$ | | test_setitem | 0.1111ms | 30.8819μs | 32.3814 KOps/s | 31.3893 KOps/s | $\color{#35bf28}+3.16\\%$ | | test_set | 0.1247ms | 30.0619μs | 33.2647 KOps/s | 31.5649 KOps/s | $\textbf{\color{#35bf28}+5.39\\%}$ | | test_set_shared | 3.8803ms | 0.2219ms | 4.5075 KOps/s | 4.4708 KOps/s | $\color{#35bf28}+0.82\\%$ | | test_update | 0.1658ms | 36.6309μs | 27.2994 KOps/s | 25.5093 KOps/s | $\textbf{\color{#35bf28}+7.02\\%}$ | | test_update_nested | 0.2261ms | 47.4724μs | 21.0649 KOps/s | 19.9409 KOps/s | $\textbf{\color{#35bf28}+5.64\\%}$ | | test_update__nested | 0.1950ms | 34.8115μs | 28.7262 KOps/s | 28.3219 KOps/s | $\color{#35bf28}+1.43\\%$ | | test_set_nested | 0.1649ms | 32.2656μs | 30.9928 KOps/s | 30.0308 KOps/s | $\color{#35bf28}+3.20\\%$ | | test_set_nested_new | 0.1400ms | 37.0331μs | 27.0029 KOps/s | 25.7790 KOps/s | $\color{#35bf28}+4.75\\%$ | | test_select | 0.1462ms | 54.8544μs | 18.2301 KOps/s | 18.1048 KOps/s | $\color{#35bf28}+0.69\\%$ | | test_select_nested | 0.1283ms | 60.5448μs | 16.5167 KOps/s | 16.6768 KOps/s | $\color{#d91a1a}-0.96\\%$ | | test_exclude_nested | 0.1628ms | 81.6546μs | 12.2467 KOps/s | 12.4947 KOps/s | $\color{#d91a1a}-1.98\\%$ | | test_empty[True] | 0.4698ms | 0.3475ms | 2.8775 KOps/s | 2.9709 KOps/s | $\color{#d91a1a}-3.14\\%$ | | test_empty[False] | 11.7895μs | 1.2930μs | 773.3822 KOps/s | 819.6408 KOps/s | $\textbf{\color{#d91a1a}-5.64\\%}$ | | test_unbind_speed | 0.7840ms | 0.3350ms | 2.9849 KOps/s | 2.9951 KOps/s | $\color{#d91a1a}-0.34\\%$ | | test_unbind_speed_stack0 | 0.5580ms | 0.3086ms | 3.2409 KOps/s | 3.0356 KOps/s | $\textbf{\color{#35bf28}+6.76\\%}$ | | test_unbind_speed_stack1 | 84.0227ms | 0.8102ms | 1.2342 KOps/s | 1.3879 KOps/s | $\textbf{\color{#d91a1a}-11.07\\%}$ | | test_split | 84.2526ms | 2.2327ms | 447.8861 Ops/s | 425.0238 Ops/s | $\textbf{\color{#35bf28}+5.38\\%}$ | | test_chunk | 83.9164ms | 2.2321ms | 448.0180 Ops/s | 420.9829 Ops/s | $\textbf{\color{#35bf28}+6.42\\%}$ | | test_creation[device0] | 0.2793ms | 0.1227ms | 8.1528 KOps/s | 8.0588 KOps/s | $\color{#35bf28}+1.17\\%$ | | test_creation_from_tensor | 3.3868ms | 0.1229ms | 8.1370 KOps/s | 8.1320 KOps/s | $\color{#35bf28}+0.06\\%$ | | test_add_one[memmap_tensor0] | 0.1872ms | 7.6793μs | 130.2203 KOps/s | 120.5144 KOps/s | $\textbf{\color{#35bf28}+8.05\\%}$ | | test_contiguous[memmap_tensor0] | 25.9180μs | 2.2112μs | 452.2454 KOps/s | 450.3832 KOps/s | $\color{#35bf28}+0.41\\%$ | | test_stack[memmap_tensor0] | 72.2750μs | 6.0014μs | 166.6291 KOps/s | 164.9375 KOps/s | $\color{#35bf28}+1.03\\%$ | | test_memmaptd_index | 1.1382ms | 0.4386ms | 2.2802 KOps/s | 2.2862 KOps/s | $\color{#d91a1a}-0.26\\%$ | | test_memmaptd_index_astensor | 0.7801ms | 0.5166ms | 1.9357 KOps/s | 1.9206 KOps/s | $\color{#35bf28}+0.79\\%$ | | test_memmaptd_index_op | 1.4163ms | 1.0637ms | 940.1244 Ops/s | 900.7616 Ops/s | $\color{#35bf28}+4.37\\%$ | | test_serialize_model | 0.1352s | 0.1283s | 7.7947 Ops/s | 7.0631 Ops/s | $\textbf{\color{#35bf28}+10.36\\%}$ | | test_serialize_model_pickle | 0.4465s | 0.3941s | 2.5377 Ops/s | 2.4965 Ops/s | $\color{#35bf28}+1.65\\%$ | | test_serialize_weights | 0.1457s | 0.1292s | 7.7397 Ops/s | 7.8621 Ops/s | $\color{#d91a1a}-1.56\\%$ | | test_serialize_weights_returnearly | 0.1858s | 0.1664s | 6.0096 Ops/s | 6.0906 Ops/s | $\color{#d91a1a}-1.33\\%$ | | test_serialize_weights_pickle | 1.0404s | 0.7489s | 1.3352 Ops/s | 2.4615 Ops/s | $\textbf{\color{#d91a1a}-45.76\\%}$ | | test_serialize_weights_filesystem | 0.1477s | 0.1448s | 6.9084 Ops/s | 6.9730 Ops/s | $\color{#d91a1a}-0.93\\%$ | | test_serialize_model_filesystem | 0.1552s | 0.1490s | 6.7123 Ops/s | 6.7188 Ops/s | $\color{#d91a1a}-0.10\\%$ | | test_reshape_pytree | 89.7170μs | 39.4581μs | 25.3434 KOps/s | 25.4025 KOps/s | $\color{#d91a1a}-0.23\\%$ | | test_reshape_td | 0.1352ms | 49.5370μs | 20.1869 KOps/s | 19.4473 KOps/s | $\color{#35bf28}+3.80\\%$ | | test_view_pytree | 95.6790μs | 39.2206μs | 25.4968 KOps/s | 25.6273 KOps/s | $\color{#d91a1a}-0.51\\%$ | | test_view_td | 0.1187ms | 55.1220μs | 18.1416 KOps/s | 17.2698 KOps/s | $\textbf{\color{#35bf28}+5.05\\%}$ | | test_unbind_pytree | 84.5980μs | 35.7377μs | 27.9816 KOps/s | 28.1659 KOps/s | $\color{#d91a1a}-0.65\\%$ | | test_unbind_td | 0.3589ms | 48.6826μs | 20.5412 KOps/s | 17.8744 KOps/s | $\textbf{\color{#35bf28}+14.92\\%}$ | | test_split_pytree | 0.1077ms | 39.1108μs | 25.5684 KOps/s | 26.4003 KOps/s | $\color{#d91a1a}-3.15\\%$ | | test_split_td | 0.5476ms | 60.5027μs | 16.5282 KOps/s | 16.1651 KOps/s | $\color{#35bf28}+2.25\\%$ | | test_add_pytree | 0.1217ms | 44.0748μs | 22.6887 KOps/s | 22.4682 KOps/s | $\color{#35bf28}+0.98\\%$ | | test_add_td | 0.1734ms | 82.1569μs | 12.1718 KOps/s | 11.2424 KOps/s | $\textbf{\color{#35bf28}+8.27\\%}$ | | test_distributed | 0.2937ms | 0.1318ms | 7.5861 KOps/s | 7.5281 KOps/s | $\color{#35bf28}+0.77\\%$ | | test_tdmodule | 40.7560μs | 17.5237μs | 57.0654 KOps/s | 57.2777 KOps/s | $\color{#d91a1a}-0.37\\%$ | | test_tdmodule_dispatch | 78.8770μs | 37.2370μs | 26.8550 KOps/s | 26.9176 KOps/s | $\color{#d91a1a}-0.23\\%$ | | test_tdseq | 41.9380μs | 19.6569μs | 50.8726 KOps/s | 52.0305 KOps/s | $\color{#d91a1a}-2.23\\%$ | | test_tdseq_dispatch | 85.8200μs | 43.0589μs | 23.2240 KOps/s | 24.0971 KOps/s | $\color{#d91a1a}-3.62\\%$ | | test_instantiation_functorch | 2.9127ms | 1.5996ms | 625.1481 Ops/s | 637.1188 Ops/s | $\color{#d91a1a}-1.88\\%$ | | test_instantiation_td | 1.8868ms | 1.1770ms | 849.5863 Ops/s | 882.3800 Ops/s | $\color{#d91a1a}-3.72\\%$ | | test_exec_functorch | 0.3198ms | 0.1801ms | 5.5526 KOps/s | 5.4188 KOps/s | $\color{#35bf28}+2.47\\%$ | | test_exec_functional_call | 0.2930ms | 0.1693ms | 5.9068 KOps/s | 5.7416 KOps/s | $\color{#35bf28}+2.88\\%$ | | test_exec_td | 0.3253ms | 0.1722ms | 5.8081 KOps/s | 5.8716 KOps/s | $\color{#d91a1a}-1.08\\%$ | | test_exec_td_decorator | 83.8653ms | 0.3218ms | 3.1079 KOps/s | 3.9247 KOps/s | $\textbf{\color{#d91a1a}-20.81\\%}$ | | test_vmap_mlp_speed[True-True] | 0.8064ms | 0.6111ms | 1.6365 KOps/s | 1.6191 KOps/s | $\color{#35bf28}+1.07\\%$ | | test_vmap_mlp_speed[True-False] | 1.3898ms | 0.6191ms | 1.6154 KOps/s | 1.6322 KOps/s | $\color{#d91a1a}-1.03\\%$ | | test_vmap_mlp_speed[False-True] | 0.7495ms | 0.4962ms | 2.0152 KOps/s | 1.9881 KOps/s | $\color{#35bf28}+1.36\\%$ | | test_vmap_mlp_speed[False-False] | 0.6995ms | 0.4954ms | 2.0185 KOps/s | 1.9735 KOps/s | $\color{#35bf28}+2.28\\%$ | | test_vmap_mlp_speed_decorator[True-True] | 1.2148ms | 0.6980ms | 1.4327 KOps/s | 1.4159 KOps/s | $\color{#35bf28}+1.18\\%$ | | test_vmap_mlp_speed_decorator[True-False] | 1.2134ms | 0.7026ms | 1.4233 KOps/s | 1.4189 KOps/s | $\color{#35bf28}+0.31\\%$ | | test_vmap_mlp_speed_decorator[False-True] | 0.8101ms | 0.5771ms | 1.7328 KOps/s | 1.7194 KOps/s | $\color{#35bf28}+0.78\\%$ | | test_vmap_mlp_speed_decorator[False-False] | 0.7292ms | 0.5776ms | 1.7314 KOps/s | 1.7054 KOps/s | $\color{#35bf28}+1.52\\%$ | | test_to_module_speed[True] | 2.8508ms | 1.8597ms | 537.7103 Ops/s | 558.4528 Ops/s | $\color{#d91a1a}-3.71\\%$ | | test_to_module_speed[False] | 4.1727ms | 1.8502ms | 540.4939 Ops/s | 565.2006 Ops/s | $\color{#d91a1a}-4.37\\%$ | | test_tc_init | 0.1001ms | 43.9544μs | 22.7508 KOps/s | 22.0774 KOps/s | $\color{#35bf28}+3.05\\%$ | | test_tc_init_nested | 0.1874ms | 87.2828μs | 11.4570 KOps/s | 11.1972 KOps/s | $\color{#35bf28}+2.32\\%$ | | test_tc_first_layer_tensor | 51.5160μs | 9.4453μs | 105.8731 KOps/s | 109.4267 KOps/s | $\color{#d91a1a}-3.25\\%$ | | test_tc_first_layer_nontensor | 40.6260μs | 9.2622μs | 107.9654 KOps/s | 108.8915 KOps/s | $\color{#d91a1a}-0.85\\%$ | | test_tc_second_layer_tensor | 78.9380μs | 2.9365μs | 340.5458 KOps/s | 356.0622 KOps/s | $\color{#d91a1a}-4.36\\%$ | | test_tc_second_layer_nontensor | 35.0760μs | 10.5991μs | 94.3479 KOps/s | 98.3132 KOps/s | $\color{#d91a1a}-4.03\\%$ | | test_unbind | 0.1090s | 13.9113ms | 71.8840 Ops/s | 69.7159 Ops/s | $\color{#35bf28}+3.11\\%$ | | test_full_like | 10.4053ms | 8.4895ms | 117.7920 Ops/s | 141.7561 Ops/s | $\textbf{\color{#d91a1a}-16.91\\%}$ | | test_zeros_like | 11.7537ms | 7.0446ms | 141.9529 Ops/s | 142.5975 Ops/s | $\color{#d91a1a}-0.45\\%$ | | test_ones_like | 16.4042ms | 8.1388ms | 122.8679 Ops/s | 133.5150 Ops/s | $\textbf{\color{#d91a1a}-7.97\\%}$ | | test_clone | 18.6706ms | 10.0841ms | 99.1663 Ops/s | 109.6168 Ops/s | $\textbf{\color{#d91a1a}-9.53\\%}$ | | test_squeeze | 71.2320μs | 15.1322μs | 66.0841 KOps/s | 69.0604 KOps/s | $\color{#d91a1a}-4.31\\%$ | | test_unsqueeze | 0.1650ms | 97.9576μs | 10.2085 KOps/s | 10.1546 KOps/s | $\color{#35bf28}+0.53\\%$ | | test_split | 0.4583ms | 0.2108ms | 4.7432 KOps/s | 4.7645 KOps/s | $\color{#d91a1a}-0.45\\%$ | | test_permute | 0.3733ms | 0.2241ms | 4.4618 KOps/s | 4.3284 KOps/s | $\color{#35bf28}+3.08\\%$ | | test_stack | 33.9953ms | 26.9606ms | 37.0912 Ops/s | 40.4623 Ops/s | $\textbf{\color{#d91a1a}-8.33\\%}$ | | test_cat | 38.1198ms | 28.0177ms | 35.6917 Ops/s | 40.1871 Ops/s | $\textbf{\color{#d91a1a}-11.19\\%}$ |
github-actions[bot]
commented
1 month ago
$\color{#D29922}\textsf{\Large\⚠\kern{0.2cm}\normalsize Warning}$ Result of GPU Benchmark Tests
Total Benchmarks: 219. Improved: $\large\color{#35bf28}25$. Worsened: $\large\color{#d91a1a}14$.
Expand to view detailed results
| Name | Max | Mean | Ops | Ops on Repo `HEAD` | Change | | -------------------------------------------------- | --------- | --------- | --------------- | ------------------ | ----------------------------------- | | test_plain_set_nested | 27.4200μs | 16.6275μs | 60.1413 KOps/s | 55.7803 KOps/s | $\textbf{\color{#35bf28}+7.82\\%}$ | | test_plain_set_stack_nested | 34.6610μs | 16.2696μs | 61.4642 KOps/s | 56.1647 KOps/s | $\textbf{\color{#35bf28}+9.44\\%}$ | | test_plain_set_nested_inplace | 37.9720μs | 17.4744μs | 57.2267 KOps/s | 52.2553 KOps/s | $\textbf{\color{#35bf28}+9.51\\%}$ | | test_plain_set_stack_nested_inplace | 45.0410μs | 17.2813μs | 57.8659 KOps/s | 52.3854 KOps/s | $\textbf{\color{#35bf28}+10.46\\%}$ | | test_items | 23.4900μs | 4.7139μs | 212.1371 KOps/s | 211.2163 KOps/s | $\color{#35bf28}+0.44\\%$ | | test_items_nested | 0.4542ms | 0.3949ms | 2.5322 KOps/s | 2.5693 KOps/s | $\color{#d91a1a}-1.44\\%$ | | test_items_nested_locked | 0.4581ms | 0.4004ms | 2.4976 KOps/s | 2.5179 KOps/s | $\color{#d91a1a}-0.81\\%$ | | test_items_nested_leaf | 0.1276ms | 87.2761μs | 11.4579 KOps/s | 11.4943 KOps/s | $\color{#d91a1a}-0.32\\%$ | | test_items_stack_nested | 0.4529ms | 0.3944ms | 2.5355 KOps/s | 2.5452 KOps/s | $\color{#d91a1a}-0.38\\%$ | | test_items_stack_nested_leaf | 0.1203ms | 88.4886μs | 11.3009 KOps/s | 11.2733 KOps/s | $\color{#35bf28}+0.24\\%$ | | test_items_stack_nested_locked | 0.4573ms | 0.4020ms | 2.4874 KOps/s | 2.5462 KOps/s | $\color{#d91a1a}-2.31\\%$ | | test_keys | 21.9100μs | 4.4491μs | 224.7628 KOps/s | 226.7229 KOps/s | $\color{#d91a1a}-0.86\\%$ | | test_keys_nested | 83.7120μs | 67.0394μs | 14.9166 KOps/s | 14.7062 KOps/s | $\color{#35bf28}+1.43\\%$ | | test_keys_nested_locked | 1.8916ms | 75.0286μs | 13.3283 KOps/s | 13.4313 KOps/s | $\color{#d91a1a}-0.77\\%$ | | test_keys_nested_leaf | 77.3810μs | 57.5560μs | 17.3744 KOps/s | 17.5021 KOps/s | $\color{#d91a1a}-0.73\\%$ | | test_keys_stack_nested | 84.0520μs | 67.4978μs | 14.8153 KOps/s | 15.0656 KOps/s | $\color{#d91a1a}-1.66\\%$ | | test_keys_stack_nested_leaf | 75.3110μs | 57.9153μs | 17.2666 KOps/s | 17.0262 KOps/s | $\color{#35bf28}+1.41\\%$ | | test_keys_stack_nested_locked | 93.4210μs | 74.6499μs | 13.3959 KOps/s | 13.4594 KOps/s | $\color{#d91a1a}-0.47\\%$ | | test_values | 9.5907μs | 1.8011μs | 555.2276 KOps/s | 555.0669 KOps/s | $\color{#35bf28}+0.03\\%$ | | test_values_nested | 97.7720μs | 34.4806μs | 29.0018 KOps/s | 29.5414 KOps/s | $\color{#d91a1a}-1.83\\%$ | | test_values_nested_locked | 51.0410μs | 35.8828μs | 27.8685 KOps/s | 27.8277 KOps/s | $\color{#35bf28}+0.15\\%$ | | test_values_nested_leaf | 50.7110μs | 30.3097μs | 32.9927 KOps/s | 33.0660 KOps/s | $\color{#d91a1a}-0.22\\%$ | | test_values_stack_nested | 58.1210μs | 35.0536μs | 28.5277 KOps/s | 28.7017 KOps/s | $\color{#d91a1a}-0.61\\%$ | | test_values_stack_nested_leaf | 46.9910μs | 31.5750μs | 31.6706 KOps/s | 32.8461 KOps/s | $\color{#d91a1a}-3.58\\%$ | | test_values_stack_nested_locked | 66.4110μs | 37.3864μs | 26.7477 KOps/s | 27.6401 KOps/s | $\color{#d91a1a}-3.23\\%$ | | test_membership | 2.1212μs | 0.5777μs | 1.7309 MOps/s | 1.7825 MOps/s | $\color{#d91a1a}-2.90\\%$ | | test_membership_nested | 17.9100μs | 2.0491μs | 488.0199 KOps/s | 486.1990 KOps/s | $\color{#35bf28}+0.37\\%$ | | test_membership_nested_leaf | 13.1505μs | 2.0007μs | 499.8188 KOps/s | 513.0497 KOps/s | $\color{#d91a1a}-2.58\\%$ | | test_membership_stacked_nested | 22.7900μs | 2.0996μs | 476.2768 KOps/s | 473.4756 KOps/s | $\color{#35bf28}+0.59\\%$ | | test_membership_stacked_nested_leaf | 15.2700μs | 2.0881μs | 478.9039 KOps/s | 479.9968 KOps/s | $\color{#d91a1a}-0.23\\%$ | | test_membership_nested_last | 21.8500μs | 3.0348μs | 329.5138 KOps/s | 335.7276 KOps/s | $\color{#d91a1a}-1.85\\%$ | | test_membership_nested_leaf_last | 15.0500μs | 2.9993μs | 333.4155 KOps/s | 333.7747 KOps/s | $\color{#d91a1a}-0.11\\%$ | | test_membership_stacked_nested_last | 25.3300μs | 3.8235μs | 261.5373 KOps/s | 326.2389 KOps/s | $\textbf{\color{#d91a1a}-19.83\\%}$ | | test_membership_stacked_nested_leaf_last | 15.1700μs | 3.7916μs | 263.7442 KOps/s | 331.4646 KOps/s | $\textbf{\color{#d91a1a}-20.43\\%}$ | | test_nested_getleaf | 25.8500μs | 8.1136μs | 123.2495 KOps/s | 124.1011 KOps/s | $\color{#d91a1a}-0.69\\%$ | | test_nested_get | 28.8100μs | 7.5317μs | 132.7729 KOps/s | 132.6260 KOps/s | $\color{#35bf28}+0.11\\%$ | | test_stacked_getleaf | 25.6900μs | 8.0805μs | 123.7546 KOps/s | 123.7242 KOps/s | $\color{#35bf28}+0.02\\%$ | | test_stacked_get | 23.5000μs | 7.5628μs | 132.2268 KOps/s | 131.9494 KOps/s | $\color{#35bf28}+0.21\\%$ | | test_nested_getitemleaf | 22.2410μs | 8.3263μs | 120.1012 KOps/s | 121.8553 KOps/s | $\color{#d91a1a}-1.44\\%$ | | test_nested_getitem | 24.0600μs | 7.6826μs | 130.1643 KOps/s | 129.3548 KOps/s | $\color{#35bf28}+0.63\\%$ | | test_stacked_getitemleaf | 24.1310μs | 8.1735μs | 122.3466 KOps/s | 121.2356 KOps/s | $\color{#35bf28}+0.92\\%$ | | test_stacked_getitem | 24.9000μs | 7.6947μs | 129.9590 KOps/s | 128.4970 KOps/s | $\color{#35bf28}+1.14\\%$ | | test_lock_nested | 4.3851ms | 0.4808ms | 2.0800 KOps/s | 2.1162 KOps/s | $\color{#d91a1a}-1.71\\%$ | | test_lock_stack_nested | 0.5020ms | 0.4363ms | 2.2917 KOps/s | 2.3101 KOps/s | $\color{#d91a1a}-0.80\\%$ | | test_unlock_nested | 0.8696ms | 0.4014ms | 2.4913 KOps/s | 2.5255 KOps/s | $\color{#d91a1a}-1.35\\%$ | | test_unlock_stack_nested | 0.4254ms | 0.3539ms | 2.8253 KOps/s | 2.8522 KOps/s | $\color{#d91a1a}-0.95\\%$ | | test_flatten_speed | 0.2139ms | 0.1085ms | 9.2139 KOps/s | 9.3483 KOps/s | $\color{#d91a1a}-1.44\\%$ | | test_unflatten_speed | 0.3890ms | 0.3027ms | 3.3033 KOps/s | 3.3528 KOps/s | $\color{#d91a1a}-1.48\\%$ | | test_common_ops | 1.7608ms | 1.3041ms | 766.8076 Ops/s | 755.4838 Ops/s | $\color{#35bf28}+1.50\\%$ | | test_creation | 16.2310μs | 2.0390μs | 490.4280 KOps/s | 500.3419 KOps/s | $\color{#d91a1a}-1.98\\%$ | | test_creation_empty | 34.2510μs | 15.9307μs | 62.7720 KOps/s | 52.4190 KOps/s | $\textbf{\color{#35bf28}+19.75\\%}$ | | test_creation_nested_1 | 38.2400μs | 17.7328μs | 56.3927 KOps/s | 48.2570 KOps/s | $\textbf{\color{#35bf28}+16.86\\%}$ | | test_creation_nested_2 | 38.1710μs | 20.8432μs | 47.9773 KOps/s | 42.5916 KOps/s | $\textbf{\color{#35bf28}+12.64\\%}$ | | test_clone | 53.7310μs | 32.3456μs | 30.9161 KOps/s | 34.1344 KOps/s | $\textbf{\color{#d91a1a}-9.43\\%}$ | | test_getitem[int] | 1.1253ms | 17.1136μs | 58.4330 KOps/s | 60.4198 KOps/s | $\color{#d91a1a}-3.29\\%$ | | test_getitem[slice_int] | 0.1488ms | 28.5179μs | 35.0656 KOps/s | 34.8834 KOps/s | $\color{#35bf28}+0.52\\%$ | | test_getitem[range] | 0.2962ms | 0.1139ms | 8.7823 KOps/s | 8.7827 KOps/s | $-0.00\\%$ | | test_getitem[tuple] | 0.1531ms | 25.5525μs | 39.1352 KOps/s | 39.1228 KOps/s | $\color{#35bf28}+0.03\\%$ | | test_getitem[list] | 0.2259ms | 0.1026ms | 9.7501 KOps/s | 9.5003 KOps/s | $\color{#35bf28}+2.63\\%$ | | test_setitem_dim[int] | 74.2120μs | 51.6020μs | 19.3791 KOps/s | 17.8856 KOps/s | $\textbf{\color{#35bf28}+8.35\\%}$ | | test_setitem_dim[slice_int] | 0.1273ms | 81.9282μs | 12.2058 KOps/s | 12.8242 KOps/s | $\color{#d91a1a}-4.82\\%$ | | test_setitem_dim[range] | 0.1727ms | 0.1451ms | 6.8935 KOps/s | 7.0253 KOps/s | $\color{#d91a1a}-1.88\\%$ | | test_setitem_dim[tuple] | 90.5820μs | 73.8534μs | 13.5403 KOps/s | 13.5004 KOps/s | $\color{#35bf28}+0.30\\%$ | | test_setitem | 72.2510μs | 46.8522μs | 21.3437 KOps/s | 22.8240 KOps/s | $\textbf{\color{#d91a1a}-6.49\\%}$ | | test_set | 68.3710μs | 45.5662μs | 21.9461 KOps/s | 23.1938 KOps/s | $\textbf{\color{#d91a1a}-5.38\\%}$ | | test_set_shared | 0.3999ms | 56.0141μs | 17.8527 KOps/s | 18.6479 KOps/s | $\color{#d91a1a}-4.26\\%$ | | test_update | 83.2010μs | 50.4002μs | 19.8412 KOps/s | 18.9368 KOps/s | $\color{#35bf28}+4.78\\%$ | | test_update_nested | 95.6010μs | 61.0634μs | 16.3764 KOps/s | 15.6692 KOps/s | $\color{#35bf28}+4.51\\%$ | | test_update__nested | 0.1022ms | 65.7648μs | 15.2057 KOps/s | 16.6525 KOps/s | $\textbf{\color{#d91a1a}-8.69\\%}$ | | test_set_nested | 82.2510μs | 48.5883μs | 20.5811 KOps/s | 21.7755 KOps/s | $\textbf{\color{#d91a1a}-5.49\\%}$ | | test_set_nested_new | 0.5492ms | 52.0438μs | 19.2146 KOps/s | 19.3044 KOps/s | $\color{#d91a1a}-0.47\\%$ | | test_select | 0.1079ms | 68.2853μs | 14.6444 KOps/s | 15.3077 KOps/s | $\color{#d91a1a}-4.33\\%$ | | test_select_nested | 79.6520μs | 54.0718μs | 18.4939 KOps/s | 18.2731 KOps/s | $\color{#35bf28}+1.21\\%$ | | test_exclude_nested | 0.1060ms | 72.3312μs | 13.8253 KOps/s | 14.0302 KOps/s | $\color{#d91a1a}-1.46\\%$ | | test_empty[True] | 0.3722ms | 0.3030ms | 3.3000 KOps/s | 3.3181 KOps/s | $\color{#d91a1a}-0.54\\%$ | | test_empty[False] | 2.5312μs | 0.9924μs | 1.0077 MOps/s | 1.0478 MOps/s | $\color{#d91a1a}-3.82\\%$ | | test_to | 64.8310μs | 38.7991μs | 25.7738 KOps/s | 26.5540 KOps/s | $\color{#d91a1a}-2.94\\%$ | | test_to_nonblocking | 48.5810μs | 23.7870μs | 42.0397 KOps/s | 42.3945 KOps/s | $\color{#d91a1a}-0.84\\%$ | | test_unbind_speed | 0.3826ms | 0.3264ms | 3.0640 KOps/s | 3.3060 KOps/s | $\textbf{\color{#d91a1a}-7.32\\%}$ | | test_unbind_speed_stack0 | 0.3973ms | 0.3164ms | 3.1607 KOps/s | 3.3078 KOps/s | $\color{#d91a1a}-4.45\\%$ | | test_unbind_speed_stack1 | 95.2872ms | 0.8771ms | 1.1402 KOps/s | 1.2525 KOps/s | $\textbf{\color{#d91a1a}-8.97\\%}$ | | test_split | 2.4179ms | 2.0861ms | 479.3662 Ops/s | 430.8863 Ops/s | $\textbf{\color{#35bf28}+11.25\\%}$ | | test_chunk | 95.1120ms | 2.5063ms | 398.9897 Ops/s | 429.7260 Ops/s | $\textbf{\color{#d91a1a}-7.15\\%}$ | | test_creation[device0] | 0.1423ms | 0.1030ms | 9.7088 KOps/s | 9.8276 KOps/s | $\color{#d91a1a}-1.21\\%$ | | test_creation_from_tensor | 0.1555ms | 0.1006ms | 9.9413 KOps/s | 10.0686 KOps/s | $\color{#d91a1a}-1.26\\%$ | | test_add_one[memmap_tensor0] | 20.9110μs | 8.8569μs | 112.9064 KOps/s | 115.6682 KOps/s | $\color{#d91a1a}-2.39\\%$ | | test_contiguous[memmap_tensor0] | 23.6810μs | 2.1782μs | 459.0896 KOps/s | 473.0169 KOps/s | $\color{#d91a1a}-2.94\\%$ | | test_stack[memmap_tensor0] | 32.5510μs | 6.7225μs | 148.7537 KOps/s | 153.9780 KOps/s | $\color{#d91a1a}-3.39\\%$ | | test_memmaptd_index | 1.0577ms | 0.4180ms | 2.3923 KOps/s | 2.4417 KOps/s | $\color{#d91a1a}-2.02\\%$ | | test_memmaptd_index_astensor | 0.9072ms | 0.4929ms | 2.0290 KOps/s | 2.0789 KOps/s | $\color{#d91a1a}-2.40\\%$ | | test_memmaptd_index_op | 1.4730ms | 1.0034ms | 996.5980 Ops/s | 960.0169 Ops/s | $\color{#35bf28}+3.81\\%$ | | test_serialize_model | 99.9471ms | 95.2779ms | 10.4956 Ops/s | 10.2492 Ops/s | $\color{#35bf28}+2.40\\%$ | | test_serialize_model_pickle | 1.3529s | 1.2384s | 0.8075 Ops/s | 0.8059 Ops/s | $\color{#35bf28}+0.21\\%$ | | test_serialize_weights | 0.1884s | 0.1033s | 9.6783 Ops/s | 10.2960 Ops/s | $\textbf{\color{#d91a1a}-6.00\\%}$ | | test_serialize_weights_returnearly | 75.3639ms | 70.1493ms | 14.2553 Ops/s | 11.5553 Ops/s | $\textbf{\color{#35bf28}+23.37\\%}$ | | test_serialize_weights_pickle | 1.4034s | 1.2454s | 0.8030 Ops/s | 0.8085 Ops/s | $\color{#d91a1a}-0.68\\%$ | | test_reshape_pytree | 64.8910μs | 38.2897μs | 26.1167 KOps/s | 26.2489 KOps/s | $\color{#d91a1a}-0.50\\%$ | | test_reshape_td | 0.2570ms | 43.7454μs | 22.8596 KOps/s | 22.5761 KOps/s | $\color{#35bf28}+1.26\\%$ | | test_view_pytree | 60.6910μs | 37.7317μs | 26.5029 KOps/s | 26.4404 KOps/s | $\color{#35bf28}+0.24\\%$ | | test_view_td | 0.2525ms | 53.7250μs | 18.6133 KOps/s | 19.7699 KOps/s | $\textbf{\color{#d91a1a}-5.85\\%}$ | | test_unbind_pytree | 68.3010μs | 37.0955μs | 26.9574 KOps/s | 27.6617 KOps/s | $\color{#d91a1a}-2.55\\%$ | | test_unbind_td | 91.6892ms | 52.8417μs | 18.9245 KOps/s | 21.7569 KOps/s | $\textbf{\color{#d91a1a}-13.02\\%}$ | | test_split_pytree | 78.9810μs | 50.6513μs | 19.7428 KOps/s | 19.6698 KOps/s | $\color{#35bf28}+0.37\\%$ | | test_split_td | 0.2648ms | 58.9889μs | 16.9523 KOps/s | 16.6808 KOps/s | $\color{#35bf28}+1.63\\%$ | | test_add_pytree | 95.3930μs | 58.0741μs | 17.2194 KOps/s | 17.4374 KOps/s | $\color{#d91a1a}-1.25\\%$ | | test_add_td | 0.3776ms | 90.2655μs | 11.0784 KOps/s | 10.5932 KOps/s | $\color{#35bf28}+4.58\\%$ | | test_compile_add_one_nested[tensordict-compile] | 0.4127ms | 0.2110ms | 4.7383 KOps/s | 4.9021 KOps/s | $\color{#d91a1a}-3.34\\%$ | | test_compile_add_one_nested[tensordict-eager] | 0.2740ms | 0.1738ms | 5.7552 KOps/s | 5.7834 KOps/s | $\color{#d91a1a}-0.49\\%$ | | test_compile_add_one_nested[pytree-compile] | 0.2130ms | 0.1430ms | 6.9919 KOps/s | 7.0930 KOps/s | $\color{#d91a1a}-1.43\\%$ | | test_compile_add_one_nested[pytree-eager] | 0.2486ms | 0.1899ms | 5.2671 KOps/s | 5.2768 KOps/s | $\color{#d91a1a}-0.18\\%$ | | test_compile_copy_nested[tensordict-compile] | 42.2100μs | 21.6576μs | 46.1732 KOps/s | 44.6901 KOps/s | $\color{#35bf28}+3.32\\%$ | | test_compile_copy_nested[tensordict-eager] | 82.3410μs | 49.9581μs | 20.0168 KOps/s | 20.7224 KOps/s | $\color{#d91a1a}-3.41\\%$ | | test_compile_copy_nested[pytree-compile] | 0.1207ms | 74.3966μs | 13.4415 KOps/s | 13.6327 KOps/s | $\color{#d91a1a}-1.40\\%$ | | test_compile_copy_nested[pytree-eager] | 92.1710μs | 61.1767μs | 16.3461 KOps/s | 16.5005 KOps/s | $\color{#d91a1a}-0.94\\%$ | | test_compile_add_one_flat[tensordict-compile] | 0.4201ms | 0.3308ms | 3.0226 KOps/s | 3.1329 KOps/s | $\color{#d91a1a}-3.52\\%$ | | test_compile_add_one_flat[tensordict-eager] | 0.3150ms | 0.2255ms | 4.4337 KOps/s | 4.4753 KOps/s | $\color{#d91a1a}-0.93\\%$ | | test_compile_add_one_flat[tensorclass-compile] | 0.1811ms | 0.1313ms | 7.6137 KOps/s | 7.9805 KOps/s | $\color{#d91a1a}-4.60\\%$ | | test_compile_add_one_flat[tensorclass-eager] | 0.1301ms | 63.2368μs | 15.8136 KOps/s | 15.9363 KOps/s | $\color{#d91a1a}-0.77\\%$ | | test_compile_add_one_flat[pytree-compile] | 0.3934ms | 0.3271ms | 3.0567 KOps/s | 3.1549 KOps/s | $\color{#d91a1a}-3.11\\%$ | | test_compile_add_one_flat[pytree-eager] | 0.6748ms | 0.6060ms | 1.6500 KOps/s | 1.6164 KOps/s | $\color{#35bf28}+2.08\\%$ | | test_compile_add_self_flat[tensordict-eager] | 0.3360ms | 0.2753ms | 3.6323 KOps/s | 3.6553 KOps/s | $\color{#d91a1a}-0.63\\%$ | | test_compile_add_self_flat[tensordict-compile] | 0.3949ms | 0.3301ms | 3.0292 KOps/s | 3.1406 KOps/s | $\color{#d91a1a}-3.55\\%$ | | test_compile_add_self_flat[tensorclass-eager] | 0.1755ms | 76.4164μs | 13.0862 KOps/s | 12.9380 KOps/s | $\color{#35bf28}+1.15\\%$ | | test_compile_add_self_flat[tensorclass-compile] | 0.1883ms | 0.1315ms | 7.6020 KOps/s | 7.9416 KOps/s | $\color{#d91a1a}-4.28\\%$ | | test_compile_add_self_flat[pytree-eager] | 0.5906ms | 0.5272ms | 1.8969 KOps/s | 1.7707 KOps/s | $\textbf{\color{#35bf28}+7.13\\%}$ | | test_compile_add_self_flat[pytree-compile] | 0.3908ms | 0.3218ms | 3.1072 KOps/s | 3.1230 KOps/s | $\color{#d91a1a}-0.51\\%$ | | test_compile_copy_flat[tensordict-compile] | 44.4810μs | 18.8745μs | 52.9815 KOps/s | 55.4555 KOps/s | $\color{#d91a1a}-4.46\\%$ | | test_compile_copy_flat[tensordict-eager] | 59.8610μs | 32.7118μs | 30.5700 KOps/s | 30.9447 KOps/s | $\color{#d91a1a}-1.21\\%$ | | test_compile_copy_flat[pytree-compile] | 0.1088ms | 76.5091μs | 13.0703 KOps/s | 13.3911 KOps/s | $\color{#d91a1a}-2.40\\%$ | | test_compile_copy_flat[pytree-eager] | 84.7210μs | 61.2903μs | 16.3158 KOps/s | 16.4438 KOps/s | $\color{#d91a1a}-0.78\\%$ | | test_compile_assign_and_add[tensordict-compile] | 2.7824ms | 0.9872ms | 1.0130 KOps/s | 1.0374 KOps/s | $\color{#d91a1a}-2.35\\%$ | | test_compile_assign_and_add[tensordict-eager] | 3.6383ms | 3.2648ms | 306.3016 Ops/s | 307.2478 Ops/s | $\color{#d91a1a}-0.31\\%$ | | test_compile_assign_and_add[pytree-compile] | 2.5631ms | 0.9253ms | 1.0807 KOps/s | 1.0787 KOps/s | $\color{#35bf28}+0.19\\%$ | | test_compile_assign_and_add[pytree-eager] | 3.3847ms | 3.1636ms | 316.0940 Ops/s | 319.4569 Ops/s | $\color{#d91a1a}-1.05\\%$ | | test_compile_indexing[tensor-tensordict-compile] | 0.1879ms | 0.1096ms | 9.1261 KOps/s | 9.4073 KOps/s | $\color{#d91a1a}-2.99\\%$ | | test_compile_indexing[tensor-tensordict-eager] | 0.2232ms | 63.9160μs | 15.6455 KOps/s | 15.3719 KOps/s | $\color{#35bf28}+1.78\\%$ | | test_compile_indexing[tensor-tensorclass-compile] | 0.1717ms | 0.1058ms | 9.4494 KOps/s | 10.1503 KOps/s | $\textbf{\color{#d91a1a}-6.91\\%}$ | | test_compile_indexing[tensor-tensorclass-eager] | 98.9010μs | 43.9990μs | 22.7278 KOps/s | 21.8880 KOps/s | $\color{#35bf28}+3.84\\%$ | | test_compile_indexing[tensor-pytree-compile] | 0.1720ms | 0.1036ms | 9.6538 KOps/s | 9.7274 KOps/s | $\color{#d91a1a}-0.76\\%$ | | test_compile_indexing[tensor-pytree-eager] | 79.8110μs | 43.9812μs | 22.7370 KOps/s | 21.0350 KOps/s | $\textbf{\color{#35bf28}+8.09\\%}$ | | test_compile_indexing[slice-tensordict-compile] | 0.1908ms | 0.1387ms | 7.2107 KOps/s | 7.3777 KOps/s | $\color{#d91a1a}-2.26\\%$ | | test_compile_indexing[slice-tensordict-eager] | 0.1862ms | 26.3717μs | 37.9194 KOps/s | 39.1969 KOps/s | $\color{#d91a1a}-3.26\\%$ | | test_compile_indexing[slice-tensorclass-compile] | 0.1794ms | 0.1288ms | 7.7666 KOps/s | 7.9562 KOps/s | $\color{#d91a1a}-2.38\\%$ | | test_compile_indexing[slice-tensorclass-eager] | 64.5710μs | 22.6478μs | 44.1544 KOps/s | 44.7776 KOps/s | $\color{#d91a1a}-1.39\\%$ | | test_compile_indexing[slice-pytree-compile] | 0.2104ms | 0.1291ms | 7.7462 KOps/s | 7.9163 KOps/s | $\color{#d91a1a}-2.15\\%$ | | test_compile_indexing[slice-pytree-eager] | 50.2810μs | 22.5431μs | 44.3595 KOps/s | 41.9962 KOps/s | $\textbf{\color{#35bf28}+5.63\\%}$ | | test_compile_indexing[int-tensordict-compile] | 0.1972ms | 0.1371ms | 7.2923 KOps/s | 7.4710 KOps/s | $\color{#d91a1a}-2.39\\%$ | | test_compile_indexing[int-tensordict-eager] | 0.5220ms | 25.4578μs | 39.2806 KOps/s | 39.6523 KOps/s | $\color{#d91a1a}-0.94\\%$ | | test_compile_indexing[int-tensorclass-compile] | 0.1809ms | 0.1290ms | 7.7526 KOps/s | 7.9787 KOps/s | $\color{#d91a1a}-2.83\\%$ | | test_compile_indexing[int-tensorclass-eager] | 55.5710μs | 22.5234μs | 44.3983 KOps/s | 45.5605 KOps/s | $\color{#d91a1a}-2.55\\%$ | | test_compile_indexing[int-pytree-compile] | 0.1965ms | 0.1281ms | 7.8060 KOps/s | 7.9830 KOps/s | $\color{#d91a1a}-2.22\\%$ | | test_compile_indexing[int-pytree-eager] | 53.7800μs | 22.1152μs | 45.2178 KOps/s | 45.6170 KOps/s | $\color{#d91a1a}-0.88\\%$ | | test_mod_add[eager] | 98.4710μs | 36.4800μs | 27.4123 KOps/s | 24.6742 KOps/s | $\textbf{\color{#35bf28}+11.10\\%}$ | | test_mod_add[compile] | 0.1238ms | 68.5807μs | 14.5814 KOps/s | 14.9464 KOps/s | $\color{#d91a1a}-2.44\\%$ | | test_mod_add[compile-overhead] | 0.2582ms | 0.1460ms | 6.8512 KOps/s | 6.9089 KOps/s | $\color{#d91a1a}-0.84\\%$ | | test_mod_wrap[eager] | 0.3487ms | 0.2444ms | 4.0918 KOps/s | 3.8263 KOps/s | $\textbf{\color{#35bf28}+6.94\\%}$ | | test_mod_wrap[compile] | 0.3540ms | 0.2923ms | 3.4207 KOps/s | 3.2672 KOps/s | $\color{#35bf28}+4.70\\%$ | | test_mod_wrap[compile-overhead] | 8.4125ms | 4.4096ms | 226.7754 Ops/s | 229.3415 Ops/s | $\color{#d91a1a}-1.12\\%$ | | test_mod_wrap_and_backward[eager] | 1.5140ms | 1.4022ms | 713.1805 Ops/s | 743.1192 Ops/s | $\color{#d91a1a}-4.03\\%$ | | test_mod_wrap_and_backward[compile] | 1.6520ms | 1.4435ms | 692.7837 Ops/s | 690.1405 Ops/s | $\color{#35bf28}+0.38\\%$ | | test_mod_wrap_and_backward[compile-overhead] | 1.4780ms | 0.9926ms | 1.0075 KOps/s | 1.0104 KOps/s | $\color{#d91a1a}-0.29\\%$ | | test_seq_add[eager] | 0.2204ms | 0.1069ms | 9.3524 KOps/s | 8.9882 KOps/s | $\color{#35bf28}+4.05\\%$ | | test_seq_add[compile] | 0.1353ms | 84.5005μs | 11.8342 KOps/s | 11.9647 KOps/s | $\color{#d91a1a}-1.09\\%$ | | test_seq_add[compile-overhead] | 0.1853ms | 0.1208ms | 8.2752 KOps/s | 8.1858 KOps/s | $\color{#35bf28}+1.09\\%$ | | test_seq_wrap[eager] | 0.5097ms | 0.4146ms | 2.4123 KOps/s | 2.3130 KOps/s | $\color{#35bf28}+4.29\\%$ | | test_seq_wrap[compile] | 1.5376ms | 0.3265ms | 3.0628 KOps/s | 3.0228 KOps/s | $\color{#35bf28}+1.32\\%$ | | test_seq_wrap[compile-overhead] | 0.3100s | 0.1483s | 6.7447 Ops/s | 6.7216 Ops/s | $\color{#35bf28}+0.34\\%$ | | test_func_call_runtime[False-eager] | 0.8236ms | 0.7351ms | 1.3603 KOps/s | 1.4055 KOps/s | $\color{#d91a1a}-3.22\\%$ | | test_func_call_runtime[False-compile] | 0.8681ms | 0.8063ms | 1.2403 KOps/s | 1.2316 KOps/s | $\color{#35bf28}+0.70\\%$ | | test_func_call_runtime[False-compile-overhead] | 0.4156ms | 0.3674ms | 2.7221 KOps/s | 2.7123 KOps/s | $\color{#35bf28}+0.36\\%$ | | test_func_call_runtime[True-eager] | 1.0569ms | 0.9681ms | 1.0330 KOps/s | 1.0298 KOps/s | $\color{#35bf28}+0.31\\%$ | | test_func_call_runtime[True-compile] | 0.9457ms | 0.8422ms | 1.1873 KOps/s | 1.1777 KOps/s | $\color{#35bf28}+0.82\\%$ | | test_func_call_runtime[True-compile-overhead] | 0.4689ms | 0.4079ms | 2.4516 KOps/s | 2.4160 KOps/s | $\color{#35bf28}+1.48\\%$ | | test_distributed | 0.2630ms | 70.4561μs | 14.1932 KOps/s | 13.7128 KOps/s | $\color{#35bf28}+3.50\\%$ | | test_tdmodule | 83.0410μs | 15.0949μs | 66.2473 KOps/s | 54.6456 KOps/s | $\textbf{\color{#35bf28}+21.23\\%}$ | | test_tdmodule_dispatch | 50.7410μs | 30.8341μs | 32.4316 KOps/s | 27.1476 KOps/s | $\textbf{\color{#35bf28}+19.46\\%}$ | | test_tdseq | 31.8400μs | 15.7014μs | 63.6886 KOps/s | 56.0165 KOps/s | $\textbf{\color{#35bf28}+13.70\\%}$ | | test_tdseq_dispatch | 50.3010μs | 33.0111μs | 30.2928 KOps/s | 25.6177 KOps/s | $\textbf{\color{#35bf28}+18.25\\%}$ | | test_instantiation_functorch | 2.0684ms | 1.9752ms | 506.2675 Ops/s | 506.5423 Ops/s | $\color{#d91a1a}-0.05\\%$ | | test_instantiation_td | 2.0188ms | 1.3179ms | 758.7708 Ops/s | 769.3446 Ops/s | $\color{#d91a1a}-1.37\\%$ | | test_exec_functorch | 0.2772ms | 0.2202ms | 4.5422 KOps/s | 4.5484 KOps/s | $\color{#d91a1a}-0.14\\%$ | | test_exec_functional_call | 0.2605ms | 0.2087ms | 4.7908 KOps/s | 4.4861 KOps/s | $\textbf{\color{#35bf28}+6.79\\%}$ | | test_exec_td | 0.2913ms | 0.2089ms | 4.7862 KOps/s | 4.6087 KOps/s | $\color{#35bf28}+3.85\\%$ | | test_exec_td_decorator | 0.4960ms | 0.2842ms | 3.5182 KOps/s | 3.3840 KOps/s | $\color{#35bf28}+3.96\\%$ | | test_vmap_mlp_speed[True-True] | 0.7758ms | 0.6420ms | 1.5577 KOps/s | 1.4706 KOps/s | $\textbf{\color{#35bf28}+5.93\\%}$ | | test_vmap_mlp_speed[True-False] | 0.7259ms | 0.6395ms | 1.5636 KOps/s | 1.4864 KOps/s | $\textbf{\color{#35bf28}+5.20\\%}$ | | test_vmap_mlp_speed[False-True] | 0.6426ms | 0.5613ms | 1.7817 KOps/s | 1.7324 KOps/s | $\color{#35bf28}+2.85\\%$ | | test_vmap_mlp_speed[False-False] | 0.6448ms | 0.5860ms | 1.7065 KOps/s | 1.6827 KOps/s | $\color{#35bf28}+1.42\\%$ | | test_vmap_mlp_speed_decorator[True-True] | 1.5539ms | 0.7235ms | 1.3821 KOps/s | 1.3546 KOps/s | $\color{#35bf28}+2.03\\%$ | | test_vmap_mlp_speed_decorator[True-False] | 0.9000ms | 0.7194ms | 1.3900 KOps/s | 1.3641 KOps/s | $\color{#35bf28}+1.89\\%$ | | test_vmap_mlp_speed_decorator[False-True] | 0.7820ms | 0.6278ms | 1.5929 KOps/s | 1.5480 KOps/s | $\color{#35bf28}+2.90\\%$ | | test_vmap_mlp_speed_decorator[False-False] | 0.7484ms | 0.6281ms | 1.5921 KOps/s | 1.5765 KOps/s | $\color{#35bf28}+0.99\\%$ | | test_vmap_transformer_speed[True-True] | 8.6877ms | 8.4784ms | 117.9464 Ops/s | 116.7469 Ops/s | $\color{#35bf28}+1.03\\%$ | | test_vmap_transformer_speed[True-False] | 8.7365ms | 8.4429ms | 118.4426 Ops/s | 116.6183 Ops/s | $\color{#35bf28}+1.56\\%$ | | test_vmap_transformer_speed[False-True] | 8.5760ms | 8.3169ms | 120.2375 Ops/s | 117.5465 Ops/s | $\color{#35bf28}+2.29\\%$ | | test_vmap_transformer_speed[False-False] | 8.7319ms | 8.3458ms | 119.8213 Ops/s | 118.4971 Ops/s | $\color{#35bf28}+1.12\\%$ | | test_vmap_transformer_speed_decorator[True-True] | 20.7067ms | 20.3157ms | 49.2230 Ops/s | 48.5215 Ops/s | $\color{#35bf28}+1.45\\%$ | | test_vmap_transformer_speed_decorator[True-False] | 21.1312ms | 20.1782ms | 49.5585 Ops/s | 48.7671 Ops/s | $\color{#35bf28}+1.62\\%$ | | test_vmap_transformer_speed_decorator[False-True] | 20.1909ms | 19.9431ms | 50.1427 Ops/s | 49.8228 Ops/s | $\color{#35bf28}+0.64\\%$ | | test_vmap_transformer_speed_decorator[False-False] | 20.8096ms | 19.9837ms | 50.0407 Ops/s | 49.0351 Ops/s | $\color{#35bf28}+2.05\\%$ | | test_to_module_speed[True] | 2.7672ms | 1.5216ms | 657.2084 Ops/s | 658.7015 Ops/s | $\color{#d91a1a}-0.23\\%$ | | test_to_module_speed[False] | 2.0350ms | 1.5052ms | 664.3767 Ops/s | 663.5525 Ops/s | $\color{#35bf28}+0.12\\%$ | | test_tc_init | 61.4110μs | 35.9707μs | 27.8004 KOps/s | 25.9630 KOps/s | $\textbf{\color{#35bf28}+7.08\\%}$ | | test_tc_init_nested | 94.8810μs | 71.8255μs | 13.9226 KOps/s | 12.3680 KOps/s | $\textbf{\color{#35bf28}+12.57\\%}$ | | test_tc_first_layer_tensor | 19.1510μs | 3.9884μs | 250.7253 KOps/s | 247.6769 KOps/s | $\color{#35bf28}+1.23\\%$ | | test_tc_first_layer_nontensor | 27.7500μs | 4.0210μs | 248.6914 KOps/s | 246.2047 KOps/s | $\color{#35bf28}+1.01\\%$ | | test_tc_second_layer_tensor | 33.7380μs | 1.3011μs | 768.5742 KOps/s | 760.1912 KOps/s | $\color{#35bf28}+1.10\\%$ | | test_tc_second_layer_nontensor | 21.5910μs | 4.6742μs | 213.9399 KOps/s | 215.8284 KOps/s | $\color{#d91a1a}-0.87\\%$ | | test_unbind | 0.3253s | 12.9979ms | 76.9356 Ops/s | 71.7861 Ops/s | $\textbf{\color{#35bf28}+7.17\\%}$ | | test_full_like | 0.6535ms | 0.5784ms | 1.7288 KOps/s | 1.7311 KOps/s | $\color{#d91a1a}-0.14\\%$ | | test_zeros_like | 0.2596ms | 0.1979ms | 5.0543 KOps/s | 5.0614 KOps/s | $\color{#d91a1a}-0.14\\%$ | | test_ones_like | 0.2257ms | 0.1976ms | 5.0597 KOps/s | 5.0651 KOps/s | $\color{#d91a1a}-0.11\\%$ | | test_clone | 0.4467ms | 0.4144ms | 2.4132 KOps/s | 2.4204 KOps/s | $\color{#d91a1a}-0.30\\%$ | | test_squeeze | 27.5810μs | 11.7168μs | 85.3477 KOps/s | 83.4446 KOps/s | $\color{#35bf28}+2.28\\%$ | | test_unsqueeze | 0.2495ms | 86.0552μs | 11.6205 KOps/s | 11.7908 KOps/s | $\color{#d91a1a}-1.44\\%$ | | test_split | 0.4329ms | 0.1791ms | 5.5835 KOps/s | 5.5562 KOps/s | $\color{#35bf28}+0.49\\%$ | | test_permute | 0.2561ms | 0.1911ms | 5.2339 KOps/s | 5.1307 KOps/s | $\color{#35bf28}+2.01\\%$ | | test_stack | 1.2633ms | 0.9275ms | 1.0782 KOps/s | 1.0961 KOps/s | $\color{#d91a1a}-1.63\\%$ | | test_cat | 1.2568ms | 1.2314ms | 812.0556 Ops/s | 812.3090 Ops/s | $\color{#d91a1a}-0.03\\%$ |
github-actions[bot]
commented
1 month ago github-actions[bot]
commented
1 month ago
$\color{#D29922}\textsf{\Large\⚠\kern{0.2cm}\normalsize Warning}$ Result of CPU Benchmark Tests
Total Benchmarks: 144. Improved: $\large\color{#35bf28}26$. Worsened: $\large\color{#d91a1a}13$.
Expand to view detailed results
| Name | Max | Mean | Ops | Ops on Repo `HEAD` | Change | | ------------------------------------------ | --------- | --------- | --------------- | ------------------ | ----------------------------------- | | test_plain_set_nested | 45.5250μs | 21.9617μs | 45.5338 KOps/s | 44.0560 KOps/s | $\color{#35bf28}+3.35\\%$ | | test_plain_set_stack_nested | 63.4780μs | 22.1794μs | 45.0869 KOps/s | 43.9914 KOps/s | $\color{#35bf28}+2.49\\%$ | | test_plain_set_nested_inplace | 89.7970μs | 23.9012μs | 41.8389 KOps/s | 40.3770 KOps/s | $\color{#35bf28}+3.62\\%$ | | test_plain_set_stack_nested_inplace | 76.5330μs | 24.1039μs | 41.4871 KOps/s | 41.0286 KOps/s | $\color{#35bf28}+1.12\\%$ | | test_items | 24.0540μs | 2.5795μs | 387.6664 KOps/s | 357.7499 KOps/s | $\textbf{\color{#35bf28}+8.36\\%}$ | | test_items_nested | 0.7392ms | 0.3611ms | 2.7695 KOps/s | 2.7502 KOps/s | $\color{#35bf28}+0.70\\%$ | | test_items_nested_locked | 2.1839ms | 0.3632ms | 2.7534 KOps/s | 2.7660 KOps/s | $\color{#d91a1a}-0.45\\%$ | | test_items_nested_leaf | 0.3255ms | 89.1020μs | 11.2231 KOps/s | 11.6143 KOps/s | $\color{#d91a1a}-3.37\\%$ | | test_items_stack_nested | 0.5639ms | 0.3645ms | 2.7434 KOps/s | 2.7288 KOps/s | $\color{#35bf28}+0.54\\%$ | | test_items_stack_nested_leaf | 0.1630ms | 85.4703μs | 11.7000 KOps/s | 11.5941 KOps/s | $\color{#35bf28}+0.91\\%$ | | test_items_stack_nested_locked | 0.7707ms | 0.3696ms | 2.7054 KOps/s | 2.7677 KOps/s | $\color{#d91a1a}-2.25\\%$ | | test_keys | 47.6290μs | 3.8536μs | 259.4956 KOps/s | 244.1452 KOps/s | $\textbf{\color{#35bf28}+6.29\\%}$ | | test_keys_nested | 0.3137ms | 0.1430ms | 6.9935 KOps/s | 6.8945 KOps/s | $\color{#35bf28}+1.44\\%$ | | test_keys_nested_locked | 1.9063ms | 0.1537ms | 6.5081 KOps/s | 6.6518 KOps/s | $\color{#d91a1a}-2.16\\%$ | | test_keys_nested_leaf | 0.2186ms | 0.1237ms | 8.0815 KOps/s | 8.0074 KOps/s | $\color{#35bf28}+0.93\\%$ | | test_keys_stack_nested | 0.2408ms | 0.1447ms | 6.9100 KOps/s | 6.8815 KOps/s | $\color{#35bf28}+0.42\\%$ | | test_keys_stack_nested_leaf | 0.2161ms | 0.1238ms | 8.0749 KOps/s | 8.0556 KOps/s | $\color{#35bf28}+0.24\\%$ | | test_keys_stack_nested_locked | 0.2599ms | 0.1493ms | 6.6962 KOps/s | 6.6383 KOps/s | $\color{#35bf28}+0.87\\%$ | | test_values | 10.3143μs | 1.1639μs | 859.1836 KOps/s | 861.2822 KOps/s | $\color{#d91a1a}-0.24\\%$ | | test_values_nested | 91.4700μs | 48.9993μs | 20.4085 KOps/s | 20.0089 KOps/s | $\color{#35bf28}+2.00\\%$ | | test_values_nested_locked | 94.2850μs | 48.7434μs | 20.5156 KOps/s | 19.9248 KOps/s | $\color{#35bf28}+2.97\\%$ | | test_values_nested_leaf | 96.3670μs | 44.7104μs | 22.3662 KOps/s | 22.3119 KOps/s | $\color{#35bf28}+0.24\\%$ | | test_values_stack_nested | 0.1002ms | 50.6312μs | 19.7507 KOps/s | 19.9275 KOps/s | $\color{#d91a1a}-0.89\\%$ | | test_values_stack_nested_leaf | 96.4800μs | 44.2706μs | 22.5884 KOps/s | 22.3222 KOps/s | $\color{#35bf28}+1.19\\%$ | | test_values_stack_nested_locked | 0.1024ms | 51.1807μs | 19.5386 KOps/s | 20.0695 KOps/s | $\color{#d91a1a}-2.65\\%$ | | test_membership | 2.8383μs | 0.7063μs | 1.4159 MOps/s | 1.1023 MOps/s | $\textbf{\color{#35bf28}+28.45\\%}$ | | test_membership_nested | 27.9420μs | 2.6925μs | 371.3982 KOps/s | 372.0381 KOps/s | $\color{#d91a1a}-0.17\\%$ | | test_membership_nested_leaf | 29.2140μs | 2.7090μs | 369.1374 KOps/s | 326.7643 KOps/s | $\textbf{\color{#35bf28}+12.97\\%}$ | | test_membership_stacked_nested | 28.4840μs | 2.6800μs | 373.1388 KOps/s | 369.4733 KOps/s | $\color{#35bf28}+0.99\\%$ | | test_membership_stacked_nested_leaf | 30.5770μs | 2.7093μs | 369.0963 KOps/s | 369.4705 KOps/s | $\color{#d91a1a}-0.10\\%$ | | test_membership_nested_last | 41.9380μs | 3.9699μs | 251.8933 KOps/s | 250.1299 KOps/s | $\color{#35bf28}+0.71\\%$ | | test_membership_nested_leaf_last | 24.2750μs | 4.0002μs | 249.9870 KOps/s | 250.9570 KOps/s | $\color{#d91a1a}-0.39\\%$ | | test_membership_stacked_nested_last | 58.9900μs | 12.9737μs | 77.0792 KOps/s | 252.3239 KOps/s | $\textbf{\color{#d91a1a}-69.45\\%}$ | | test_membership_stacked_nested_leaf_last | 43.1510μs | 13.1713μs | 75.9226 KOps/s | 248.8684 KOps/s | $\textbf{\color{#d91a1a}-69.49\\%}$ | | test_nested_getleaf | 57.0260μs | 11.0861μs | 90.2031 KOps/s | 95.1283 KOps/s | $\textbf{\color{#d91a1a}-5.18\\%}$ | | test_nested_get | 51.8170μs | 10.6157μs | 94.1997 KOps/s | 99.3060 KOps/s | $\textbf{\color{#d91a1a}-5.14\\%}$ | | test_stacked_getleaf | 50.0640μs | 10.9301μs | 91.4908 KOps/s | 94.0257 KOps/s | $\color{#d91a1a}-2.70\\%$ | | test_stacked_get | 33.6420μs | 10.4294μs | 95.8824 KOps/s | 99.0112 KOps/s | $\color{#d91a1a}-3.16\\%$ | | test_nested_getitemleaf | 53.8700μs | 11.3772μs | 87.8949 KOps/s | 90.0638 KOps/s | $\color{#d91a1a}-2.41\\%$ | | test_nested_getitem | 60.3820μs | 10.5287μs | 94.9789 KOps/s | 97.7026 KOps/s | $\color{#d91a1a}-2.79\\%$ | | test_stacked_getitemleaf | 40.4650μs | 11.2858μs | 88.6073 KOps/s | 89.9988 KOps/s | $\color{#d91a1a}-1.55\\%$ | | test_stacked_getitem | 56.8350μs | 10.5348μs | 94.9237 KOps/s | 97.4208 KOps/s | $\color{#d91a1a}-2.56\\%$ | | test_lock_nested | 3.0446ms | 0.5262ms | 1.9006 KOps/s | 1.6648 KOps/s | $\textbf{\color{#35bf28}+14.16\\%}$ | | test_lock_stack_nested | 0.6106ms | 0.4677ms | 2.1381 KOps/s | 2.0306 KOps/s | $\textbf{\color{#35bf28}+5.29\\%}$ | | test_unlock_nested | 0.9455ms | 0.4449ms | 2.2478 KOps/s | 1.9542 KOps/s | $\textbf{\color{#35bf28}+15.03\\%}$ | | test_unlock_stack_nested | 0.5492ms | 0.3819ms | 2.6186 KOps/s | 2.4484 KOps/s | $\textbf{\color{#35bf28}+6.95\\%}$ | | test_flatten_speed | 0.2396ms | 0.1078ms | 9.2736 KOps/s | 9.3572 KOps/s | $\color{#d91a1a}-0.89\\%$ | | test_unflatten_speed | 0.6077ms | 0.4529ms | 2.2078 KOps/s | 2.2503 KOps/s | $\color{#d91a1a}-1.89\\%$ | | test_common_ops | 5.3983ms | 1.1482ms | 870.9199 Ops/s | 850.5103 Ops/s | $\color{#35bf28}+2.40\\%$ | | test_creation | 26.3890μs | 2.5970μs | 385.0580 KOps/s | 399.5637 KOps/s | $\color{#d91a1a}-3.63\\%$ | | test_creation_empty | 65.9130μs | 18.9621μs | 52.7368 KOps/s | 48.3839 KOps/s | $\textbf{\color{#35bf28}+9.00\\%}$ | | test_creation_nested_1 | 50.9250μs | 22.1614μs | 45.1235 KOps/s | 42.3411 KOps/s | $\textbf{\color{#35bf28}+6.57\\%}$ | | test_creation_nested_2 | 90.2990μs | 26.4394μs | 37.8224 KOps/s | 35.5785 KOps/s | $\textbf{\color{#35bf28}+6.31\\%}$ | | test_clone | 87.3230μs | 17.8868μs | 55.9072 KOps/s | 55.9399 KOps/s | $\color{#d91a1a}-0.06\\%$ | | test_getitem[int] | 0.9763ms | 13.3332μs | 75.0008 KOps/s | 76.8919 KOps/s | $\color{#d91a1a}-2.46\\%$ | | test_getitem[slice_int] | 0.1541ms | 34.0167μs | 29.3973 KOps/s | 30.5000 KOps/s | $\color{#d91a1a}-3.62\\%$ | | test_getitem[range] | 0.4018ms | 57.1003μs | 17.5131 KOps/s | 17.5687 KOps/s | $\color{#d91a1a}-0.32\\%$ | | test_getitem[tuple] | 0.1322ms | 27.2979μs | 36.6328 KOps/s | 36.5675 KOps/s | $\color{#35bf28}+0.18\\%$ | | test_getitem[list] | 0.2565ms | 52.0509μs | 19.2120 KOps/s | 19.3084 KOps/s | $\color{#d91a1a}-0.50\\%$ | | test_setitem_dim[int] | 57.7170μs | 32.7733μs | 30.5126 KOps/s | 26.6784 KOps/s | $\textbf{\color{#35bf28}+14.37\\%}$ | | test_setitem_dim[slice_int] | 0.1685ms | 70.7185μs | 14.1406 KOps/s | 13.1765 KOps/s | $\textbf{\color{#35bf28}+7.32\\%}$ | | test_setitem_dim[range] | 0.1389ms | 92.2327μs | 10.8421 KOps/s | 10.1312 KOps/s | $\textbf{\color{#35bf28}+7.02\\%}$ | | test_setitem_dim[tuple] | 0.1077ms | 58.9015μs | 16.9775 KOps/s | 15.6625 KOps/s | $\textbf{\color{#35bf28}+8.40\\%}$ | | test_setitem | 0.1111ms | 30.8819μs | 32.3814 KOps/s | 31.3893 KOps/s | $\color{#35bf28}+3.16\\%$ | | test_set | 0.1247ms | 30.0619μs | 33.2647 KOps/s | 31.5649 KOps/s | $\textbf{\color{#35bf28}+5.39\\%}$ | | test_set_shared | 3.8803ms | 0.2219ms | 4.5075 KOps/s | 4.4708 KOps/s | $\color{#35bf28}+0.82\\%$ | | test_update | 0.1658ms | 36.6309μs | 27.2994 KOps/s | 25.5093 KOps/s | $\textbf{\color{#35bf28}+7.02\\%}$ | | test_update_nested | 0.2261ms | 47.4724μs | 21.0649 KOps/s | 19.9409 KOps/s | $\textbf{\color{#35bf28}+5.64\\%}$ | | test_update__nested | 0.1950ms | 34.8115μs | 28.7262 KOps/s | 28.3219 KOps/s | $\color{#35bf28}+1.43\\%$ | | test_set_nested | 0.1649ms | 32.2656μs | 30.9928 KOps/s | 30.0308 KOps/s | $\color{#35bf28}+3.20\\%$ | | test_set_nested_new | 0.1400ms | 37.0331μs | 27.0029 KOps/s | 25.7790 KOps/s | $\color{#35bf28}+4.75\\%$ | | test_select | 0.1462ms | 54.8544μs | 18.2301 KOps/s | 18.1048 KOps/s | $\color{#35bf28}+0.69\\%$ | | test_select_nested | 0.1283ms | 60.5448μs | 16.5167 KOps/s | 16.6768 KOps/s | $\color{#d91a1a}-0.96\\%$ | | test_exclude_nested | 0.1628ms | 81.6546μs | 12.2467 KOps/s | 12.4947 KOps/s | $\color{#d91a1a}-1.98\\%$ | | test_empty[True] | 0.4698ms | 0.3475ms | 2.8775 KOps/s | 2.9709 KOps/s | $\color{#d91a1a}-3.14\\%$ | | test_empty[False] | 11.7895μs | 1.2930μs | 773.3822 KOps/s | 819.6408 KOps/s | $\textbf{\color{#d91a1a}-5.64\\%}$ | | test_unbind_speed | 0.7840ms | 0.3350ms | 2.9849 KOps/s | 2.9951 KOps/s | $\color{#d91a1a}-0.34\\%$ | | test_unbind_speed_stack0 | 0.5580ms | 0.3086ms | 3.2409 KOps/s | 3.0356 KOps/s | $\textbf{\color{#35bf28}+6.76\\%}$ | | test_unbind_speed_stack1 | 84.0227ms | 0.8102ms | 1.2342 KOps/s | 1.3879 KOps/s | $\textbf{\color{#d91a1a}-11.07\\%}$ | | test_split | 84.2526ms | 2.2327ms | 447.8861 Ops/s | 425.0238 Ops/s | $\textbf{\color{#35bf28}+5.38\\%}$ | | test_chunk | 83.9164ms | 2.2321ms | 448.0180 Ops/s | 420.9829 Ops/s | $\textbf{\color{#35bf28}+6.42\\%}$ | | test_creation[device0] | 0.2793ms | 0.1227ms | 8.1528 KOps/s | 8.0588 KOps/s | $\color{#35bf28}+1.17\\%$ | | test_creation_from_tensor | 3.3868ms | 0.1229ms | 8.1370 KOps/s | 8.1320 KOps/s | $\color{#35bf28}+0.06\\%$ | | test_add_one[memmap_tensor0] | 0.1872ms | 7.6793μs | 130.2203 KOps/s | 120.5144 KOps/s | $\textbf{\color{#35bf28}+8.05\\%}$ | | test_contiguous[memmap_tensor0] | 25.9180μs | 2.2112μs | 452.2454 KOps/s | 450.3832 KOps/s | $\color{#35bf28}+0.41\\%$ | | test_stack[memmap_tensor0] | 72.2750μs | 6.0014μs | 166.6291 KOps/s | 164.9375 KOps/s | $\color{#35bf28}+1.03\\%$ | | test_memmaptd_index | 1.1382ms | 0.4386ms | 2.2802 KOps/s | 2.2862 KOps/s | $\color{#d91a1a}-0.26\\%$ | | test_memmaptd_index_astensor | 0.7801ms | 0.5166ms | 1.9357 KOps/s | 1.9206 KOps/s | $\color{#35bf28}+0.79\\%$ | | test_memmaptd_index_op | 1.4163ms | 1.0637ms | 940.1244 Ops/s | 900.7616 Ops/s | $\color{#35bf28}+4.37\\%$ | | test_serialize_model | 0.1352s | 0.1283s | 7.7947 Ops/s | 7.0631 Ops/s | $\textbf{\color{#35bf28}+10.36\\%}$ | | test_serialize_model_pickle | 0.4465s | 0.3941s | 2.5377 Ops/s | 2.4965 Ops/s | $\color{#35bf28}+1.65\\%$ | | test_serialize_weights | 0.1457s | 0.1292s | 7.7397 Ops/s | 7.8621 Ops/s | $\color{#d91a1a}-1.56\\%$ | | test_serialize_weights_returnearly | 0.1858s | 0.1664s | 6.0096 Ops/s | 6.0906 Ops/s | $\color{#d91a1a}-1.33\\%$ | | test_serialize_weights_pickle | 1.0404s | 0.7489s | 1.3352 Ops/s | 2.4615 Ops/s | $\textbf{\color{#d91a1a}-45.76\\%}$ | | test_serialize_weights_filesystem | 0.1477s | 0.1448s | 6.9084 Ops/s | 6.9730 Ops/s | $\color{#d91a1a}-0.93\\%$ | | test_serialize_model_filesystem | 0.1552s | 0.1490s | 6.7123 Ops/s | 6.7188 Ops/s | $\color{#d91a1a}-0.10\\%$ | | test_reshape_pytree | 89.7170μs | 39.4581μs | 25.3434 KOps/s | 25.4025 KOps/s | $\color{#d91a1a}-0.23\\%$ | | test_reshape_td | 0.1352ms | 49.5370μs | 20.1869 KOps/s | 19.4473 KOps/s | $\color{#35bf28}+3.80\\%$ | | test_view_pytree | 95.6790μs | 39.2206μs | 25.4968 KOps/s | 25.6273 KOps/s | $\color{#d91a1a}-0.51\\%$ | | test_view_td | 0.1187ms | 55.1220μs | 18.1416 KOps/s | 17.2698 KOps/s | $\textbf{\color{#35bf28}+5.05\\%}$ | | test_unbind_pytree | 84.5980μs | 35.7377μs | 27.9816 KOps/s | 28.1659 KOps/s | $\color{#d91a1a}-0.65\\%$ | | test_unbind_td | 0.3589ms | 48.6826μs | 20.5412 KOps/s | 17.8744 KOps/s | $\textbf{\color{#35bf28}+14.92\\%}$ | | test_split_pytree | 0.1077ms | 39.1108μs | 25.5684 KOps/s | 26.4003 KOps/s | $\color{#d91a1a}-3.15\\%$ | | test_split_td | 0.5476ms | 60.5027μs | 16.5282 KOps/s | 16.1651 KOps/s | $\color{#35bf28}+2.25\\%$ | | test_add_pytree | 0.1217ms | 44.0748μs | 22.6887 KOps/s | 22.4682 KOps/s | $\color{#35bf28}+0.98\\%$ | | test_add_td | 0.1734ms | 82.1569μs | 12.1718 KOps/s | 11.2424 KOps/s | $\textbf{\color{#35bf28}+8.27\\%}$ | | test_distributed | 0.2937ms | 0.1318ms | 7.5861 KOps/s | 7.5281 KOps/s | $\color{#35bf28}+0.77\\%$ | | test_tdmodule | 40.7560μs | 17.5237μs | 57.0654 KOps/s | 57.2777 KOps/s | $\color{#d91a1a}-0.37\\%$ | | test_tdmodule_dispatch | 78.8770μs | 37.2370μs | 26.8550 KOps/s | 26.9176 KOps/s | $\color{#d91a1a}-0.23\\%$ | | test_tdseq | 41.9380μs | 19.6569μs | 50.8726 KOps/s | 52.0305 KOps/s | $\color{#d91a1a}-2.23\\%$ | | test_tdseq_dispatch | 85.8200μs | 43.0589μs | 23.2240 KOps/s | 24.0971 KOps/s | $\color{#d91a1a}-3.62\\%$ | | test_instantiation_functorch | 2.9127ms | 1.5996ms | 625.1481 Ops/s | 637.1188 Ops/s | $\color{#d91a1a}-1.88\\%$ | | test_instantiation_td | 1.8868ms | 1.1770ms | 849.5863 Ops/s | 882.3800 Ops/s | $\color{#d91a1a}-3.72\\%$ | | test_exec_functorch | 0.3198ms | 0.1801ms | 5.5526 KOps/s | 5.4188 KOps/s | $\color{#35bf28}+2.47\\%$ | | test_exec_functional_call | 0.2930ms | 0.1693ms | 5.9068 KOps/s | 5.7416 KOps/s | $\color{#35bf28}+2.88\\%$ | | test_exec_td | 0.3253ms | 0.1722ms | 5.8081 KOps/s | 5.8716 KOps/s | $\color{#d91a1a}-1.08\\%$ | | test_exec_td_decorator | 83.8653ms | 0.3218ms | 3.1079 KOps/s | 3.9247 KOps/s | $\textbf{\color{#d91a1a}-20.81\\%}$ | | test_vmap_mlp_speed[True-True] | 0.8064ms | 0.6111ms | 1.6365 KOps/s | 1.6191 KOps/s | $\color{#35bf28}+1.07\\%$ | | test_vmap_mlp_speed[True-False] | 1.3898ms | 0.6191ms | 1.6154 KOps/s | 1.6322 KOps/s | $\color{#d91a1a}-1.03\\%$ | | test_vmap_mlp_speed[False-True] | 0.7495ms | 0.4962ms | 2.0152 KOps/s | 1.9881 KOps/s | $\color{#35bf28}+1.36\\%$ | | test_vmap_mlp_speed[False-False] | 0.6995ms | 0.4954ms | 2.0185 KOps/s | 1.9735 KOps/s | $\color{#35bf28}+2.28\\%$ | | test_vmap_mlp_speed_decorator[True-True] | 1.2148ms | 0.6980ms | 1.4327 KOps/s | 1.4159 KOps/s | $\color{#35bf28}+1.18\\%$ | | test_vmap_mlp_speed_decorator[True-False] | 1.2134ms | 0.7026ms | 1.4233 KOps/s | 1.4189 KOps/s | $\color{#35bf28}+0.31\\%$ | | test_vmap_mlp_speed_decorator[False-True] | 0.8101ms | 0.5771ms | 1.7328 KOps/s | 1.7194 KOps/s | $\color{#35bf28}+0.78\\%$ | | test_vmap_mlp_speed_decorator[False-False] | 0.7292ms | 0.5776ms | 1.7314 KOps/s | 1.7054 KOps/s | $\color{#35bf28}+1.52\\%$ | | test_to_module_speed[True] | 2.8508ms | 1.8597ms | 537.7103 Ops/s | 558.4528 Ops/s | $\color{#d91a1a}-3.71\\%$ | | test_to_module_speed[False] | 4.1727ms | 1.8502ms | 540.4939 Ops/s | 565.2006 Ops/s | $\color{#d91a1a}-4.37\\%$ | | test_tc_init | 0.1001ms | 43.9544μs | 22.7508 KOps/s | 22.0774 KOps/s | $\color{#35bf28}+3.05\\%$ | | test_tc_init_nested | 0.1874ms | 87.2828μs | 11.4570 KOps/s | 11.1972 KOps/s | $\color{#35bf28}+2.32\\%$ | | test_tc_first_layer_tensor | 51.5160μs | 9.4453μs | 105.8731 KOps/s | 109.4267 KOps/s | $\color{#d91a1a}-3.25\\%$ | | test_tc_first_layer_nontensor | 40.6260μs | 9.2622μs | 107.9654 KOps/s | 108.8915 KOps/s | $\color{#d91a1a}-0.85\\%$ | | test_tc_second_layer_tensor | 78.9380μs | 2.9365μs | 340.5458 KOps/s | 356.0622 KOps/s | $\color{#d91a1a}-4.36\\%$ | | test_tc_second_layer_nontensor | 35.0760μs | 10.5991μs | 94.3479 KOps/s | 98.3132 KOps/s | $\color{#d91a1a}-4.03\\%$ | | test_unbind | 0.1090s | 13.9113ms | 71.8840 Ops/s | 69.7159 Ops/s | $\color{#35bf28}+3.11\\%$ | | test_full_like | 10.4053ms | 8.4895ms | 117.7920 Ops/s | 141.7561 Ops/s | $\textbf{\color{#d91a1a}-16.91\\%}$ | | test_zeros_like | 11.7537ms | 7.0446ms | 141.9529 Ops/s | 142.5975 Ops/s | $\color{#d91a1a}-0.45\\%$ | | test_ones_like | 16.4042ms | 8.1388ms | 122.8679 Ops/s | 133.5150 Ops/s | $\textbf{\color{#d91a1a}-7.97\\%}$ | | test_clone | 18.6706ms | 10.0841ms | 99.1663 Ops/s | 109.6168 Ops/s | $\textbf{\color{#d91a1a}-9.53\\%}$ | | test_squeeze | 71.2320μs | 15.1322μs | 66.0841 KOps/s | 69.0604 KOps/s | $\color{#d91a1a}-4.31\\%$ | | test_unsqueeze | 0.1650ms | 97.9576μs | 10.2085 KOps/s | 10.1546 KOps/s | $\color{#35bf28}+0.53\\%$ | | test_split | 0.4583ms | 0.2108ms | 4.7432 KOps/s | 4.7645 KOps/s | $\color{#d91a1a}-0.45\\%$ | | test_permute | 0.3733ms | 0.2241ms | 4.4618 KOps/s | 4.3284 KOps/s | $\color{#35bf28}+3.08\\%$ | | test_stack | 33.9953ms | 26.9606ms | 37.0912 Ops/s | 40.4623 Ops/s | $\textbf{\color{#d91a1a}-8.33\\%}$ | | test_cat | 38.1198ms | 28.0177ms | 35.6917 Ops/s | 40.1871 Ops/s | $\textbf{\color{#d91a1a}-11.19\\%}$ |