-
Notifications
You must be signed in to change notification settings - Fork 112
[BugFix,Setup] Don't track _version.py #1522
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Open
vmoens
wants to merge
6
commits into
main
Choose a base branch
from
fix-version
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Open
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
| Name | Max | Mean | Ops | Ops on Repo HEAD
|
Change |
|---|---|---|---|---|---|
| test_plain_set_nested | 1.2672ms | 15.1242μs | 66.1194 KOps/s | 65.5893 KOps/s | |
| test_plain_set_stack_nested | 45.3800μs | 15.2875μs | 65.4128 KOps/s | 65.4222 KOps/s | |
| test_plain_set_nested_inplace | 50.4110μs | 16.6324μs | 60.1237 KOps/s | 59.4441 KOps/s | |
| test_plain_set_stack_nested_inplace | 54.4510μs | 16.7428μs | 59.7270 KOps/s | 60.6259 KOps/s | |
| test_items | 27.2400μs | 5.8908μs | 169.7556 KOps/s | 170.7797 KOps/s | |
| test_items_nested | 0.5895ms | 0.5332ms | 1.8754 KOps/s | 1.8448 KOps/s | |
| test_items_nested_locked | 0.6017ms | 0.5353ms | 1.8682 KOps/s | 1.8239 KOps/s | |
| test_items_nested_leaf | 0.1270ms | 95.8633μs | 10.4315 KOps/s | 10.3055 KOps/s | |
| test_items_stack_nested | 0.6000ms | 0.5380ms | 1.8586 KOps/s | 1.8478 KOps/s | |
| test_items_stack_nested_leaf | 0.1332ms | 97.2613μs | 10.2816 KOps/s | 10.4690 KOps/s | |
| test_items_stack_nested_locked | 0.6178ms | 0.5438ms | 1.8389 KOps/s | 1.8192 KOps/s | |
| test_keys | 32.8610μs | 4.1983μs | 238.1939 KOps/s | 236.0240 KOps/s | |
| test_keys_nested | 0.1817ms | 0.1190ms | 8.4039 KOps/s | 8.3219 KOps/s | |
| test_keys_nested_locked | 2.2306ms | 0.1299ms | 7.7001 KOps/s | 7.6563 KOps/s | |
| test_keys_nested_leaf | 0.1444ms | 0.1113ms | 8.9816 KOps/s | 9.0219 KOps/s | |
| test_keys_stack_nested | 0.1507ms | 0.1205ms | 8.3012 KOps/s | 8.2214 KOps/s | |
| test_keys_stack_nested_leaf | 0.1568ms | 0.1103ms | 9.0627 KOps/s | 8.9547 KOps/s | |
| test_keys_stack_nested_locked | 0.1746ms | 0.1284ms | 7.7866 KOps/s | 7.6608 KOps/s | |
| test_values | 7.0942μs | 1.0273μs | 973.4250 KOps/s | 975.0223 KOps/s | |
| test_values_nested | 79.5310μs | 48.1646μs | 20.7622 KOps/s | 20.6793 KOps/s | |
| test_values_nested_locked | 91.2720μs | 51.5907μs | 19.3833 KOps/s | 19.3539 KOps/s | |
| test_values_nested_leaf | 95.0110μs | 54.8445μs | 18.2334 KOps/s | 18.2672 KOps/s | |
| test_values_stack_nested | 92.2720μs | 48.3441μs | 20.6850 KOps/s | 20.7411 KOps/s | |
| test_values_stack_nested_leaf | 87.4720μs | 54.4559μs | 18.3635 KOps/s | 18.3059 KOps/s | |
| test_values_stack_nested_locked | 88.8310μs | 51.0264μs | 19.5977 KOps/s | 19.3676 KOps/s | |
| test_membership | 5.0302μs | 0.8342μs | 1.1988 MOps/s | 1.1797 MOps/s | |
| test_membership_nested | 29.7210μs | 3.1948μs | 313.0100 KOps/s | 313.9586 KOps/s | |
| test_membership_nested_leaf | 32.1800μs | 3.1719μs | 315.2726 KOps/s | 311.7922 KOps/s | |
| test_membership_stacked_nested | 49.8810μs | 3.2062μs | 311.8954 KOps/s | 311.2010 KOps/s | |
| test_membership_stacked_nested_leaf | 32.9410μs | 3.1938μs | 313.1033 KOps/s | 312.3754 KOps/s | |
| test_membership_nested_last | 36.2000μs | 4.6851μs | 213.4406 KOps/s | 214.3302 KOps/s | |
| test_membership_nested_leaf_last | 33.3100μs | 4.6567μs | 214.7428 KOps/s | 212.2912 KOps/s | |
| test_membership_stacked_nested_last | 34.1510μs | 4.6147μs | 216.7007 KOps/s | 213.2944 KOps/s | |
| test_membership_stacked_nested_leaf_last | 40.0410μs | 4.6512μs | 215.0003 KOps/s | 213.4710 KOps/s | |
| test_nested_getleaf | 47.1510μs | 21.5264μs | 46.4546 KOps/s | 46.7229 KOps/s | |
| test_nested_get | 54.8410μs | 20.1641μs | 49.5931 KOps/s | 49.0842 KOps/s | |
| test_stacked_getleaf | 50.9310μs | 21.2083μs | 47.1512 KOps/s | 46.1854 KOps/s | |
| test_stacked_get | 48.2410μs | 20.6273μs | 48.4795 KOps/s | 49.0351 KOps/s | |
| test_nested_getitemleaf | 42.5610μs | 22.1987μs | 45.0476 KOps/s | 45.2324 KOps/s | |
| test_nested_getitem | 48.1710μs | 20.6102μs | 48.5196 KOps/s | 47.4921 KOps/s | |
| test_stacked_getitemleaf | 48.2210μs | 21.9133μs | 45.6344 KOps/s | 45.3497 KOps/s | |
| test_stacked_getitem | 50.2510μs | 21.1391μs | 47.3056 KOps/s | 47.2836 KOps/s | |
| test_lock_nested | 4.4982ms | 0.4783ms | 2.0905 KOps/s | 2.1164 KOps/s | |
| test_lock_stack_nested | 0.5315ms | 0.4766ms | 2.0982 KOps/s | 2.0894 KOps/s | |
| test_unlock_nested | 0.4892ms | 0.3851ms | 2.5964 KOps/s | 2.6190 KOps/s | |
| test_unlock_stack_nested | 0.4429ms | 0.3826ms | 2.6137 KOps/s | 2.6043 KOps/s | |
| test_flatten_speed | 0.1671ms | 0.1235ms | 8.0970 KOps/s | 8.1719 KOps/s | |
| test_unflatten_speed | 0.6401ms | 0.5971ms | 1.6749 KOps/s | 1.6806 KOps/s | |
| test_common_ops | 0.9202ms | 0.7520ms | 1.3298 KOps/s | 1.3374 KOps/s | |
| test_creation | 0.1284ms | 2.5463μs | 392.7285 KOps/s | 399.6175 KOps/s | |
| test_creation_empty | 26.9800μs | 8.8783μs | 112.6346 KOps/s | 112.9448 KOps/s | |
| test_creation_nested_1 | 43.0110μs | 11.7647μs | 85.0004 KOps/s | 84.5625 KOps/s | |
| test_creation_nested_2 | 38.5210μs | 15.6663μs | 63.8314 KOps/s | 63.7925 KOps/s | |
| test_clone | 64.6410μs | 13.6553μs | 73.2316 KOps/s | 74.3578 KOps/s | |
| test_getitem[int] | 1.1534ms | 14.1656μs | 70.5934 KOps/s | 70.6727 KOps/s | |
| test_getitem[slice_int] | 0.1422ms | 24.6401μs | 40.5843 KOps/s | 40.3205 KOps/s | |
| test_getitem[range] | 0.1760ms | 59.7531μs | 16.7355 KOps/s | 16.8892 KOps/s | |
| test_getitem[tuple] | 0.1464ms | 24.5515μs | 40.7307 KOps/s | 40.7942 KOps/s | |
| test_getitem[list] | 0.1803ms | 54.3997μs | 18.3824 KOps/s | 18.6608 KOps/s | |
| test_setitem_dim[int] | 44.1210μs | 24.9388μs | 40.0982 KOps/s | 40.5969 KOps/s | |
| test_setitem_dim[slice_int] | 83.6920μs | 47.8401μs | 20.9029 KOps/s | 21.9964 KOps/s | |
| test_setitem_dim[range] | 0.1116ms | 87.0954μs | 11.4817 KOps/s | 11.5620 KOps/s | |
| test_setitem_dim[tuple] | 81.3520μs | 41.8788μs | 23.8784 KOps/s | 23.7192 KOps/s | |
| test_setitem | 53.8310μs | 18.3067μs | 54.6248 KOps/s | 54.9308 KOps/s | |
| test_set | 51.1410μs | 17.7837μs | 56.2313 KOps/s | 57.5294 KOps/s | |
| test_set_shared | 0.4962ms | 0.2124ms | 4.7087 KOps/s | 4.8431 KOps/s | |
| test_update | 0.2592ms | 22.7443μs | 43.9671 KOps/s | 44.6996 KOps/s | |
| test_update_nested | 62.9710μs | 35.1940μs | 28.4139 KOps/s | 28.3612 KOps/s | |
| test_update__nested | 0.4849ms | 35.4928μs | 28.1747 KOps/s | 27.8321 KOps/s | |
| test_set_nested | 55.7010μs | 19.7585μs | 50.6111 KOps/s | 50.6462 KOps/s | |
| test_set_nested_new | 62.5110μs | 25.4164μs | 39.3446 KOps/s | 39.4571 KOps/s | |
| test_select | 83.7820μs | 43.4064μs | 23.0381 KOps/s | 23.1319 KOps/s | |
| test_select_nested | 0.1088ms | 75.3455μs | 13.2722 KOps/s | 13.3213 KOps/s | |
| test_exclude_nested | 0.1338ms | 97.6580μs | 10.2398 KOps/s | 10.3259 KOps/s | |
| test_empty[True] | 0.4932ms | 0.4369ms | 2.2889 KOps/s | 2.2796 KOps/s | |
| test_empty[False] | 9.6402μs | 1.3393μs | 746.6752 KOps/s | 754.5441 KOps/s | |
| test_to | 0.1024ms | 71.4038μs | 14.0049 KOps/s | 13.7250 KOps/s | |
| test_to_nonblocking | 0.1300ms | 64.7699μs | 15.4393 KOps/s | 15.2958 KOps/s | |
| test_unbind_speed | 0.3597ms | 0.3279ms | 3.0497 KOps/s | 3.0973 KOps/s | |
| test_unbind_speed_stack0 | 0.3779ms | 0.3267ms | 3.0612 KOps/s | 3.1290 KOps/s | |
| test_unbind_speed_stack1 | 98.3744ms | 0.8571ms | 1.1668 KOps/s | 1.1670 KOps/s | |
| test_split | 98.3313ms | 1.2794ms | 781.5892 Ops/s | 656.5179 Ops/s | |
| test_chunk | 98.6563ms | 1.2169ms | 821.7447 Ops/s | 919.5290 Ops/s | |
| test_consolidate[False-None] | 4.0067ms | 3.8973ms | 256.5904 Ops/s | 256.3129 Ops/s | |
| test_consolidate[default-None] | 2.2712ms | 2.0657ms | 484.1012 Ops/s | 452.2193 Ops/s | |
| test_consolidate[reduce-overhead-None] | 2.0529ms | 1.9755ms | 506.2121 Ops/s | 470.8125 Ops/s | |
| test_consolidate_njt[False-None] | 0.1814s | 10.4820ms | 95.4013 Ops/s | 112.3211 Ops/s | |
| test_to[False-False-None] | 2.2266ms | 2.0791ms | 480.9703 Ops/s | 480.6035 Ops/s | |
| test_to[True-False-None] | 2.1776ms | 1.8991ms | 526.5755 Ops/s | 531.9751 Ops/s | |
| test_to[within-False-None] | 6.1503ms | 5.8746ms | 170.2241 Ops/s | 170.4826 Ops/s | |
| test_to[True-default-None] | 12.0961ms | 11.8682ms | 84.2588 Ops/s | 83.1961 Ops/s | |
| test_to_njt[False-False-None] | 8.9979ms | 8.6870ms | 115.1139 Ops/s | 116.3231 Ops/s | |
| test_to_njt[True-False-None] | 7.8328ms | 7.4902ms | 133.5082 Ops/s | 134.5819 Ops/s | |
| test_to_njt[within-False-None] | 17.1176ms | 16.5586ms | 60.3916 Ops/s | 60.3414 Ops/s | |
| test_creation[device0] | 0.3518ms | 0.1099ms | 9.0957 KOps/s | 8.9421 KOps/s | |
| test_creation_from_tensor | 0.3525ms | 0.1126ms | 8.8798 KOps/s | 8.9754 KOps/s | |
| test_add_one[memmap_tensor0] | 0.1682ms | 6.9902μs | 143.0571 KOps/s | 148.9746 KOps/s | |
| test_contiguous[memmap_tensor0] | 15.1010μs | 0.6973μs | 1.4341 MOps/s | 1.9921 MOps/s | |
| test_stack[memmap_tensor0] | 31.1300μs | 4.6664μs | 214.2997 KOps/s | 214.2838 KOps/s | |
| test_memmaptd_index | 1.1028ms | 0.2824ms | 3.5417 KOps/s | 3.4910 KOps/s | |
| test_memmaptd_index_astensor | 0.5376ms | 0.3750ms | 2.6665 KOps/s | 2.6322 KOps/s | |
| test_memmaptd_index_op | 0.8382ms | 0.6359ms | 1.5725 KOps/s | 1.5980 KOps/s | |
| test_serialize_model | 0.1353s | 0.1337s | 7.4770 Ops/s | 7.4874 Ops/s | |
| test_serialize_model_pickle | 1.3494s | 1.2113s | 0.8256 Ops/s | 0.8360 Ops/s | |
| test_serialize_weights | 0.3083s | 0.1577s | 6.3425 Ops/s | 7.5288 Ops/s | |
| test_serialize_weights_returnearly | 0.4038s | 68.4441ms | 14.6105 Ops/s | 19.4232 Ops/s | |
| test_serialize_weights_pickle | 1.3711s | 1.2229s | 0.8177 Ops/s | 0.8371 Ops/s | |
| test_reshape_pytree | 0.3495ms | 33.4207μs | 29.9216 KOps/s | 30.0233 KOps/s | |
| test_reshape_td | 71.0310μs | 38.5414μs | 25.9461 KOps/s | 25.9868 KOps/s | |
| test_view_pytree | 0.2318ms | 34.3357μs | 29.1242 KOps/s | 30.3719 KOps/s | |
| test_view_td | 91.7720μs | 47.8658μs | 20.8918 KOps/s | 21.5276 KOps/s | |
| test_unbind_pytree | 0.2409ms | 38.8746μs | 25.7237 KOps/s | 26.3224 KOps/s | |
| test_unbind_td | 0.1101ms | 48.3421μs | 20.6859 KOps/s | 20.4814 KOps/s | |
| test_split_pytree | 0.2534ms | 44.0046μs | 22.7249 KOps/s | 22.3095 KOps/s | |
| test_split_td | 0.1230ms | 65.6361μs | 15.2355 KOps/s | 15.4837 KOps/s | |
| test_add_pytree | 0.2493ms | 46.6216μs | 21.4493 KOps/s | 22.6480 KOps/s | |
| test_add_td | 99.4410μs | 57.6361μs | 17.3502 KOps/s | 18.4798 KOps/s | |
| test_compile_add_one_nested[tensordict-compile] | 0.3292ms | 0.1902ms | 5.2566 KOps/s | 5.4879 KOps/s | |
| test_compile_add_one_nested[tensordict-eager] | 0.3939ms | 0.1913ms | 5.2272 KOps/s | 5.1452 KOps/s | |
| test_compile_add_one_nested[pytree-compile] | 0.3103ms | 0.1520ms | 6.5770 KOps/s | 6.5165 KOps/s | |
| test_compile_add_one_nested[pytree-eager] | 0.4352ms | 0.1874ms | 5.3365 KOps/s | 5.3210 KOps/s | |
| test_compile_copy_nested[tensordict-compile] | 66.5810μs | 28.0086μs | 35.7034 KOps/s | 33.0834 KOps/s | |
| test_compile_copy_nested[tensordict-eager] | 80.0010μs | 52.8469μs | 18.9226 KOps/s | 19.1669 KOps/s | |
| test_compile_copy_nested[pytree-compile] | 0.2549ms | 14.4295μs | 69.3026 KOps/s | 66.0117 KOps/s | |
| test_compile_copy_nested[pytree-eager] | 0.3912ms | 76.4222μs | 13.0852 KOps/s | 13.0904 KOps/s | |
| test_compile_add_one_flat[tensordict-compile] | 0.2443ms | 0.2071ms | 4.8297 KOps/s | 4.7232 KOps/s | |
| test_compile_add_one_flat[tensordict-eager] | 0.3184ms | 0.2597ms | 3.8507 KOps/s | 3.8400 KOps/s | |
| test_compile_add_one_flat[tensorclass-compile] | 0.2121ms | 0.1538ms | 6.5035 KOps/s | 6.4301 KOps/s | |
| test_compile_add_one_flat[tensorclass-eager] | 0.1135ms | 71.5855μs | 13.9693 KOps/s | 14.1071 KOps/s | |
| test_compile_add_one_flat[pytree-compile] | 0.2555ms | 0.2013ms | 4.9676 KOps/s | 4.8171 KOps/s | |
| test_compile_add_one_flat[pytree-eager] | 0.7918ms | 0.5379ms | 1.8590 KOps/s | 1.8348 KOps/s | |
| test_compile_add_self_flat[tensordict-eager] | 0.4833ms | 0.3150ms | 3.1742 KOps/s | 3.1690 KOps/s | |
| test_compile_add_self_flat[tensordict-compile] | 0.2587ms | 0.2089ms | 4.7875 KOps/s | 4.6606 KOps/s | |
| test_compile_add_self_flat[tensorclass-eager] | 0.1288ms | 86.8762μs | 11.5106 KOps/s | 11.6286 KOps/s | |
| test_compile_add_self_flat[tensorclass-compile] | 0.3069ms | 0.1560ms | 6.4101 KOps/s | 6.1188 KOps/s | |
| test_compile_add_self_flat[pytree-eager] | 0.6429ms | 0.4437ms | 2.2538 KOps/s | 2.2411 KOps/s | |
| test_compile_add_self_flat[pytree-compile] | 0.2431ms | 0.2028ms | 4.9315 KOps/s | 4.6668 KOps/s | |
| test_compile_copy_flat[tensordict-compile] | 0.3462ms | 24.2457μs | 41.2445 KOps/s | 38.5138 KOps/s | |
| test_compile_copy_flat[tensordict-eager] | 0.1032ms | 42.7428μs | 23.3957 KOps/s | 23.6243 KOps/s | |
| test_compile_copy_flat[pytree-compile] | 0.1784ms | 20.5209μs | 48.7307 KOps/s | 45.6227 KOps/s | |
| test_compile_copy_flat[pytree-eager] | 0.3651ms | 70.5347μs | 14.1774 KOps/s | 14.2737 KOps/s | |
| test_compile_assign_and_add[tensordict-compile] | 2.0762ms | 0.2115ms | 4.7279 KOps/s | 4.5827 KOps/s | |
| test_compile_assign_and_add[tensordict-eager] | 3.4339ms | 3.3246ms | 300.7871 Ops/s | 304.2215 Ops/s | |
| test_compile_assign_and_add[pytree-compile] | 2.0711ms | 0.2100ms | 4.7617 KOps/s | 4.7509 KOps/s | |
| test_compile_assign_and_add[pytree-eager] | 3.0485ms | 2.8344ms | 352.8039 Ops/s | 353.6499 Ops/s | |
| test_compile_indexing[tensor-tensordict-compile] | 0.1887ms | 0.1440ms | 6.9452 KOps/s | 6.7213 KOps/s | |
| test_compile_indexing[tensor-tensordict-eager] | 0.3014ms | 65.5699μs | 15.2509 KOps/s | 15.1132 KOps/s | |
| test_compile_indexing[tensor-tensorclass-compile] | 0.1960ms | 0.1368ms | 7.3112 KOps/s | 7.3095 KOps/s | |
| test_compile_indexing[tensor-tensorclass-eager] | 0.2530ms | 46.4412μs | 21.5326 KOps/s | 21.6254 KOps/s | |
| test_compile_indexing[tensor-pytree-compile] | 0.1872ms | 0.1376ms | 7.2669 KOps/s | 7.2742 KOps/s | |
| test_compile_indexing[tensor-pytree-eager] | 0.2812ms | 46.6987μs | 21.4139 KOps/s | 21.5536 KOps/s | |
| test_compile_indexing[slice-tensordict-compile] | 0.1526ms | 87.5884μs | 11.4170 KOps/s | 11.3527 KOps/s | |
| test_compile_indexing[slice-tensordict-eager] | 0.2077ms | 27.2419μs | 36.7081 KOps/s | 36.5620 KOps/s | |
| test_compile_indexing[slice-tensorclass-compile] | 1.9633ms | 82.5303μs | 12.1168 KOps/s | 12.1009 KOps/s | |
| test_compile_indexing[slice-tensorclass-eager] | 0.2374ms | 24.1667μs | 41.3792 KOps/s | 41.9325 KOps/s | |
| test_compile_indexing[slice-pytree-compile] | 0.2667ms | 82.6716μs | 12.0961 KOps/s | 12.0385 KOps/s | |
| test_compile_indexing[slice-pytree-eager] | 0.2303ms | 23.9110μs | 41.8218 KOps/s | 42.0775 KOps/s | |
| test_compile_indexing[int-tensordict-compile] | 0.1723ms | 89.3565μs | 11.1911 KOps/s | 11.1778 KOps/s | |
| test_compile_indexing[int-tensordict-eager] | 0.2649ms | 26.9242μs | 37.1413 KOps/s | 37.2007 KOps/s | |
| test_compile_indexing[int-tensorclass-compile] | 0.1446ms | 82.3194μs | 12.1478 KOps/s | 12.1451 KOps/s | |
| test_compile_indexing[int-tensorclass-eager] | 0.2382ms | 23.9704μs | 41.7181 KOps/s | 42.3907 KOps/s | |
| test_compile_indexing[int-pytree-compile] | 0.1425ms | 82.2179μs | 12.1628 KOps/s | 12.1250 KOps/s | |
| test_compile_indexing[int-pytree-eager] | 0.2463ms | 24.0109μs | 41.6477 KOps/s | 42.0233 KOps/s | |
| test_mod_add[eager] | 0.1081ms | 52.3306μs | 19.1093 KOps/s | 19.4862 KOps/s | |
| test_mod_add[compile] | 0.2282ms | 0.1532ms | 6.5281 KOps/s | 6.5676 KOps/s | |
| test_mod_add[compile-overhead] | 0.2908ms | 0.1996ms | 5.0100 KOps/s | 5.0222 KOps/s | |
| test_mod_wrap[eager] | 0.3786ms | 0.3082ms | 3.2445 KOps/s | 3.2752 KOps/s | |
| test_mod_wrap[compile] | 0.4753ms | 0.4033ms | 2.4794 KOps/s | 2.3714 KOps/s | |
| test_mod_wrap[compile-overhead] | 7.6735ms | 3.9805ms | 251.2277 Ops/s | 264.8440 Ops/s | |
| test_mod_wrap_and_backward[eager] | 1.7205ms | 1.5652ms | 638.8945 Ops/s | 639.1920 Ops/s | |
| test_mod_wrap_and_backward[compile] | 1.8362ms | 1.6292ms | 613.8074 Ops/s | 579.7397 Ops/s | |
| test_mod_wrap_and_backward[compile-overhead] | 1.3512ms | 0.9903ms | 1.0098 KOps/s | 900.5606 Ops/s | |
| test_seq_add[eager] | 0.2206ms | 0.1587ms | 6.3029 KOps/s | 6.1008 KOps/s | |
| test_seq_add[compile] | 0.2561ms | 0.1629ms | 6.1393 KOps/s | 6.1869 KOps/s | |
| test_seq_add[compile-overhead] | 0.2608ms | 0.2068ms | 4.8352 KOps/s | 4.9133 KOps/s | |
| test_seq_wrap[eager] | 0.6590ms | 0.5445ms | 1.8364 KOps/s | 1.8537 KOps/s | |
| test_seq_wrap[compile] | 0.5264ms | 0.4231ms | 2.3638 KOps/s | 2.3923 KOps/s | |
| test_seq_wrap[compile-overhead] | 0.4121ms | 0.3243ms | 3.0831 KOps/s | 3.1631 KOps/s | |
| test_func_call_runtime[False-eager] | 0.9939ms | 0.8857ms | 1.1290 KOps/s | 1.1357 KOps/s | |
| test_func_call_runtime[False-compile] | 1.2022ms | 0.9550ms | 1.0472 KOps/s | 1.0592 KOps/s | |
| test_func_call_runtime[False-compile-overhead] | 0.5576ms | 0.4990ms | 2.0038 KOps/s | 1.9909 KOps/s | |
| test_func_call_runtime[True-eager] | 1.4621ms | 1.1333ms | 882.3825 Ops/s | 887.1753 Ops/s | |
| test_func_call_runtime[True-compile] | 1.0939ms | 0.9683ms | 1.0327 KOps/s | 1.0316 KOps/s | |
| test_func_call_runtime[True-compile-overhead] | 0.6122ms | 0.5222ms | 1.9150 KOps/s | 1.9044 KOps/s | |
| test_func_call_cm_runtime[False-eager] | 0.9449ms | 0.8724ms | 1.1462 KOps/s | 1.0615 KOps/s | |
| test_func_call_cm_runtime[False-compile] | 1.1055ms | 0.9508ms | 1.0518 KOps/s | 1.0467 KOps/s | |
| test_func_call_cm_runtime[False-compile-overhead] | 0.5580ms | 0.5033ms | 1.9869 KOps/s | 1.9811 KOps/s | |
| test_func_call_cm_runtime[True-eager] | 1.3840ms | 1.2761ms | 783.6468 Ops/s | 782.0963 Ops/s | |
| test_func_call_cm_runtime[True-compile] | 1.3871ms | 1.0096ms | 990.4652 Ops/s | 997.5946 Ops/s | |
| test_func_call_cm_runtime[True-compile-overhead] | 0.5969ms | 0.5561ms | 1.7983 KOps/s | 1.8000 KOps/s | |
| test_vmap_func_call_cm_runtime[eager] | 2.8817ms | 2.3820ms | 419.8119 Ops/s | 417.9607 Ops/s | |
| test_vmap_func_call_cm_runtime[compile] | 1.0984ms | 1.0083ms | 991.7944 Ops/s | 973.1936 Ops/s | |
| test_vmap_func_call_cm_runtime[compile-overhead] | 0.5897ms | 0.5503ms | 1.8170 KOps/s | 1.8079 KOps/s | |
| test_distributed | 0.5587ms | 0.1548ms | 6.4610 KOps/s | 6.3242 KOps/s | |
| test_tdmodule | 0.3793ms | 28.4911μs | 35.0986 KOps/s | 34.2453 KOps/s | |
| test_tdmodule_dispatch | 78.9120μs | 48.0498μs | 20.8117 KOps/s | 20.5821 KOps/s | |
| test_tdseq | 48.8410μs | 27.1559μs | 36.8244 KOps/s | 37.0513 KOps/s | |
| test_tdseq_dispatch | 71.2710μs | 49.9358μs | 20.0257 KOps/s | 19.7039 KOps/s | |
| test_instantiation_functorch | 2.1836ms | 2.0952ms | 477.2725 Ops/s | 480.7769 Ops/s | |
| test_exec_functorch | 0.2327ms | 0.1853ms | 5.3961 KOps/s | 5.3796 KOps/s | |
| test_exec_functional_call | 0.2102ms | 0.1637ms | 6.1069 KOps/s | 6.0510 KOps/s | |
| test_exec_td_decorator | 0.4448ms | 0.2432ms | 4.1115 KOps/s | 4.1539 KOps/s | |
| test_vmap_mlp_speed_decorator[True-True] | 0.9709ms | 0.8111ms | 1.2329 KOps/s | 1.2191 KOps/s | |
| test_vmap_mlp_speed_decorator[True-False] | 0.9469ms | 0.8102ms | 1.2342 KOps/s | 1.2262 KOps/s | |
| test_vmap_mlp_speed_decorator[False-True] | 0.8612ms | 0.6944ms | 1.4401 KOps/s | 1.4347 KOps/s | |
| test_vmap_mlp_speed_decorator[False-False] | 0.8410ms | 0.6930ms | 1.4430 KOps/s | 1.4397 KOps/s | |
| test_vmap_transformer_speed_decorator[True-True] | 21.4307ms | 20.7313ms | 48.2362 Ops/s | 48.0491 Ops/s | |
| test_vmap_transformer_speed_decorator[True-False] | 20.8642ms | 20.7312ms | 48.2365 Ops/s | 47.9757 Ops/s | |
| test_vmap_transformer_speed_decorator[False-True] | 20.6732ms | 20.5308ms | 48.7072 Ops/s | 48.4882 Ops/s | |
| test_vmap_transformer_speed_decorator[False-False] | 20.6646ms | 20.5269ms | 48.7165 Ops/s | 48.4786 Ops/s | |
| test_to_module_speed[True] | 1.6223ms | 1.4862ms | 672.8411 Ops/s | 674.9048 Ops/s | |
| test_to_module_speed[False] | 1.5484ms | 1.4476ms | 690.7770 Ops/s | 686.1443 Ops/s | |
| test_tc_init | 79.8710μs | 51.6989μs | 19.3428 KOps/s | 19.1220 KOps/s | |
| test_tc_init_tensor_only | 35.1810μs | 14.9051μs | 67.0911 KOps/s | 66.1858 KOps/s | |
| test_tc_init_nested | 0.1592ms | 0.1030ms | 9.7120 KOps/s | 9.5135 KOps/s | |
| test_tc_first_layer_tensor | 14.3900μs | 1.7943μs | 557.3245 KOps/s | 585.7536 KOps/s | |
| test_tc_first_layer_tensor_only | 4.4329μs | 0.6991μs | 1.4303 MOps/s | 1.4602 MOps/s | |
| test_tc_first_layer_tensor_set | 38.9710μs | 4.1966μs | 238.2883 KOps/s | 238.3849 KOps/s | |
| test_tc_first_layer_tensor_only_set | 24.2800μs | 3.1536μs | 317.0935 KOps/s | 313.5868 KOps/s | |
| test_tc_first_layer_nontensor | 61.0220μs | 5.9188μs | 168.9533 KOps/s | 170.4199 KOps/s | |
| test_tc_second_layer_tensor | 32.4400μs | 4.3572μs | 229.5050 KOps/s | 235.2589 KOps/s | |
| test_tc_second_layer_nontensor | 33.5710μs | 8.4796μs | 117.9296 KOps/s | 118.3397 KOps/s | |
| test_unbind | 0.2554s | 15.4433ms | 64.7528 Ops/s | 59.9356 Ops/s | |
| test_full_like | 4.6249ms | 4.4129ms | 226.6105 Ops/s | 226.6057 Ops/s | |
| test_zeros_like | 4.5695ms | 4.3872ms | 227.9349 Ops/s | 227.3738 Ops/s | |
| test_ones_like | 4.5205ms | 4.3889ms | 227.8451 Ops/s | 226.6025 Ops/s | |
| test_clone | 7.1661ms | 6.6713ms | 149.8965 Ops/s | 151.8467 Ops/s | |
| test_squeeze | 63.5610μs | 14.3908μs | 69.4890 KOps/s | 65.3442 KOps/s | |
| test_unsqueeze | 0.1637ms | 0.1097ms | 9.1189 KOps/s | 8.6063 KOps/s | |
| test_split | 0.2736ms | 0.1885ms | 5.3057 KOps/s | 5.1969 KOps/s | |
| test_permute | 0.2694ms | 0.2095ms | 4.7725 KOps/s | 4.7397 KOps/s | |
| test_stack | 52.3651ms | 51.6961ms | 19.3438 Ops/s | 19.2884 Ops/s | |
| test_cat | 43.4887ms | 43.2620ms | 23.1150 Ops/s | 19.2816 Ops/s |
|
| Name | Max | Mean | Ops | Ops on Repo HEAD
|
Change |
|---|---|---|---|---|---|
| test_plain_set_nested | 36.1810μs | 14.3795μs | 69.5434 KOps/s | 69.2776 KOps/s | |
| test_plain_set_stack_nested | 41.2410μs | 14.8484μs | 67.3473 KOps/s | 67.8986 KOps/s | |
| test_plain_set_nested_inplace | 39.2210μs | 16.1619μs | 61.8738 KOps/s | 61.9207 KOps/s | |
| test_plain_set_stack_nested_inplace | 41.8610μs | 16.0730μs | 62.2162 KOps/s | 62.5696 KOps/s | |
| test_items | 51.9620μs | 5.3954μs | 185.3445 KOps/s | 179.1740 KOps/s | |
| test_items_nested | 0.5725ms | 0.5217ms | 1.9168 KOps/s | 1.9162 KOps/s | |
| test_items_nested_locked | 0.6553ms | 0.5214ms | 1.9181 KOps/s | 1.8993 KOps/s | |
| test_items_nested_leaf | 0.1381ms | 91.2291μs | 10.9614 KOps/s | 10.8965 KOps/s | |
| test_items_stack_nested | 0.5779ms | 0.5252ms | 1.9042 KOps/s | 1.9219 KOps/s | |
| test_items_stack_nested_leaf | 0.1292ms | 92.5309μs | 10.8072 KOps/s | 10.6837 KOps/s | |
| test_items_stack_nested_locked | 0.5735ms | 0.5243ms | 1.9074 KOps/s | 1.8980 KOps/s | |
| test_keys | 29.6510μs | 4.1219μs | 242.6040 KOps/s | 243.5408 KOps/s | |
| test_keys_nested | 0.1694ms | 0.1174ms | 8.5208 KOps/s | 8.4972 KOps/s | |
| test_keys_nested_locked | 2.3297ms | 0.1273ms | 7.8553 KOps/s | 7.8682 KOps/s | |
| test_keys_nested_leaf | 0.1409ms | 0.1093ms | 9.1477 KOps/s | 9.2612 KOps/s | |
| test_keys_stack_nested | 0.1739ms | 0.1192ms | 8.3893 KOps/s | 8.4718 KOps/s | |
| test_keys_stack_nested_leaf | 0.1512ms | 0.1093ms | 9.1494 KOps/s | 9.2075 KOps/s | |
| test_keys_stack_nested_locked | 0.1803ms | 0.1265ms | 7.9080 KOps/s | 7.8672 KOps/s | |
| test_values | 6.0622μs | 1.0059μs | 994.1492 KOps/s | 1.0014 MOps/s | |
| test_values_nested | 77.6320μs | 47.3339μs | 21.1265 KOps/s | 21.4440 KOps/s | |
| test_values_nested_locked | 0.5149ms | 49.6030μs | 20.1601 KOps/s | 20.0786 KOps/s | |
| test_values_nested_leaf | 81.9820μs | 53.7397μs | 18.6082 KOps/s | 18.8595 KOps/s | |
| test_values_stack_nested | 81.1120μs | 47.3010μs | 21.1412 KOps/s | 21.3656 KOps/s | |
| test_values_stack_nested_leaf | 79.8320μs | 54.1780μs | 18.4577 KOps/s | 18.8657 KOps/s | |
| test_values_stack_nested_locked | 89.1420μs | 50.2610μs | 19.8962 KOps/s | 20.0436 KOps/s | |
| test_membership | 4.8567μs | 0.8202μs | 1.2192 MOps/s | 1.2368 MOps/s | |
| test_membership_nested | 25.9610μs | 2.9322μs | 341.0408 KOps/s | 336.2573 KOps/s | |
| test_membership_nested_leaf | 43.3610μs | 2.8718μs | 348.2144 KOps/s | 335.5470 KOps/s | |
| test_membership_stacked_nested | 38.7110μs | 2.9757μs | 336.0590 KOps/s | 332.0329 KOps/s | |
| test_membership_stacked_nested_leaf | 23.4400μs | 2.9747μs | 336.1681 KOps/s | 335.7393 KOps/s | |
| test_membership_nested_last | 28.7800μs | 4.3504μs | 229.8628 KOps/s | 228.6517 KOps/s | |
| test_membership_nested_leaf_last | 36.6010μs | 4.3685μs | 228.9124 KOps/s | 228.3957 KOps/s | |
| test_membership_stacked_nested_last | 34.6510μs | 4.3522μs | 229.7688 KOps/s | 228.5418 KOps/s | |
| test_membership_stacked_nested_leaf_last | 30.3310μs | 4.3426μs | 230.2774 KOps/s | 228.5319 KOps/s | |
| test_nested_getleaf | 48.7210μs | 20.3653μs | 49.1030 KOps/s | 48.5282 KOps/s | |
| test_nested_get | 56.6310μs | 19.3829μs | 51.5919 KOps/s | 51.0501 KOps/s | |
| test_stacked_getleaf | 45.3910μs | 20.1031μs | 49.7435 KOps/s | 48.9794 KOps/s | |
| test_stacked_get | 48.0020μs | 19.4175μs | 51.5000 KOps/s | 50.7677 KOps/s | |
| test_nested_getitemleaf | 41.3310μs | 20.8528μs | 47.9551 KOps/s | 46.5943 KOps/s | |
| test_nested_getitem | 43.9410μs | 19.4467μs | 51.4227 KOps/s | 49.7401 KOps/s | |
| test_stacked_getitemleaf | 69.3910μs | 20.8857μs | 47.8797 KOps/s | 47.3343 KOps/s | |
| test_stacked_getitem | 46.5110μs | 19.7796μs | 50.5572 KOps/s | 49.3673 KOps/s | |
| test_lock_nested | 4.5375ms | 0.4579ms | 2.1840 KOps/s | 2.1716 KOps/s | |
| test_lock_stack_nested | 0.5352ms | 0.4570ms | 2.1880 KOps/s | 2.1642 KOps/s | |
| test_unlock_nested | 0.4747ms | 0.3700ms | 2.7024 KOps/s | 2.7065 KOps/s | |
| test_unlock_stack_nested | 0.4493ms | 0.3648ms | 2.7416 KOps/s | 2.6916 KOps/s | |
| test_flatten_speed | 0.1532ms | 0.1158ms | 8.6356 KOps/s | 8.5553 KOps/s | |
| test_unflatten_speed | 0.6305ms | 0.5725ms | 1.7466 KOps/s | 1.7688 KOps/s | |
| test_common_ops | 0.8606ms | 0.7143ms | 1.4000 KOps/s | 1.3776 KOps/s | |
| test_creation | 0.1182ms | 2.3790μs | 420.3397 KOps/s | 425.1122 KOps/s | |
| test_creation_empty | 29.3910μs | 8.4299μs | 118.6251 KOps/s | 120.0296 KOps/s | |
| test_creation_nested_1 | 36.1610μs | 11.3015μs | 88.4841 KOps/s | 89.0684 KOps/s | |
| test_creation_nested_2 | 36.8510μs | 14.9239μs | 67.0067 KOps/s | 68.2339 KOps/s | |
| test_clone | 36.1510μs | 12.7762μs | 78.2707 KOps/s | 77.3366 KOps/s | |
| test_getitem[int] | 1.2315ms | 13.7295μs | 72.8359 KOps/s | 74.1076 KOps/s | |
| test_getitem[slice_int] | 0.1377ms | 24.0169μs | 41.6373 KOps/s | 41.8487 KOps/s | |
| test_getitem[range] | 0.1705ms | 57.1377μs | 17.5016 KOps/s | 17.3074 KOps/s | |
| test_getitem[tuple] | 0.1455ms | 23.7084μs | 42.1792 KOps/s | 43.0130 KOps/s | |
| test_getitem[list] | 0.1781ms | 50.7813μs | 19.6923 KOps/s | 18.9536 KOps/s | |
| test_setitem_dim[int] | 52.6510μs | 23.3394μs | 42.8460 KOps/s | 38.7392 KOps/s | |
| test_setitem_dim[slice_int] | 66.2220μs | 43.7040μs | 22.8812 KOps/s | 21.7690 KOps/s | |
| test_setitem_dim[range] | 0.1247ms | 83.2574μs | 12.0109 KOps/s | 12.0645 KOps/s | |
| test_setitem_dim[tuple] | 63.7910μs | 41.3703μs | 24.1719 KOps/s | 24.6026 KOps/s | |
| test_setitem | 54.1410μs | 17.3158μs | 57.7507 KOps/s | 56.5730 KOps/s | |
| test_set | 80.4920μs | 16.4643μs | 60.7374 KOps/s | 56.4426 KOps/s | |
| test_set_shared | 0.6373ms | 0.2035ms | 4.9151 KOps/s | 4.7564 KOps/s | |
| test_update | 0.3022ms | 21.7056μs | 46.0711 KOps/s | 43.4274 KOps/s | |
| test_update_nested | 67.7110μs | 33.5679μs | 29.7903 KOps/s | 28.2043 KOps/s | |
| test_update__nested | 0.4976ms | 33.0038μs | 30.2995 KOps/s | 27.6524 KOps/s | |
| test_set_nested | 53.9510μs | 18.2351μs | 54.8393 KOps/s | 48.2878 KOps/s | |
| test_set_nested_new | 61.7710μs | 23.4310μs | 42.6785 KOps/s | 38.5015 KOps/s | |
| test_select | 70.9410μs | 41.2462μs | 24.2446 KOps/s | 22.6876 KOps/s | |
| test_select_nested | 0.1054ms | 70.6616μs | 14.1520 KOps/s | 14.0222 KOps/s | |
| test_exclude_nested | 0.1290ms | 91.9671μs | 10.8735 KOps/s | 10.8720 KOps/s | |
| test_empty[True] | 0.6319ms | 0.4237ms | 2.3603 KOps/s | 2.3510 KOps/s | |
| test_empty[False] | 7.3253μs | 1.2644μs | 790.8790 KOps/s | 789.8249 KOps/s | |
| test_to | 99.8120μs | 70.1353μs | 14.2582 KOps/s | 13.9924 KOps/s | |
| test_to_nonblocking | 0.1076ms | 63.5622μs | 15.7326 KOps/s | 15.7164 KOps/s | |
| test_unbind_speed | 0.3442ms | 0.3148ms | 3.1763 KOps/s | 3.1318 KOps/s | |
| test_unbind_speed_stack0 | 0.3640ms | 0.3136ms | 3.1890 KOps/s | 3.2595 KOps/s | |
| test_unbind_speed_stack1 | 98.3470ms | 1.0277ms | 973.0012 Ops/s | 1.1837 KOps/s | |
| test_split | 1.1494ms | 1.1092ms | 901.5705 Ops/s | 669.2272 Ops/s | |
| test_chunk | 98.3701ms | 1.1674ms | 856.6320 Ops/s | 945.8171 Ops/s | |
| test_consolidate[False-None] | 3.8441ms | 3.7360ms | 267.6647 Ops/s | 267.3755 Ops/s | |
| test_consolidate[default-None] | 2.0332ms | 1.9586ms | 510.5656 Ops/s | 486.4697 Ops/s | |
| test_consolidate[reduce-overhead-None] | 2.0017ms | 1.9019ms | 525.7817 Ops/s | 498.1231 Ops/s | |
| test_consolidate_njt[False-None] | 8.8328ms | 8.6469ms | 115.6480 Ops/s | 117.4154 Ops/s | |
| test_to[False-False-None] | 2.1387ms | 2.0525ms | 487.2061 Ops/s | 491.8994 Ops/s | |
| test_to[True-False-None] | 2.0935ms | 1.8256ms | 547.7632 Ops/s | 557.9573 Ops/s | |
| test_to[within-False-None] | 5.7121ms | 5.5859ms | 179.0234 Ops/s | 179.9524 Ops/s | |
| test_to[True-default-None] | 11.8545ms | 11.6691ms | 85.6962 Ops/s | 85.6138 Ops/s | |
| test_to_njt[False-False-None] | 8.5395ms | 8.4165ms | 118.8141 Ops/s | 118.6959 Ops/s | |
| test_to_njt[True-False-None] | 7.3226ms | 7.1869ms | 139.1429 Ops/s | 138.2658 Ops/s | |
| test_to_njt[within-False-None] | 16.0376ms | 15.9720ms | 62.6097 Ops/s | 62.4341 Ops/s | |
| test_creation[device0] | 0.5468ms | 0.1096ms | 9.1221 KOps/s | 9.3401 KOps/s | |
| test_creation_from_tensor | 0.4778ms | 0.1108ms | 9.0279 KOps/s | 9.0710 KOps/s | |
| test_add_one[memmap_tensor0] | 0.1386ms | 6.4173μs | 155.8295 KOps/s | 151.8855 KOps/s | |
| test_contiguous[memmap_tensor0] | 25.5010μs | 0.6275μs | 1.5937 MOps/s | 2.0951 MOps/s | |
| test_stack[memmap_tensor0] | 38.1010μs | 4.4386μs | 225.2971 KOps/s | 220.5517 KOps/s | |
| test_memmaptd_index | 1.1429ms | 0.2687ms | 3.7214 KOps/s | 3.7297 KOps/s | |
| test_memmaptd_index_astensor | 0.5187ms | 0.3562ms | 2.8071 KOps/s | 2.8044 KOps/s | |
| test_memmaptd_index_op | 0.8187ms | 0.6212ms | 1.6097 KOps/s | 1.6686 KOps/s | |
| test_serialize_model | 0.1344s | 0.1331s | 7.5139 Ops/s | 7.4946 Ops/s | |
| test_serialize_model_pickle | 1.9912s | 1.3209s | 0.7571 Ops/s | 0.8256 Ops/s | |
| test_serialize_weights | 0.1345s | 0.1330s | 7.5164 Ops/s | 7.5427 Ops/s | |
| test_serialize_weights_returnearly | 0.4118s | 68.4682ms | 14.6053 Ops/s | 10.3058 Ops/s | |
| test_serialize_weights_pickle | 1.3633s | 1.1895s | 0.8407 Ops/s | 0.8140 Ops/s | |
| test_reshape_pytree | 0.3666ms | 31.9193μs | 31.3290 KOps/s | 30.5308 KOps/s | |
| test_reshape_td | 65.2120μs | 37.1020μs | 26.9527 KOps/s | 25.7193 KOps/s | |
| test_view_pytree | 0.2245ms | 31.7507μs | 31.4954 KOps/s | 31.4115 KOps/s | |
| test_view_td | 78.4420μs | 44.4169μs | 22.5140 KOps/s | 22.8264 KOps/s | |
| test_unbind_pytree | 0.2553ms | 36.7668μs | 27.1985 KOps/s | 27.0996 KOps/s | |
| test_unbind_td | 95.2820μs | 47.1380μs | 21.2143 KOps/s | 21.2393 KOps/s | |
| test_split_pytree | 0.2466ms | 42.3784μs | 23.5969 KOps/s | 23.3425 KOps/s | |
| test_split_td | 0.2226ms | 63.6161μs | 15.7193 KOps/s | 15.6853 KOps/s | |
| test_add_pytree | 0.2335ms | 43.3816μs | 23.0512 KOps/s | 22.6806 KOps/s | |
| test_add_td | 91.7320μs | 52.4306μs | 19.0728 KOps/s | 18.1431 KOps/s | |
| test_compile_add_one_nested[tensordict-compile] | 0.2616ms | 0.1785ms | 5.6008 KOps/s | 5.2942 KOps/s | |
| test_compile_add_one_nested[tensordict-eager] | 0.2676ms | 0.1872ms | 5.3415 KOps/s | 5.3625 KOps/s | |
| test_compile_add_one_nested[pytree-compile] | 0.2079ms | 0.1487ms | 6.7243 KOps/s | 6.6502 KOps/s | |
| test_compile_add_one_nested[pytree-eager] | 0.4356ms | 0.1859ms | 5.3798 KOps/s | 5.4883 KOps/s | |
| test_compile_copy_nested[tensordict-compile] | 0.1119ms | 27.1330μs | 36.8554 KOps/s | 34.1316 KOps/s | |
| test_compile_copy_nested[tensordict-eager] | 90.7520μs | 50.1801μs | 19.9282 KOps/s | 19.6275 KOps/s | |
| test_compile_copy_nested[pytree-compile] | 0.2226ms | 14.6970μs | 68.0409 KOps/s | 70.9228 KOps/s | |
| test_compile_copy_nested[pytree-eager] | 0.3938ms | 72.0625μs | 13.8768 KOps/s | 13.7175 KOps/s | |
| test_compile_add_one_flat[tensordict-compile] | 0.4765ms | 0.2058ms | 4.8594 KOps/s | 4.6243 KOps/s | |
| test_compile_add_one_flat[tensordict-eager] | 0.3140ms | 0.2603ms | 3.8413 KOps/s | 3.9197 KOps/s | |
| test_compile_add_one_flat[tensorclass-compile] | 0.2331ms | 0.1497ms | 6.6785 KOps/s | 6.5390 KOps/s | |
| test_compile_add_one_flat[tensorclass-eager] | 0.1113ms | 69.8865μs | 14.3089 KOps/s | 14.4734 KOps/s | |
| test_compile_add_one_flat[pytree-compile] | 0.2504ms | 0.1999ms | 5.0023 KOps/s | 4.6723 KOps/s | |
| test_compile_add_one_flat[pytree-eager] | 0.8248ms | 0.5490ms | 1.8214 KOps/s | 1.8820 KOps/s | |
| test_compile_add_self_flat[tensordict-eager] | 0.3827ms | 0.3134ms | 3.1905 KOps/s | 3.2575 KOps/s | |
| test_compile_add_self_flat[tensordict-compile] | 0.2980ms | 0.2069ms | 4.8341 KOps/s | 4.7509 KOps/s | |
| test_compile_add_self_flat[tensorclass-eager] | 0.1487ms | 85.0529μs | 11.7574 KOps/s | 11.9066 KOps/s | |
| test_compile_add_self_flat[tensorclass-compile] | 0.2260ms | 0.1521ms | 6.5764 KOps/s | 6.2770 KOps/s | |
| test_compile_add_self_flat[pytree-eager] | 0.7129ms | 0.4519ms | 2.2129 KOps/s | 2.2234 KOps/s | |
| test_compile_add_self_flat[pytree-compile] | 0.4262ms | 0.2019ms | 4.9527 KOps/s | 4.7392 KOps/s | |
| test_compile_copy_flat[tensordict-compile] | 84.8120μs | 24.7381μs | 40.4235 KOps/s | 39.9665 KOps/s | |
| test_compile_copy_flat[tensordict-eager] | 81.2920μs | 40.2774μs | 24.8278 KOps/s | 24.9779 KOps/s | |
| test_compile_copy_flat[pytree-compile] | 0.2117ms | 21.4821μs | 46.5504 KOps/s | 48.0703 KOps/s | |
| test_compile_copy_flat[pytree-eager] | 0.3720ms | 67.0117μs | 14.9228 KOps/s | 14.7609 KOps/s | |
| test_compile_assign_and_add[tensordict-compile] | 2.0200ms | 0.2216ms | 4.5118 KOps/s | 4.7465 KOps/s | |
| test_compile_assign_and_add[tensordict-eager] | 3.4784ms | 3.3002ms | 303.0100 Ops/s | 308.3063 Ops/s | |
| test_compile_assign_and_add[pytree-compile] | 1.9927ms | 0.2063ms | 4.8463 KOps/s | 4.8542 KOps/s | |
| test_compile_assign_and_add[pytree-eager] | 3.0013ms | 2.8741ms | 347.9342 Ops/s | 351.5362 Ops/s | |
| test_compile_indexing[tensor-tensordict-compile] | 0.2625ms | 0.1400ms | 7.1426 KOps/s | 7.0176 KOps/s | |
| test_compile_indexing[tensor-tensordict-eager] | 0.2994ms | 64.9482μs | 15.3969 KOps/s | 14.9114 KOps/s | |
| test_compile_indexing[tensor-tensorclass-compile] | 0.2499ms | 0.1342ms | 7.4491 KOps/s | 7.2595 KOps/s | |
| test_compile_indexing[tensor-tensorclass-eager] | 0.2520ms | 48.1992μs | 20.7472 KOps/s | 22.3252 KOps/s | |
| test_compile_indexing[tensor-pytree-compile] | 0.1839ms | 0.1371ms | 7.2945 KOps/s | 7.1189 KOps/s | |
| test_compile_indexing[tensor-pytree-eager] | 0.2636ms | 47.8578μs | 20.8952 KOps/s | 21.9743 KOps/s | |
| test_compile_indexing[slice-tensordict-compile] | 0.1792ms | 88.0008μs | 11.3635 KOps/s | 11.2292 KOps/s | |
| test_compile_indexing[slice-tensordict-eager] | 0.2097ms | 27.2925μs | 36.6401 KOps/s | 38.3256 KOps/s | |
| test_compile_indexing[slice-tensorclass-compile] | 0.2014ms | 79.3674μs | 12.5996 KOps/s | 11.7864 KOps/s | |
| test_compile_indexing[slice-tensorclass-eager] | 0.2453ms | 22.7666μs | 43.9240 KOps/s | 43.4666 KOps/s | |
| test_compile_indexing[slice-pytree-compile] | 0.1227ms | 81.0244μs | 12.3420 KOps/s | 11.7209 KOps/s | |
| test_compile_indexing[slice-pytree-eager] | 0.2433ms | 22.6895μs | 44.0733 KOps/s | 43.5381 KOps/s | |
| test_compile_indexing[int-tensordict-compile] | 0.1459ms | 86.6908μs | 11.5353 KOps/s | 11.1676 KOps/s | |
| test_compile_indexing[int-tensordict-eager] | 0.2945ms | 26.0689μs | 38.3598 KOps/s | 38.7209 KOps/s | |
| test_compile_indexing[int-tensorclass-compile] | 0.1275ms | 83.3465μs | 11.9981 KOps/s | 12.2813 KOps/s | |
| test_compile_indexing[int-tensorclass-eager] | 0.2430ms | 22.7700μs | 43.9174 KOps/s | 43.5914 KOps/s | |
| test_compile_indexing[int-pytree-compile] | 0.1258ms | 83.4943μs | 11.9769 KOps/s | 12.2675 KOps/s | |
| test_compile_indexing[int-pytree-eager] | 0.2623ms | 22.6882μs | 44.0758 KOps/s | 43.7336 KOps/s | |
| test_mod_add[eager] | 86.5920μs | 51.0542μs | 19.5870 KOps/s | 20.0696 KOps/s | |
| test_mod_add[compile] | 0.2729ms | 0.1473ms | 6.7879 KOps/s | 6.7258 KOps/s | |
| test_mod_add[compile-overhead] | 0.2812ms | 0.1936ms | 5.1644 KOps/s | 5.1573 KOps/s | |
| test_mod_wrap[eager] | 0.3898ms | 0.3070ms | 3.2574 KOps/s | 3.2857 KOps/s | |
| test_mod_wrap[compile] | 0.5112ms | 0.4083ms | 2.4492 KOps/s | 2.4033 KOps/s | |
| test_mod_wrap[compile-overhead] | 7.4023ms | 3.8857ms | 257.3567 Ops/s | 252.8893 Ops/s | |
| test_mod_wrap_and_backward[eager] | 1.7815ms | 1.5645ms | 639.1840 Ops/s | 637.2718 Ops/s | |
| test_mod_wrap_and_backward[compile] | 1.8114ms | 1.5954ms | 626.7842 Ops/s | 628.2634 Ops/s | |
| test_mod_wrap_and_backward[compile-overhead] | 1.3240ms | 0.9749ms | 1.0257 KOps/s | 914.6850 Ops/s | |
| test_seq_add[eager] | 0.2137ms | 0.1560ms | 6.4085 KOps/s | 6.2443 KOps/s | |
| test_seq_add[compile] | 0.2231ms | 0.1571ms | 6.3648 KOps/s | 6.0276 KOps/s | |
| test_seq_add[compile-overhead] | 0.2555ms | 0.1966ms | 5.0855 KOps/s | 4.9710 KOps/s | |
| test_seq_wrap[eager] | 0.6032ms | 0.5325ms | 1.8778 KOps/s | 1.8630 KOps/s | |
| test_seq_wrap[compile] | 0.5336ms | 0.4091ms | 2.4442 KOps/s | 2.4409 KOps/s | |
| test_seq_wrap[compile-overhead] | 0.3655ms | 0.3055ms | 3.2729 KOps/s | 3.1449 KOps/s | |
| test_func_call_runtime[False-eager] | 0.9445ms | 0.8774ms | 1.1397 KOps/s | 1.1487 KOps/s | |
| test_func_call_runtime[False-compile] | 0.9795ms | 0.9254ms | 1.0806 KOps/s | 1.0830 KOps/s | |
| test_func_call_runtime[False-compile-overhead] | 0.5205ms | 0.4790ms | 2.0876 KOps/s | 2.0710 KOps/s | |
| test_func_call_runtime[True-eager] | 1.1846ms | 1.1135ms | 898.0767 Ops/s | 904.4352 Ops/s | |
| test_func_call_runtime[True-compile] | 1.0823ms | 0.9497ms | 1.0530 KOps/s | 1.0532 KOps/s | |
| test_func_call_runtime[True-compile-overhead] | 0.5582ms | 0.5005ms | 1.9981 KOps/s | 1.9867 KOps/s | |
| test_func_call_cm_runtime[False-eager] | 0.9404ms | 0.8703ms | 1.1491 KOps/s | 1.1177 KOps/s | |
| test_func_call_cm_runtime[False-compile] | 0.9851ms | 0.9275ms | 1.0781 KOps/s | 1.0778 KOps/s | |
| test_func_call_cm_runtime[False-compile-overhead] | 0.5331ms | 0.4832ms | 2.0694 KOps/s | 2.0680 KOps/s | |
| test_func_call_cm_runtime[True-eager] | 1.3546ms | 1.2488ms | 800.7457 Ops/s | 801.7922 Ops/s | |
| test_func_call_cm_runtime[True-compile] | 1.1096ms | 0.9836ms | 1.0167 KOps/s | 1.0266 KOps/s | |
| test_func_call_cm_runtime[True-compile-overhead] | 0.5865ms | 0.5371ms | 1.8619 KOps/s | 1.8615 KOps/s | |
| test_vmap_func_call_cm_runtime[eager] | 2.8278ms | 2.3405ms | 427.2626 Ops/s | 426.9873 Ops/s | |
| test_vmap_func_call_cm_runtime[compile] | 1.2315ms | 1.0004ms | 999.5913 Ops/s | 1.0113 KOps/s | |
| test_vmap_func_call_cm_runtime[compile-overhead] | 0.5867ms | 0.5298ms | 1.8875 KOps/s | 1.8662 KOps/s | |
| test_distributed | 0.6235ms | 0.1541ms | 6.4891 KOps/s | 6.5037 KOps/s | |
| test_tdmodule | 57.4510μs | 27.0025μs | 37.0337 KOps/s | 36.0894 KOps/s | |
| test_tdmodule_dispatch | 76.5120μs | 46.8466μs | 21.3463 KOps/s | 20.9852 KOps/s | |
| test_tdseq | 55.1110μs | 26.4536μs | 37.8021 KOps/s | 37.3613 KOps/s | |
| test_tdseq_dispatch | 80.6210μs | 48.8660μs | 20.4641 KOps/s | 20.2293 KOps/s | |
| test_instantiation_functorch | 2.0722ms | 1.9959ms | 501.0226 Ops/s | 501.2577 Ops/s | |
| test_exec_functorch | 0.2318ms | 0.1798ms | 5.5628 KOps/s | 5.6076 KOps/s | |
| test_exec_functional_call | 0.2073ms | 0.1619ms | 6.1752 KOps/s | 6.2972 KOps/s | |
| test_exec_td_decorator | 0.4404ms | 0.2356ms | 4.2441 KOps/s | 4.2935 KOps/s | |
| test_vmap_mlp_speed_decorator[True-True] | 0.9625ms | 0.7924ms | 1.2619 KOps/s | 1.2552 KOps/s | |
| test_vmap_mlp_speed_decorator[True-False] | 0.9438ms | 0.7905ms | 1.2650 KOps/s | 1.2547 KOps/s | |
| test_vmap_mlp_speed_decorator[False-True] | 0.8507ms | 0.6802ms | 1.4702 KOps/s | 1.4634 KOps/s | |
| test_vmap_mlp_speed_decorator[False-False] | 0.8285ms | 0.6810ms | 1.4684 KOps/s | 1.4630 KOps/s | |
| test_vmap_transformer_speed_decorator[True-True] | 21.2097ms | 20.4146ms | 48.9846 Ops/s | 48.9236 Ops/s | |
| test_vmap_transformer_speed_decorator[True-False] | 20.5765ms | 20.4014ms | 49.0162 Ops/s | 48.9069 Ops/s | |
| test_vmap_transformer_speed_decorator[False-True] | 20.7909ms | 20.1993ms | 49.5068 Ops/s | 49.3894 Ops/s | |
| test_vmap_transformer_speed_decorator[False-False] | 20.8776ms | 20.2196ms | 49.4569 Ops/s | 49.3782 Ops/s | |
| test_to_module_speed[True] | 2.0025ms | 1.4168ms | 705.8277 Ops/s | 701.4850 Ops/s | |
| test_to_module_speed[False] | 1.4800ms | 1.3855ms | 721.7845 Ops/s | 711.8276 Ops/s | |
| test_tc_init | 0.1384ms | 50.0976μs | 19.9611 KOps/s | 19.6128 KOps/s | |
| test_tc_init_tensor_only | 43.8310μs | 14.2709μs | 70.0727 KOps/s | 70.0639 KOps/s | |
| test_tc_init_nested | 0.1342ms | 0.1003ms | 9.9722 KOps/s | 9.9108 KOps/s | |
| test_tc_first_layer_tensor | 19.6300μs | 1.7181μs | 582.0248 KOps/s | 586.3775 KOps/s | |
| test_tc_first_layer_tensor_only | 3.6662μs | 0.6803μs | 1.4700 MOps/s | 1.4917 MOps/s | |
| test_tc_first_layer_tensor_set | 38.0610μs | 3.9710μs | 251.8254 KOps/s | 249.0557 KOps/s | |
| test_tc_first_layer_tensor_only_set | 17.0705μs | 2.8845μs | 346.6756 KOps/s | 331.0258 KOps/s | |
| test_tc_first_layer_nontensor | 32.6100μs | 5.7455μs | 174.0492 KOps/s | 175.8050 KOps/s | |
| test_tc_second_layer_tensor | 6.8010μs | 4.0931μs | 244.3155 KOps/s | 247.2552 KOps/s | |
| test_tc_second_layer_nontensor | 39.1510μs | 7.9333μs | 126.0509 KOps/s | 124.6369 KOps/s | |
| test_unbind | 0.2568s | 14.0367ms | 71.2416 Ops/s | 69.2286 Ops/s | |
| test_full_like | 5.5622ms | 4.3826ms | 228.1729 Ops/s | 95.1243 Ops/s | |
| test_zeros_like | 4.6711ms | 4.3588ms | 229.4232 Ops/s | 73.8915 Ops/s | |
| test_ones_like | 4.7952ms | 4.3678ms | 228.9470 Ops/s | 73.7455 Ops/s | |
| test_clone | 11.4924ms | 9.2208ms | 108.4500 Ops/s | 67.2668 Ops/s | |
| test_squeeze | 0.1825ms | 14.0259μs | 71.2966 KOps/s | 71.8611 KOps/s | |
| test_unsqueeze | 0.1596ms | 0.1047ms | 9.5471 KOps/s | 9.4316 KOps/s | |
| test_split | 0.3708ms | 0.1788ms | 5.5929 KOps/s | 5.5819 KOps/s | |
| test_permute | 0.2458ms | 0.1964ms | 5.0904 KOps/s | 5.0865 KOps/s | |
| test_stack | 51.2319ms | 50.9619ms | 19.6225 Ops/s | 19.5850 Ops/s | |
| test_cat | 54.1195ms | 51.3384ms | 19.4786 Ops/s | 19.6366 Ops/s |
331e5c1 to
65c037d
Compare
This file is auto-generated by setuptools_scm at build time and should not be tracked in version control (as stated in the file header itself). The version includes +g<sha> suffix based on the current git commit, which is already tested by test_setup.py::test_install_strategies_version.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Labels
bug
Something isn't working
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
setup
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This file is auto-generated by setuptools_scm at build time and should
not be tracked in version control (as stated in the file header itself).
The version includes +g suffix based on the current git commit,
which is already tested by test_setup.py::test_install_strategies_version.