Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Minor] Typo in lazy legacy warnings #639

Merged
merged 3 commits into from
Jan 26, 2024
Merged

[Minor] Typo in lazy legacy warnings #639

merged 3 commits into from
Jan 26, 2024

Conversation

vmoens
Copy link
Contributor

@vmoens vmoens commented Jan 26, 2024

No description provided.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Jan 26, 2024
@vmoens vmoens added the documentation Improvements or additions to documentation label Jan 26, 2024
@vmoens vmoens merged commit a5a7ab5 into main Jan 26, 2024
20 of 23 checks passed
@vmoens vmoens deleted the lazylegacy-typo branch January 26, 2024 11:03
Copy link

$\color{#D29922}\textsf{\Large⚠\kern{0.2cm}\normalsize Warning}$ Result of CPU Benchmark Tests

Total Benchmarks: 124. Improved: $\large\color{#35bf28}3$. Worsened: $\large\color{#d91a1a}13$.

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_plain_set_nested 39.3530μs 16.7492μs 59.7043 KOps/s 59.5339 KOps/s $\color{#35bf28}+0.29\%$
test_plain_set_stack_nested 0.1994ms 0.1459ms 6.8518 KOps/s 6.9326 KOps/s $\color{#d91a1a}-1.17\%$
test_plain_set_nested_inplace 55.1040μs 19.6001μs 51.0201 KOps/s 51.9830 KOps/s $\color{#d91a1a}-1.85\%$
test_plain_set_stack_nested_inplace 0.3369ms 0.1786ms 5.6006 KOps/s 5.6355 KOps/s $\color{#d91a1a}-0.62\%$
test_items 19.9680μs 2.5618μs 390.3491 KOps/s 395.2781 KOps/s $\color{#d91a1a}-1.25\%$
test_items_nested 0.4715ms 0.2715ms 3.6829 KOps/s 3.6669 KOps/s $\color{#35bf28}+0.44\%$
test_items_nested_locked 1.8175ms 0.2785ms 3.5907 KOps/s 3.6569 KOps/s $\color{#d91a1a}-1.81\%$
test_items_nested_leaf 0.3501ms 0.1683ms 5.9408 KOps/s 5.9643 KOps/s $\color{#d91a1a}-0.39\%$
test_items_stack_nested 1.6844ms 1.3194ms 757.9482 Ops/s 762.2831 Ops/s $\color{#d91a1a}-0.57\%$
test_items_stack_nested_leaf 1.7733ms 1.1965ms 835.7920 Ops/s 850.7508 Ops/s $\color{#d91a1a}-1.76\%$
test_items_stack_nested_locked 1.0346ms 0.8638ms 1.1576 KOps/s 1.1166 KOps/s $\color{#35bf28}+3.67\%$
test_keys 25.1570μs 3.8407μs 260.3682 KOps/s 260.6242 KOps/s $\color{#d91a1a}-0.10\%$
test_keys_nested 67.5358ms 0.1611ms 6.2065 KOps/s 6.7170 KOps/s $\textbf{\color{#d91a1a}-7.60\%}$
test_keys_nested_locked 0.2548ms 0.1533ms 6.5241 KOps/s 6.5081 KOps/s $\color{#35bf28}+0.25\%$
test_keys_nested_leaf 2.5858ms 0.1322ms 7.5652 KOps/s 7.6448 KOps/s $\color{#d91a1a}-1.04\%$
test_keys_stack_nested 1.5165ms 1.2706ms 787.0180 Ops/s 799.4296 Ops/s $\color{#d91a1a}-1.55\%$
test_keys_stack_nested_leaf 1.5371ms 1.2600ms 793.6502 Ops/s 800.2200 Ops/s $\color{#d91a1a}-0.82\%$
test_keys_stack_nested_locked 1.0565ms 0.8049ms 1.2424 KOps/s 1.2388 KOps/s $\color{#35bf28}+0.29\%$
test_values 9.7808μs 1.1693μs 855.2265 KOps/s 858.0372 KOps/s $\color{#d91a1a}-0.33\%$
test_values_nested 0.1004ms 53.7521μs 18.6039 KOps/s 18.9906 KOps/s $\color{#d91a1a}-2.04\%$
test_values_nested_locked 97.7020μs 53.4052μs 18.7248 KOps/s 18.8519 KOps/s $\color{#d91a1a}-0.67\%$
test_values_nested_leaf 98.6420μs 47.3850μs 21.1037 KOps/s 21.6082 KOps/s $\color{#d91a1a}-2.33\%$
test_values_stack_nested 1.6864ms 1.0327ms 968.3371 Ops/s 964.6173 Ops/s $\color{#35bf28}+0.39\%$
test_values_stack_nested_leaf 1.7972ms 1.0503ms 952.1052 Ops/s 989.9208 Ops/s $\color{#d91a1a}-3.82\%$
test_values_stack_nested_locked 0.7658ms 0.6005ms 1.6654 KOps/s 1.6784 KOps/s $\color{#d91a1a}-0.78\%$
test_membership 25.8980μs 1.3602μs 735.2011 KOps/s 735.2369 KOps/s $-0.00\%$
test_membership_nested 32.7410μs 3.5382μs 282.6330 KOps/s 296.2924 KOps/s $\color{#d91a1a}-4.61\%$
test_membership_nested_leaf 47.8390μs 3.6393μs 274.7762 KOps/s 292.7246 KOps/s $\textbf{\color{#d91a1a}-6.13\%}$
test_membership_stacked_nested 34.0740μs 12.3127μs 81.2173 KOps/s 84.2800 KOps/s $\color{#d91a1a}-3.63\%$
test_membership_stacked_nested_leaf 57.3450μs 12.3432μs 81.0161 KOps/s 83.6428 KOps/s $\color{#d91a1a}-3.14\%$
test_membership_nested_last 24.6460μs 6.8222μs 146.5811 KOps/s 150.8491 KOps/s $\color{#d91a1a}-2.83\%$
test_membership_nested_leaf_last 37.9510μs 6.9125μs 144.6653 KOps/s 147.7371 KOps/s $\color{#d91a1a}-2.08\%$
test_membership_stacked_nested_last 0.3255ms 0.1805ms 5.5387 KOps/s 5.6005 KOps/s $\color{#d91a1a}-1.10\%$
test_membership_stacked_nested_leaf_last 48.4100μs 14.5008μs 68.9618 KOps/s 71.3157 KOps/s $\color{#d91a1a}-3.30\%$
test_nested_getleaf 34.9150μs 10.9185μs 91.5878 KOps/s 93.3993 KOps/s $\color{#d91a1a}-1.94\%$
test_nested_get 54.3320μs 10.4668μs 95.5398 KOps/s 98.7042 KOps/s $\color{#d91a1a}-3.21\%$
test_stacked_getleaf 0.6933ms 0.3949ms 2.5322 KOps/s 2.5317 KOps/s $\color{#35bf28}+0.02\%$
test_stacked_get 0.6214ms 0.3647ms 2.7418 KOps/s 2.7556 KOps/s $\color{#d91a1a}-0.50\%$
test_nested_getitemleaf 38.9530μs 12.5643μs 79.5905 KOps/s 82.6560 KOps/s $\color{#d91a1a}-3.71\%$
test_nested_getitem 41.6670μs 11.8937μs 84.0784 KOps/s 87.3074 KOps/s $\color{#d91a1a}-3.70\%$
test_stacked_getitemleaf 0.5969ms 0.4003ms 2.4980 KOps/s 2.5333 KOps/s $\color{#d91a1a}-1.40\%$
test_stacked_getitem 0.5357ms 0.3717ms 2.6907 KOps/s 2.7400 KOps/s $\color{#d91a1a}-1.80\%$
test_lock_nested 1.0557ms 0.3434ms 2.9117 KOps/s 3.0262 KOps/s $\color{#d91a1a}-3.78\%$
test_lock_stack_nested 85.6840ms 5.9345ms 168.5065 Ops/s 177.1203 Ops/s $\color{#d91a1a}-4.86\%$
test_unlock_nested 1.1130ms 0.3453ms 2.8963 KOps/s 2.5184 KOps/s $\textbf{\color{#35bf28}+15.01\%}$
test_unlock_stack_nested 93.3684ms 6.1841ms 161.7051 Ops/s 172.1958 Ops/s $\textbf{\color{#d91a1a}-6.09\%}$
test_flatten_speed 0.6232ms 0.3725ms 2.6848 KOps/s 2.6971 KOps/s $\color{#d91a1a}-0.45\%$
test_unflatten_speed 0.7274ms 0.4709ms 2.1237 KOps/s 2.1518 KOps/s $\color{#d91a1a}-1.31\%$
test_common_ops 5.2594ms 0.6953ms 1.4383 KOps/s 1.5144 KOps/s $\textbf{\color{#d91a1a}-5.03\%}$
test_creation 17.2520μs 1.8689μs 535.0849 KOps/s 523.9184 KOps/s $\color{#35bf28}+2.13\%$
test_creation_empty 39.1530μs 10.0651μs 99.3530 KOps/s 104.0580 KOps/s $\color{#d91a1a}-4.52\%$
test_creation_nested_1 60.6720μs 12.6280μs 79.1894 KOps/s 82.0911 KOps/s $\color{#d91a1a}-3.53\%$
test_creation_nested_2 38.3820μs 15.8733μs 62.9989 KOps/s 63.3637 KOps/s $\color{#d91a1a}-0.58\%$
test_clone 0.1392ms 13.0449μs 76.6585 KOps/s 78.4074 KOps/s $\color{#d91a1a}-2.23\%$
test_getitem[int] 39.9350μs 10.9372μs 91.4314 KOps/s 91.8545 KOps/s $\color{#d91a1a}-0.46\%$
test_getitem[slice_int] 65.6720μs 21.9031μs 45.6557 KOps/s 45.6271 KOps/s $\color{#35bf28}+0.06\%$
test_getitem[range] 0.2643ms 40.5941μs 24.6341 KOps/s 24.8423 KOps/s $\color{#d91a1a}-0.84\%$
test_getitem[tuple] 55.1230μs 18.1125μs 55.2105 KOps/s 55.4686 KOps/s $\color{#d91a1a}-0.47\%$
test_getitem[list] 0.2913ms 35.4674μs 28.1949 KOps/s 28.2062 KOps/s $\color{#d91a1a}-0.04\%$
test_setitem_dim[int] 51.5460μs 29.3911μs 34.0239 KOps/s 34.7174 KOps/s $\color{#d91a1a}-2.00\%$
test_setitem_dim[slice_int] 0.1001ms 55.7847μs 17.9260 KOps/s 18.2255 KOps/s $\color{#d91a1a}-1.64\%$
test_setitem_dim[range] 0.1437ms 74.6727μs 13.3918 KOps/s 13.8870 KOps/s $\color{#d91a1a}-3.57\%$
test_setitem_dim[tuple] 65.2510μs 43.7932μs 22.8346 KOps/s 23.3789 KOps/s $\color{#d91a1a}-2.33\%$
test_setitem 0.1313ms 19.4738μs 51.3511 KOps/s 53.7883 KOps/s $\color{#d91a1a}-4.53\%$
test_set 0.1596ms 19.0162μs 52.5866 KOps/s 55.0468 KOps/s $\color{#d91a1a}-4.47\%$
test_set_shared 3.5077ms 0.1444ms 6.9234 KOps/s 7.3284 KOps/s $\textbf{\color{#d91a1a}-5.53\%}$
test_update 0.2789ms 21.5312μs 46.4442 KOps/s 48.0749 KOps/s $\color{#d91a1a}-3.39\%$
test_update_nested 0.1533ms 29.1242μs 34.3357 KOps/s 35.6060 KOps/s $\color{#d91a1a}-3.57\%$
test_set_nested 0.1357ms 20.7599μs 48.1697 KOps/s 50.0089 KOps/s $\color{#d91a1a}-3.68\%$
test_set_nested_new 0.1455ms 24.9178μs 40.1320 KOps/s 42.3971 KOps/s $\textbf{\color{#d91a1a}-5.34\%}$
test_select 0.1264ms 38.1484μs 26.2134 KOps/s 27.2133 KOps/s $\color{#d91a1a}-3.67\%$
test_select_nested 0.1081ms 57.6112μs 17.3577 KOps/s 16.7960 KOps/s $\color{#35bf28}+3.34\%$
test_exclude_nested 0.1974ms 0.1201ms 8.3235 KOps/s 8.3670 KOps/s $\color{#d91a1a}-0.52\%$
test_empty[True] 0.7211ms 0.4153ms 2.4082 KOps/s 2.3716 KOps/s $\color{#35bf28}+1.54\%$
test_empty[False] 6.5662μs 1.0432μs 958.5498 KOps/s 964.4166 KOps/s $\color{#d91a1a}-0.61\%$
test_unbind_speed 0.3051ms 0.2445ms 4.0899 KOps/s 4.1429 KOps/s $\color{#d91a1a}-1.28\%$
test_unbind_speed_stack0 91.6690ms 3.4906ms 286.4869 Ops/s 339.6950 Ops/s $\textbf{\color{#d91a1a}-15.66\%}$
test_unbind_speed_stack1 33.5120μs 1.9739μs 506.6144 KOps/s 491.5648 KOps/s $\color{#35bf28}+3.06\%$
test_split 2.4497ms 1.4737ms 678.5702 Ops/s 677.9892 Ops/s $\color{#35bf28}+0.09\%$
test_chunk 82.8420ms 1.5864ms 630.3580 Ops/s 679.6796 Ops/s $\textbf{\color{#d91a1a}-7.26\%}$
test_creation[device0] 0.2167ms 0.1026ms 9.7434 KOps/s 9.6967 KOps/s $\color{#35bf28}+0.48\%$
test_creation_from_tensor 4.7910ms 82.1976μs 12.1658 KOps/s 12.4579 KOps/s $\color{#d91a1a}-2.34\%$
test_add_one[memmap_tensor0] 0.5253ms 5.3651μs 186.3901 KOps/s 188.9783 KOps/s $\color{#d91a1a}-1.37\%$
test_contiguous[memmap_tensor0] 13.7250μs 0.6418μs 1.5582 MOps/s 1.5683 MOps/s $\color{#d91a1a}-0.64\%$
test_stack[memmap_tensor0] 0.1561ms 3.6294μs 275.5278 KOps/s 284.9383 KOps/s $\color{#d91a1a}-3.30\%$
test_memmaptd_index 0.8332ms 0.2231ms 4.4818 KOps/s 4.6487 KOps/s $\color{#d91a1a}-3.59\%$
test_memmaptd_index_astensor 0.6866ms 0.2884ms 3.4674 KOps/s 3.6220 KOps/s $\color{#d91a1a}-4.27\%$
test_memmaptd_index_op 0.9486ms 0.5708ms 1.7518 KOps/s 1.8026 KOps/s $\color{#d91a1a}-2.82\%$
test_serialize_model 0.1806s 0.1153s 8.6736 Ops/s 9.7369 Ops/s $\textbf{\color{#d91a1a}-10.92\%}$
test_serialize_model_pickle 0.4670s 0.3776s 2.6481 Ops/s 2.6412 Ops/s $\color{#35bf28}+0.26\%$
test_serialize_weights 0.1817s 0.1080s 9.2603 Ops/s 9.3527 Ops/s $\color{#d91a1a}-0.99\%$
test_serialize_weights_returnearly 0.2028s 0.1327s 7.5383 Ops/s 7.0670 Ops/s $\textbf{\color{#35bf28}+6.67\%}$
test_serialize_weights_pickle 0.9008s 0.5908s 1.6925 Ops/s 1.5559 Ops/s $\textbf{\color{#35bf28}+8.78\%}$
test_serialize_weights_filesystem 99.4190ms 92.7924ms 10.7767 Ops/s 11.0325 Ops/s $\color{#d91a1a}-2.32\%$
test_serialize_model_filesystem 0.1049s 95.7638ms 10.4424 Ops/s 9.9673 Ops/s $\color{#35bf28}+4.77\%$
test_reshape_pytree 49.4830μs 23.4162μs 42.7055 KOps/s 42.3914 KOps/s $\color{#35bf28}+0.74\%$
test_reshape_td 65.4730μs 30.0393μs 33.2898 KOps/s 34.0678 KOps/s $\color{#d91a1a}-2.28\%$
test_view_pytree 88.9870μs 23.6916μs 42.2090 KOps/s 43.2104 KOps/s $\color{#d91a1a}-2.32\%$
test_view_td 76.2087ms 11.5165μs 86.8317 KOps/s 88.5743 KOps/s $\color{#d91a1a}-1.97\%$
test_unbind_pytree 68.4280μs 26.4340μs 37.8300 KOps/s 38.3720 KOps/s $\color{#d91a1a}-1.41\%$
test_unbind_td 0.4866ms 39.2926μs 25.4501 KOps/s 28.1556 KOps/s $\textbf{\color{#d91a1a}-9.61\%}$
test_split_pytree 70.3410μs 26.0317μs 38.4147 KOps/s 38.6379 KOps/s $\color{#d91a1a}-0.58\%$
test_split_td 0.1125ms 39.6989μs 25.1896 KOps/s 25.0518 KOps/s $\color{#35bf28}+0.55\%$
test_add_pytree 81.7030μs 32.6724μs 30.6069 KOps/s 30.1233 KOps/s $\color{#35bf28}+1.61\%$
test_add_td 0.1017ms 50.2464μs 19.9019 KOps/s 19.9959 KOps/s $\color{#d91a1a}-0.47\%$
test_distributed 0.2414ms 99.6183μs 10.0383 KOps/s 9.7346 KOps/s $\color{#35bf28}+3.12\%$
test_tdmodule 0.3355ms 22.6646μs 44.1216 KOps/s 47.0420 KOps/s $\textbf{\color{#d91a1a}-6.21\%}$
test_tdmodule_dispatch 0.2564ms 44.0267μs 22.7135 KOps/s 23.2336 KOps/s $\color{#d91a1a}-2.24\%$
test_tdseq 0.1398ms 25.2359μs 39.6261 KOps/s 40.6619 KOps/s $\color{#d91a1a}-2.55\%$
test_tdseq_dispatch 0.5214ms 47.8334μs 20.9059 KOps/s 22.0731 KOps/s $\textbf{\color{#d91a1a}-5.29\%}$
test_instantiation_functorch 2.2774ms 1.2835ms 779.1016 Ops/s 764.9180 Ops/s $\color{#35bf28}+1.85\%$
test_instantiation_td 0.1045s 1.1102ms 900.7301 Ops/s 1.0076 KOps/s $\textbf{\color{#d91a1a}-10.60\%}$
test_exec_functorch 0.2855ms 0.1602ms 6.2410 KOps/s 6.3131 KOps/s $\color{#d91a1a}-1.14\%$
test_exec_functional_call 0.2936ms 0.1502ms 6.6557 KOps/s 6.6735 KOps/s $\color{#d91a1a}-0.27\%$
test_exec_td 0.4027ms 0.1447ms 6.9114 KOps/s 6.9435 KOps/s $\color{#d91a1a}-0.46\%$
test_exec_td_decorator 0.9840ms 0.1769ms 5.6517 KOps/s 5.6414 KOps/s $\color{#35bf28}+0.18\%$
test_vmap_mlp_speed[True-True] 1.3028ms 0.8707ms 1.1485 KOps/s 1.1404 KOps/s $\color{#35bf28}+0.71\%$
test_vmap_mlp_speed[True-False] 0.8247ms 0.4723ms 2.1175 KOps/s 2.1106 KOps/s $\color{#35bf28}+0.32\%$
test_vmap_mlp_speed[False-True] 1.2889ms 0.7728ms 1.2939 KOps/s 1.3088 KOps/s $\color{#d91a1a}-1.13\%$
test_vmap_mlp_speed[False-False] 0.6237ms 0.3824ms 2.6154 KOps/s 2.5665 KOps/s $\color{#35bf28}+1.90\%$
test_vmap_mlp_speed_decorator[True-True] 2.8424ms 2.2735ms 439.8562 Ops/s 432.1335 Ops/s $\color{#35bf28}+1.79\%$
test_vmap_mlp_speed_decorator[True-False] 0.9637ms 0.5185ms 1.9285 KOps/s 1.9286 KOps/s $-0.00\%$
test_vmap_mlp_speed_decorator[False-True] 2.3973ms 1.8518ms 540.0270 Ops/s 545.1101 Ops/s $\color{#d91a1a}-0.93\%$
test_vmap_mlp_speed_decorator[False-False] 0.7646ms 0.3955ms 2.5285 KOps/s 2.5048 KOps/s $\color{#35bf28}+0.95\%$

Copy link

$\color{#D29922}\textsf{\Large⚠\kern{0.2cm}\normalsize Warning}$ Result of GPU Benchmark Tests

Total Benchmarks: 132. Improved: $\large\color{#35bf28}8$. Worsened: $\large\color{#d91a1a}28$.

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_plain_set_nested 0.4810ms 14.5445μs 68.7546 KOps/s 76.0798 KOps/s $\textbf{\color{#d91a1a}-9.63\%}$
test_plain_set_stack_nested 0.1542ms 0.1203ms 8.3142 KOps/s 8.4065 KOps/s $\color{#d91a1a}-1.10\%$
test_plain_set_nested_inplace 34.3310μs 16.0056μs 62.4782 KOps/s 69.0887 KOps/s $\textbf{\color{#d91a1a}-9.57\%}$
test_plain_set_stack_nested_inplace 0.1833ms 0.1502ms 6.6594 KOps/s 6.6959 KOps/s $\color{#d91a1a}-0.55\%$
test_items 29.2610μs 4.8147μs 207.6992 KOps/s 209.5815 KOps/s $\color{#d91a1a}-0.90\%$
test_items_nested 0.3631ms 0.3419ms 2.9245 KOps/s 2.9032 KOps/s $\color{#35bf28}+0.73\%$
test_items_nested_locked 0.3641ms 0.3453ms 2.8961 KOps/s 2.8785 KOps/s $\color{#35bf28}+0.61\%$
test_items_nested_leaf 0.2370ms 0.2030ms 4.9260 KOps/s 4.8982 KOps/s $\color{#35bf28}+0.57\%$
test_items_stack_nested 1.3946ms 1.3262ms 754.0275 Ops/s 750.5382 Ops/s $\color{#35bf28}+0.46\%$
test_items_stack_nested_leaf 1.2162ms 1.1613ms 861.0839 Ops/s 857.5540 Ops/s $\color{#35bf28}+0.41\%$
test_items_stack_nested_locked 1.9790ms 0.9043ms 1.1058 KOps/s 1.0878 KOps/s $\color{#35bf28}+1.66\%$
test_keys 31.6510μs 4.6382μs 215.5999 KOps/s 216.8733 KOps/s $\color{#d91a1a}-0.59\%$
test_keys_nested 0.5147ms 94.2108μs 10.6145 KOps/s 10.5789 KOps/s $\color{#35bf28}+0.34\%$
test_keys_nested_locked 0.1271ms 97.5002μs 10.2564 KOps/s 10.2494 KOps/s $\color{#35bf28}+0.07\%$
test_keys_nested_leaf 0.1850ms 77.5996μs 12.8867 KOps/s 12.7422 KOps/s $\color{#35bf28}+1.13\%$
test_keys_stack_nested 1.2049ms 1.1565ms 864.6412 Ops/s 847.4171 Ops/s $\color{#35bf28}+2.03\%$
test_keys_stack_nested_leaf 1.1940ms 1.1534ms 867.0108 Ops/s 851.5679 Ops/s $\color{#35bf28}+1.81\%$
test_keys_stack_nested_locked 0.7811ms 0.7346ms 1.3613 KOps/s 1.3319 KOps/s $\color{#35bf28}+2.20\%$
test_values 9.7103μs 1.9008μs 526.1076 KOps/s 528.7463 KOps/s $\color{#d91a1a}-0.50\%$
test_values_nested 71.3520μs 44.9670μs 22.2385 KOps/s 22.0725 KOps/s $\color{#35bf28}+0.75\%$
test_values_nested_locked 68.4510μs 47.4054μs 21.0946 KOps/s 20.9700 KOps/s $\color{#35bf28}+0.59\%$
test_values_nested_leaf 63.7810μs 39.5922μs 25.2575 KOps/s 25.1776 KOps/s $\color{#35bf28}+0.32\%$
test_values_stack_nested 0.9992ms 0.9613ms 1.0402 KOps/s 1.0416 KOps/s $\color{#d91a1a}-0.14\%$
test_values_stack_nested_leaf 1.0183ms 0.9629ms 1.0386 KOps/s 1.0399 KOps/s $\color{#d91a1a}-0.13\%$
test_values_stack_nested_locked 0.6932ms 0.5723ms 1.7472 KOps/s 1.6702 KOps/s $\color{#35bf28}+4.61\%$
test_membership 5.1442μs 0.9492μs 1.0535 MOps/s 1.0630 MOps/s $\color{#d91a1a}-0.89\%$
test_membership_nested 33.9000μs 2.9274μs 341.6003 KOps/s 340.9489 KOps/s $\color{#35bf28}+0.19\%$
test_membership_nested_leaf 36.4210μs 2.9412μs 340.0003 KOps/s 343.1402 KOps/s $\color{#d91a1a}-0.92\%$
test_membership_stacked_nested 36.1910μs 11.3768μs 87.8984 KOps/s 87.6787 KOps/s $\color{#35bf28}+0.25\%$
test_membership_stacked_nested_leaf 30.1910μs 11.4607μs 87.2551 KOps/s 88.4431 KOps/s $\color{#d91a1a}-1.34\%$
test_membership_nested_last 38.9210μs 5.3952μs 185.3489 KOps/s 186.6776 KOps/s $\color{#d91a1a}-0.71\%$
test_membership_nested_leaf_last 37.4300μs 5.4454μs 183.6417 KOps/s 185.6955 KOps/s $\color{#d91a1a}-1.11\%$
test_membership_stacked_nested_last 0.1951ms 0.1570ms 6.3685 KOps/s 6.3396 KOps/s $\color{#35bf28}+0.46\%$
test_membership_stacked_nested_leaf_last 33.0600μs 13.1723μs 75.9170 KOps/s 75.9868 KOps/s $\color{#d91a1a}-0.09\%$
test_nested_getleaf 30.3310μs 8.4971μs 117.6877 KOps/s 118.4625 KOps/s $\color{#d91a1a}-0.65\%$
test_nested_get 35.4610μs 8.0222μs 124.6537 KOps/s 125.5327 KOps/s $\color{#d91a1a}-0.70\%$
test_stacked_getleaf 0.3700ms 0.3284ms 3.0452 KOps/s 2.8401 KOps/s $\textbf{\color{#35bf28}+7.22\%}$
test_stacked_get 0.3670ms 0.2929ms 3.4137 KOps/s 3.2900 KOps/s $\color{#35bf28}+3.76\%$
test_nested_getitemleaf 26.6310μs 9.8604μs 101.4156 KOps/s 101.5338 KOps/s $\color{#d91a1a}-0.12\%$
test_nested_getitem 36.6400μs 9.4651μs 105.6508 KOps/s 106.3326 KOps/s $\color{#d91a1a}-0.64\%$
test_stacked_getitemleaf 0.3735ms 0.3317ms 3.0146 KOps/s 2.8735 KOps/s $\color{#35bf28}+4.91\%$
test_stacked_getitem 0.3258ms 0.2988ms 3.3469 KOps/s 3.0902 KOps/s $\textbf{\color{#35bf28}+8.31\%}$
test_lock_nested 0.7454ms 0.3564ms 2.8058 KOps/s 2.7938 KOps/s $\color{#35bf28}+0.43\%$
test_lock_stack_nested 85.2188ms 6.2671ms 159.5641 Ops/s 158.5380 Ops/s $\color{#35bf28}+0.65\%$
test_unlock_nested 79.6022ms 0.4340ms 2.3042 KOps/s 2.8241 KOps/s $\textbf{\color{#d91a1a}-18.41\%}$
test_unlock_stack_nested 85.8806ms 6.3485ms 157.5176 Ops/s 154.4805 Ops/s $\color{#35bf28}+1.97\%$
test_flatten_speed 0.6492ms 0.2616ms 3.8221 KOps/s 3.8032 KOps/s $\color{#35bf28}+0.50\%$
test_unflatten_speed 0.4011ms 0.3640ms 2.7472 KOps/s 2.7639 KOps/s $\color{#d91a1a}-0.61\%$
test_common_ops 1.0920ms 0.6479ms 1.5435 KOps/s 1.6742 KOps/s $\textbf{\color{#d91a1a}-7.80\%}$
test_creation 34.6210μs 1.5271μs 654.8230 KOps/s 643.6188 KOps/s $\color{#35bf28}+1.74\%$
test_creation_empty 23.7400μs 10.1971μs 98.0672 KOps/s 136.1621 KOps/s $\textbf{\color{#d91a1a}-27.98\%}$
test_creation_nested_1 46.4610μs 11.9742μs 83.5128 KOps/s 109.8378 KOps/s $\textbf{\color{#d91a1a}-23.97\%}$
test_creation_nested_2 37.6000μs 14.3631μs 69.6229 KOps/s 86.9214 KOps/s $\textbf{\color{#d91a1a}-19.90\%}$
test_clone 67.7810μs 13.9651μs 71.6072 KOps/s 72.5977 KOps/s $\color{#d91a1a}-1.36\%$
test_getitem[int] 29.1300μs 10.9496μs 91.3273 KOps/s 90.1069 KOps/s $\color{#35bf28}+1.35\%$
test_getitem[slice_int] 54.8200μs 21.9107μs 45.6399 KOps/s 45.6402 KOps/s $-0.00\%$
test_getitem[range] 0.1634ms 38.2780μs 26.1247 KOps/s 26.5266 KOps/s $\color{#d91a1a}-1.52\%$
test_getitem[tuple] 38.9600μs 18.9184μs 52.8585 KOps/s 52.0785 KOps/s $\color{#35bf28}+1.50\%$
test_getitem[list] 0.2065ms 36.6110μs 27.3142 KOps/s 28.3823 KOps/s $\color{#d91a1a}-3.76\%$
test_setitem_dim[int] 52.5910μs 30.5266μs 32.7584 KOps/s 37.4828 KOps/s $\textbf{\color{#d91a1a}-12.60\%}$
test_setitem_dim[slice_int] 71.3520μs 50.7859μs 19.6905 KOps/s 20.9883 KOps/s $\textbf{\color{#d91a1a}-6.18\%}$
test_setitem_dim[range] 87.7110μs 65.2369μs 15.3287 KOps/s 16.3008 KOps/s $\textbf{\color{#d91a1a}-5.96\%}$
test_setitem_dim[tuple] 67.5720μs 44.0267μs 22.7135 KOps/s 24.0438 KOps/s $\textbf{\color{#d91a1a}-5.53\%}$
test_setitem 60.2320μs 20.2691μs 49.3363 KOps/s 53.8789 KOps/s $\textbf{\color{#d91a1a}-8.43\%}$
test_set 82.0020μs 19.7878μs 50.5362 KOps/s 55.0008 KOps/s $\textbf{\color{#d91a1a}-8.12\%}$
test_set_shared 2.6663ms 0.1054ms 9.4894 KOps/s 9.5756 KOps/s $\color{#d91a1a}-0.90\%$
test_update 86.8120μs 23.1356μs 43.2234 KOps/s 50.1304 KOps/s $\textbf{\color{#d91a1a}-13.78\%}$
test_update_nested 94.1510μs 29.9093μs 33.4344 KOps/s 37.8433 KOps/s $\textbf{\color{#d91a1a}-11.65\%}$
test_set_nested 75.6510μs 20.8846μs 47.8823 KOps/s 51.8021 KOps/s $\textbf{\color{#d91a1a}-7.57\%}$
test_set_nested_new 79.8010μs 23.7461μs 42.1123 KOps/s 45.5145 KOps/s $\textbf{\color{#d91a1a}-7.47\%}$
test_select 89.8010μs 36.9648μs 27.0528 KOps/s 28.6398 KOps/s $\textbf{\color{#d91a1a}-5.54\%}$
test_select_nested 88.5010μs 53.6068μs 18.6544 KOps/s 18.8114 KOps/s $\color{#d91a1a}-0.83\%$
test_exclude_nested 0.1334ms 0.1140ms 8.7740 KOps/s 8.6608 KOps/s $\color{#35bf28}+1.31\%$
test_empty[True] 0.4173ms 0.3852ms 2.5959 KOps/s 2.6007 KOps/s $\color{#d91a1a}-0.18\%$
test_empty[False] 3.0271μs 0.8530μs 1.1724 MOps/s 1.1726 MOps/s $\color{#d91a1a}-0.02\%$
test_to 74.4120μs 53.6788μs 18.6293 KOps/s 18.9118 KOps/s $\color{#d91a1a}-1.49\%$
test_to_nonblocking 57.5010μs 34.0919μs 29.3325 KOps/s 29.9192 KOps/s $\color{#d91a1a}-1.96\%$
test_unbind_speed 0.2931ms 0.2686ms 3.7234 KOps/s 3.7161 KOps/s $\color{#35bf28}+0.19\%$
test_unbind_speed_stack0 86.8907ms 3.7856ms 264.1593 Ops/s 293.0726 Ops/s $\textbf{\color{#d91a1a}-9.87\%}$
test_unbind_speed_stack1 16.4267μs 1.7172μs 582.3545 KOps/s 538.6784 KOps/s $\textbf{\color{#35bf28}+8.11\%}$
test_split 79.2545ms 1.7386ms 575.1697 Ops/s 566.9620 Ops/s $\color{#35bf28}+1.45\%$
test_chunk 1.7038ms 1.5482ms 645.8938 Ops/s 587.8645 Ops/s $\textbf{\color{#35bf28}+9.87\%}$
test_creation[device0] 0.1333ms 72.4277μs 13.8069 KOps/s 13.8333 KOps/s $\color{#d91a1a}-0.19\%$
test_creation_from_tensor 0.1299ms 57.6572μs 17.3439 KOps/s 18.6229 KOps/s $\textbf{\color{#d91a1a}-6.87\%}$
test_add_one[memmap_tensor0] 0.2843ms 7.0324μs 142.1994 KOps/s 148.2598 KOps/s $\color{#d91a1a}-4.09\%$
test_contiguous[memmap_tensor0] 16.7600μs 0.6258μs 1.5980 MOps/s 1.5826 MOps/s $\color{#35bf28}+0.97\%$
test_stack[memmap_tensor0] 42.9520μs 4.7041μs 212.5819 KOps/s 221.9713 KOps/s $\color{#d91a1a}-4.23\%$
test_memmaptd_index 1.0562ms 0.2818ms 3.5491 KOps/s 3.6289 KOps/s $\color{#d91a1a}-2.20\%$
test_memmaptd_index_astensor 0.5785ms 0.3374ms 2.9641 KOps/s 3.0098 KOps/s $\color{#d91a1a}-1.52\%$
test_memmaptd_index_op 0.9195ms 0.6679ms 1.4972 KOps/s 1.6386 KOps/s $\textbf{\color{#d91a1a}-8.63\%}$
test_serialize_model 93.3255ms 89.5259ms 11.1699 Ops/s 9.5639 Ops/s $\textbf{\color{#35bf28}+16.79\%}$
test_serialize_model_pickle 1.3511s 1.2362s 0.8090 Ops/s 0.8083 Ops/s $\color{#35bf28}+0.09\%$
test_serialize_weights 0.1704s 96.3186ms 10.3822 Ops/s 9.6873 Ops/s $\textbf{\color{#35bf28}+7.17\%}$
test_serialize_weights_returnearly 0.2797s 82.0418ms 12.1889 Ops/s 14.1705 Ops/s $\textbf{\color{#d91a1a}-13.98\%}$
test_serialize_weights_pickle 1.4105s 1.2441s 0.8038 Ops/s 0.8028 Ops/s $\color{#35bf28}+0.13\%$
test_reshape_pytree 51.1420μs 25.3956μs 39.3770 KOps/s 39.7993 KOps/s $\color{#d91a1a}-1.06\%$
test_reshape_td 63.6610μs 29.7233μs 33.6436 KOps/s 33.8562 KOps/s $\color{#d91a1a}-0.63\%$
test_view_pytree 50.9900μs 25.1517μs 39.7587 KOps/s 40.6785 KOps/s $\color{#d91a1a}-2.26\%$
test_view_td 0.3333ms 6.7918μs 147.2372 KOps/s 91.7713 KOps/s $\textbf{\color{#35bf28}+60.44\%}$
test_unbind_pytree 59.3720μs 30.6266μs 32.6513 KOps/s 32.6917 KOps/s $\color{#d91a1a}-0.12\%$
test_unbind_td 0.1094ms 41.0255μs 24.3751 KOps/s 24.6550 KOps/s $\color{#d91a1a}-1.14\%$
test_split_pytree 46.0000μs 29.4279μs 33.9814 KOps/s 34.0336 KOps/s $\color{#d91a1a}-0.15\%$
test_split_td 0.1087ms 39.4745μs 25.3328 KOps/s 25.4362 KOps/s $\color{#d91a1a}-0.41\%$
test_add_pytree 54.3520μs 35.8743μs 27.8751 KOps/s 27.9534 KOps/s $\color{#d91a1a}-0.28\%$
test_add_td 90.0930μs 51.9268μs 19.2579 KOps/s 21.0035 KOps/s $\textbf{\color{#d91a1a}-8.31\%}$
test_distributed 0.1877ms 69.4833μs 14.3920 KOps/s 13.8582 KOps/s $\color{#35bf28}+3.85\%$
test_tdmodule 34.2800μs 18.9877μs 52.6658 KOps/s 57.0918 KOps/s $\textbf{\color{#d91a1a}-7.75\%}$
test_tdmodule_dispatch 0.1507ms 38.9052μs 25.7035 KOps/s 28.1798 KOps/s $\textbf{\color{#d91a1a}-8.79\%}$
test_tdseq 37.4400μs 21.7242μs 46.0316 KOps/s 49.5775 KOps/s $\textbf{\color{#d91a1a}-7.15\%}$
test_tdseq_dispatch 57.5210μs 41.3540μs 24.1815 KOps/s 26.1847 KOps/s $\textbf{\color{#d91a1a}-7.65\%}$
test_instantiation_functorch 1.7909ms 1.6852ms 593.3898 Ops/s 597.2127 Ops/s $\color{#d91a1a}-0.64\%$
test_instantiation_td 1.6820ms 1.1706ms 854.2349 Ops/s 852.7088 Ops/s $\color{#35bf28}+0.18\%$
test_exec_functorch 0.2150ms 0.1612ms 6.2026 KOps/s 6.2397 KOps/s $\color{#d91a1a}-0.59\%$
test_exec_functional_call 0.2227ms 0.1580ms 6.3297 KOps/s 6.4300 KOps/s $\color{#d91a1a}-1.56\%$
test_exec_td 0.1832ms 0.1485ms 6.7325 KOps/s 6.6977 KOps/s $\color{#35bf28}+0.52\%$
test_exec_td_decorator 0.6854ms 0.1867ms 5.3563 KOps/s 5.3934 KOps/s $\color{#d91a1a}-0.69\%$
test_vmap_mlp_speed[True-True] 1.0894ms 1.0215ms 978.9827 Ops/s 964.8123 Ops/s $\color{#35bf28}+1.47\%$
test_vmap_mlp_speed[True-False] 0.6369ms 0.5822ms 1.7176 KOps/s 1.7141 KOps/s $\color{#35bf28}+0.21\%$
test_vmap_mlp_speed[False-True] 1.0592ms 0.9576ms 1.0442 KOps/s 1.0455 KOps/s $\color{#d91a1a}-0.12\%$
test_vmap_mlp_speed[False-False] 0.7104ms 0.5255ms 1.9029 KOps/s 1.9478 KOps/s $\color{#d91a1a}-2.31\%$
test_vmap_mlp_speed_decorator[True-True] 2.8185ms 2.3551ms 424.6020 Ops/s 433.6278 Ops/s $\color{#d91a1a}-2.08\%$
test_vmap_mlp_speed_decorator[True-False] 1.1296ms 0.6290ms 1.5899 KOps/s 1.5937 KOps/s $\color{#d91a1a}-0.24\%$
test_vmap_mlp_speed_decorator[False-True] 2.3631ms 1.9755ms 506.2029 Ops/s 516.7777 Ops/s $\color{#d91a1a}-2.05\%$
test_vmap_mlp_speed_decorator[False-False] 0.9286ms 0.5267ms 1.8987 KOps/s 1.8938 KOps/s $\color{#35bf28}+0.26\%$
test_vmap_transformer_speed[True-True] 12.5714ms 11.9555ms 83.6435 Ops/s 83.1960 Ops/s $\color{#35bf28}+0.54\%$
test_vmap_transformer_speed[True-False] 7.8047ms 7.7582ms 128.8955 Ops/s 128.4234 Ops/s $\color{#35bf28}+0.37\%$
test_vmap_transformer_speed[False-True] 11.9442ms 11.8405ms 84.4556 Ops/s 84.0842 Ops/s $\color{#35bf28}+0.44\%$
test_vmap_transformer_speed[False-False] 7.7015ms 7.6567ms 130.6052 Ops/s 128.4068 Ops/s $\color{#35bf28}+1.71\%$
test_vmap_transformer_speed_decorator[True-True] 73.6240ms 72.7930ms 13.7376 Ops/s 12.4961 Ops/s $\textbf{\color{#35bf28}+9.93\%}$
test_vmap_transformer_speed_decorator[True-False] 20.0627ms 18.5657ms 53.8627 Ops/s 53.5694 Ops/s $\color{#35bf28}+0.55\%$
test_vmap_transformer_speed_decorator[False-True] 0.1957s 73.6759ms 13.5730 Ops/s 15.3959 Ops/s $\textbf{\color{#d91a1a}-11.84\%}$
test_vmap_transformer_speed_decorator[False-False] 19.8141ms 18.1518ms 55.0909 Ops/s 54.7598 Ops/s $\color{#35bf28}+0.60\%$

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. documentation Improvements or additions to documentation
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants