-
Notifications
You must be signed in to change notification settings - Fork 79
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
[BugFix] Fix key errors catch in del_ and related #949
Merged
Conversation
This file contains 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 | 43.2910μs | 20.4215μs | 48.9681 KOps/s | 46.3783 KOps/s | |
test_plain_set_stack_nested | 61.9770μs | 20.2975μs | 49.2672 KOps/s | 45.9695 KOps/s | |
test_plain_set_nested_inplace | 0.1116ms | 22.2549μs | 44.9340 KOps/s | 41.7829 KOps/s | |
test_plain_set_stack_nested_inplace | 64.8920μs | 22.0791μs | 45.2918 KOps/s | 41.8313 KOps/s | |
test_items | 43.2610μs | 2.8378μs | 352.3846 KOps/s | 344.3875 KOps/s | |
test_items_nested | 0.6298ms | 0.3356ms | 2.9799 KOps/s | 2.9614 KOps/s | |
test_items_nested_locked | 2.0927ms | 0.3363ms | 2.9732 KOps/s | 2.9612 KOps/s | |
test_items_nested_leaf | 0.1612ms | 82.7943μs | 12.0781 KOps/s | 11.4996 KOps/s | |
test_items_stack_nested | 0.4803ms | 0.3375ms | 2.9628 KOps/s | 2.9596 KOps/s | |
test_items_stack_nested_leaf | 0.1437ms | 83.2166μs | 12.0168 KOps/s | 11.6539 KOps/s | |
test_items_stack_nested_locked | 0.4536ms | 0.3370ms | 2.9671 KOps/s | 2.9336 KOps/s | |
test_keys | 17.0620μs | 3.8989μs | 256.4840 KOps/s | 258.1308 KOps/s | |
test_keys_nested | 0.2556ms | 0.1459ms | 6.8517 KOps/s | 6.9433 KOps/s | |
test_keys_nested_locked | 0.6967ms | 0.1503ms | 6.6517 KOps/s | 6.6566 KOps/s | |
test_keys_nested_leaf | 0.2071ms | 0.1238ms | 8.0752 KOps/s | 8.1896 KOps/s | |
test_keys_stack_nested | 0.2407ms | 0.1443ms | 6.9314 KOps/s | 7.0326 KOps/s | |
test_keys_stack_nested_leaf | 0.2104ms | 0.1236ms | 8.0928 KOps/s | 8.2110 KOps/s | |
test_keys_stack_nested_locked | 0.2620ms | 0.1482ms | 6.7478 KOps/s | 6.7271 KOps/s | |
test_values | 9.7383μs | 1.1839μs | 844.6981 KOps/s | 830.7463 KOps/s | |
test_values_nested | 0.1041ms | 50.0423μs | 19.9831 KOps/s | 19.9054 KOps/s | |
test_values_nested_locked | 0.1482ms | 49.9878μs | 20.0049 KOps/s | 19.7941 KOps/s | |
test_values_nested_leaf | 0.2593ms | 46.1246μs | 21.6804 KOps/s | 21.9702 KOps/s | |
test_values_stack_nested | 96.1500μs | 50.4550μs | 19.8196 KOps/s | 18.7006 KOps/s | |
test_values_stack_nested_leaf | 84.5690μs | 44.6460μs | 22.3984 KOps/s | 22.2292 KOps/s | |
test_values_stack_nested_locked | 89.9490μs | 49.8425μs | 20.0632 KOps/s | 19.3394 KOps/s | |
test_membership | 28.5130μs | 0.8967μs | 1.1152 MOps/s | 1.3291 MOps/s | |
test_membership_nested | 38.2410μs | 2.6439μs | 378.2362 KOps/s | 385.3611 KOps/s | |
test_membership_nested_leaf | 37.2590μs | 2.6255μs | 380.8746 KOps/s | 386.0094 KOps/s | |
test_membership_stacked_nested | 40.5660μs | 2.5981μs | 384.8998 KOps/s | 391.5853 KOps/s | |
test_membership_stacked_nested_leaf | 25.6790μs | 2.6128μs | 382.7316 KOps/s | 371.7759 KOps/s | |
test_membership_nested_last | 45.0540μs | 3.8794μs | 257.7713 KOps/s | 257.5443 KOps/s | |
test_membership_nested_leaf_last | 44.2330μs | 3.8777μs | 257.8839 KOps/s | 255.3631 KOps/s | |
test_membership_stacked_nested_last | 38.8930μs | 4.4773μs | 223.3489 KOps/s | 78.4338 KOps/s | |
test_membership_stacked_nested_leaf_last | 22.4320μs | 4.5059μs | 221.9330 KOps/s | 78.7940 KOps/s | |
test_nested_getleaf | 0.1392ms | 10.3650μs | 96.4789 KOps/s | 97.0786 KOps/s | |
test_nested_get | 50.7850μs | 9.7878μs | 102.1675 KOps/s | 101.6232 KOps/s | |
test_stacked_getleaf | 48.5810μs | 10.2787μs | 97.2884 KOps/s | 98.0587 KOps/s | |
test_stacked_get | 52.6390μs | 9.6771μs | 103.3370 KOps/s | 103.2847 KOps/s | |
test_nested_getitemleaf | 54.3620μs | 10.9584μs | 91.2541 KOps/s | 92.3470 KOps/s | |
test_nested_getitem | 52.4080μs | 10.0341μs | 99.6606 KOps/s | 99.2984 KOps/s | |
test_stacked_getitemleaf | 55.7240μs | 10.7845μs | 92.7257 KOps/s | 93.4255 KOps/s | |
test_stacked_getitem | 46.4070μs | 10.0661μs | 99.3430 KOps/s | 100.4335 KOps/s | |
test_lock_nested | 0.9917ms | 0.4900ms | 2.0407 KOps/s | 1.9830 KOps/s | |
test_lock_stack_nested | 0.6098ms | 0.4647ms | 2.1517 KOps/s | 2.2550 KOps/s | |
test_unlock_nested | 0.8412ms | 0.4179ms | 2.3930 KOps/s | 2.3919 KOps/s | |
test_unlock_stack_nested | 0.6401ms | 0.3829ms | 2.6115 KOps/s | 2.7709 KOps/s | |
test_flatten_speed | 0.6970ms | 0.1034ms | 9.6689 KOps/s | 9.3803 KOps/s | |
test_unflatten_speed | 1.0993ms | 0.4459ms | 2.2427 KOps/s | 2.2610 KOps/s | |
test_common_ops | 3.7236ms | 1.0428ms | 958.9345 Ops/s | 929.9035 Ops/s | |
test_creation | 71.1340μs | 1.9933μs | 501.6922 KOps/s | 497.2550 KOps/s | |
test_creation_empty | 49.8530μs | 15.8451μs | 63.1111 KOps/s | 58.5004 KOps/s | |
test_creation_nested_1 | 61.8160μs | 19.3219μs | 51.7547 KOps/s | 50.1420 KOps/s | |
test_creation_nested_2 | 73.0170μs | 23.0488μs | 43.3863 KOps/s | 40.7223 KOps/s | |
test_clone | 0.1712ms | 15.9417μs | 62.7287 KOps/s | 59.9635 KOps/s | |
test_getitem[int] | 0.9935ms | 15.9784μs | 62.5846 KOps/s | 59.5759 KOps/s | |
test_getitem[slice_int] | 0.1544ms | 30.0103μs | 33.3219 KOps/s | 31.8245 KOps/s | |
test_getitem[range] | 0.2112ms | 54.7741μs | 18.2568 KOps/s | 17.2567 KOps/s | |
test_getitem[tuple] | 0.1289ms | 24.2881μs | 41.1724 KOps/s | 39.9400 KOps/s | |
test_getitem[list] | 0.2396ms | 49.6361μs | 20.1466 KOps/s | 18.6984 KOps/s | |
test_setitem_dim[int] | 0.1028ms | 34.9952μs | 28.5754 KOps/s | 23.5377 KOps/s | |
test_setitem_dim[slice_int] | 90.5790μs | 64.1871μs | 15.5795 KOps/s | 13.6528 KOps/s | |
test_setitem_dim[range] | 0.1308ms | 85.5115μs | 11.6943 KOps/s | 10.7904 KOps/s | |
test_setitem_dim[tuple] | 0.1223ms | 52.4406μs | 19.0692 KOps/s | 16.7280 KOps/s | |
test_setitem | 0.1592ms | 26.8798μs | 37.2027 KOps/s | 34.8045 KOps/s | |
test_set | 0.1698ms | 26.0386μs | 38.4045 KOps/s | 35.7405 KOps/s | |
test_set_shared | 4.0539ms | 0.2130ms | 4.6943 KOps/s | 4.6502 KOps/s | |
test_update | 0.2046ms | 32.0697μs | 31.1820 KOps/s | 28.6524 KOps/s | |
test_update_nested | 0.2459ms | 41.3497μs | 24.1839 KOps/s | 22.4153 KOps/s | |
test_update__nested | 0.2184ms | 32.5418μs | 30.7297 KOps/s | 29.7227 KOps/s | |
test_set_nested | 0.2133ms | 28.5357μs | 35.0439 KOps/s | 32.7841 KOps/s | |
test_set_nested_new | 0.2140ms | 33.6771μs | 29.6938 KOps/s | 28.7686 KOps/s | |
test_select | 0.2104ms | 51.1324μs | 19.5571 KOps/s | 19.5089 KOps/s | |
test_select_nested | 0.1299ms | 59.0665μs | 16.9301 KOps/s | 16.8982 KOps/s | |
test_exclude_nested | 0.1464ms | 77.4816μs | 12.9063 KOps/s | 12.9510 KOps/s | |
test_empty[True] | 0.5984ms | 0.3208ms | 3.1169 KOps/s | 3.1145 KOps/s | |
test_empty[False] | 11.3438μs | 1.1795μs | 847.8019 KOps/s | 873.8751 KOps/s | |
test_unbind_speed | 0.3865ms | 0.3011ms | 3.3217 KOps/s | 3.2296 KOps/s | |
test_unbind_speed_stack0 | 0.3877ms | 0.3029ms | 3.3014 KOps/s | 3.4343 KOps/s | |
test_unbind_speed_stack1 | 88.6472ms | 0.8002ms | 1.2497 KOps/s | 1.4194 KOps/s | |
test_split | 94.8164ms | 2.1162ms | 472.5507 Ops/s | 470.2207 Ops/s | |
test_chunk | 85.0241ms | 2.0940ms | 477.5567 Ops/s | 465.6367 Ops/s | |
test_creation[device0] | 0.2517ms | 0.1187ms | 8.4267 KOps/s | 8.6044 KOps/s | |
test_creation_from_tensor | 3.8643ms | 0.1195ms | 8.3672 KOps/s | 8.3155 KOps/s | |
test_add_one[memmap_tensor0] | 0.1906ms | 7.5253μs | 132.8856 KOps/s | 133.8634 KOps/s | |
test_contiguous[memmap_tensor0] | 22.3020μs | 2.0350μs | 491.3953 KOps/s | 496.0874 KOps/s | |
test_stack[memmap_tensor0] | 57.7480μs | 5.5395μs | 180.5231 KOps/s | 173.6984 KOps/s | |
test_memmaptd_index | 1.0664ms | 0.3998ms | 2.5011 KOps/s | 2.5175 KOps/s | |
test_memmaptd_index_astensor | 0.7767ms | 0.4763ms | 2.0994 KOps/s | 2.0929 KOps/s | |
test_memmaptd_index_op | 1.3487ms | 0.9586ms | 1.0432 KOps/s | 988.2844 Ops/s | |
test_serialize_model | 0.1258s | 0.1172s | 8.5342 Ops/s | 7.3445 Ops/s | |
test_serialize_model_pickle | 0.4473s | 0.3941s | 2.5377 Ops/s | 2.5558 Ops/s | |
test_serialize_weights | 0.2042s | 0.1259s | 7.9424 Ops/s | 8.7809 Ops/s | |
test_serialize_weights_returnearly | 0.1813s | 0.1647s | 6.0724 Ops/s | 6.3367 Ops/s | |
test_serialize_weights_pickle | 0.5391s | 0.4176s | 2.3944 Ops/s | 2.3889 Ops/s | |
test_serialize_weights_filesystem | 0.1467s | 0.1419s | 7.0460 Ops/s | 6.7865 Ops/s | |
test_serialize_model_filesystem | 0.2343s | 0.1617s | 6.1853 Ops/s | 6.0128 Ops/s | |
test_reshape_pytree | 77.2050μs | 39.7409μs | 25.1630 KOps/s | 25.1755 KOps/s | |
test_reshape_td | 0.1047ms | 45.6144μs | 21.9229 KOps/s | 21.7396 KOps/s | |
test_view_pytree | 0.1092ms | 39.6533μs | 25.2186 KOps/s | 25.8400 KOps/s | |
test_view_td | 0.1085ms | 52.6678μs | 18.9869 KOps/s | 19.0634 KOps/s | |
test_unbind_pytree | 94.0670μs | 36.9393μs | 27.0714 KOps/s | 27.3803 KOps/s | |
test_unbind_td | 0.4459ms | 44.6064μs | 22.4183 KOps/s | 22.0280 KOps/s | |
test_split_pytree | 94.1770μs | 39.1501μs | 25.5427 KOps/s | 25.4084 KOps/s | |
test_split_td | 0.5415ms | 56.6313μs | 17.6581 KOps/s | 16.8392 KOps/s | |
test_add_pytree | 95.2180μs | 46.7377μs | 21.3960 KOps/s | 21.7705 KOps/s | |
test_add_td | 0.2227ms | 76.4767μs | 13.0759 KOps/s | 12.0024 KOps/s | |
test_compile_add_one_nested[tensordict-compile] | 0.1137ms | 53.4658μs | 18.7036 KOps/s | 18.5361 KOps/s | |
test_compile_add_one_nested[tensordict-eager] | 0.3597ms | 0.1907ms | 5.2452 KOps/s | 5.1942 KOps/s | |
test_compile_add_one_nested[pytree-compile] | 0.2519ms | 54.8658μs | 18.2263 KOps/s | 18.3123 KOps/s | |
test_compile_add_one_nested[pytree-eager] | 0.3809ms | 0.1430ms | 6.9923 KOps/s | 6.9124 KOps/s | |
test_compile_copy_nested[tensordict-compile] | 80.3900μs | 20.9964μs | 47.6271 KOps/s | 49.2664 KOps/s | |
test_compile_copy_nested[tensordict-eager] | 0.1489ms | 64.7886μs | 15.4348 KOps/s | 15.5546 KOps/s | |
test_compile_copy_nested[pytree-compile] | 0.1910ms | 77.0364μs | 12.9809 KOps/s | 12.6117 KOps/s | |
test_compile_copy_nested[pytree-eager] | 0.1490ms | 70.3100μs | 14.2227 KOps/s | 14.0811 KOps/s | |
test_compile_add_one_flat[tensordict-compile] | 0.3629ms | 0.1765ms | 5.6651 KOps/s | 5.4929 KOps/s | |
test_compile_add_one_flat[tensordict-eager] | 0.3507ms | 0.1932ms | 5.1771 KOps/s | 5.1285 KOps/s | |
test_compile_add_one_flat[tensorclass-compile] | 0.1180ms | 38.0608μs | 26.2738 KOps/s | 24.8888 KOps/s | |
test_compile_add_one_flat[tensorclass-eager] | 1.2508ms | 70.2846μs | 14.2279 KOps/s | 14.2839 KOps/s | |
test_compile_add_one_flat[pytree-compile] | 0.3558ms | 0.1745ms | 5.7291 KOps/s | 5.7362 KOps/s | |
test_compile_add_one_flat[pytree-eager] | 0.5700ms | 0.2954ms | 3.3851 KOps/s | 3.4353 KOps/s | |
test_compile_add_self_flat[tensordict-eager] | 0.3790ms | 0.2057ms | 4.8606 KOps/s | 4.8384 KOps/s | |
test_compile_add_self_flat[tensordict-compile] | 0.4411ms | 0.1815ms | 5.5109 KOps/s | 5.5676 KOps/s | |
test_compile_add_self_flat[tensorclass-eager] | 0.1782ms | 63.3651μs | 15.7816 KOps/s | 15.7890 KOps/s | |
test_compile_add_self_flat[tensorclass-compile] | 0.1084ms | 39.7013μs | 25.1881 KOps/s | 25.0913 KOps/s | |
test_compile_add_self_flat[pytree-eager] | 0.5183ms | 0.2418ms | 4.1355 KOps/s | 4.1880 KOps/s | |
test_compile_add_self_flat[pytree-compile] | 0.3811ms | 0.1758ms | 5.6872 KOps/s | 5.7784 KOps/s | |
test_compile_copy_flat[tensordict-compile] | 0.2522ms | 0.1116ms | 8.9632 KOps/s | 9.0439 KOps/s | |
test_compile_copy_flat[tensordict-eager] | 0.1578ms | 55.5051μs | 18.0163 KOps/s | 17.6781 KOps/s | |
test_compile_copy_flat[pytree-compile] | 0.1611ms | 80.4475μs | 12.4305 KOps/s | 12.4877 KOps/s | |
test_compile_copy_flat[pytree-eager] | 0.1483ms | 71.4729μs | 13.9913 KOps/s | 13.9944 KOps/s | |
test_compile_assign_and_add[tensordict-compile] | 0.3223ms | 0.1860ms | 5.3772 KOps/s | 5.1978 KOps/s | |
test_compile_assign_and_add[tensordict-eager] | 2.1066ms | 1.6111ms | 620.6788 Ops/s | 621.2653 Ops/s | |
test_compile_assign_and_add[pytree-compile] | 0.2961ms | 0.1850ms | 5.4065 KOps/s | 5.2940 KOps/s | |
test_compile_assign_and_add[pytree-eager] | 1.1893ms | 1.0845ms | 922.0713 Ops/s | 928.7287 Ops/s | |
test_compile_assign_and_add_stack[compile] | 0.7172ms | 0.4079ms | 2.4516 KOps/s | 2.3635 KOps/s | |
test_compile_assign_and_add_stack[eager] | 4.6936ms | 3.5142ms | 284.5606 Ops/s | 266.2274 Ops/s | |
test_compile_indexing[tensor-tensordict-compile] | 75.2510μs | 31.7662μs | 31.4800 KOps/s | 29.9551 KOps/s | |
test_compile_indexing[tensor-tensordict-eager] | 0.5343ms | 46.6917μs | 21.4171 KOps/s | 20.3941 KOps/s | |
test_compile_indexing[tensor-tensorclass-compile] | 95.6700μs | 27.2698μs | 36.6706 KOps/s | 34.4839 KOps/s | |
test_compile_indexing[tensor-tensorclass-eager] | 77.1040μs | 29.4768μs | 33.9250 KOps/s | 32.9703 KOps/s | |
test_compile_indexing[tensor-pytree-compile] | 66.1640μs | 27.5272μs | 36.3277 KOps/s | 35.1065 KOps/s | |
test_compile_indexing[tensor-pytree-eager] | 99.8570μs | 30.1371μs | 33.1817 KOps/s | 32.5023 KOps/s | |
test_compile_indexing[slice-tensordict-compile] | 0.1401ms | 72.3254μs | 13.8264 KOps/s | 13.7259 KOps/s | |
test_compile_indexing[slice-tensordict-eager] | 0.5143ms | 27.7588μs | 36.0246 KOps/s | 34.2903 KOps/s | |
test_compile_indexing[slice-tensorclass-compile] | 0.1268ms | 67.5126μs | 14.8120 KOps/s | 14.7743 KOps/s | |
test_compile_indexing[slice-tensorclass-eager] | 64.4410μs | 23.2559μs | 42.9998 KOps/s | 41.6414 KOps/s | |
test_compile_indexing[slice-pytree-compile] | 0.1287ms | 67.4111μs | 14.8344 KOps/s | 14.9024 KOps/s | |
test_compile_indexing[slice-pytree-eager] | 64.6110μs | 23.7830μs | 42.0469 KOps/s | 41.3567 KOps/s | |
test_compile_indexing[int-tensordict-compile] | 0.1851ms | 72.6271μs | 13.7690 KOps/s | 13.6728 KOps/s | |
test_compile_indexing[int-tensordict-eager] | 0.8063ms | 26.8225μs | 37.2821 KOps/s | 34.5557 KOps/s | |
test_compile_indexing[int-tensorclass-compile] | 0.1258ms | 67.1639μs | 14.8890 KOps/s | 14.9593 KOps/s | |
test_compile_indexing[int-tensorclass-eager] | 94.1570μs | 23.9327μs | 41.7839 KOps/s | 41.9857 KOps/s | |
test_compile_indexing[int-pytree-compile] | 0.1271ms | 68.2310μs | 14.6561 KOps/s | 14.8315 KOps/s | |
test_compile_indexing[int-pytree-eager] | 72.1850μs | 23.3147μs | 42.8914 KOps/s | 41.9388 KOps/s | |
test_mod_add[eager] | 79.2590μs | 23.2114μs | 43.0823 KOps/s | 40.0653 KOps/s | |
test_mod_add[compile] | 75.2710μs | 35.8077μs | 27.9270 KOps/s | 27.9887 KOps/s | |
test_mod_add[compile-overhead] | 80.8410μs | 35.7859μs | 27.9440 KOps/s | 27.3220 KOps/s | |
test_mod_wrap[eager] | 0.3428ms | 0.2033ms | 4.9193 KOps/s | 4.7390 KOps/s | |
test_mod_wrap[compile] | 1.8039ms | 0.2232ms | 4.4793 KOps/s | 4.4255 KOps/s | |
test_mod_wrap[compile-overhead] | 0.3490ms | 0.2188ms | 4.5710 KOps/s | 4.4863 KOps/s | |
test_mod_wrap_and_backward[eager] | 15.7445ms | 11.4894ms | 87.0366 Ops/s | 90.5281 Ops/s | |
test_mod_wrap_and_backward[compile] | 18.2774ms | 11.4990ms | 86.9644 Ops/s | 85.6493 Ops/s | |
test_mod_wrap_and_backward[compile-overhead] | 12.5507ms | 11.0135ms | 90.7977 Ops/s | 88.0312 Ops/s | |
test_seq_add[eager] | 0.2058ms | 85.4704μs | 11.7000 KOps/s | 11.5474 KOps/s | |
test_seq_add[compile] | 0.1590ms | 61.1234μs | 16.3604 KOps/s | 16.1403 KOps/s | |
test_seq_add[compile-overhead] | 0.1580ms | 59.2490μs | 16.8779 KOps/s | 16.6987 KOps/s | |
test_seq_wrap[eager] | 0.6212ms | 0.3611ms | 2.7695 KOps/s | 2.6555 KOps/s | |
test_seq_wrap[compile] | 0.6386ms | 0.2569ms | 3.8924 KOps/s | 3.8317 KOps/s | |
test_seq_wrap[compile-overhead] | 0.4823ms | 0.2578ms | 3.8792 KOps/s | 3.7915 KOps/s | |
test_func_call_runtime[False-eager] | 0.7546ms | 0.5202ms | 1.9224 KOps/s | 1.8556 KOps/s | |
test_func_call_runtime[False-compile] | 0.6031ms | 0.4880ms | 2.0491 KOps/s | 2.0230 KOps/s | |
test_func_call_runtime[False-compile-overhead] | 0.8500ms | 0.4949ms | 2.0205 KOps/s | 2.0233 KOps/s | |
test_func_call_runtime[True-eager] | 0.9981ms | 0.7568ms | 1.3214 KOps/s | 1.3123 KOps/s | |
test_func_call_runtime[True-compile] | 0.7137ms | 0.5079ms | 1.9688 KOps/s | 1.9824 KOps/s | |
test_func_call_runtime[True-compile-overhead] | 0.6197ms | 0.5020ms | 1.9920 KOps/s | 1.9778 KOps/s | |
test_func_call_cm_runtime[False-eager] | 0.8342ms | 0.5223ms | 1.9147 KOps/s | 1.8654 KOps/s | |
test_func_call_cm_runtime[False-compile] | 0.8968ms | 0.4883ms | 2.0481 KOps/s | 2.0228 KOps/s | |
test_func_call_cm_runtime[False-compile-overhead] | 0.8475ms | 0.4959ms | 2.0167 KOps/s | 2.0396 KOps/s | |
test_func_call_cm_runtime[True-eager] | 1.8671ms | 0.8854ms | 1.1294 KOps/s | 1.1131 KOps/s | |
test_func_call_cm_runtime[True-compile] | 1.0361ms | 0.8326ms | 1.2011 KOps/s | 1.1838 KOps/s | |
test_func_call_cm_runtime[True-compile-overhead] | 1.1460ms | 0.8343ms | 1.1987 KOps/s | 1.1782 KOps/s | |
test_distributed | 0.2630ms | 0.1316ms | 7.6003 KOps/s | 7.4735 KOps/s | |
test_tdmodule | 0.1152ms | 15.7503μs | 63.4911 KOps/s | 61.9537 KOps/s | |
test_tdmodule_dispatch | 64.3610μs | 32.5601μs | 30.7125 KOps/s | 29.4324 KOps/s | |
test_tdseq | 35.0150μs | 17.3696μs | 57.5719 KOps/s | 55.8549 KOps/s | |
test_tdseq_dispatch | 55.0330μs | 36.4852μs | 27.4083 KOps/s | 26.1534 KOps/s | |
test_instantiation_functorch | 3.3944ms | 1.6304ms | 613.3564 Ops/s | 601.2891 Ops/s | |
test_instantiation_td | 2.1214ms | 1.1861ms | 843.0851 Ops/s | 791.4699 Ops/s | |
test_exec_functorch | 0.4119ms | 0.1804ms | 5.5440 KOps/s | 5.6017 KOps/s | |
test_exec_functional_call | 0.4297ms | 0.1685ms | 5.9362 KOps/s | 5.6475 KOps/s | |
test_exec_td | 0.2622ms | 0.1680ms | 5.9523 KOps/s | 5.6481 KOps/s | |
test_exec_td_decorator | 0.6191ms | 0.2200ms | 4.5445 KOps/s | 4.3563 KOps/s | |
test_vmap_mlp_speed[True-True] | 1.0376ms | 0.5877ms | 1.7016 KOps/s | 1.7235 KOps/s | |
test_vmap_mlp_speed[True-False] | 0.8373ms | 0.5495ms | 1.8199 KOps/s | 1.7366 KOps/s | |
test_vmap_mlp_speed[False-True] | 0.6717ms | 0.4599ms | 2.1745 KOps/s | 2.0814 KOps/s | |
test_vmap_mlp_speed[False-False] | 0.8613ms | 0.4600ms | 2.1741 KOps/s | 2.0848 KOps/s | |
test_vmap_mlp_speed_decorator[True-True] | 1.6463ms | 0.6089ms | 1.6423 KOps/s | 1.5625 KOps/s | |
test_vmap_mlp_speed_decorator[True-False] | 0.9111ms | 0.6119ms | 1.6344 KOps/s | 1.5614 KOps/s | |
test_vmap_mlp_speed_decorator[False-True] | 0.7003ms | 0.5075ms | 1.9703 KOps/s | 1.8888 KOps/s | |
test_vmap_mlp_speed_decorator[False-False] | 0.8020ms | 0.5097ms | 1.9620 KOps/s | 1.8918 KOps/s | |
test_to_module_speed[True] | 1.4693ms | 1.3286ms | 752.6454 Ops/s | 750.6283 Ops/s | |
test_to_module_speed[False] | 2.1599ms | 1.3014ms | 768.3747 Ops/s | 772.8137 Ops/s | |
test_tc_init | 73.7580μs | 39.9998μs | 25.0001 KOps/s | 22.8738 KOps/s | |
test_tc_init_nested | 0.1744ms | 81.2075μs | 12.3141 KOps/s | 11.5424 KOps/s | |
test_tc_first_layer_tensor | 16.3910μs | 1.4816μs | 674.9378 KOps/s | 699.1996 KOps/s | |
test_tc_first_layer_nontensor | 40.8670μs | 4.2410μs | 235.7949 KOps/s | 237.8705 KOps/s | |
test_tc_second_layer_tensor | 35.7270μs | 2.7370μs | 365.3675 KOps/s | 373.1309 KOps/s | |
test_tc_second_layer_nontensor | 43.3110μs | 5.6300μs | 177.6204 KOps/s | 180.7704 KOps/s | |
test_unbind | 0.4564s | 13.5764ms | 73.6571 Ops/s | 63.4948 Ops/s | |
test_full_like | 8.5545ms | 7.2875ms | 137.2203 Ops/s | 129.8637 Ops/s | |
test_zeros_like | 12.9535ms | 6.1047ms | 163.8090 Ops/s | 139.0960 Ops/s | |
test_ones_like | 11.1263ms | 7.2176ms | 138.5508 Ops/s | 133.6635 Ops/s | |
test_clone | 14.5764ms | 8.8739ms | 112.6895 Ops/s | 105.8527 Ops/s | |
test_squeeze | 67.9970μs | 12.7180μs | 78.6288 KOps/s | 77.9008 KOps/s | |
test_unsqueeze | 0.1667ms | 93.2144μs | 10.7280 KOps/s | 10.8044 KOps/s | |
test_split | 0.4915ms | 0.1959ms | 5.1056 KOps/s | 4.9895 KOps/s | |
test_permute | 0.3724ms | 0.2167ms | 4.6144 KOps/s | 4.6427 KOps/s | |
test_stack | 29.9852ms | 24.5840ms | 40.6769 Ops/s | 40.8632 Ops/s | |
test_cat | 29.2294ms | 24.1843ms | 41.3492 Ops/s | 41.3114 Ops/s |
|
Name | Max | Mean | Ops | Ops on Repo HEAD
|
Change |
---|---|---|---|---|---|
test_plain_set_nested | 0.2136ms | 17.2169μs | 58.0823 KOps/s | 56.7701 KOps/s | |
test_plain_set_stack_nested | 47.5420μs | 17.3889μs | 57.5078 KOps/s | 56.7947 KOps/s | |
test_plain_set_nested_inplace | 37.9400μs | 18.4224μs | 54.2816 KOps/s | 53.6176 KOps/s | |
test_plain_set_stack_nested_inplace | 38.1910μs | 18.3493μs | 54.4980 KOps/s | 53.6903 KOps/s | |
test_items | 20.1200μs | 4.6859μs | 213.4053 KOps/s | 214.9177 KOps/s | |
test_items_nested | 0.3871ms | 0.3646ms | 2.7426 KOps/s | 2.7172 KOps/s | |
test_items_nested_locked | 0.3849ms | 0.3665ms | 2.7283 KOps/s | 2.7410 KOps/s | |
test_items_nested_leaf | 98.3720μs | 83.4151μs | 11.9882 KOps/s | 12.0034 KOps/s | |
test_items_stack_nested | 0.4210ms | 0.3637ms | 2.7494 KOps/s | 2.7279 KOps/s | |
test_items_stack_nested_leaf | 0.1106ms | 84.8553μs | 11.7848 KOps/s | 11.6379 KOps/s | |
test_items_stack_nested_locked | 0.4303ms | 0.3676ms | 2.7205 KOps/s | 2.6912 KOps/s | |
test_keys | 55.2410μs | 4.3439μs | 230.2081 KOps/s | 228.9047 KOps/s | |
test_keys_nested | 90.4220μs | 68.0473μs | 14.6957 KOps/s | 14.9204 KOps/s | |
test_keys_nested_locked | 0.6780ms | 73.3770μs | 13.6283 KOps/s | 13.8419 KOps/s | |
test_keys_nested_leaf | 0.1324ms | 57.7602μs | 17.3130 KOps/s | 17.4193 KOps/s | |
test_keys_stack_nested | 81.4010μs | 66.0026μs | 15.1509 KOps/s | 14.9087 KOps/s | |
test_keys_stack_nested_leaf | 75.8620μs | 56.5235μs | 17.6918 KOps/s | 17.4036 KOps/s | |
test_keys_stack_nested_locked | 94.4020μs | 72.4944μs | 13.7942 KOps/s | 13.7991 KOps/s | |
test_values | 54.6077μs | 1.7589μs | 568.5518 KOps/s | 567.6354 KOps/s | |
test_values_nested | 0.2008ms | 33.5595μs | 29.7978 KOps/s | 29.7261 KOps/s | |
test_values_nested_locked | 0.2145ms | 35.6210μs | 28.0733 KOps/s | 27.7723 KOps/s | |
test_values_nested_leaf | 0.2241ms | 30.1077μs | 33.2141 KOps/s | 33.3149 KOps/s | |
test_values_stack_nested | 62.6910μs | 33.8292μs | 29.5603 KOps/s | 29.0511 KOps/s | |
test_values_stack_nested_leaf | 0.1552ms | 30.1228μs | 33.1974 KOps/s | 32.4236 KOps/s | |
test_values_stack_nested_locked | 63.4010μs | 35.9549μs | 27.8126 KOps/s | 27.1858 KOps/s | |
test_membership | 1.5351μs | 0.5438μs | 1.8389 MOps/s | 1.8315 MOps/s | |
test_membership_nested | 22.2910μs | 1.9628μs | 509.4771 KOps/s | 513.4141 KOps/s | |
test_membership_nested_leaf | 15.1655μs | 1.9001μs | 526.2812 KOps/s | 525.6319 KOps/s | |
test_membership_stacked_nested | 20.6610μs | 1.9650μs | 508.9116 KOps/s | 500.9530 KOps/s | |
test_membership_stacked_nested_leaf | 15.7910μs | 1.9655μs | 508.7772 KOps/s | 508.6424 KOps/s | |
test_membership_nested_last | 32.2700μs | 2.9028μs | 344.4976 KOps/s | 338.8273 KOps/s | |
test_membership_nested_leaf_last | 14.6800μs | 2.8957μs | 345.3432 KOps/s | 338.4676 KOps/s | |
test_membership_stacked_nested_last | 19.8700μs | 2.8936μs | 345.5844 KOps/s | 343.6328 KOps/s | |
test_membership_stacked_nested_leaf_last | 32.1110μs | 2.9045μs | 344.2884 KOps/s | 345.6425 KOps/s | |
test_nested_getleaf | 73.7010μs | 7.7808μs | 128.5208 KOps/s | 128.6441 KOps/s | |
test_nested_get | 29.1800μs | 7.2956μs | 137.0685 KOps/s | 137.8333 KOps/s | |
test_stacked_getleaf | 36.0110μs | 7.8112μs | 128.0220 KOps/s | 128.8614 KOps/s | |
test_stacked_get | 27.5200μs | 7.3358μs | 136.3187 KOps/s | 137.3503 KOps/s | |
test_nested_getitemleaf | 32.0900μs | 8.1414μs | 122.8288 KOps/s | 124.2273 KOps/s | |
test_nested_getitem | 22.0700μs | 7.6562μs | 130.6134 KOps/s | 131.1123 KOps/s | |
test_stacked_getitemleaf | 37.3610μs | 8.1037μs | 123.4006 KOps/s | 123.1867 KOps/s | |
test_stacked_getitem | 0.1865ms | 7.6559μs | 130.6174 KOps/s | 131.5814 KOps/s | |
test_lock_nested | 10.3945ms | 0.4768ms | 2.0973 KOps/s | 2.1396 KOps/s | |
test_lock_stack_nested | 0.6264ms | 0.4335ms | 2.3066 KOps/s | 2.3177 KOps/s | |
test_unlock_nested | 0.8778ms | 0.3883ms | 2.5753 KOps/s | 2.5831 KOps/s | |
test_unlock_stack_nested | 0.4484ms | 0.3512ms | 2.8472 KOps/s | 2.8612 KOps/s | |
test_flatten_speed | 0.4968ms | 0.1031ms | 9.6972 KOps/s | 9.4467 KOps/s | |
test_unflatten_speed | 0.3621ms | 0.3075ms | 3.2523 KOps/s | 3.5246 KOps/s | |
test_common_ops | 1.6893ms | 1.3382ms | 747.2975 Ops/s | 757.1431 Ops/s | |
test_creation | 16.7700μs | 1.6227μs | 616.2576 KOps/s | 610.5394 KOps/s | |
test_creation_empty | 45.3010μs | 17.7641μs | 56.2935 KOps/s | 54.3382 KOps/s | |
test_creation_nested_1 | 93.7420μs | 20.1641μs | 49.5932 KOps/s | 49.0392 KOps/s | |
test_creation_nested_2 | 52.5610μs | 22.5920μs | 44.2635 KOps/s | 41.6181 KOps/s | |
test_clone | 0.1943ms | 29.7814μs | 33.5780 KOps/s | 33.9857 KOps/s | |
test_getitem[int] | 1.1320ms | 17.6010μs | 56.8149 KOps/s | 58.8920 KOps/s | |
test_getitem[slice_int] | 0.1737ms | 29.0158μs | 34.4639 KOps/s | 35.3066 KOps/s | |
test_getitem[range] | 0.2836ms | 0.1135ms | 8.8074 KOps/s | 8.9732 KOps/s | |
test_getitem[tuple] | 0.1914ms | 24.6405μs | 40.5836 KOps/s | 39.0169 KOps/s | |
test_getitem[list] | 95.9818ms | 0.1144ms | 8.7391 KOps/s | 9.9116 KOps/s | |
test_setitem_dim[int] | 0.1830ms | 53.4119μs | 18.7224 KOps/s | 18.3540 KOps/s | |
test_setitem_dim[slice_int] | 0.2451ms | 78.7656μs | 12.6959 KOps/s | 12.5158 KOps/s | |
test_setitem_dim[range] | 0.2849ms | 0.1401ms | 7.1398 KOps/s | 7.1197 KOps/s | |
test_setitem_dim[tuple] | 0.2195ms | 71.7385μs | 13.9395 KOps/s | 14.1057 KOps/s | |
test_setitem | 0.1901ms | 42.7861μs | 23.3721 KOps/s | 21.4645 KOps/s | |
test_set | 0.2303ms | 44.8899μs | 22.2767 KOps/s | 23.8903 KOps/s | |
test_set_shared | 0.3751ms | 53.9315μs | 18.5420 KOps/s | 19.1392 KOps/s | |
test_update | 0.2484ms | 54.5564μs | 18.3296 KOps/s | 19.5024 KOps/s | |
test_update_nested | 0.2433ms | 58.4477μs | 17.1093 KOps/s | 16.9106 KOps/s | |
test_update__nested | 0.2389ms | 59.4432μs | 16.8228 KOps/s | 16.7286 KOps/s | |
test_set_nested | 0.2278ms | 44.9113μs | 22.2661 KOps/s | 22.5020 KOps/s | |
test_set_nested_new | 0.1957ms | 49.6812μs | 20.1283 KOps/s | 20.8100 KOps/s | |
test_select | 0.2441ms | 63.2552μs | 15.8090 KOps/s | 15.8854 KOps/s | |
test_select_nested | 0.4485ms | 52.4109μs | 19.0800 KOps/s | 19.6837 KOps/s | |
test_exclude_nested | 0.1017ms | 69.4567μs | 14.3975 KOps/s | 14.4786 KOps/s | |
test_empty[True] | 0.3281ms | 0.2801ms | 3.5703 KOps/s | 3.5877 KOps/s | |
test_empty[False] | 3.1591μs | 0.8547μs | 1.1700 MOps/s | 1.1613 MOps/s | |
test_to | 65.9920μs | 26.4181μs | 37.8529 KOps/s | 37.3161 KOps/s | |
test_to_nonblocking | 50.7710μs | 25.4027μs | 39.3658 KOps/s | 39.5193 KOps/s | |
test_unbind_speed | 0.4440ms | 0.3046ms | 3.2833 KOps/s | 3.3185 KOps/s | |
test_unbind_speed_stack0 | 0.4405ms | 0.2973ms | 3.3639 KOps/s | 3.3052 KOps/s | |
test_unbind_speed_stack1 | 92.9822ms | 0.7651ms | 1.3069 KOps/s | 1.2860 KOps/s | |
test_split | 94.7434ms | 2.3404ms | 427.2818 Ops/s | 425.7900 Ops/s | |
test_chunk | 97.5985ms | 2.3113ms | 432.6617 Ops/s | 426.5588 Ops/s | |
test_creation[device0] | 0.2305ms | 0.1037ms | 9.6409 KOps/s | 9.6383 KOps/s | |
test_creation_from_tensor | 0.2975ms | 0.1051ms | 9.5165 KOps/s | 9.9773 KOps/s | |
test_add_one[memmap_tensor0] | 0.1793ms | 8.9914μs | 111.2172 KOps/s | 115.2092 KOps/s | |
test_contiguous[memmap_tensor0] | 19.5910μs | 2.1310μs | 469.2525 KOps/s | 463.0900 KOps/s | |
test_stack[memmap_tensor0] | 0.1818ms | 6.5226μs | 153.3123 KOps/s | 153.2391 KOps/s | |
test_memmaptd_index | 1.2376ms | 0.4408ms | 2.2687 KOps/s | 2.3335 KOps/s | |
test_memmaptd_index_astensor | 0.7893ms | 0.4984ms | 2.0063 KOps/s | 2.0150 KOps/s | |
test_memmaptd_index_op | 1.5829ms | 1.0799ms | 925.9898 Ops/s | 949.4534 Ops/s | |
test_serialize_model | 93.6437ms | 89.4080ms | 11.1847 Ops/s | 10.8952 Ops/s | |
test_serialize_model_pickle | 1.3611s | 1.2392s | 0.8070 Ops/s | 0.8059 Ops/s | |
test_serialize_weights | 0.1944s | 98.1708ms | 10.1863 Ops/s | 11.0027 Ops/s | |
test_serialize_weights_returnearly | 0.2060s | 66.0288ms | 15.1449 Ops/s | 16.2317 Ops/s | |
test_serialize_weights_pickle | 1.3526s | 1.2372s | 0.8083 Ops/s | 0.8082 Ops/s | |
test_reshape_pytree | 0.2417ms | 39.2029μs | 25.5083 KOps/s | 25.0249 KOps/s | |
test_reshape_td | 0.1929ms | 44.5621μs | 22.4406 KOps/s | 21.7982 KOps/s | |
test_view_pytree | 0.2623ms | 36.8663μs | 27.1250 KOps/s | 27.1621 KOps/s | |
test_view_td | 0.1563ms | 49.4277μs | 20.2316 KOps/s | 21.0259 KOps/s | |
test_unbind_pytree | 0.2283ms | 36.3860μs | 27.4831 KOps/s | 27.3808 KOps/s | |
test_unbind_td | 0.4183ms | 44.4948μs | 22.4746 KOps/s | 22.3745 KOps/s | |
test_split_pytree | 0.3617ms | 50.6613μs | 19.7389 KOps/s | 20.1031 KOps/s | |
test_split_td | 96.7878ms | 68.3424μs | 14.6322 KOps/s | 14.2006 KOps/s | |
test_add_pytree | 0.2051ms | 55.5188μs | 18.0119 KOps/s | 17.6380 KOps/s | |
test_add_td | 0.3095ms | 94.9253μs | 10.5346 KOps/s | 10.7080 KOps/s | |
test_compile_add_one_nested[tensordict-compile] | 0.4164ms | 0.2147ms | 4.6570 KOps/s | 4.6742 KOps/s | |
test_compile_add_one_nested[tensordict-eager] | 0.3281ms | 0.1742ms | 5.7416 KOps/s | 5.8165 KOps/s | |
test_compile_add_one_nested[pytree-compile] | 0.3134ms | 0.1475ms | 6.7781 KOps/s | 6.6084 KOps/s | |
test_compile_add_one_nested[pytree-eager] | 0.3775ms | 0.1850ms | 5.4062 KOps/s | 5.3694 KOps/s | |
test_compile_copy_nested[tensordict-compile] | 0.1482ms | 21.9123μs | 45.6365 KOps/s | 45.7991 KOps/s | |
test_compile_copy_nested[tensordict-eager] | 0.1924ms | 47.6843μs | 20.9713 KOps/s | 21.2852 KOps/s | |
test_compile_copy_nested[pytree-compile] | 0.1392ms | 72.3732μs | 13.8173 KOps/s | 13.7376 KOps/s | |
test_compile_copy_nested[pytree-eager] | 0.2055ms | 59.0302μs | 16.9405 KOps/s | 16.8294 KOps/s | |
test_compile_add_one_flat[tensordict-compile] | 0.6141ms | 0.3301ms | 3.0295 KOps/s | 3.0250 KOps/s | |
test_compile_add_one_flat[tensordict-eager] | 0.3912ms | 0.2210ms | 4.5249 KOps/s | 4.5404 KOps/s | |
test_compile_add_one_flat[tensorclass-compile] | 0.3023ms | 0.1337ms | 7.4801 KOps/s | 7.1980 KOps/s | |
test_compile_add_one_flat[tensorclass-eager] | 0.2505ms | 65.4559μs | 15.2775 KOps/s | 16.2403 KOps/s | |
test_compile_add_one_flat[pytree-compile] | 0.4671ms | 0.3278ms | 3.0510 KOps/s | 3.0384 KOps/s | |
test_compile_add_one_flat[pytree-eager] | 0.7895ms | 0.6078ms | 1.6453 KOps/s | 1.6514 KOps/s | |
test_compile_add_self_flat[tensordict-eager] | 0.4159ms | 0.2674ms | 3.7399 KOps/s | 3.7341 KOps/s | |
test_compile_add_self_flat[tensordict-compile] | 0.4404ms | 0.3325ms | 3.0074 KOps/s | 2.9983 KOps/s | |
test_compile_add_self_flat[tensorclass-eager] | 0.2212ms | 75.6263μs | 13.2229 KOps/s | 13.5227 KOps/s | |
test_compile_add_self_flat[tensorclass-compile] | 0.2977ms | 0.1369ms | 7.3046 KOps/s | 7.3658 KOps/s | |
test_compile_add_self_flat[pytree-eager] | 0.6885ms | 0.5065ms | 1.9742 KOps/s | 1.9426 KOps/s | |
test_compile_add_self_flat[pytree-compile] | 0.4634ms | 0.3272ms | 3.0564 KOps/s | 3.0412 KOps/s | |
test_compile_copy_flat[tensordict-compile] | 0.1594ms | 18.9706μs | 52.7131 KOps/s | 55.4228 KOps/s | |
test_compile_copy_flat[tensordict-eager] | 54.4010μs | 31.4898μs | 31.7563 KOps/s | 30.9740 KOps/s | |
test_compile_copy_flat[pytree-compile] | 0.1093ms | 75.9363μs | 13.1689 KOps/s | 13.2406 KOps/s | |
test_compile_copy_flat[pytree-eager] | 87.8110μs | 59.9541μs | 16.6794 KOps/s | 16.4994 KOps/s | |
test_compile_assign_and_add[tensordict-compile] | 2.4594ms | 0.8388ms | 1.1922 KOps/s | 1.0906 KOps/s | |
test_compile_assign_and_add[tensordict-eager] | 3.6365ms | 3.2962ms | 303.3772 Ops/s | 307.3915 Ops/s | |
test_compile_assign_and_add[pytree-compile] | 2.4248ms | 0.8281ms | 1.2075 KOps/s | 1.1123 KOps/s | |
test_compile_assign_and_add[pytree-eager] | 3.5212ms | 3.1627ms | 316.1895 Ops/s | 310.6465 Ops/s | |
test_compile_indexing[tensor-tensordict-compile] | 0.2634ms | 0.1150ms | 8.6967 KOps/s | 8.7516 KOps/s | |
test_compile_indexing[tensor-tensordict-eager] | 0.2439ms | 65.4362μs | 15.2821 KOps/s | 15.3250 KOps/s | |
test_compile_indexing[tensor-tensorclass-compile] | 0.3008ms | 0.1047ms | 9.5499 KOps/s | 9.1015 KOps/s | |
test_compile_indexing[tensor-tensorclass-eager] | 0.2128ms | 45.4191μs | 22.0172 KOps/s | 20.8227 KOps/s | |
test_compile_indexing[tensor-pytree-compile] | 0.2812ms | 0.1054ms | 9.4850 KOps/s | 8.9701 KOps/s | |
test_compile_indexing[tensor-pytree-eager] | 0.2493ms | 45.9047μs | 21.7843 KOps/s | 21.9498 KOps/s | |
test_compile_indexing[slice-tensordict-compile] | 0.3045ms | 0.1424ms | 7.0234 KOps/s | 7.0913 KOps/s | |
test_compile_indexing[slice-tensordict-eager] | 0.1766ms | 25.7466μs | 38.8401 KOps/s | 38.5085 KOps/s | |
test_compile_indexing[slice-tensorclass-compile] | 0.3034ms | 0.1371ms | 7.2927 KOps/s | 7.5084 KOps/s | |
test_compile_indexing[slice-tensorclass-eager] | 86.6720μs | 22.2821μs | 44.8791 KOps/s | 45.0132 KOps/s | |
test_compile_indexing[slice-pytree-compile] | 0.2955ms | 0.1360ms | 7.3552 KOps/s | 7.5240 KOps/s | |
test_compile_indexing[slice-pytree-eager] | 58.6110μs | 21.9333μs | 45.5927 KOps/s | 43.3829 KOps/s | |
test_compile_indexing[int-tensordict-compile] | 0.2987ms | 0.1407ms | 7.1096 KOps/s | 6.7736 KOps/s | |
test_compile_indexing[int-tensordict-eager] | 0.4991ms | 25.1119μs | 39.8217 KOps/s | 33.5131 KOps/s | |
test_compile_indexing[int-tensorclass-compile] | 0.2871ms | 0.1327ms | 7.5343 KOps/s | 7.1579 KOps/s | |
test_compile_indexing[int-tensorclass-eager] | 51.4710μs | 21.8064μs | 45.8580 KOps/s | 41.9299 KOps/s | |
test_compile_indexing[int-pytree-compile] | 0.3143ms | 0.1356ms | 7.3758 KOps/s | 7.5322 KOps/s | |
test_compile_indexing[int-pytree-eager] | 52.9610μs | 21.6448μs | 46.2006 KOps/s | 45.1448 KOps/s | |
test_mod_add[eager] | 0.1945ms | 33.9559μs | 29.4500 KOps/s | 29.0980 KOps/s | |
test_mod_add[compile] | 0.2190ms | 70.6468μs | 14.1549 KOps/s | 14.2204 KOps/s | |
test_mod_add[compile-overhead] | 0.2752ms | 0.1414ms | 7.0700 KOps/s | 6.6876 KOps/s | |
test_mod_wrap[eager] | 0.4354ms | 0.2464ms | 4.0585 KOps/s | 4.0616 KOps/s | |
test_mod_wrap[compile] | 1.0979ms | 0.2925ms | 3.4183 KOps/s | 3.3071 KOps/s | |
test_mod_wrap[compile-overhead] | 8.2095ms | 4.3297ms | 230.9615 Ops/s | 228.4069 Ops/s | |
test_mod_wrap_and_backward[eager] | 1.6594ms | 1.3414ms | 745.5121 Ops/s | 690.9436 Ops/s | |
test_mod_wrap_and_backward[compile] | 1.5019ms | 1.3336ms | 749.8301 Ops/s | 678.1777 Ops/s | |
test_mod_wrap_and_backward[compile-overhead] | 1.3301ms | 0.9294ms | 1.0760 KOps/s | 986.2470 Ops/s | |
test_seq_add[eager] | 0.2622ms | 98.2476μs | 10.1784 KOps/s | 10.0941 KOps/s | |
test_seq_add[compile] | 0.2350ms | 81.9381μs | 12.2043 KOps/s | 12.2446 KOps/s | |
test_seq_add[compile-overhead] | 0.2887ms | 0.1176ms | 8.5063 KOps/s | 8.5024 KOps/s | |
test_seq_wrap[eager] | 0.5338ms | 0.3885ms | 2.5737 KOps/s | 2.5631 KOps/s | |
test_seq_wrap[compile] | 0.5113ms | 0.3136ms | 3.1891 KOps/s | 3.0016 KOps/s | |
test_seq_wrap[compile-overhead] | 0.3725ms | 0.2274ms | 4.3977 KOps/s | 4.3137 KOps/s | |
test_func_call_runtime[False-eager] | 0.9424ms | 0.7571ms | 1.3208 KOps/s | 1.3417 KOps/s | |
test_func_call_runtime[False-compile] | 0.9851ms | 0.8097ms | 1.2350 KOps/s | 1.2533 KOps/s | |
test_func_call_runtime[False-compile-overhead] | 0.5050ms | 0.3710ms | 2.6953 KOps/s | 2.7140 KOps/s | |
test_func_call_runtime[True-eager] | 1.0944ms | 0.9243ms | 1.0819 KOps/s | 1.0788 KOps/s | |
test_func_call_runtime[True-compile] | 1.0117ms | 0.8395ms | 1.1912 KOps/s | 1.1800 KOps/s | |
test_func_call_runtime[True-compile-overhead] | 0.5797ms | 0.4143ms | 2.4138 KOps/s | 2.3934 KOps/s | |
test_func_call_cm_runtime[False-eager] | 0.9366ms | 0.7696ms | 1.2993 KOps/s | 1.2830 KOps/s | |
test_func_call_cm_runtime[False-compile] | 1.0328ms | 0.8315ms | 1.2026 KOps/s | 1.2250 KOps/s | |
test_func_call_cm_runtime[False-compile-overhead] | 0.5491ms | 0.3775ms | 2.6490 KOps/s | 2.7043 KOps/s | |
test_func_call_cm_runtime[True-eager] | 1.2124ms | 1.0282ms | 972.5455 Ops/s | 960.3480 Ops/s | |
test_func_call_cm_runtime[True-compile] | 1.1753ms | 0.9987ms | 1.0013 KOps/s | 953.1430 Ops/s | |
test_func_call_cm_runtime[True-compile-overhead] | 1.2474ms | 1.0018ms | 998.1961 Ops/s | 993.1546 Ops/s | |
test_distributed | 0.3308ms | 72.2553μs | 13.8398 KOps/s | 13.7449 KOps/s | |
test_tdmodule | 0.1431ms | 17.0830μs | 58.5377 KOps/s | 59.0100 KOps/s | |
test_tdmodule_dispatch | 0.1739ms | 34.8234μs | 28.7163 KOps/s | 30.0280 KOps/s | |
test_tdseq | 35.6600μs | 17.5558μs | 56.9612 KOps/s | 55.7614 KOps/s | |
test_tdseq_dispatch | 0.1751ms | 36.7719μs | 27.1947 KOps/s | 26.8795 KOps/s | |
test_instantiation_functorch | 2.2072ms | 1.9792ms | 505.2628 Ops/s | 499.2107 Ops/s | |
test_instantiation_td | 2.2386ms | 1.2898ms | 775.3355 Ops/s | 761.3316 Ops/s | |
test_exec_functorch | 0.3673ms | 0.2240ms | 4.4634 KOps/s | 4.6677 KOps/s | |
test_exec_functional_call | 0.4125ms | 0.2221ms | 4.5030 KOps/s | 4.6723 KOps/s | |
test_exec_td | 0.4173ms | 0.2308ms | 4.3330 KOps/s | 4.6054 KOps/s | |
test_exec_td_decorator | 1.1341ms | 0.2820ms | 3.5461 KOps/s | 3.6516 KOps/s | |
test_vmap_mlp_speed[True-True] | 0.8991ms | 0.6649ms | 1.5041 KOps/s | 1.5332 KOps/s | |
test_vmap_mlp_speed[True-False] | 0.8598ms | 0.6602ms | 1.5148 KOps/s | 1.5372 KOps/s | |
test_vmap_mlp_speed[False-True] | 0.9875ms | 0.5591ms | 1.7886 KOps/s | 1.7342 KOps/s | |
test_vmap_mlp_speed[False-False] | 0.7774ms | 0.5663ms | 1.7658 KOps/s | 1.7527 KOps/s | |
test_vmap_mlp_speed_decorator[True-True] | 0.8829ms | 0.6902ms | 1.4488 KOps/s | 1.4364 KOps/s | |
test_vmap_mlp_speed_decorator[True-False] | 1.0267ms | 0.6927ms | 1.4436 KOps/s | 1.4317 KOps/s | |
test_vmap_mlp_speed_decorator[False-True] | 0.8083ms | 0.6096ms | 1.6405 KOps/s | 1.6490 KOps/s | |
test_vmap_mlp_speed_decorator[False-False] | 0.8297ms | 0.5996ms | 1.6679 KOps/s | 1.6450 KOps/s | |
test_vmap_transformer_speed[True-True] | 9.0748ms | 8.7052ms | 114.8745 Ops/s | 114.9565 Ops/s | |
test_vmap_transformer_speed[True-False] | 8.9616ms | 8.6127ms | 116.1071 Ops/s | 114.8879 Ops/s | |
test_vmap_transformer_speed[False-True] | 8.8609ms | 8.5276ms | 117.2666 Ops/s | 115.9097 Ops/s | |
test_vmap_transformer_speed[False-False] | 8.7791ms | 8.4967ms | 117.6921 Ops/s | 115.2166 Ops/s | |
test_vmap_transformer_speed_decorator[True-True] | 20.6089ms | 20.2913ms | 49.2823 Ops/s | 49.0385 Ops/s | |
test_vmap_transformer_speed_decorator[True-False] | 21.2659ms | 20.3498ms | 49.1405 Ops/s | 49.0973 Ops/s | |
test_vmap_transformer_speed_decorator[False-True] | 20.5133ms | 20.1686ms | 49.5820 Ops/s | 49.6145 Ops/s | |
test_vmap_transformer_speed_decorator[False-False] | 20.9968ms | 20.1569ms | 49.6107 Ops/s | 48.9257 Ops/s | |
test_to_module_speed[True] | 1.2964ms | 1.1414ms | 876.1322 Ops/s | 888.2681 Ops/s | |
test_to_module_speed[False] | 1.6293ms | 1.1202ms | 892.6889 Ops/s | 900.5753 Ops/s | |
test_tc_init | 0.2404ms | 41.2442μs | 24.2458 KOps/s | 25.4787 KOps/s | |
test_tc_init_nested | 0.1360ms | 81.7382μs | 12.2342 KOps/s | 12.6262 KOps/s | |
test_tc_first_layer_tensor | 3.1914μs | 0.7775μs | 1.2861 MOps/s | 1.2964 MOps/s | |
test_tc_first_layer_nontensor | 17.2810μs | 2.5605μs | 390.5417 KOps/s | 396.8387 KOps/s | |
test_tc_second_layer_tensor | 41.2607μs | 1.6252μs | 615.2926 KOps/s | 630.3589 KOps/s | |
test_tc_second_layer_nontensor | 17.5710μs | 3.3825μs | 295.6432 KOps/s | 301.3063 KOps/s | |
test_unbind | 0.1854s | 10.5794ms | 94.5238 Ops/s | 61.4573 Ops/s | |
test_full_like | 0.1728s | 0.6648ms | 1.5043 KOps/s | 1.7292 KOps/s | |
test_zeros_like | 0.4153ms | 0.1981ms | 5.0475 KOps/s | 5.0451 KOps/s | |
test_ones_like | 0.3867ms | 0.1981ms | 5.0491 KOps/s | 5.0501 KOps/s | |
test_clone | 0.5733ms | 0.4153ms | 2.4078 KOps/s | 2.4148 KOps/s | |
test_squeeze | 0.1208ms | 10.6293μs | 94.0794 KOps/s | 93.1845 KOps/s | |
test_unsqueeze | 0.2169ms | 79.8577μs | 12.5223 KOps/s | 12.5163 KOps/s | |
test_split | 0.4352ms | 0.1704ms | 5.8676 KOps/s | 5.7597 KOps/s | |
test_permute | 0.3679ms | 0.1946ms | 5.1399 KOps/s | 5.3531 KOps/s | |
test_stack | 1.3996ms | 0.8651ms | 1.1560 KOps/s | 1.1155 KOps/s | |
test_cat | 1.3612ms | 1.2324ms | 811.4076 Ops/s | 811.4126 Ops/s |
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.
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.
No description provided.