[BugFix] Fix CI breakage from torch.squeeze and dynamo changes - #1738
Merged
Conversation
Recent torch releases (stable and nightly) broke main's CI in two independent ways: - torch.squeeze no longer accepts dim=None; call torch.squeeze(td) in the tests instead (equivalent semantics on all torch versions). This exposed a missing dim default in _UnsqueezedTensorDict._legacy_squeeze, which the base class and LazyStackedTensorDict already have. - dynamo now re-dispatches a slot wrapper stored as a class attribute (cls.__setattr_parent__ = object.__setattr__) through the custom __setattr__, so tensorclass _setattr recursed infinitely under torch.compile (InternalTorchDynamoError: RecursionError). Call object.__setattr__(self, ...) directly instead; dynamo handles that form. The __setattr_parent__ class attribute is kept for backward compatibility. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Contributor
|
| Name | Max | Mean | Ops | Ops on Repo HEAD
|
Change |
|---|---|---|---|---|---|
| test_collector_preallocated_write[stack_out-cpu-32-2] | 0.2258ms | 0.1716ms | 5.8268 KOps/s | 5.9796 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-32-8] | 0.3628ms | 0.3017ms | 3.3146 KOps/s | 3.3917 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-32-32] | 0.8823ms | 0.7766ms | 1.2876 KOps/s | 1.2994 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-2] | 0.3855ms | 0.3397ms | 2.9437 KOps/s | 3.0216 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-8] | 1.2878ms | 1.1460ms | 872.6181 Ops/s | 924.2077 Ops/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-32] | 7.7208ms | 7.3852ms | 135.4065 Ops/s | 137.1701 Ops/s | |
| test_collector_preallocated_write[stack_out-cuda-32-2] | 0.2838ms | 0.2520ms | 3.9678 KOps/s | 4.0374 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-32-8] | 0.5421ms | 0.3935ms | 2.5410 KOps/s | 2.7161 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-32-32] | 0.9082ms | 0.8267ms | 1.2096 KOps/s | 1.2237 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-2] | 0.3410ms | 0.2663ms | 3.7552 KOps/s | 4.0643 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-8] | 0.4669ms | 0.3928ms | 2.5458 KOps/s | 2.7097 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-32] | 0.9939ms | 0.8847ms | 1.1303 KOps/s | 1.1430 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-2] | 0.2741ms | 0.2188ms | 4.5707 KOps/s | 4.7465 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-8] | 1.0428ms | 0.8358ms | 1.1964 KOps/s | 1.2356 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-32] | 3.5914ms | 3.3251ms | 300.7425 Ops/s | 315.0856 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-2] | 0.4380ms | 0.3827ms | 2.6127 KOps/s | 2.7686 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-8] | 1.8708ms | 1.7350ms | 576.3679 Ops/s | 629.4853 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-32] | 11.4965ms | 10.2148ms | 97.8973 Ops/s | 99.8423 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-2] | 0.4665ms | 0.4102ms | 2.4380 KOps/s | 2.4842 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-8] | 1.6454ms | 1.5637ms | 639.5145 Ops/s | 668.3652 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-32] | 6.2906ms | 6.1217ms | 163.3539 Ops/s | 167.5197 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-2] | 0.5066ms | 0.4079ms | 2.4516 KOps/s | 2.4520 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-8] | 1.6808ms | 1.5823ms | 631.9922 Ops/s | 684.4466 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-32] | 6.7169ms | 6.1863ms | 161.6462 Ops/s | 166.3232 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-32-2] | 0.6002ms | 0.1705ms | 5.8637 KOps/s | 5.8831 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-32-8] | 0.7165ms | 0.6661ms | 1.5013 KOps/s | 1.5162 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-32-32] | 2.7172ms | 2.6461ms | 377.9157 Ops/s | 382.6439 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-2] | 0.4238ms | 0.3190ms | 3.1344 KOps/s | 3.1246 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-8] | 1.6870ms | 1.4551ms | 687.2388 Ops/s | 692.4047 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-32] | 11.3459ms | 9.6425ms | 103.7074 Ops/s | 105.7533 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-32-2] | 0.3934ms | 0.3211ms | 3.1145 KOps/s | 3.1763 KOps/s | |
| test_collector_preallocated_write[copy_at-cuda-32-8] | 1.2835ms | 1.1952ms | 836.6850 Ops/s | 855.5349 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-32-32] | 4.8701ms | 4.6863ms | 213.3864 Ops/s | 218.3614 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-2] | 0.4085ms | 0.3240ms | 3.0869 KOps/s | 3.1663 KOps/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-8] | 1.2661ms | 1.2048ms | 830.0149 Ops/s | 849.0579 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-32] | 4.9198ms | 4.7402ms | 210.9610 Ops/s | 216.6304 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] | 0.1773ms | 0.1217ms | 8.2198 KOps/s | 8.6773 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] | 0.5655ms | 0.4757ms | 2.1022 KOps/s | 2.2113 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] | 1.9761ms | 1.9030ms | 525.4766 Ops/s | 557.0945 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] | 0.3504ms | 0.2718ms | 3.6787 KOps/s | 3.8157 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] | 1.4636ms | 1.2806ms | 780.8745 Ops/s | 818.4294 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] | 8.8188ms | 8.5325ms | 117.1992 Ops/s | 117.8057 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] | 0.3828ms | 0.2973ms | 3.3636 KOps/s | 3.5468 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] | 1.1587ms | 1.1030ms | 906.6581 Ops/s | 953.0303 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] | 4.4564ms | 4.3241ms | 231.2615 Ops/s | 248.9245 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] | 0.3702ms | 0.2973ms | 3.3634 KOps/s | 3.6824 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] | 1.1961ms | 1.1042ms | 905.6414 Ops/s | 1.0127 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] | 4.4939ms | 4.3295ms | 230.9724 Ops/s | 246.5537 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-128-128] | 16.4117ms | 16.0463ms | 62.3196 Ops/s | 62.3935 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-128-256] | 32.8539ms | 32.6657ms | 30.6131 Ops/s | 30.8453 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-4096-128] | 89.3057ms | 85.6144ms | 11.6803 Ops/s | 11.5258 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-4096-256] | 0.1979s | 0.1972s | 5.0699 Ops/s | 4.9089 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-128-128] | 13.7478ms | 13.5752ms | 73.6636 Ops/s | 73.9005 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-128-256] | 26.7273ms | 26.4721ms | 37.7756 Ops/s | 38.1031 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-4096-128] | 14.0639ms | 13.7688ms | 72.6281 Ops/s | 73.2099 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-4096-256] | 27.3217ms | 26.7658ms | 37.3612 Ops/s | 37.2480 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] | 52.3172ms | 52.0010ms | 19.2304 Ops/s | 19.8952 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] | 0.1042s | 0.1039s | 9.6273 Ops/s | 9.9135 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] | 0.1458s | 0.1453s | 6.8807 Ops/s | 6.6922 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] | 0.3272s | 0.3251s | 3.0758 Ops/s | 3.0266 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] | 79.8734ms | 79.0076ms | 12.6570 Ops/s | 13.7383 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] | 0.1597s | 0.1589s | 6.2943 Ops/s | 6.6194 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] | 80.5627ms | 80.1209ms | 12.4811 Ops/s | 13.0140 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] | 0.1615s | 0.1607s | 6.2235 Ops/s | 6.4429 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-128-128] | 30.1222ms | 29.9977ms | 33.3359 Ops/s | 33.8150 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-128-256] | 60.5761ms | 60.4027ms | 16.5556 Ops/s | 16.7416 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-4096-128] | 0.1200s | 0.1181s | 8.4648 Ops/s | 8.5621 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-4096-256] | 0.2678s | 0.2665s | 3.7521 Ops/s | 3.8309 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-128-128] | 43.3093ms | 43.0016ms | 23.2549 Ops/s | 23.8505 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-128-256] | 86.4153ms | 85.9614ms | 11.6331 Ops/s | 11.8116 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-4096-128] | 43.6268ms | 43.2242ms | 23.1352 Ops/s | 23.5863 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-4096-256] | 87.5054ms | 86.7751ms | 11.5240 Ops/s | 11.7497 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] | 20.9300ms | 20.7938ms | 48.0913 Ops/s | 51.1267 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] | 42.6177ms | 42.4644ms | 23.5492 Ops/s | 25.0300 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] | 0.1086s | 0.1070s | 9.3466 Ops/s | 9.5119 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] | 0.2484s | 0.2434s | 4.1082 Ops/s | 4.2389 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] | 39.7101ms | 37.6879ms | 26.5337 Ops/s | 28.4753 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] | 79.4609ms | 78.5566ms | 12.7297 Ops/s | 13.7913 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] | 39.9985ms | 39.4433ms | 25.3528 Ops/s | 28.2294 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] | 79.3981ms | 78.6612ms | 12.7127 Ops/s | 13.5534 Ops/s | |
| test_plain_set_nested | 43.0010μs | 15.0417μs | 66.4818 KOps/s | 66.7704 KOps/s | |
| test_plain_set_stack_nested | 55.1620μs | 14.7843μs | 67.6392 KOps/s | 68.4356 KOps/s | |
| test_plain_set_nested_inplace | 74.6220μs | 16.2394μs | 61.5788 KOps/s | 61.2799 KOps/s | |
| test_plain_set_stack_nested_inplace | 0.1151ms | 15.6605μs | 63.8549 KOps/s | 62.4683 KOps/s | |
| test_items | 24.6210μs | 5.5284μs | 180.8841 KOps/s | 180.2662 KOps/s | |
| test_items_nested | 0.5242ms | 0.4506ms | 2.2195 KOps/s | 2.2190 KOps/s | |
| test_items_nested_locked | 0.5191ms | 0.4557ms | 2.1946 KOps/s | 2.1828 KOps/s | |
| test_items_nested_leaf | 0.1476ms | 95.1144μs | 10.5137 KOps/s | 10.5546 KOps/s | |
| test_items_stack_nested | 0.5381ms | 0.4518ms | 2.2136 KOps/s | 2.1981 KOps/s | |
| test_items_stack_nested_leaf | 0.1451ms | 94.6463μs | 10.5656 KOps/s | 10.4720 KOps/s | |
| test_items_stack_nested_locked | 0.5254ms | 0.4557ms | 2.1946 KOps/s | 2.2433 KOps/s | |
| test_keys | 43.0510μs | 4.1488μs | 241.0323 KOps/s | 241.2258 KOps/s | |
| test_keys_nested | 0.1619ms | 0.1273ms | 7.8536 KOps/s | 7.7484 KOps/s | |
| test_keys_nested_locked | 0.7780ms | 0.1354ms | 7.3862 KOps/s | 7.2851 KOps/s | |
| test_keys_nested_leaf | 0.1664ms | 0.1182ms | 8.4595 KOps/s | 8.3374 KOps/s | |
| test_keys_stack_nested | 0.1960ms | 0.1270ms | 7.8722 KOps/s | 7.7501 KOps/s | |
| test_keys_stack_nested_leaf | 0.1704ms | 0.1175ms | 8.5122 KOps/s | 8.4770 KOps/s | |
| test_keys_stack_nested_locked | 0.1895ms | 0.1347ms | 7.4222 KOps/s | 7.3665 KOps/s | |
| test_values | 6.9780μs | 0.9996μs | 1.0004 MOps/s | 998.2585 KOps/s | |
| test_values_nested | 0.1048ms | 51.4884μs | 19.4219 KOps/s | 19.1841 KOps/s | |
| test_values_nested_locked | 94.8820μs | 54.5878μs | 18.3191 KOps/s | 18.1662 KOps/s | |
| test_values_nested_leaf | 0.1023ms | 59.3110μs | 16.8603 KOps/s | 16.4980 KOps/s | |
| test_values_stack_nested | 0.1032ms | 51.3825μs | 19.4619 KOps/s | 19.0554 KOps/s | |
| test_values_stack_nested_leaf | 99.0620μs | 59.0733μs | 16.9281 KOps/s | 16.5950 KOps/s | |
| test_values_stack_nested_locked | 0.1183ms | 54.6712μs | 18.2912 KOps/s | 17.9964 KOps/s | |
| test_membership | 13.4920μs | 0.8065μs | 1.2399 MOps/s | 1.2298 MOps/s | |
| test_membership_nested | 52.9020μs | 2.8712μs | 348.2815 KOps/s | 348.4179 KOps/s | |
| test_membership_nested_leaf | 70.9020μs | 2.8961μs | 345.2876 KOps/s | 361.9308 KOps/s | |
| test_membership_stacked_nested | 50.7510μs | 2.7994μs | 357.2184 KOps/s | 362.9079 KOps/s | |
| test_membership_stacked_nested_leaf | 89.5220μs | 2.6701μs | 374.5140 KOps/s | 361.9187 KOps/s | |
| test_membership_nested_last | 34.6200μs | 4.3645μs | 229.1205 KOps/s | 229.5882 KOps/s | |
| test_membership_nested_leaf_last | 31.9100μs | 4.3847μs | 228.0642 KOps/s | 230.1617 KOps/s | |
| test_membership_stacked_nested_last | 47.9610μs | 4.1919μs | 238.5553 KOps/s | 240.0597 KOps/s | |
| test_membership_stacked_nested_leaf_last | 0.1038ms | 4.0348μs | 247.8412 KOps/s | 241.8268 KOps/s | |
| test_nested_getleaf | 94.1820μs | 20.2070μs | 49.4877 KOps/s | 50.2183 KOps/s | |
| test_nested_get | 59.4220μs | 18.9374μs | 52.8055 KOps/s | 50.6050 KOps/s | |
| test_stacked_getleaf | 76.1020μs | 19.9805μs | 50.0488 KOps/s | 48.6164 KOps/s | |
| test_stacked_get | 58.2410μs | 18.9277μs | 52.8326 KOps/s | 50.3444 KOps/s | |
| test_nested_getitemleaf | 81.7820μs | 21.1315μs | 47.3228 KOps/s | 46.4075 KOps/s | |
| test_nested_getitem | 94.8320μs | 20.3405μs | 49.1630 KOps/s | 49.1703 KOps/s | |
| test_stacked_getitemleaf | 80.2520μs | 20.8097μs | 48.0546 KOps/s | 47.4142 KOps/s | |
| test_stacked_getitem | 71.3720μs | 19.6850μs | 50.8000 KOps/s | 49.8378 KOps/s | |
| test_lock_nested | 7.9286ms | 0.5030ms | 1.9880 KOps/s | 2.0282 KOps/s | |
| test_lock_stack_nested | 0.7048ms | 0.5024ms | 1.9906 KOps/s | 2.0305 KOps/s | |
| test_unlock_nested | 0.4741ms | 0.3932ms | 2.5434 KOps/s | 2.5793 KOps/s | |
| test_unlock_stack_nested | 0.5803ms | 0.3991ms | 2.5056 KOps/s | 2.5593 KOps/s | |
| test_flatten_speed | 0.2080ms | 0.1181ms | 8.4654 KOps/s | 8.2925 KOps/s | |
| test_unflatten_speed | 0.6578ms | 0.5455ms | 1.8331 KOps/s | 1.7985 KOps/s | |
| test_common_ops | 0.8925ms | 0.6988ms | 1.4310 KOps/s | 1.4539 KOps/s | |
| test_creation | 0.1204ms | 3.0930μs | 323.3110 KOps/s | 322.0560 KOps/s | |
| test_creation_empty | 46.1210μs | 6.8491μs | 146.0041 KOps/s | 146.8671 KOps/s | |
| test_creation_nested_1 | 53.1910μs | 11.3004μs | 88.4925 KOps/s | 88.6718 KOps/s | |
| test_creation_nested_2 | 71.5610μs | 12.7899μs | 78.1868 KOps/s | 75.6392 KOps/s | |
| test_creation_many_keys[10] | 91.6820μs | 20.4337μs | 48.9388 KOps/s | 50.3539 KOps/s | |
| test_creation_many_keys[50] | 0.1326ms | 88.3561μs | 11.3178 KOps/s | 11.4337 KOps/s | |
| test_creation_many_keys[100] | 0.2559ms | 0.1748ms | 5.7192 KOps/s | 5.8273 KOps/s | |
| test_creation_nested_many_keys[10] | 80.1920μs | 44.4036μs | 22.5207 KOps/s | 22.5230 KOps/s | |
| test_creation_nested_many_keys[50] | 0.2253ms | 0.1814ms | 5.5130 KOps/s | 5.5528 KOps/s | |
| test_clone | 41.8810μs | 13.0625μs | 76.5552 KOps/s | 78.6055 KOps/s | |
| test_getitem[int] | 1.5257ms | 15.3507μs | 65.1438 KOps/s | 66.9493 KOps/s | |
| test_getitem[slice_int] | 0.1412ms | 25.1074μs | 39.8289 KOps/s | 41.0657 KOps/s | |
| test_getitem[range] | 0.1709ms | 62.0810μs | 16.1080 KOps/s | 16.3179 KOps/s | |
| test_getitem[tuple] | 0.1459ms | 24.1675μs | 41.3780 KOps/s | 42.7686 KOps/s | |
| test_getitem[list] | 0.1869ms | 58.4215μs | 17.1170 KOps/s | 17.8020 KOps/s | |
| test_setitem_dim[int] | 56.1120μs | 25.3686μs | 39.4189 KOps/s | 40.7370 KOps/s | |
| test_setitem_dim[slice_int] | 81.4520μs | 42.1697μs | 23.7137 KOps/s | 24.5634 KOps/s | |
| test_setitem_dim[range] | 0.1506ms | 96.6282μs | 10.3489 KOps/s | 10.8133 KOps/s | |
| test_setitem_dim[tuple] | 63.6010μs | 40.0923μs | 24.9424 KOps/s | 25.7850 KOps/s | |
| test_setitem | 73.8420μs | 17.3762μs | 57.5499 KOps/s | 59.3279 KOps/s | |
| test_set | 59.8310μs | 16.6005μs | 60.2390 KOps/s | 61.4711 KOps/s | |
| test_set_shared | 0.5311ms | 0.2116ms | 4.7262 KOps/s | 4.9224 KOps/s | |
| test_update | 0.3878ms | 21.6406μs | 46.2095 KOps/s | 48.3696 KOps/s | |
| test_update_nested | 78.8420μs | 32.6593μs | 30.6191 KOps/s | 30.2602 KOps/s | |
| test_update__nested | 0.4678ms | 32.8317μs | 30.4584 KOps/s | 31.0455 KOps/s | |
| test_set_nested | 51.3310μs | 18.2070μs | 54.9239 KOps/s | 57.8045 KOps/s | |
| test_set_nested_new | 56.6720μs | 22.9287μs | 43.6134 KOps/s | 44.3979 KOps/s | |
| test_select | 88.7920μs | 39.3510μs | 25.4123 KOps/s | 25.8747 KOps/s | |
| test_select_nested | 0.1406ms | 72.1534μs | 13.8594 KOps/s | 13.8248 KOps/s | |
| test_exclude_nested | 0.1410ms | 88.1964μs | 11.3383 KOps/s | 11.6543 KOps/s | |
| test_empty[True] | 0.4239ms | 0.3844ms | 2.6014 KOps/s | 2.5981 KOps/s | |
| test_empty[False] | 20.5555μs | 1.3723μs | 728.7213 KOps/s | 737.5943 KOps/s | |
| test_to | 0.1674ms | 72.9308μs | 13.7116 KOps/s | 14.6507 KOps/s | |
| test_to_nonblocking | 0.1308ms | 69.7228μs | 14.3425 KOps/s | 14.7455 KOps/s | |
| test_unbind_speed | 0.3934ms | 0.3315ms | 3.0163 KOps/s | 3.0346 KOps/s | |
| test_unbind_speed_stack0 | 0.4947ms | 0.3307ms | 3.0243 KOps/s | 3.0624 KOps/s | |
| test_unbind_speed_stack1 | 0.1157s | 0.8382ms | 1.1930 KOps/s | 915.4151 Ops/s | |
| test_split | 0.1160s | 1.4198ms | 704.3299 Ops/s | 789.0804 Ops/s | |
| test_chunk | 1.3326ms | 1.1989ms | 834.0662 Ops/s | 822.4250 Ops/s | |
| test_to_cpu_blocking | 20.7702ms | 19.8220ms | 50.4491 Ops/s | 52.8636 Ops/s | |
| test_to_cpu_global_sync | 0.1282s | 13.2556ms | 75.4398 Ops/s | 88.0967 Ops/s | |
| test_to_cpu_event_sync | 13.4640ms | 12.8024ms | 78.1105 Ops/s | 81.8809 Ops/s | |
| test_to_cpu_default | 13.2996ms | 12.7860ms | 78.2103 Ops/s | 82.0535 Ops/s | |
| test_consolidate[False-None] | 4.3398ms | 4.1315ms | 242.0403 Ops/s | 246.9013 Ops/s | |
| test_consolidate[default-None] | 3.3117ms | 2.3270ms | 429.7419 Ops/s | 447.6800 Ops/s | |
| test_consolidate[reduce-overhead-None] | 3.3071ms | 2.2386ms | 446.7072 Ops/s | 464.5347 Ops/s | |
| test_consolidate_njt[False-None] | 5.6512ms | 5.1009ms | 196.0439 Ops/s | 196.6273 Ops/s | |
| test_to[False-False-None] | 2.3827ms | 2.1830ms | 458.0756 Ops/s | 464.3660 Ops/s | |
| test_to[True-False-None] | 2.1402ms | 1.9799ms | 505.0662 Ops/s | 529.5242 Ops/s | |
| test_to[within-False-None] | 6.5624ms | 6.2583ms | 159.7887 Ops/s | 165.2191 Ops/s | |
| test_to[True-default-None] | 10.6021ms | 10.4065ms | 96.0940 Ops/s | 97.0803 Ops/s | |
| test_to_njt[False-False-None] | 7.0723ms | 6.9607ms | 143.6643 Ops/s | 143.8565 Ops/s | |
| test_to_njt[True-False-None] | 4.0616ms | 3.8585ms | 259.1648 Ops/s | 258.8952 Ops/s | |
| test_to_njt[within-False-None] | 9.2276ms | 9.0582ms | 110.3974 Ops/s | 110.5148 Ops/s | |
| test_creation[device0] | 0.4127ms | 0.1152ms | 8.6835 KOps/s | 8.7903 KOps/s | |
| test_creation_from_tensor | 0.4333ms | 0.1133ms | 8.8238 KOps/s | 8.9207 KOps/s | |
| test_add_one[memmap_tensor0] | 0.3554ms | 6.4541μs | 154.9391 KOps/s | 165.3632 KOps/s | |
| test_contiguous[memmap_tensor0] | 24.3110μs | 0.6641μs | 1.5057 MOps/s | 2.1350 MOps/s | |
| test_stack[memmap_tensor0] | 81.9920μs | 4.2378μs | 235.9732 KOps/s | 250.4462 KOps/s | |
| test_memmaptd_index | 1.2574ms | 0.2773ms | 3.6065 KOps/s | 3.7257 KOps/s | |
| test_memmaptd_index_astensor | 0.5311ms | 0.3778ms | 2.6469 KOps/s | 2.7131 KOps/s | |
| test_memmaptd_index_op | 0.9975ms | 0.6351ms | 1.5745 KOps/s | 1.6381 KOps/s | |
| test_serialize_model | 0.1394s | 0.1356s | 7.3769 Ops/s | 7.4017 Ops/s | |
| test_serialize_model_pickle | 1.3473s | 1.2099s | 0.8265 Ops/s | 0.8254 Ops/s | |
| test_serialize_weights | 0.3240s | 0.1626s | 6.1494 Ops/s | 7.4762 Ops/s | |
| test_serialize_weights_returnearly | 71.0103ms | 62.6889ms | 15.9518 Ops/s | 15.0964 Ops/s | |
| test_serialize_weights_pickle | 1.3495s | 1.2132s | 0.8243 Ops/s | 0.8228 Ops/s | |
| test_reshape_pytree | 0.2150ms | 32.1303μs | 31.1233 KOps/s | 31.6521 KOps/s | |
| test_reshape_td | 84.7920μs | 45.0354μs | 22.2048 KOps/s | 22.3129 KOps/s | |
| test_view_pytree | 0.2190ms | 31.7097μs | 31.5361 KOps/s | 32.1295 KOps/s | |
| test_view_td | 0.1006ms | 54.1151μs | 18.4791 KOps/s | 18.3671 KOps/s | |
| test_unbind_pytree | 0.2380ms | 35.5483μs | 28.1308 KOps/s | 28.3260 KOps/s | |
| test_unbind_td | 0.1016ms | 50.3400μs | 19.8649 KOps/s | 20.3935 KOps/s | |
| test_split_pytree | 0.2604ms | 41.2716μs | 24.2297 KOps/s | 24.5160 KOps/s | |
| test_split_td | 0.1284ms | 71.5755μs | 13.9713 KOps/s | 14.0060 KOps/s | |
| test_add_pytree | 0.2236ms | 40.2217μs | 24.8622 KOps/s | 25.5486 KOps/s | |
| test_add_td | 91.0720μs | 53.7932μs | 18.5897 KOps/s | 18.6748 KOps/s | |
| test_compile_add_one_nested[tensordict-compile] | 0.3744ms | 0.2726ms | 3.6682 KOps/s | 3.8289 KOps/s | |
| test_compile_add_one_nested[tensordict-eager] | 0.4366ms | 0.2011ms | 4.9737 KOps/s | 4.9136 KOps/s | |
| test_compile_add_one_nested[pytree-compile] | 0.4298ms | 0.1572ms | 6.3593 KOps/s | 6.3773 KOps/s | |
| test_compile_add_one_nested[pytree-eager] | 0.4880ms | 0.1771ms | 5.6460 KOps/s | 5.6483 KOps/s | |
| test_compile_copy_nested[tensordict-compile] | 0.1195ms | 23.2760μs | 42.9626 KOps/s | 48.1883 KOps/s | |
| test_compile_copy_nested[tensordict-eager] | 0.1254ms | 52.0422μs | 19.2152 KOps/s | 18.6033 KOps/s | |
| test_compile_copy_nested[pytree-compile] | 0.1837ms | 20.1863μs | 49.5385 KOps/s | 52.9490 KOps/s | |
| test_compile_copy_nested[pytree-eager] | 0.3528ms | 65.3686μs | 15.2979 KOps/s | 15.1374 KOps/s | |
| test_compile_add_one_flat[tensordict-compile] | 0.3725ms | 0.3033ms | 3.2971 KOps/s | 3.1675 KOps/s | |
| test_compile_add_one_flat[tensordict-eager] | 0.3393ms | 0.2867ms | 3.4876 KOps/s | 3.3424 KOps/s | |
| test_compile_add_one_flat[tensorclass-compile] | 0.3030ms | 0.2082ms | 4.8028 KOps/s | 4.5452 KOps/s | |
| test_compile_add_one_flat[tensorclass-eager] | 0.1696ms | 75.3950μs | 13.2635 KOps/s | 13.0332 KOps/s | |
| test_compile_add_one_flat[pytree-compile] | 0.5495ms | 0.2398ms | 4.1693 KOps/s | 4.0580 KOps/s | |
| test_compile_add_one_flat[pytree-eager] | 0.7987ms | 0.5299ms | 1.8871 KOps/s | 1.9148 KOps/s | |
| test_compile_add_self_flat[tensordict-eager] | 0.4147ms | 0.3393ms | 2.9470 KOps/s | 2.8986 KOps/s | |
| test_compile_add_self_flat[tensordict-compile] | 0.5717ms | 0.3059ms | 3.2687 KOps/s | 3.1103 KOps/s | |
| test_compile_add_self_flat[tensorclass-eager] | 0.1609ms | 92.3284μs | 10.8309 KOps/s | 10.9868 KOps/s | |
| test_compile_add_self_flat[tensorclass-compile] | 0.3651ms | 0.2132ms | 4.6910 KOps/s | 4.5653 KOps/s | |
| test_compile_add_self_flat[pytree-eager] | 0.7212ms | 0.4418ms | 2.2633 KOps/s | 2.3390 KOps/s | |
| test_compile_add_self_flat[pytree-compile] | 0.6506ms | 0.2414ms | 4.1433 KOps/s | 4.0793 KOps/s | |
| test_compile_copy_flat[tensordict-compile] | 90.4420μs | 27.5619μs | 36.2820 KOps/s | 37.2264 KOps/s | |
| test_compile_copy_flat[tensordict-eager] | 91.5720μs | 39.7265μs | 25.1721 KOps/s | 24.6628 KOps/s | |
| test_compile_copy_flat[pytree-compile] | 0.2362ms | 26.3230μs | 37.9897 KOps/s | 38.9448 KOps/s | |
| test_compile_copy_flat[pytree-eager] | 0.3335ms | 50.7081μs | 19.7207 KOps/s | 19.6989 KOps/s | |
| test_compile_assign_and_add[tensordict-compile] | 2.1296ms | 0.2511ms | 3.9831 KOps/s | 3.6057 KOps/s | |
| test_compile_assign_and_add[tensordict-eager] | 3.7087ms | 3.3664ms | 297.0569 Ops/s | 299.9770 Ops/s | |
| test_compile_assign_and_add[pytree-compile] | 1.9753ms | 0.1929ms | 5.1827 KOps/s | 5.0881 KOps/s | |
| test_compile_assign_and_add[pytree-eager] | 3.0256ms | 2.7978ms | 357.4276 Ops/s | 368.7954 Ops/s | |
| test_compile_indexing[tensor-tensordict-compile] | 0.3130ms | 0.1571ms | 6.3672 KOps/s | 6.2064 KOps/s | |
| test_compile_indexing[tensor-tensordict-eager] | 0.2269s | 97.2627μs | 10.2814 KOps/s | 13.0735 KOps/s | |
| test_compile_indexing[tensor-tensorclass-compile] | 0.2962ms | 0.1360ms | 7.3512 KOps/s | 7.4540 KOps/s | |
| test_compile_indexing[tensor-tensorclass-eager] | 0.2993ms | 44.6857μs | 22.3785 KOps/s | 22.8628 KOps/s | |
| test_compile_indexing[tensor-pytree-compile] | 0.2346ms | 0.1386ms | 7.2150 KOps/s | 7.1308 KOps/s | |
| test_compile_indexing[tensor-pytree-eager] | 0.2773ms | 44.7273μs | 22.3577 KOps/s | 21.9268 KOps/s | |
| test_compile_indexing[slice-tensordict-compile] | 0.1879ms | 96.4485μs | 10.3682 KOps/s | 11.4893 KOps/s | |
| test_compile_indexing[slice-tensordict-eager] | 0.2238ms | 27.7248μs | 36.0688 KOps/s | 36.4824 KOps/s | |
| test_compile_indexing[slice-tensorclass-compile] | 0.1844ms | 73.5929μs | 13.5883 KOps/s | 14.3271 KOps/s | |
| test_compile_indexing[slice-tensorclass-eager] | 0.2610ms | 22.2830μs | 44.8773 KOps/s | 46.0702 KOps/s | |
| test_compile_indexing[slice-pytree-compile] | 0.1348ms | 73.8886μs | 13.5339 KOps/s | 14.1673 KOps/s | |
| test_compile_indexing[slice-pytree-eager] | 0.2571ms | 22.3831μs | 44.6765 KOps/s | 46.4440 KOps/s | |
| test_compile_indexing[int-tensordict-compile] | 0.2149ms | 90.4303μs | 11.0582 KOps/s | 11.1287 KOps/s | |
| test_compile_indexing[int-tensordict-eager] | 0.2115ms | 27.8480μs | 35.9093 KOps/s | 35.0750 KOps/s | |
| test_compile_indexing[int-tensorclass-compile] | 0.1175ms | 73.1917μs | 13.6628 KOps/s | 14.1990 KOps/s | |
| test_compile_indexing[int-tensorclass-eager] | 0.2757ms | 22.3541μs | 44.7345 KOps/s | 46.6525 KOps/s | |
| test_compile_indexing[int-pytree-compile] | 0.1150ms | 73.2765μs | 13.6469 KOps/s | 14.2749 KOps/s | |
| test_compile_indexing[int-pytree-eager] | 0.2746ms | 22.3781μs | 44.6865 KOps/s | 46.4519 KOps/s | |
| test_compile_replace[single-eager] | 0.1104ms | 48.2702μs | 20.7167 KOps/s | 21.3433 KOps/s | |
| test_compile_replace[single-compile] | 0.2540ms | 0.1449ms | 6.8993 KOps/s | 6.8986 KOps/s | |
| test_compile_replace[multi-eager] | 0.6735ms | 0.5656ms | 1.7682 KOps/s | 1.8015 KOps/s | |
| test_compile_replace[multi-compile] | 0.2871ms | 0.1559ms | 6.4123 KOps/s | 6.4544 KOps/s | |
| test_compile_tc_getattr_20[eager] | 0.2322ms | 0.1752ms | 5.7066 KOps/s | 5.8294 KOps/s | |
| test_compile_tc_getattr_20[compile] | 0.3741ms | 0.1749ms | 5.7185 KOps/s | 5.7204 KOps/s | |
| test_compile_clone_shallow[20-eager] | 48.2310μs | 18.8706μs | 52.9925 KOps/s | 52.6748 KOps/s | |
| test_compile_clone_shallow[20-compile] | 90.6420μs | 24.4249μs | 40.9418 KOps/s | 43.8041 KOps/s | |
| test_compile_clone_shallow[40-eager] | 71.1620μs | 32.3842μs | 30.8792 KOps/s | 30.2467 KOps/s | |
| test_compile_clone_shallow[40-compile] | 89.0420μs | 26.1088μs | 38.3013 KOps/s | 40.7200 KOps/s | |
| test_compile_clone_shallow[80-eager] | 98.1320μs | 60.1197μs | 16.6335 KOps/s | 16.3018 KOps/s | |
| test_compile_clone_shallow[80-compile] | 0.1397ms | 30.1314μs | 33.1880 KOps/s | 35.6658 KOps/s | |
| test_compile_update_inplace[eager] | 0.2360ms | 61.7103μs | 16.2048 KOps/s | 16.2893 KOps/s | |
| test_compile_update_inplace[compile] | 0.4228ms | 0.1194ms | 8.3785 KOps/s | 7.8479 KOps/s | |
| test_mod_add[eager] | 0.2943ms | 0.2250ms | 4.4445 KOps/s | 4.4503 KOps/s | |
| test_mod_add[compile] | 0.3034ms | 0.1206ms | 8.2890 KOps/s | 8.2827 KOps/s | |
| test_mod_add[compile-overhead] | 0.3390ms | 0.1965ms | 5.0891 KOps/s | 4.9530 KOps/s | |
| test_mod_wrap[eager] | 0.5542ms | 0.4678ms | 2.1378 KOps/s | 2.1131 KOps/s | |
| test_mod_wrap[compile] | 0.4537ms | 0.3521ms | 2.8399 KOps/s | 2.9174 KOps/s | |
| test_mod_wrap[compile-overhead] | 7.1107ms | 3.9991ms | 250.0535 Ops/s | 250.3029 Ops/s | |
| test_mod_wrap_and_backward[eager] | 1.8452ms | 1.6871ms | 592.7454 Ops/s | 606.9798 Ops/s | |
| test_mod_wrap_and_backward[compile] | 2.4221ms | 1.3639ms | 733.2043 Ops/s | 755.8591 Ops/s | |
| test_mod_wrap_and_backward[compile-overhead] | 1.5632ms | 0.9509ms | 1.0516 KOps/s | 1.0503 KOps/s | |
| test_seq_add[eager] | 0.6018ms | 0.5313ms | 1.8821 KOps/s | 1.9923 KOps/s | |
| test_seq_add[compile] | 0.2794ms | 0.1390ms | 7.1957 KOps/s | 7.4942 KOps/s | |
| test_seq_add[compile-overhead] | 0.3674ms | 0.2101ms | 4.7586 KOps/s | 4.8320 KOps/s | |
| test_seq_wrap[eager] | 5.4607ms | 1.4614ms | 684.2791 Ops/s | 715.2053 Ops/s | |
| test_seq_wrap[compile] | 0.4772ms | 0.3785ms | 2.6422 KOps/s | 2.6839 KOps/s | |
| test_seq_wrap[compile-overhead] | 0.4513ms | 0.3206ms | 3.1189 KOps/s | 3.0523 KOps/s | |
| test_func_call_runtime[False-eager] | 0.9236ms | 0.8241ms | 1.2134 KOps/s | 1.2546 KOps/s | |
| test_func_call_runtime[False-compile] | 0.8861ms | 0.7612ms | 1.3136 KOps/s | 1.3145 KOps/s | |
| test_func_call_runtime[False-compile-overhead] | 0.6677ms | 0.5044ms | 1.9824 KOps/s | 1.9652 KOps/s | |
| test_func_call_runtime[True-eager] | 1.1919ms | 1.0582ms | 944.9943 Ops/s | 953.6883 Ops/s | |
| test_func_call_runtime[True-compile] | 0.9189ms | 0.7903ms | 1.2653 KOps/s | 1.3033 KOps/s | |
| test_func_call_runtime[True-compile-overhead] | 0.6892ms | 0.5205ms | 1.9212 KOps/s | 1.9126 KOps/s | |
| test_func_call_cm_runtime[False-eager] | 0.9114ms | 0.7992ms | 1.2512 KOps/s | 1.1893 KOps/s | |
| test_func_call_cm_runtime[False-compile] | 0.9017ms | 0.7439ms | 1.3443 KOps/s | 1.3447 KOps/s | |
| test_func_call_cm_runtime[False-compile-overhead] | 0.6323ms | 0.4831ms | 2.0701 KOps/s | 2.0413 KOps/s | |
| test_func_call_cm_runtime[True-eager] | 1.3129ms | 1.2042ms | 830.3980 Ops/s | 816.3635 Ops/s | |
| test_func_call_cm_runtime[True-compile] | 0.9056ms | 0.8470ms | 1.1806 KOps/s | 1.1761 KOps/s | |
| test_func_call_cm_runtime[True-compile-overhead] | 1.1312ms | 0.8670ms | 1.1534 KOps/s | 1.1460 KOps/s | |
| test_vmap_func_call_cm_runtime[eager] | 3.1176ms | 2.6445ms | 378.1499 Ops/s | 377.8182 Ops/s | |
| test_vmap_func_call_cm_runtime[compile] | 4.2015ms | 3.6400ms | 274.7241 Ops/s | 275.0944 Ops/s | |
| test_vmap_func_call_cm_runtime[compile-overhead] | 4.1818ms | 3.6373ms | 274.9259 Ops/s | 276.1017 Ops/s | |
| test_distributed | 1.7920ms | 0.1532ms | 6.5290 KOps/s | 6.9867 KOps/s | |
| test_tdmodule | 0.8701ms | 0.1901ms | 5.2607 KOps/s | 5.1803 KOps/s | |
| test_tdmodule_dispatch | 0.4082ms | 0.3700ms | 2.7030 KOps/s | 2.6953 KOps/s | |
| test_tdseq | 45.5810μs | 26.1338μs | 38.2646 KOps/s | 37.3816 KOps/s | |
| test_tdseq_dispatch | 0.2852ms | 0.2189ms | 4.5693 KOps/s | 4.6324 KOps/s | |
| test_instantiation_functorch | 2.1297ms | 2.0368ms | 490.9736 Ops/s | 496.0275 Ops/s | |
| test_exec_functorch | 0.2389ms | 0.1750ms | 5.7140 KOps/s | 5.7925 KOps/s | |
| test_exec_functional_call | 0.4520ms | 0.1542ms | 6.4837 KOps/s | 6.5261 KOps/s | |
| test_exec_td_decorator | 0.5083ms | 0.2433ms | 4.1101 KOps/s | 4.1878 KOps/s | |
| test_vmap_mlp_speed_decorator[True-True] | 1.3131ms | 1.1216ms | 891.6158 Ops/s | 901.5051 Ops/s | |
| test_vmap_mlp_speed_decorator[True-False] | 1.3279ms | 1.1146ms | 897.1509 Ops/s | 906.4150 Ops/s | |
| test_vmap_mlp_speed_decorator[False-True] | 1.0487ms | 0.8174ms | 1.2234 KOps/s | 1.2304 KOps/s | |
| test_vmap_mlp_speed_decorator[False-False] | 1.0475ms | 0.8157ms | 1.2259 KOps/s | 1.2298 KOps/s | |
| test_vmap_transformer_speed_decorator[True-True] | 26.1534ms | 24.8122ms | 40.3028 Ops/s | 40.6002 Ops/s | |
| test_vmap_transformer_speed_decorator[True-False] | 24.9987ms | 24.6694ms | 40.5361 Ops/s | 40.7605 Ops/s | |
| test_vmap_transformer_speed_decorator[False-True] | 24.5098ms | 24.3435ms | 41.0787 Ops/s | 41.2391 Ops/s | |
| test_vmap_transformer_speed_decorator[False-False] | 24.6431ms | 24.2788ms | 41.1881 Ops/s | 41.3889 Ops/s | |
| test_to_module_speed[True] | 1.7723ms | 1.6056ms | 622.8285 Ops/s | 618.1857 Ops/s | |
| test_to_module_speed[False] | 1.6564ms | 1.5683ms | 637.6256 Ops/s | 627.5912 Ops/s | |
| test_to_module_plain_tensor_speed[None] | 1.8104ms | 1.6805ms | 595.0657 Ops/s | 579.4612 Ops/s | |
| test_to_module_plain_tensor_speed[False] | 1.6576ms | 1.5468ms | 646.4815 Ops/s | 637.9864 Ops/s | |
| test_to_module_plain_tensor_speed[True] | 3.3904ms | 3.2994ms | 303.0830 Ops/s | 297.8372 Ops/s | |
| test_tc_init | 78.7510μs | 43.6831μs | 22.8922 KOps/s | 22.8067 KOps/s | |
| test_tc_init_tensor_only | 36.6410μs | 9.2338μs | 108.2975 KOps/s | 107.6091 KOps/s | |
| test_tc_init_nested | 0.1404ms | 87.2005μs | 11.4678 KOps/s | 11.3823 KOps/s | |
| test_tc_init_many_fields | 52.0210μs | 16.9276μs | 59.0750 KOps/s | 59.0272 KOps/s | |
| test_tc_first_layer_tensor | 22.4810μs | 1.7709μs | 564.6743 KOps/s | 574.4616 KOps/s | |
| test_tc_first_layer_tensor_only | 3.7316μs | 0.3920μs | 2.5509 MOps/s | 2.5474 MOps/s | |
| test_tc_first_layer_tensor_set | 55.7520μs | 4.1612μs | 240.3157 KOps/s | 243.7797 KOps/s | |
| test_tc_first_layer_tensor_only_set | 25.4410μs | 3.4249μs | 291.9791 KOps/s | 294.5996 KOps/s | |
| test_tc_first_layer_nontensor | 37.4300μs | 5.9645μs | 167.6576 KOps/s | 168.0723 KOps/s | |
| test_tc_second_layer_tensor | 34.3410μs | 4.3005μs | 232.5326 KOps/s | 237.1293 KOps/s | |
| test_tc_second_layer_nontensor | 74.9120μs | 8.4503μs | 118.3390 KOps/s | 119.5837 KOps/s | |
| test_unbind | 0.2879s | 16.4270ms | 60.8752 Ops/s | 65.1749 Ops/s | |
| test_full_like | 4.5076ms | 4.3755ms | 228.5455 Ops/s | 229.0102 Ops/s | |
| test_zeros_like | 4.5205ms | 4.3642ms | 229.1364 Ops/s | 229.0254 Ops/s | |
| test_ones_like | 5.0132ms | 4.2853ms | 233.3541 Ops/s | 229.2059 Ops/s | |
| test_clone | 6.9126ms | 6.4872ms | 154.1486 Ops/s | 154.7675 Ops/s | |
| test_squeeze | 73.1620μs | 13.9448μs | 71.7112 KOps/s | 71.3813 KOps/s | |
| test_unsqueeze | 0.1771ms | 0.1080ms | 9.2590 KOps/s | 8.9359 KOps/s | |
| test_split | 0.3955ms | 0.1899ms | 5.2657 KOps/s | 5.3461 KOps/s | |
| test_permute | 0.2540ms | 0.1923ms | 5.2006 KOps/s | 5.2173 KOps/s | |
| test_stack | 51.2222ms | 50.8740ms | 19.6564 Ops/s | 37.6447 Ops/s | |
| test_cat | 51.4524ms | 50.9227ms | 19.6376 Ops/s | 37.6902 Ops/s | |
| test_sequential_tensordict | 0.6323ms | 0.5626ms | 1.7775 KOps/s | 1.7700 KOps/s | |
| test_sequential_graph_module | 0.1788ms | 0.1182ms | 8.4634 KOps/s | 8.8009 KOps/s | |
| test_nested_tensordict | 0.9187ms | 0.8159ms | 1.2256 KOps/s | 1.2426 KOps/s | |
| test_nested_graph_module | 0.2096ms | 0.1275ms | 7.8457 KOps/s | 8.3339 KOps/s |
Contributor
|
| Name | Max | Mean | Ops | Ops on Repo HEAD
|
Change |
|---|---|---|---|---|---|
| test_collector_preallocated_write[stack_out-cpu-32-2] | 0.2026ms | 0.1638ms | 6.1063 KOps/s | 6.1118 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-32-8] | 0.3501ms | 0.2918ms | 3.4265 KOps/s | 3.3826 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-32-32] | 0.9522ms | 0.7756ms | 1.2893 KOps/s | 1.2607 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-2] | 0.3937ms | 0.3393ms | 2.9471 KOps/s | 2.8880 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-8] | 1.2720ms | 1.1725ms | 852.8733 Ops/s | 897.8922 Ops/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-32] | 7.5411ms | 7.3024ms | 136.9422 Ops/s | 135.5724 Ops/s | |
| test_collector_preallocated_write[stack_out-cuda-32-2] | 0.3054ms | 0.2437ms | 4.1038 KOps/s | 4.0297 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-32-8] | 0.5273ms | 0.3644ms | 2.7441 KOps/s | 2.6711 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-32-32] | 0.9570ms | 0.8196ms | 1.2202 KOps/s | 1.2003 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-2] | 0.3701ms | 0.2429ms | 4.1163 KOps/s | 3.8403 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-8] | 0.4216ms | 0.3648ms | 2.7411 KOps/s | 2.6870 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-32] | 0.9676ms | 0.8812ms | 1.1348 KOps/s | 1.1188 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-2] | 0.2573ms | 0.2034ms | 4.9169 KOps/s | 4.7695 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-8] | 0.8875ms | 0.7875ms | 1.2699 KOps/s | 1.2370 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-32] | 3.2653ms | 3.1201ms | 320.5010 Ops/s | 310.5712 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-2] | 0.5102ms | 0.3563ms | 2.8065 KOps/s | 2.7687 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-8] | 1.9098ms | 1.6641ms | 600.9212 Ops/s | 606.3159 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-32] | 10.1947ms | 9.9962ms | 100.0381 Ops/s | 99.7095 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-2] | 0.4396ms | 0.3934ms | 2.5421 KOps/s | 2.6217 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-8] | 1.5263ms | 1.4267ms | 700.8980 Ops/s | 697.3675 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-32] | 5.8513ms | 5.5229ms | 181.0640 Ops/s | 175.7247 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-2] | 0.4385ms | 0.3759ms | 2.6603 KOps/s | 2.6119 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-8] | 1.5573ms | 1.4072ms | 710.6226 Ops/s | 697.3773 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-32] | 6.0467ms | 5.6137ms | 178.1361 Ops/s | 178.2748 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-32-2] | 0.6147ms | 0.1676ms | 5.9658 KOps/s | 5.8865 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-32-8] | 0.7429ms | 0.6448ms | 1.5509 KOps/s | 1.5339 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-32-32] | 2.6601ms | 2.5544ms | 391.4885 Ops/s | 386.1273 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-2] | 0.3898ms | 0.3075ms | 3.2526 KOps/s | 3.2185 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-8] | 1.6799ms | 1.4566ms | 686.5438 Ops/s | 687.0702 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-32] | 9.5101ms | 9.4042ms | 106.3355 Ops/s | 106.7669 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-32-2] | 0.3987ms | 0.3115ms | 3.2099 KOps/s | 3.1503 KOps/s | |
| test_collector_preallocated_write[copy_at-cuda-32-8] | 1.2224ms | 1.1624ms | 860.3247 Ops/s | 849.7260 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-32-32] | 4.7415ms | 4.5371ms | 220.4071 Ops/s | 217.8125 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-2] | 0.3888ms | 0.3146ms | 3.1782 KOps/s | 3.1601 KOps/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-8] | 1.2584ms | 1.1734ms | 852.2173 Ops/s | 848.2685 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-32] | 4.7653ms | 4.6016ms | 217.3143 Ops/s | 215.8388 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] | 0.1531ms | 0.1173ms | 8.5277 KOps/s | 8.5482 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] | 0.5654ms | 0.4564ms | 2.1910 KOps/s | 2.1771 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] | 1.8777ms | 1.8206ms | 549.2690 Ops/s | 547.1257 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] | 0.3423ms | 0.2536ms | 3.9437 KOps/s | 4.0663 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] | 1.4120ms | 1.2164ms | 822.1047 Ops/s | 837.6081 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] | 8.6382ms | 8.4393ms | 118.4935 Ops/s | 119.5684 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] | 0.3205ms | 0.2633ms | 3.7976 KOps/s | 3.7960 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] | 1.0236ms | 0.9738ms | 1.0269 KOps/s | 1.0238 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] | 3.9614ms | 3.8221ms | 261.6385 Ops/s | 261.8401 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] | 0.3066ms | 0.2633ms | 3.7984 KOps/s | 3.8114 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] | 1.1128ms | 0.9795ms | 1.0209 KOps/s | 1.0205 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] | 3.9427ms | 3.7980ms | 263.2982 Ops/s | 261.0746 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-128-128] | 16.6390ms | 16.4546ms | 60.7732 Ops/s | 60.7384 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-128-256] | 33.3115ms | 33.1202ms | 30.1930 Ops/s | 30.1766 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-4096-128] | 84.0279ms | 83.2545ms | 12.0114 Ops/s | 11.9156 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-4096-256] | 0.1963s | 0.1959s | 5.1047 Ops/s | 5.1238 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-128-128] | 13.9510ms | 13.8056ms | 72.4345 Ops/s | 72.1969 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-128-256] | 27.3242ms | 27.0333ms | 36.9914 Ops/s | 36.8686 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-4096-128] | 14.0882ms | 13.9891ms | 71.4845 Ops/s | 71.0874 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-4096-256] | 27.4797ms | 27.3428ms | 36.5727 Ops/s | 36.4427 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] | 50.6343ms | 50.1633ms | 19.9349 Ops/s | 19.8307 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] | 0.1013s | 0.1011s | 9.8941 Ops/s | 9.8937 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] | 0.1439s | 0.1437s | 6.9587 Ops/s | 6.9236 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] | 0.3158s | 0.3148s | 3.1764 Ops/s | 3.2098 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] | 74.8074ms | 72.3741ms | 13.8171 Ops/s | 14.0570 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] | 0.1443s | 0.1430s | 6.9916 Ops/s | 6.7600 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] | 72.2666ms | 71.9263ms | 13.9031 Ops/s | 13.7553 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] | 0.1446s | 0.1440s | 6.9423 Ops/s | 6.7557 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-128-128] | 30.0569ms | 29.8915ms | 33.4543 Ops/s | 33.4835 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-128-256] | 60.5619ms | 60.4515ms | 16.5422 Ops/s | 16.5979 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-4096-128] | 0.1173s | 0.1169s | 8.5560 Ops/s | 8.4324 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-4096-256] | 0.2630s | 0.2627s | 3.8064 Ops/s | 3.8354 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-128-128] | 42.7939ms | 42.2225ms | 23.6840 Ops/s | 23.5404 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-128-256] | 84.6149ms | 84.2400ms | 11.8708 Ops/s | 11.8095 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-4096-128] | 44.1592ms | 42.6258ms | 23.4600 Ops/s | 23.3498 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-4096-256] | 85.3583ms | 84.7683ms | 11.7969 Ops/s | 11.7009 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] | 20.0846ms | 19.9528ms | 50.1182 Ops/s | 49.6973 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] | 41.0197ms | 40.8685ms | 24.4687 Ops/s | 24.5073 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] | 0.1068s | 0.1064s | 9.3993 Ops/s | 9.5007 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] | 0.2393s | 0.2387s | 4.1901 Ops/s | 4.2406 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] | 35.4111ms | 34.8868ms | 28.6641 Ops/s | 28.3757 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] | 70.0743ms | 69.7352ms | 14.3400 Ops/s | 14.1587 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] | 35.6324ms | 35.0438ms | 28.5357 Ops/s | 28.2489 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] | 71.0791ms | 70.6911ms | 14.1460 Ops/s | 14.1379 Ops/s | |
| test_plain_set_nested | 39.3610μs | 15.1016μs | 66.2180 KOps/s | 66.6550 KOps/s | |
| test_plain_set_stack_nested | 53.8810μs | 14.8257μs | 67.4504 KOps/s | 67.7436 KOps/s | |
| test_plain_set_nested_inplace | 49.8120μs | 16.5846μs | 60.2969 KOps/s | 60.7528 KOps/s | |
| test_plain_set_stack_nested_inplace | 95.4220μs | 16.0815μs | 62.1833 KOps/s | 63.1301 KOps/s | |
| test_items | 43.0810μs | 6.0399μs | 165.5644 KOps/s | 167.3585 KOps/s | |
| test_items_nested | 0.4836ms | 0.4573ms | 2.1866 KOps/s | 2.1771 KOps/s | |
| test_items_nested_locked | 0.4886ms | 0.4638ms | 2.1562 KOps/s | 2.1682 KOps/s | |
| test_items_nested_leaf | 0.1634ms | 99.4909μs | 10.0512 KOps/s | 10.1695 KOps/s | |
| test_items_stack_nested | 0.5490ms | 0.4617ms | 2.1659 KOps/s | 2.1745 KOps/s | |
| test_items_stack_nested_leaf | 0.1386ms | 98.6019μs | 10.1418 KOps/s | 10.1043 KOps/s | |
| test_items_stack_nested_locked | 0.4881ms | 0.4603ms | 2.1723 KOps/s | 2.1711 KOps/s | |
| test_keys | 34.2310μs | 4.2116μs | 237.4401 KOps/s | 239.0159 KOps/s | |
| test_keys_nested | 0.2373ms | 0.1281ms | 7.8077 KOps/s | 7.8092 KOps/s | |
| test_keys_nested_locked | 0.7072ms | 0.1367ms | 7.3156 KOps/s | 7.2772 KOps/s | |
| test_keys_nested_leaf | 0.1578ms | 0.1196ms | 8.3626 KOps/s | 8.3493 KOps/s | |
| test_keys_stack_nested | 0.1588ms | 0.1269ms | 7.8828 KOps/s | 7.8351 KOps/s | |
| test_keys_stack_nested_leaf | 0.1738ms | 0.1175ms | 8.5102 KOps/s | 8.4176 KOps/s | |
| test_keys_stack_nested_locked | 0.1851ms | 0.1353ms | 7.3918 KOps/s | 7.3492 KOps/s | |
| test_values | 6.0762μs | 1.0574μs | 945.7147 KOps/s | 982.8822 KOps/s | |
| test_values_nested | 83.4420μs | 52.1610μs | 19.1714 KOps/s | 19.0025 KOps/s | |
| test_values_nested_locked | 86.1720μs | 55.8807μs | 17.8953 KOps/s | 17.7499 KOps/s | |
| test_values_nested_leaf | 0.1388ms | 59.8198μs | 16.7169 KOps/s | 16.4772 KOps/s | |
| test_values_stack_nested | 85.6320μs | 52.0213μs | 19.2229 KOps/s | 18.9660 KOps/s | |
| test_values_stack_nested_leaf | 93.3630μs | 59.7428μs | 16.7384 KOps/s | 16.5600 KOps/s | |
| test_values_stack_nested_locked | 89.5320μs | 55.9290μs | 17.8798 KOps/s | 17.9821 KOps/s | |
| test_membership | 16.6000μs | 0.9859μs | 1.0143 MOps/s | 1.1963 MOps/s | |
| test_membership_nested | 77.8820μs | 2.9926μs | 334.1612 KOps/s | 335.8096 KOps/s | |
| test_membership_nested_leaf | 36.8710μs | 2.9350μs | 340.7136 KOps/s | 334.8795 KOps/s | |
| test_membership_stacked_nested | 32.0810μs | 2.8497μs | 350.9122 KOps/s | 347.8977 KOps/s | |
| test_membership_stacked_nested_leaf | 34.9110μs | 2.7506μs | 363.5521 KOps/s | 346.9217 KOps/s | |
| test_membership_nested_last | 29.2310μs | 4.4447μs | 224.9886 KOps/s | 223.8521 KOps/s | |
| test_membership_nested_leaf_last | 34.9710μs | 4.4410μs | 225.1739 KOps/s | 223.0761 KOps/s | |
| test_membership_stacked_nested_last | 54.1820μs | 4.2669μs | 234.3641 KOps/s | 233.5273 KOps/s | |
| test_membership_stacked_nested_leaf_last | 45.4010μs | 4.2871μs | 233.2597 KOps/s | 231.7243 KOps/s | |
| test_nested_getleaf | 99.0330μs | 21.4186μs | 46.6884 KOps/s | 46.5363 KOps/s | |
| test_nested_get | 66.5420μs | 20.1901μs | 49.5293 KOps/s | 49.0927 KOps/s | |
| test_stacked_getleaf | 66.8320μs | 20.9179μs | 47.8059 KOps/s | 47.6324 KOps/s | |
| test_stacked_get | 45.8410μs | 19.8582μs | 50.3571 KOps/s | 49.2961 KOps/s | |
| test_nested_getitemleaf | 52.3710μs | 22.2547μs | 44.9343 KOps/s | 45.3155 KOps/s | |
| test_nested_getitem | 50.9210μs | 21.0484μs | 47.5095 KOps/s | 48.6207 KOps/s | |
| test_stacked_getitemleaf | 90.8420μs | 21.4304μs | 46.6627 KOps/s | 46.3423 KOps/s | |
| test_stacked_getitem | 56.1820μs | 20.9038μs | 47.8383 KOps/s | 48.3412 KOps/s | |
| test_lock_nested | 7.9012ms | 0.5075ms | 1.9706 KOps/s | 1.9883 KOps/s | |
| test_lock_stack_nested | 0.5340ms | 0.5029ms | 1.9886 KOps/s | 1.9875 KOps/s | |
| test_unlock_nested | 0.4716ms | 0.3943ms | 2.5361 KOps/s | 2.5357 KOps/s | |
| test_unlock_stack_nested | 0.4340ms | 0.3961ms | 2.5246 KOps/s | 2.5118 KOps/s | |
| test_flatten_speed | 0.1926ms | 0.1224ms | 8.1719 KOps/s | 8.0161 KOps/s | |
| test_unflatten_speed | 0.6382ms | 0.5634ms | 1.7751 KOps/s | 1.7743 KOps/s | |
| test_common_ops | 0.8306ms | 0.6894ms | 1.4505 KOps/s | 1.4251 KOps/s | |
| test_creation | 0.1432ms | 3.1781μs | 314.6496 KOps/s | 313.4424 KOps/s | |
| test_creation_empty | 29.0510μs | 7.0054μs | 142.7475 KOps/s | 141.9341 KOps/s | |
| test_creation_nested_1 | 43.6110μs | 11.4718μs | 87.1706 KOps/s | 87.0222 KOps/s | |
| test_creation_nested_2 | 69.3020μs | 12.7793μs | 78.2515 KOps/s | 74.1485 KOps/s | |
| test_creation_many_keys[10] | 90.6720μs | 21.1344μs | 47.3162 KOps/s | 47.2723 KOps/s | |
| test_creation_many_keys[50] | 0.1300ms | 90.8465μs | 11.0076 KOps/s | 11.0518 KOps/s | |
| test_creation_many_keys[100] | 0.2401ms | 0.1770ms | 5.6496 KOps/s | 5.5799 KOps/s | |
| test_creation_nested_many_keys[10] | 73.5220μs | 45.0409μs | 22.2020 KOps/s | 22.1023 KOps/s | |
| test_creation_nested_many_keys[50] | 0.2758ms | 0.1825ms | 5.4780 KOps/s | 5.3816 KOps/s | |
| test_clone | 43.5410μs | 12.5726μs | 79.5383 KOps/s | 78.1346 KOps/s | |
| test_getitem[int] | 1.4913ms | 15.1995μs | 65.7918 KOps/s | 65.5876 KOps/s | |
| test_getitem[slice_int] | 0.1380ms | 24.2953μs | 41.1603 KOps/s | 40.8061 KOps/s | |
| test_getitem[range] | 0.1718ms | 61.3422μs | 16.3020 KOps/s | 16.3790 KOps/s | |
| test_getitem[tuple] | 0.1433ms | 23.7662μs | 42.0765 KOps/s | 41.7271 KOps/s | |
| test_getitem[list] | 0.1833ms | 56.5608μs | 17.6801 KOps/s | 17.7713 KOps/s | |
| test_setitem_dim[int] | 46.6720μs | 24.4507μs | 40.8987 KOps/s | 39.8805 KOps/s | |
| test_setitem_dim[slice_int] | 60.8810μs | 40.4498μs | 24.7220 KOps/s | 24.3067 KOps/s | |
| test_setitem_dim[range] | 0.1320ms | 91.7241μs | 10.9023 KOps/s | 10.9031 KOps/s | |
| test_setitem_dim[tuple] | 60.7020μs | 38.3148μs | 26.0996 KOps/s | 25.7576 KOps/s | |
| test_setitem | 50.1210μs | 16.9003μs | 59.1706 KOps/s | 58.8399 KOps/s | |
| test_set | 49.6210μs | 16.1038μs | 62.0973 KOps/s | 62.1266 KOps/s | |
| test_set_shared | 0.5041ms | 0.2074ms | 4.8212 KOps/s | 4.8981 KOps/s | |
| test_update | 0.2233ms | 20.9588μs | 47.7126 KOps/s | 47.7180 KOps/s | |
| test_update_nested | 65.2710μs | 32.6907μs | 30.5898 KOps/s | 30.5604 KOps/s | |
| test_update__nested | 0.4948ms | 32.2515μs | 31.0063 KOps/s | 30.9386 KOps/s | |
| test_set_nested | 52.5220μs | 17.8728μs | 55.9508 KOps/s | 55.8807 KOps/s | |
| test_set_nested_new | 54.4410μs | 22.4102μs | 44.6226 KOps/s | 44.5546 KOps/s | |
| test_select | 74.0820μs | 38.9869μs | 25.6496 KOps/s | 25.5292 KOps/s | |
| test_select_nested | 0.1499ms | 74.0191μs | 13.5100 KOps/s | 13.2793 KOps/s | |
| test_exclude_nested | 0.1188ms | 91.1906μs | 10.9660 KOps/s | 10.9142 KOps/s | |
| test_empty[True] | 0.5641ms | 0.3956ms | 2.5276 KOps/s | 2.5267 KOps/s | |
| test_empty[False] | 10.8702μs | 1.3810μs | 724.0879 KOps/s | 719.5970 KOps/s | |
| test_to | 95.4020μs | 69.9284μs | 14.3004 KOps/s | 14.4045 KOps/s | |
| test_to_nonblocking | 0.1797ms | 68.8434μs | 14.5257 KOps/s | 14.4744 KOps/s | |
| test_unbind_speed | 0.3816ms | 0.3357ms | 2.9788 KOps/s | 2.9710 KOps/s | |
| test_unbind_speed_stack0 | 0.4158ms | 0.3280ms | 3.0490 KOps/s | 3.0236 KOps/s | |
| test_unbind_speed_stack1 | 0.1160s | 0.8561ms | 1.1681 KOps/s | 1.1581 KOps/s | |
| test_split | 0.1162s | 1.4510ms | 689.1861 Ops/s | 628.5945 Ops/s | |
| test_chunk | 1.2761ms | 1.2362ms | 808.9089 Ops/s | 808.5336 Ops/s | |
| test_to_cpu_blocking | 19.7183ms | 19.0347ms | 52.5356 Ops/s | 52.3018 Ops/s | |
| test_to_cpu_global_sync | 0.1274s | 12.7549ms | 78.4012 Ops/s | 87.6981 Ops/s | |
| test_to_cpu_event_sync | 12.4563ms | 12.2100ms | 81.8999 Ops/s | 81.4788 Ops/s | |
| test_to_cpu_default | 12.4927ms | 12.2322ms | 81.7515 Ops/s | 72.4867 Ops/s | |
| test_consolidate[False-None] | 4.2695ms | 4.1606ms | 240.3500 Ops/s | 240.2222 Ops/s | |
| test_consolidate[default-None] | 3.2320ms | 2.2194ms | 450.5668 Ops/s | 439.4906 Ops/s | |
| test_consolidate[reduce-overhead-None] | 3.1663ms | 2.1372ms | 467.8992 Ops/s | 453.8302 Ops/s | |
| test_consolidate_njt[False-None] | 5.3287ms | 5.1154ms | 195.4871 Ops/s | 193.8404 Ops/s | |
| test_to[False-False-None] | 2.2946ms | 2.1703ms | 460.7604 Ops/s | 461.1275 Ops/s | |
| test_to[True-False-None] | 2.2816ms | 1.9449ms | 514.1559 Ops/s | 520.3926 Ops/s | |
| test_to[within-False-None] | 6.3337ms | 6.1871ms | 161.6278 Ops/s | 161.8054 Ops/s | |
| test_to[True-default-None] | 10.4471ms | 10.1609ms | 98.4162 Ops/s | 97.3473 Ops/s | |
| test_to_njt[False-False-None] | 7.0221ms | 6.9099ms | 144.7200 Ops/s | 142.8496 Ops/s | |
| test_to_njt[True-False-None] | 4.0524ms | 3.8264ms | 261.3420 Ops/s | 259.7634 Ops/s | |
| test_to_njt[within-False-None] | 9.1428ms | 9.0421ms | 110.5937 Ops/s | 109.9783 Ops/s | |
| test_creation[device0] | 0.3966ms | 0.1145ms | 8.7364 KOps/s | 8.7832 KOps/s | |
| test_creation_from_tensor | 0.4428ms | 0.1154ms | 8.6692 KOps/s | 8.8991 KOps/s | |
| test_add_one[memmap_tensor0] | 0.2783ms | 5.9797μs | 167.2323 KOps/s | 165.8843 KOps/s | |
| test_contiguous[memmap_tensor0] | 46.1710μs | 0.6777μs | 1.4756 MOps/s | 2.0475 MOps/s | |
| test_stack[memmap_tensor0] | 87.2820μs | 4.0662μs | 245.9285 KOps/s | 246.8994 KOps/s | |
| test_memmaptd_index | 1.2995ms | 0.2716ms | 3.6815 KOps/s | 3.6956 KOps/s | |
| test_memmaptd_index_astensor | 0.5845ms | 0.3762ms | 2.6582 KOps/s | 2.6666 KOps/s | |
| test_memmaptd_index_op | 1.0510ms | 0.6126ms | 1.6323 KOps/s | 1.6179 KOps/s | |
| test_serialize_model | 0.1366s | 0.1351s | 7.4045 Ops/s | 7.4016 Ops/s | |
| test_serialize_model_pickle | 1.3719s | 1.2148s | 0.8232 Ops/s | 0.8257 Ops/s | |
| test_serialize_weights | 0.1351s | 0.1339s | 7.4687 Ops/s | 7.4615 Ops/s | |
| test_serialize_weights_returnearly | 65.3601ms | 61.0061ms | 16.3918 Ops/s | 15.7440 Ops/s | |
| test_serialize_weights_pickle | 1.3529s | 1.2301s | 0.8129 Ops/s | 0.8058 Ops/s | |
| test_reshape_pytree | 0.2177ms | 32.1132μs | 31.1398 KOps/s | 31.0988 KOps/s | |
| test_reshape_td | 83.1820μs | 44.7056μs | 22.3686 KOps/s | 22.2905 KOps/s | |
| test_view_pytree | 0.2233ms | 31.7762μs | 31.4701 KOps/s | 31.5284 KOps/s | |
| test_view_td | 93.0620μs | 54.6974μs | 18.2824 KOps/s | 18.3793 KOps/s | |
| test_unbind_pytree | 0.2409ms | 35.9694μs | 27.8014 KOps/s | 27.6387 KOps/s | |
| test_unbind_td | 0.1047ms | 50.3350μs | 19.8669 KOps/s | 19.7261 KOps/s | |
| test_split_pytree | 0.2579ms | 41.1366μs | 24.3092 KOps/s | 24.0016 KOps/s | |
| test_split_td | 0.1869ms | 71.6506μs | 13.9566 KOps/s | 13.9117 KOps/s | |
| test_add_pytree | 0.1942ms | 39.9941μs | 25.0037 KOps/s | 25.3966 KOps/s | |
| test_add_td | 0.1575ms | 51.8660μs | 19.2804 KOps/s | 18.7976 KOps/s | |
| test_compile_add_one_nested[tensordict-compile] | 0.4160ms | 0.2735ms | 3.6557 KOps/s | 3.6950 KOps/s | |
| test_compile_add_one_nested[tensordict-eager] | 0.2828ms | 0.1998ms | 5.0060 KOps/s | 4.9870 KOps/s | |
| test_compile_add_one_nested[pytree-compile] | 0.2153ms | 0.1549ms | 6.4538 KOps/s | 6.4854 KOps/s | |
| test_compile_add_one_nested[pytree-eager] | 0.5038ms | 0.1774ms | 5.6368 KOps/s | 5.5980 KOps/s | |
| test_compile_copy_nested[tensordict-compile] | 68.3110μs | 22.2735μs | 44.8963 KOps/s | 47.7853 KOps/s | |
| test_compile_copy_nested[tensordict-eager] | 0.3480ms | 52.7712μs | 18.9497 KOps/s | 18.3785 KOps/s | |
| test_compile_copy_nested[pytree-compile] | 0.2111ms | 19.0068μs | 52.6126 KOps/s | 51.6753 KOps/s | |
| test_compile_copy_nested[pytree-eager] | 0.3645ms | 66.8358μs | 14.9620 KOps/s | 14.8748 KOps/s | |
| test_compile_add_one_flat[tensordict-compile] | 0.5255ms | 0.3077ms | 3.2503 KOps/s | 3.1128 KOps/s | |
| test_compile_add_one_flat[tensordict-eager] | 0.3769ms | 0.2870ms | 3.4845 KOps/s | 3.4236 KOps/s | |
| test_compile_add_one_flat[tensorclass-compile] | 0.3907ms | 0.2080ms | 4.8079 KOps/s | 4.6247 KOps/s | |
| test_compile_add_one_flat[tensorclass-eager] | 0.1883ms | 75.8524μs | 13.1835 KOps/s | 13.2804 KOps/s | |
| test_compile_add_one_flat[pytree-compile] | 0.5000ms | 0.2422ms | 4.1288 KOps/s | 4.0427 KOps/s | |
| test_compile_add_one_flat[pytree-eager] | 0.8989ms | 0.5318ms | 1.8804 KOps/s | 1.8960 KOps/s | |
| test_compile_add_self_flat[tensordict-eager] | 0.5066ms | 0.3378ms | 2.9607 KOps/s | 2.9377 KOps/s | |
| test_compile_add_self_flat[tensordict-compile] | 0.3529ms | 0.3059ms | 3.2690 KOps/s | 3.0842 KOps/s | |
| test_compile_add_self_flat[tensorclass-eager] | 0.1376ms | 92.1215μs | 10.8552 KOps/s | 10.7675 KOps/s | |
| test_compile_add_self_flat[tensorclass-compile] | 0.3702ms | 0.2093ms | 4.7785 KOps/s | 4.3086 KOps/s | |
| test_compile_add_self_flat[pytree-eager] | 0.6786ms | 0.4246ms | 2.3553 KOps/s | 2.3118 KOps/s | |
| test_compile_add_self_flat[pytree-compile] | 0.3630ms | 0.2415ms | 4.1411 KOps/s | 4.0476 KOps/s | |
| test_compile_copy_flat[tensordict-compile] | 81.0720μs | 25.9266μs | 38.5705 KOps/s | 34.9578 KOps/s | |
| test_compile_copy_flat[tensordict-eager] | 70.6620μs | 40.3637μs | 24.7747 KOps/s | 24.3667 KOps/s | |
| test_compile_copy_flat[pytree-compile] | 0.1779ms | 25.3528μs | 39.4434 KOps/s | 36.7388 KOps/s | |
| test_compile_copy_flat[pytree-eager] | 0.3330ms | 51.6608μs | 19.3570 KOps/s | 19.4329 KOps/s | |
| test_compile_assign_and_add[tensordict-compile] | 2.0722ms | 0.2535ms | 3.9444 KOps/s | 3.5260 KOps/s | |
| test_compile_assign_and_add[tensordict-eager] | 3.5631ms | 3.3012ms | 302.9235 Ops/s | 289.8383 Ops/s | |
| test_compile_assign_and_add[pytree-compile] | 1.9352ms | 0.1937ms | 5.1619 KOps/s | 5.0499 KOps/s | |
| test_compile_assign_and_add[pytree-eager] | 2.9610ms | 2.7598ms | 362.3489 Ops/s | 353.7553 Ops/s | |
| test_compile_indexing[tensor-tensordict-compile] | 0.3212ms | 0.1534ms | 6.5177 KOps/s | 6.3881 KOps/s | |
| test_compile_indexing[tensor-tensordict-eager] | 0.3729ms | 73.3334μs | 13.6363 KOps/s | 13.7407 KOps/s | |
| test_compile_indexing[tensor-tensorclass-compile] | 0.2883ms | 0.1335ms | 7.4923 KOps/s | 7.4719 KOps/s | |
| test_compile_indexing[tensor-tensorclass-eager] | 0.3266ms | 43.0397μs | 23.2343 KOps/s | 21.9554 KOps/s | |
| test_compile_indexing[tensor-pytree-compile] | 0.1753ms | 0.1333ms | 7.5021 KOps/s | 7.1807 KOps/s | |
| test_compile_indexing[tensor-pytree-eager] | 0.2726ms | 43.0097μs | 23.2505 KOps/s | 21.8328 KOps/s | |
| test_compile_indexing[slice-tensordict-compile] | 0.1884ms | 0.1023ms | 9.7748 KOps/s | 10.8065 KOps/s | |
| test_compile_indexing[slice-tensordict-eager] | 0.2129ms | 27.2708μs | 36.6692 KOps/s | 36.8382 KOps/s | |
| test_compile_indexing[slice-tensorclass-compile] | 0.1815ms | 71.2554μs | 14.0340 KOps/s | 13.7215 KOps/s | |
| test_compile_indexing[slice-tensorclass-eager] | 0.2557ms | 21.9401μs | 45.5785 KOps/s | 45.8129 KOps/s | |
| test_compile_indexing[slice-pytree-compile] | 0.1716ms | 71.4898μs | 13.9880 KOps/s | 13.7819 KOps/s | |
| test_compile_indexing[slice-pytree-eager] | 0.2700ms | 21.8627μs | 45.7399 KOps/s | 46.0165 KOps/s | |
| test_compile_indexing[int-tensordict-compile] | 0.1474ms | 88.8700μs | 11.2524 KOps/s | 10.9454 KOps/s | |
| test_compile_indexing[int-tensordict-eager] | 0.2480ms | 27.2761μs | 36.6621 KOps/s | 36.8868 KOps/s | |
| test_compile_indexing[int-tensorclass-compile] | 0.1167ms | 71.4443μs | 13.9969 KOps/s | 13.8764 KOps/s | |
| test_compile_indexing[int-tensorclass-eager] | 0.2512ms | 21.8649μs | 45.7354 KOps/s | 45.9765 KOps/s | |
| test_compile_indexing[int-pytree-compile] | 0.1083ms | 71.7554μs | 13.9362 KOps/s | 13.7099 KOps/s | |
| test_compile_indexing[int-pytree-eager] | 0.2639ms | 21.8882μs | 45.6868 KOps/s | 46.0712 KOps/s | |
| test_compile_replace[single-eager] | 94.8520μs | 46.8338μs | 21.3521 KOps/s | 21.2723 KOps/s | |
| test_compile_replace[single-compile] | 0.2494ms | 0.1434ms | 6.9731 KOps/s | 6.8606 KOps/s | |
| test_compile_replace[multi-eager] | 0.6608ms | 0.5619ms | 1.7796 KOps/s | 1.8090 KOps/s | |
| test_compile_replace[multi-compile] | 0.2687ms | 0.1548ms | 6.4592 KOps/s | 6.1855 KOps/s | |
| test_compile_tc_getattr_20[eager] | 0.2315ms | 0.1678ms | 5.9582 KOps/s | 6.0043 KOps/s | |
| test_compile_tc_getattr_20[compile] | 0.2364ms | 0.1735ms | 5.7646 KOps/s | 5.6260 KOps/s | |
| test_compile_clone_shallow[20-eager] | 51.1010μs | 18.7659μs | 53.2882 KOps/s | 53.1053 KOps/s | |
| test_compile_clone_shallow[20-compile] | 84.6720μs | 22.5501μs | 44.3456 KOps/s | 43.6115 KOps/s | |
| test_compile_clone_shallow[40-eager] | 67.4920μs | 32.6886μs | 30.5917 KOps/s | 30.7082 KOps/s | |
| test_compile_clone_shallow[40-compile] | 79.1820μs | 25.0317μs | 39.9493 KOps/s | 40.5435 KOps/s | |
| test_compile_clone_shallow[80-eager] | 0.1375ms | 60.6022μs | 16.5010 KOps/s | 16.0822 KOps/s | |
| test_compile_clone_shallow[80-compile] | 82.8220μs | 28.8421μs | 34.6716 KOps/s | 34.9723 KOps/s | |
| test_compile_update_inplace[eager] | 0.1067ms | 61.2322μs | 16.3313 KOps/s | 16.3366 KOps/s | |
| test_compile_update_inplace[compile] | 0.2044ms | 0.1185ms | 8.4423 KOps/s | 8.1802 KOps/s | |
| test_mod_add[eager] | 0.2837ms | 0.2255ms | 4.4349 KOps/s | 4.4392 KOps/s | |
| test_mod_add[compile] | 0.3431ms | 0.1176ms | 8.5048 KOps/s | 8.0927 KOps/s | |
| test_mod_add[compile-overhead] | 0.3510ms | 0.1930ms | 5.1810 KOps/s | 4.9533 KOps/s | |
| test_mod_wrap[eager] | 0.5498ms | 0.4626ms | 2.1616 KOps/s | 2.1152 KOps/s | |
| test_mod_wrap[compile] | 0.4255ms | 0.3493ms | 2.8632 KOps/s | 2.8008 KOps/s | |
| test_mod_wrap[compile-overhead] | 7.1568ms | 4.0315ms | 248.0469 Ops/s | 250.0441 Ops/s | |
| test_mod_wrap_and_backward[eager] | 1.7794ms | 1.6441ms | 608.2474 Ops/s | 602.9962 Ops/s | |
| test_mod_wrap_and_backward[compile] | 1.8656ms | 1.3220ms | 756.4205 Ops/s | 757.4690 Ops/s | |
| test_mod_wrap_and_backward[compile-overhead] | 1.6044ms | 0.9334ms | 1.0714 KOps/s | 1.0536 KOps/s | |
| test_seq_add[eager] | 0.6575ms | 0.5047ms | 1.9813 KOps/s | 1.9638 KOps/s | |
| test_seq_add[compile] | 0.2504ms | 0.1309ms | 7.6373 KOps/s | 7.5091 KOps/s | |
| test_seq_add[compile-overhead] | 0.2488ms | 0.2031ms | 4.9238 KOps/s | 4.6706 KOps/s | |
| test_seq_wrap[eager] | 1.5199ms | 1.3963ms | 716.1579 Ops/s | 702.0431 Ops/s | |
| test_seq_wrap[compile] | 0.4141ms | 0.3571ms | 2.8002 KOps/s | 2.7122 KOps/s | |
| test_seq_wrap[compile-overhead] | 0.4479ms | 0.3166ms | 3.1583 KOps/s | 3.0936 KOps/s | |
| test_func_call_runtime[False-eager] | 0.9264ms | 0.7902ms | 1.2656 KOps/s | 1.2574 KOps/s | |
| test_func_call_runtime[False-compile] | 0.8094ms | 0.7502ms | 1.3331 KOps/s | 1.3234 KOps/s | |
| test_func_call_runtime[False-compile-overhead] | 0.6820ms | 0.5071ms | 1.9718 KOps/s | 1.9558 KOps/s | |
| test_func_call_runtime[True-eager] | 1.1366ms | 1.0518ms | 950.7953 Ops/s | 948.2524 Ops/s | |
| test_func_call_runtime[True-compile] | 0.8246ms | 0.7651ms | 1.3070 KOps/s | 1.2957 KOps/s | |
| test_func_call_runtime[True-compile-overhead] | 0.6855ms | 0.5211ms | 1.9190 KOps/s | 1.8823 KOps/s | |
| test_func_call_cm_runtime[False-eager] | 0.8582ms | 0.7889ms | 1.2677 KOps/s | 1.2328 KOps/s | |
| test_func_call_cm_runtime[False-compile] | 0.8571ms | 0.7367ms | 1.3575 KOps/s | 1.3092 KOps/s | |
| test_func_call_cm_runtime[False-compile-overhead] | 0.5386ms | 0.4875ms | 2.0511 KOps/s | 1.9992 KOps/s | |
| test_func_call_cm_runtime[True-eager] | 1.3022ms | 1.2011ms | 832.5378 Ops/s | 827.0374 Ops/s | |
| test_func_call_cm_runtime[True-compile] | 0.9189ms | 0.8450ms | 1.1834 KOps/s | 1.1735 KOps/s | |
| test_func_call_cm_runtime[True-compile-overhead] | 1.2172ms | 0.8494ms | 1.1773 KOps/s | 1.1278 KOps/s | |
| test_vmap_func_call_cm_runtime[eager] | 3.2066ms | 2.6568ms | 376.3885 Ops/s | 370.5090 Ops/s | |
| test_vmap_func_call_cm_runtime[compile] | 4.1051ms | 3.6186ms | 276.3496 Ops/s | 273.0181 Ops/s | |
| test_vmap_func_call_cm_runtime[compile-overhead] | 4.1027ms | 3.6250ms | 275.8594 Ops/s | 273.5968 Ops/s | |
| test_distributed | 2.0271ms | 0.1579ms | 6.3328 KOps/s | 6.8830 KOps/s | |
| test_tdmodule | 0.7088ms | 0.1944ms | 5.1453 KOps/s | 5.1257 KOps/s | |
| test_tdmodule_dispatch | 0.4545ms | 0.3716ms | 2.6914 KOps/s | 2.6592 KOps/s | |
| test_tdseq | 58.2220μs | 26.7656μs | 37.3613 KOps/s | 37.1285 KOps/s | |
| test_tdseq_dispatch | 0.2531ms | 0.2180ms | 4.5875 KOps/s | 4.5531 KOps/s | |
| test_instantiation_functorch | 2.1420ms | 2.0164ms | 495.9361 Ops/s | 488.8597 Ops/s | |
| test_exec_functorch | 0.2761ms | 0.1711ms | 5.8435 KOps/s | 5.8237 KOps/s | |
| test_exec_functional_call | 0.4316ms | 0.1538ms | 6.5016 KOps/s | 6.4799 KOps/s | |
| test_exec_td_decorator | 0.5188ms | 0.2413ms | 4.1444 KOps/s | 4.1152 KOps/s | |
| test_vmap_mlp_speed_decorator[True-True] | 1.2742ms | 1.1082ms | 902.3711 Ops/s | 894.7788 Ops/s | |
| test_vmap_mlp_speed_decorator[True-False] | 1.3003ms | 1.1045ms | 905.3932 Ops/s | 898.3223 Ops/s | |
| test_vmap_mlp_speed_decorator[False-True] | 1.0538ms | 0.8127ms | 1.2305 KOps/s | 1.2248 KOps/s | |
| test_vmap_mlp_speed_decorator[False-False] | 1.0313ms | 0.8095ms | 1.2353 KOps/s | 1.2209 KOps/s | |
| test_vmap_transformer_speed_decorator[True-True] | 24.9638ms | 24.6350ms | 40.5926 Ops/s | 40.2484 Ops/s | |
| test_vmap_transformer_speed_decorator[True-False] | 24.9625ms | 24.5415ms | 40.7473 Ops/s | 40.3679 Ops/s | |
| test_vmap_transformer_speed_decorator[False-True] | 24.4068ms | 24.1729ms | 41.3687 Ops/s | 40.9923 Ops/s | |
| test_vmap_transformer_speed_decorator[False-False] | 24.2900ms | 24.1452ms | 41.4161 Ops/s | 40.6880 Ops/s | |
| test_to_module_speed[True] | 1.7449ms | 1.6401ms | 609.7365 Ops/s | 599.3109 Ops/s | |
| test_to_module_speed[False] | 1.7340ms | 1.6245ms | 615.5808 Ops/s | 613.0135 Ops/s | |
| test_to_module_plain_tensor_speed[None] | 2.1849ms | 1.7336ms | 576.8230 Ops/s | 573.6866 Ops/s | |
| test_to_module_plain_tensor_speed[False] | 3.1062ms | 1.5784ms | 633.5539 Ops/s | 630.5065 Ops/s | |
| test_to_module_plain_tensor_speed[True] | 3.5746ms | 3.3976ms | 294.3231 Ops/s | 293.6560 Ops/s | |
| test_tc_init | 76.0410μs | 44.0803μs | 22.6859 KOps/s | 22.7166 KOps/s | |
| test_tc_init_tensor_only | 32.6610μs | 9.5053μs | 105.2049 KOps/s | 103.7246 KOps/s | |
| test_tc_init_nested | 0.1238ms | 87.9917μs | 11.3647 KOps/s | 11.3784 KOps/s | |
| test_tc_init_many_fields | 47.7610μs | 17.4162μs | 57.4177 KOps/s | 57.6294 KOps/s | |
| test_tc_first_layer_tensor | 18.9900μs | 1.8386μs | 543.8916 KOps/s | 563.3037 KOps/s | |
| test_tc_first_layer_tensor_only | 1.5265μs | 0.4070μs | 2.4571 MOps/s | 2.4623 MOps/s | |
| test_tc_first_layer_tensor_set | 29.9010μs | 4.2212μs | 236.8999 KOps/s | 238.0836 KOps/s | |
| test_tc_first_layer_tensor_only_set | 33.6110μs | 3.6060μs | 277.3130 KOps/s | 286.3240 KOps/s | |
| test_tc_first_layer_nontensor | 30.9710μs | 6.0662μs | 164.8483 KOps/s | 161.8546 KOps/s | |
| test_tc_second_layer_tensor | 26.7800μs | 4.3821μs | 228.1997 KOps/s | 226.4190 KOps/s | |
| test_tc_second_layer_nontensor | 33.7910μs | 8.6706μs | 115.3321 KOps/s | 115.6475 KOps/s | |
| test_unbind | 0.2924s | 13.5110ms | 74.0138 Ops/s | 64.4355 Ops/s | |
| test_full_like | 4.5646ms | 4.3856ms | 228.0177 Ops/s | 228.3497 Ops/s | |
| test_zeros_like | 4.5530ms | 4.3596ms | 229.3769 Ops/s | 229.1156 Ops/s | |
| test_ones_like | 4.5351ms | 4.3659ms | 229.0488 Ops/s | 228.8923 Ops/s | |
| test_clone | 6.9707ms | 6.4656ms | 154.6639 Ops/s | 156.3746 Ops/s | |
| test_squeeze | 76.1320μs | 14.2768μs | 70.0436 KOps/s | 70.8100 KOps/s | |
| test_unsqueeze | 0.1735ms | 0.1131ms | 8.8396 KOps/s | 8.9688 KOps/s | |
| test_split | 0.2668ms | 0.1898ms | 5.2686 KOps/s | 5.3607 KOps/s | |
| test_permute | 0.3217ms | 0.1945ms | 5.1425 KOps/s | 5.2116 KOps/s | |
| test_stack | 27.1452ms | 26.6587ms | 37.5111 Ops/s | 55.4376 Ops/s | |
| test_cat | 19.1717ms | 18.2143ms | 54.9018 Ops/s | 55.5535 Ops/s | |
| test_sequential_tensordict | 0.7127ms | 0.5650ms | 1.7699 KOps/s | 1.7572 KOps/s | |
| test_sequential_graph_module | 0.2360ms | 0.1164ms | 8.5886 KOps/s | 9.1089 KOps/s | |
| test_nested_tensordict | 0.9300ms | 0.8071ms | 1.2390 KOps/s | 1.2423 KOps/s | |
| test_nested_graph_module | 0.1695ms | 0.1275ms | 7.8454 KOps/s | 8.2217 KOps/s |
2 tasks
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
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
Description
Main's CI went red on 2026-07-09 (e.g. this run) after recent torch releases; both stable and nightly jobs fail with the same 49 tests. Two independent root causes, both fixed here:
1.
torch.squeezeno longer acceptsdim=None(40 failures)test_squeeze_with_noneandtest_squeeze_with_none_legacycalledtorch.squeeze(td, dim=None), which current torch rejects at the argument-parser level even for plain tensors. The tests now calltorch.squeeze(td), which has identical semantics on every torch version.Dropping the explicit
dim=Noneexposed a real library inconsistency:_UnsqueezedTensorDict._legacy_squeezewas missing thedim=Nonedefault thatTensorDictBase._legacy_squeezeandLazyStackedTensorDict._legacy_squeezealready have, sotd.squeeze()with no argument raised aTypeErroron that class in legacy mode. Fixed by adding the default.2. dynamo recursion in tensorclass
__setattr__(9 failures)The tensorclass machinery stores
cls.__setattr_parent__ = object.__setattr__and callsself.__setattr_parent__(key, value)to perform raw attribute writes. Current dynamo re-dispatches a slot wrapper stored as a class attribute through the type's custom__setattr__, so_setattrrecursed into itself infinitely undertorch.compile(InternalTorchDynamoError: RecursionError: maximum recursion depth exceeded). The call sites now invokeobject.__setattr__(self, key, value)directly, which dynamo special-cases correctly and which is equivalent in eager mode. The__setattr_parent__class attribute is kept for backward compatibility.Testing
All 49 previously failing tests pass locally on torch 2.14.0.dev20260708: the full squeeze family (95 tests), the 9 compile tests, plus full runs of
test/tensordict/test_methods.py(6441 passed),test/compile/test_compile.py,test/tensorclass,test/nn, andtest/tensordict/test_generic.py. The single remaining localtest_compile.pyfailure (TestTTDDynamoCompatibility::test_td_subclass_arithmetic_without_registration) pre-exists on main unchanged and does not appear in the Linux CI failure set.Generated with Claude Code