[Doc] Refresh TensorDict README for 0.13 - #1721
Merged
Merged
Conversation
vmoens
force-pushed
the
codex/readme-refresh-0.13
branch
from
June 4, 2026 14:05
4d13ac2 to
5506b2a
Compare
Contributor
|
| Name | Max | Mean | Ops | Ops on Repo HEAD
|
Change |
|---|---|---|---|---|---|
| test_collector_preallocated_write[stack_out-cpu-32-2] | 0.2190ms | 0.1617ms | 6.1844 KOps/s | 6.0229 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-32-8] | 0.3395ms | 0.2859ms | 3.4978 KOps/s | 3.4459 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-32-32] | 0.8219ms | 0.7463ms | 1.3400 KOps/s | 1.3198 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-2] | 0.4740ms | 0.3346ms | 2.9885 KOps/s | 2.9028 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-8] | 1.2587ms | 1.1388ms | 878.1497 Ops/s | 924.7807 Ops/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-32] | 7.5220ms | 7.2568ms | 137.8020 Ops/s | 136.3594 Ops/s | |
| test_collector_preallocated_write[stack_out-cuda-32-2] | 0.2798ms | 0.2398ms | 4.1706 KOps/s | 3.9089 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-32-8] | 0.4549ms | 0.3656ms | 2.7354 KOps/s | 2.7136 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-32-32] | 0.8811ms | 0.7943ms | 1.2590 KOps/s | 1.2541 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-2] | 0.3795ms | 0.2549ms | 3.9232 KOps/s | 3.9123 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-8] | 0.4527ms | 0.3716ms | 2.6912 KOps/s | 2.7444 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-32] | 0.9440ms | 0.8569ms | 1.1670 KOps/s | 1.1557 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-2] | 0.2771ms | 0.2167ms | 4.6148 KOps/s | 4.7030 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-8] | 0.9834ms | 0.8191ms | 1.2208 KOps/s | 1.1777 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-32] | 3.3627ms | 3.2244ms | 310.1357 Ops/s | 309.9366 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-2] | 0.4724ms | 0.3658ms | 2.7340 KOps/s | 2.6745 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-8] | 1.9219ms | 1.7000ms | 588.2448 Ops/s | 610.0313 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-32] | 10.2419ms | 10.0494ms | 99.5081 Ops/s | 98.3733 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-2] | 0.4416ms | 0.3860ms | 2.5906 KOps/s | 2.5229 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-8] | 1.6021ms | 1.4429ms | 693.0285 Ops/s | 685.6119 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-32] | 5.7956ms | 5.6478ms | 177.0607 Ops/s | 174.6005 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-2] | 0.4696ms | 0.3873ms | 2.5819 KOps/s | 2.5501 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-8] | 1.6539ms | 1.4505ms | 689.4317 Ops/s | 682.6471 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-32] | 5.8813ms | 5.6867ms | 175.8504 Ops/s | 174.5376 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-32-2] | 0.6011ms | 0.1684ms | 5.9383 KOps/s | 5.8638 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-32-8] | 0.7303ms | 0.6512ms | 1.5357 KOps/s | 1.5005 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-32-32] | 2.7319ms | 2.6022ms | 384.2858 Ops/s | 379.9325 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-2] | 0.3983ms | 0.3072ms | 3.2548 KOps/s | 3.2172 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-8] | 1.9478ms | 1.5034ms | 665.1448 Ops/s | 681.5286 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-32] | 9.7880ms | 9.4761ms | 105.5284 Ops/s | 106.7550 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-32-2] | 0.4041ms | 0.3185ms | 3.1397 KOps/s | 3.1033 KOps/s | |
| test_collector_preallocated_write[copy_at-cuda-32-8] | 1.2927ms | 1.1888ms | 841.2110 Ops/s | 831.6142 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-32-32] | 4.8197ms | 4.6564ms | 214.7583 Ops/s | 211.7915 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-2] | 0.4031ms | 0.3205ms | 3.1197 KOps/s | 3.0816 KOps/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-8] | 1.2938ms | 1.1928ms | 838.3303 Ops/s | 826.2019 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-32] | 4.8427ms | 4.6792ms | 213.7130 Ops/s | 211.0837 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] | 0.1483ms | 0.1154ms | 8.6678 KOps/s | 8.6127 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] | 0.5252ms | 0.4520ms | 2.2122 KOps/s | 2.1907 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] | 1.9097ms | 1.7997ms | 555.6567 Ops/s | 551.4438 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] | 0.3850ms | 0.2435ms | 4.1071 KOps/s | 4.0181 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] | 1.6767ms | 1.2264ms | 815.3690 Ops/s | 830.7674 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] | 8.7775ms | 8.5197ms | 117.3748 Ops/s | 117.4982 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] | 0.3359ms | 0.2719ms | 3.6778 KOps/s | 3.6499 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] | 1.1047ms | 1.0129ms | 987.3105 Ops/s | 978.8340 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] | 4.1183ms | 3.9652ms | 252.1971 Ops/s | 249.6830 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] | 0.3498ms | 0.2715ms | 3.6826 KOps/s | 3.6427 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] | 1.0991ms | 1.0114ms | 988.7048 Ops/s | 970.1529 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] | 4.2038ms | 3.9752ms | 251.5614 Ops/s | 249.5549 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-128-128] | 16.2219ms | 15.8647ms | 63.0330 Ops/s | 63.0464 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-128-256] | 31.8754ms | 31.7285ms | 31.5174 Ops/s | 31.3896 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-4096-128] | 83.4315ms | 83.0988ms | 12.0339 Ops/s | 12.0338 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-4096-256] | 0.1986s | 0.1984s | 5.0403 Ops/s | 5.1196 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-128-128] | 13.3360ms | 13.1196ms | 76.2220 Ops/s | 75.6014 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-128-256] | 26.1283ms | 25.8981ms | 38.6129 Ops/s | 38.8084 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-4096-128] | 13.5276ms | 13.3122ms | 75.1190 Ops/s | 74.6967 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-4096-256] | 26.5005ms | 26.2683ms | 38.0687 Ops/s | 38.0982 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] | 51.4995ms | 50.9128ms | 19.6414 Ops/s | 18.4902 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] | 0.1030s | 0.1022s | 9.7885 Ops/s | 9.1851 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] | 0.1462s | 0.1450s | 6.8945 Ops/s | 7.0536 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] | 0.3275s | 0.3261s | 3.0667 Ops/s | 3.1685 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] | 73.4315ms | 72.7659ms | 13.7427 Ops/s | 12.9201 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] | 0.1469s | 0.1460s | 6.8501 Ops/s | 6.4340 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] | 74.5429ms | 73.7783ms | 13.5541 Ops/s | 12.5990 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] | 0.1479s | 0.1473s | 6.7867 Ops/s | 6.4121 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-128-128] | 30.1226ms | 29.7204ms | 33.6469 Ops/s | 33.5192 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-128-256] | 60.7460ms | 60.4367ms | 16.5462 Ops/s | 16.6681 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-4096-128] | 0.1202s | 0.1198s | 8.3466 Ops/s | 8.5761 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-4096-256] | 0.2640s | 0.2633s | 3.7976 Ops/s | 3.8356 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-128-128] | 42.5801ms | 42.4331ms | 23.5665 Ops/s | 23.2944 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-128-256] | 85.8638ms | 85.5223ms | 11.6929 Ops/s | 11.6720 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-4096-128] | 42.9909ms | 42.6562ms | 23.4432 Ops/s | 23.1630 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-4096-256] | 86.0841ms | 85.8620ms | 11.6466 Ops/s | 11.5928 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] | 20.0511ms | 19.9000ms | 50.2512 Ops/s | 50.5648 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] | 41.0881ms | 40.7522ms | 24.5385 Ops/s | 24.6906 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] | 0.1076s | 0.1073s | 9.3167 Ops/s | 9.5220 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] | 0.2408s | 0.2401s | 4.1656 Ops/s | 4.2253 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] | 38.6911ms | 37.1047ms | 26.9508 Ops/s | 27.2872 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] | 77.5143ms | 74.1582ms | 13.4847 Ops/s | 13.6799 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] | 38.7022ms | 37.0715ms | 26.9749 Ops/s | 27.1522 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] | 77.4455ms | 75.4323ms | 13.2569 Ops/s | 13.6536 Ops/s | |
| test_plain_set_nested | 46.9130μs | 15.0161μs | 66.5952 KOps/s | 66.7874 KOps/s | |
| test_plain_set_stack_nested | 40.2120μs | 14.8082μs | 67.5304 KOps/s | 67.9997 KOps/s | |
| test_plain_set_nested_inplace | 45.6920μs | 16.3394μs | 61.2019 KOps/s | 59.9999 KOps/s | |
| test_plain_set_stack_nested_inplace | 40.6320μs | 16.0512μs | 62.3007 KOps/s | 61.2068 KOps/s | |
| test_items | 27.0420μs | 5.5308μs | 180.8047 KOps/s | 181.6392 KOps/s | |
| test_items_nested | 0.5174ms | 0.4539ms | 2.2031 KOps/s | 2.2144 KOps/s | |
| test_items_nested_locked | 0.5219ms | 0.4523ms | 2.2112 KOps/s | 2.2099 KOps/s | |
| test_items_nested_leaf | 0.1516ms | 92.0373μs | 10.8652 KOps/s | 10.8167 KOps/s | |
| test_items_stack_nested | 0.5473ms | 0.4523ms | 2.2110 KOps/s | 2.2299 KOps/s | |
| test_items_stack_nested_leaf | 0.1492ms | 90.8555μs | 11.0065 KOps/s | 10.8142 KOps/s | |
| test_items_stack_nested_locked | 0.5750ms | 0.4597ms | 2.1752 KOps/s | 2.2162 KOps/s | |
| test_keys | 27.2710μs | 4.1262μs | 242.3515 KOps/s | 240.6518 KOps/s | |
| test_keys_nested | 0.1723ms | 0.1308ms | 7.6476 KOps/s | 7.6599 KOps/s | |
| test_keys_nested_locked | 2.1260ms | 0.1395ms | 7.1662 KOps/s | 7.2723 KOps/s | |
| test_keys_nested_leaf | 0.1643ms | 0.1216ms | 8.2212 KOps/s | 8.3180 KOps/s | |
| test_keys_stack_nested | 0.1637ms | 0.1301ms | 7.6870 KOps/s | 7.6689 KOps/s | |
| test_keys_stack_nested_leaf | 0.1681ms | 0.1212ms | 8.2496 KOps/s | 8.3850 KOps/s | |
| test_keys_stack_nested_locked | 0.1870ms | 0.1383ms | 7.2327 KOps/s | 7.2340 KOps/s | |
| test_values | 7.0064μs | 0.9967μs | 1.0033 MOps/s | 996.5500 KOps/s | |
| test_values_nested | 80.8850μs | 53.4838μs | 18.6973 KOps/s | 19.1564 KOps/s | |
| test_values_nested_locked | 82.8150μs | 56.3874μs | 17.7345 KOps/s | 18.0643 KOps/s | |
| test_values_nested_leaf | 89.4860μs | 59.9954μs | 16.6679 KOps/s | 16.7445 KOps/s | |
| test_values_stack_nested | 79.2950μs | 53.1679μs | 18.8083 KOps/s | 19.2114 KOps/s | |
| test_values_stack_nested_leaf | 88.7050μs | 60.1194μs | 16.6336 KOps/s | 16.7209 KOps/s | |
| test_values_stack_nested_locked | 81.3750μs | 56.1988μs | 17.7940 KOps/s | 18.0513 KOps/s | |
| test_membership | 4.6337μs | 0.7943μs | 1.2590 MOps/s | 1.2738 MOps/s | |
| test_membership_nested | 25.3820μs | 2.8082μs | 356.1042 KOps/s | 355.9579 KOps/s | |
| test_membership_nested_leaf | 28.6920μs | 2.8155μs | 355.1754 KOps/s | 354.6083 KOps/s | |
| test_membership_stacked_nested | 34.2420μs | 2.7186μs | 367.8312 KOps/s | 360.2936 KOps/s | |
| test_membership_stacked_nested_leaf | 36.5530μs | 2.7039μs | 369.8313 KOps/s | 366.7718 KOps/s | |
| test_membership_nested_last | 20.7620μs | 4.2291μs | 236.4542 KOps/s | 236.9663 KOps/s | |
| test_membership_nested_leaf_last | 30.1420μs | 4.2292μs | 236.4531 KOps/s | 238.7855 KOps/s | |
| test_membership_stacked_nested_last | 38.0820μs | 4.1671μs | 239.9770 KOps/s | 242.5931 KOps/s | |
| test_membership_stacked_nested_leaf_last | 34.1420μs | 4.1337μs | 241.9141 KOps/s | 244.1975 KOps/s | |
| test_nested_getleaf | 62.8240μs | 20.6284μs | 48.4768 KOps/s | 48.2951 KOps/s | |
| test_nested_get | 49.6730μs | 19.6491μs | 50.8928 KOps/s | 51.7701 KOps/s | |
| test_stacked_getleaf | 0.1005ms | 20.4941μs | 48.7946 KOps/s | 48.2918 KOps/s | |
| test_stacked_get | 37.6120μs | 19.7017μs | 50.7570 KOps/s | 50.6074 KOps/s | |
| test_nested_getitemleaf | 47.8330μs | 21.2204μs | 47.1245 KOps/s | 46.8872 KOps/s | |
| test_nested_getitem | 56.3040μs | 20.2838μs | 49.3004 KOps/s | 49.3139 KOps/s | |
| test_stacked_getitemleaf | 48.1130μs | 21.2109μs | 47.1455 KOps/s | 47.3659 KOps/s | |
| test_stacked_getitem | 46.7030μs | 20.2079μs | 49.4857 KOps/s | 49.1738 KOps/s | |
| test_lock_nested | 0.5553ms | 0.4777ms | 2.0933 KOps/s | 2.0507 KOps/s | |
| test_lock_stack_nested | 0.5776ms | 0.4844ms | 2.0646 KOps/s | 2.0623 KOps/s | |
| test_unlock_nested | 0.4991ms | 0.3802ms | 2.6302 KOps/s | 2.6232 KOps/s | |
| test_unlock_stack_nested | 0.4682ms | 0.3820ms | 2.6176 KOps/s | 2.5977 KOps/s | |
| test_flatten_speed | 0.1905ms | 0.1155ms | 8.6573 KOps/s | 8.6271 KOps/s | |
| test_unflatten_speed | 0.6187ms | 0.5500ms | 1.8180 KOps/s | 1.8078 KOps/s | |
| test_common_ops | 0.1000s | 1.6274ms | 614.4859 Ops/s | 263.2751 Ops/s | |
| test_creation | 0.1261ms | 2.9968μs | 333.6887 KOps/s | 336.8858 KOps/s | |
| test_creation_empty | 34.6330μs | 6.5858μs | 151.8426 KOps/s | 151.3372 KOps/s | |
| test_creation_nested_1 | 30.1310μs | 10.8575μs | 92.1026 KOps/s | 91.3834 KOps/s | |
| test_creation_nested_2 | 38.5620μs | 12.9027μs | 77.5030 KOps/s | 77.2844 KOps/s | |
| test_creation_many_keys[10] | 48.5730μs | 19.6618μs | 50.8601 KOps/s | 50.9966 KOps/s | |
| test_creation_many_keys[50] | 0.1133ms | 84.9659μs | 11.7694 KOps/s | 11.7746 KOps/s | |
| test_creation_many_keys[100] | 0.2136ms | 0.1658ms | 6.0330 KOps/s | 6.0368 KOps/s | |
| test_creation_nested_many_keys[10] | 65.0040μs | 42.5515μs | 23.5009 KOps/s | 23.5484 KOps/s | |
| test_creation_nested_many_keys[50] | 0.2205ms | 0.1732ms | 5.7743 KOps/s | 5.7556 KOps/s | |
| test_clone | 40.5430μs | 12.9003μs | 77.5179 KOps/s | 77.4721 KOps/s | |
| test_getitem[int] | 1.4976ms | 14.5046μs | 68.9437 KOps/s | 68.6267 KOps/s | |
| test_getitem[slice_int] | 0.1343ms | 24.0929μs | 41.5060 KOps/s | 41.9667 KOps/s | |
| test_getitem[range] | 0.1651ms | 61.1990μs | 16.3401 KOps/s | 15.9732 KOps/s | |
| test_getitem[tuple] | 0.1458ms | 23.2254μs | 43.0563 KOps/s | 41.9413 KOps/s | |
| test_getitem[list] | 0.1769ms | 57.8320μs | 17.2915 KOps/s | 17.2905 KOps/s | |
| test_setitem_dim[int] | 44.1830μs | 24.5020μs | 40.8131 KOps/s | 39.2894 KOps/s | |
| test_setitem_dim[slice_int] | 60.5430μs | 40.8677μs | 24.4692 KOps/s | 23.9854 KOps/s | |
| test_setitem_dim[range] | 0.1168ms | 91.7437μs | 10.8999 KOps/s | 10.7671 KOps/s | |
| test_setitem_dim[tuple] | 61.2840μs | 38.6052μs | 25.9032 KOps/s | 25.1706 KOps/s | |
| test_setitem | 46.3330μs | 17.2590μs | 57.9408 KOps/s | 57.7035 KOps/s | |
| test_set | 50.3030μs | 16.6155μs | 60.1846 KOps/s | 60.4581 KOps/s | |
| test_set_shared | 0.6354ms | 0.2069ms | 4.8339 KOps/s | 4.8311 KOps/s | |
| test_update | 63.5140μs | 21.1954μs | 47.1799 KOps/s | 46.7612 KOps/s | |
| test_update_nested | 65.2140μs | 32.5921μs | 30.6823 KOps/s | 30.1190 KOps/s | |
| test_update__nested | 0.4563ms | 32.3190μs | 30.9415 KOps/s | 30.8616 KOps/s | |
| test_set_nested | 49.2530μs | 18.4127μs | 54.3105 KOps/s | 53.7419 KOps/s | |
| test_set_nested_new | 49.7830μs | 23.1736μs | 43.1526 KOps/s | 42.5983 KOps/s | |
| test_select | 70.9550μs | 39.0347μs | 25.6182 KOps/s | 25.2217 KOps/s | |
| test_select_nested | 92.8660μs | 70.9945μs | 14.0856 KOps/s | 14.1278 KOps/s | |
| test_exclude_nested | 0.1239ms | 87.3740μs | 11.4451 KOps/s | 11.5829 KOps/s | |
| test_empty[True] | 0.5243ms | 0.3905ms | 2.5609 KOps/s | 2.5840 KOps/s | |
| test_empty[False] | 6.7978μs | 1.3148μs | 760.5708 KOps/s | 759.0103 KOps/s | |
| test_to | 0.1334ms | 67.6549μs | 14.7809 KOps/s | 14.8064 KOps/s | |
| test_to_nonblocking | 0.1063ms | 67.1710μs | 14.8874 KOps/s | 14.5779 KOps/s | |
| test_unbind_speed | 0.3602ms | 0.3227ms | 3.0985 KOps/s | 3.0859 KOps/s | |
| test_unbind_speed_stack0 | 0.3648ms | 0.3217ms | 3.1082 KOps/s | 3.1524 KOps/s | |
| test_unbind_speed_stack1 | 0.1157s | 0.8278ms | 1.2080 KOps/s | 938.4833 Ops/s | |
| test_split | 0.1150s | 1.2251ms | 816.2918 Ops/s | 919.5146 Ops/s | |
| test_chunk | 1.1036ms | 1.0491ms | 953.2223 Ops/s | 953.2012 Ops/s | |
| test_to_cpu_blocking | 20.0983ms | 19.9823ms | 50.0442 Ops/s | 49.9143 Ops/s | |
| test_to_cpu_global_sync | 11.8096ms | 11.6700ms | 85.6899 Ops/s | 75.5154 Ops/s | |
| test_to_cpu_event_sync | 12.9321ms | 12.6511ms | 79.0442 Ops/s | 78.2504 Ops/s | |
| test_to_cpu_default | 12.9124ms | 12.6404ms | 79.1117 Ops/s | 78.3228 Ops/s | |
| test_consolidate[False-None] | 4.2680ms | 4.0314ms | 248.0547 Ops/s | 246.1325 Ops/s | |
| test_consolidate[default-None] | 3.4874ms | 2.0902ms | 478.4278 Ops/s | 466.7460 Ops/s | |
| test_consolidate[reduce-overhead-None] | 3.2942ms | 1.9993ms | 500.1875 Ops/s | 479.1723 Ops/s | |
| test_consolidate_njt[False-None] | 9.1928ms | 8.3443ms | 119.8424 Ops/s | 119.6906 Ops/s | |
| test_to[False-False-None] | 2.1852ms | 2.1335ms | 468.7061 Ops/s | 465.5011 Ops/s | |
| test_to[True-False-None] | 2.1144ms | 1.8334ms | 545.4389 Ops/s | 531.5536 Ops/s | |
| test_to[within-False-None] | 6.0907ms | 5.9704ms | 167.4922 Ops/s | 164.7493 Ops/s | |
| test_to[True-default-None] | 10.8504ms | 10.4951ms | 95.2829 Ops/s | 94.2831 Ops/s | |
| test_to_njt[False-False-None] | 8.7352ms | 8.4070ms | 118.9488 Ops/s | 117.6245 Ops/s | |
| test_to_njt[True-False-None] | 7.0226ms | 6.7877ms | 147.3244 Ops/s | 148.8487 Ops/s | |
| test_to_njt[within-False-None] | 15.4016ms | 14.9790ms | 66.7600 Ops/s | 62.3933 Ops/s | |
| test_creation[device0] | 0.4016ms | 0.1196ms | 8.3625 KOps/s | 8.2628 KOps/s | |
| test_creation_from_tensor | 0.5239ms | 0.1166ms | 8.5766 KOps/s | 8.4812 KOps/s | |
| test_add_one[memmap_tensor0] | 83.2050μs | 6.4903μs | 154.0768 KOps/s | 152.3362 KOps/s | |
| test_contiguous[memmap_tensor0] | 25.9820μs | 0.5928μs | 1.6870 MOps/s | 2.2991 MOps/s | |
| test_stack[memmap_tensor0] | 26.9920μs | 3.9972μs | 250.1767 KOps/s | 246.1269 KOps/s | |
| test_memmaptd_index | 1.1431ms | 0.2655ms | 3.7669 KOps/s | 3.7096 KOps/s | |
| test_memmaptd_index_astensor | 0.5185ms | 0.3654ms | 2.7369 KOps/s | 2.7173 KOps/s | |
| test_memmaptd_index_op | 0.7933ms | 0.6140ms | 1.6286 KOps/s | 1.5280 KOps/s | |
| test_serialize_model | 0.1344s | 0.1328s | 7.5321 Ops/s | 7.3876 Ops/s | |
| test_serialize_model_pickle | 1.3471s | 1.2097s | 0.8267 Ops/s | 0.8221 Ops/s | |
| test_serialize_weights | 0.3168s | 0.1591s | 6.2835 Ops/s | 7.4592 Ops/s | |
| test_serialize_weights_returnearly | 84.6801ms | 63.7906ms | 15.6763 Ops/s | 11.0849 Ops/s | |
| test_serialize_weights_pickle | 1.3693s | 1.2131s | 0.8243 Ops/s | 0.8244 Ops/s | |
| test_reshape_pytree | 0.2027ms | 31.2605μs | 31.9893 KOps/s | 31.9769 KOps/s | |
| test_reshape_td | 87.4460μs | 43.5612μs | 22.9562 KOps/s | 22.9969 KOps/s | |
| test_view_pytree | 0.2234ms | 30.7498μs | 32.5205 KOps/s | 32.6981 KOps/s | |
| test_view_td | 94.5250μs | 51.4983μs | 19.4181 KOps/s | 19.8461 KOps/s | |
| test_unbind_pytree | 0.2516ms | 36.0494μs | 27.7397 KOps/s | 28.4268 KOps/s | |
| test_unbind_td | 0.1453ms | 49.0661μs | 20.3807 KOps/s | 20.6126 KOps/s | |
| test_split_pytree | 0.2442ms | 40.8931μs | 24.4540 KOps/s | 24.4135 KOps/s | |
| test_split_td | 0.1051ms | 61.8975μs | 16.1557 KOps/s | 16.4622 KOps/s | |
| test_add_pytree | 0.2279ms | 42.1837μs | 23.7059 KOps/s | 24.7350 KOps/s | |
| test_add_td | 96.0460μs | 53.6143μs | 18.6517 KOps/s | 18.7667 KOps/s | |
| test_compile_add_one_nested[tensordict-compile] | 0.5120ms | 0.2177ms | 4.5940 KOps/s | 4.4415 KOps/s | |
| test_compile_add_one_nested[tensordict-eager] | 20.2780ms | 0.2038ms | 4.9070 KOps/s | 5.1507 KOps/s | |
| test_compile_add_one_nested[pytree-compile] | 0.3634ms | 0.1299ms | 7.7002 KOps/s | 7.4694 KOps/s | |
| test_compile_add_one_nested[pytree-eager] | 0.4010ms | 0.1724ms | 5.8016 KOps/s | 5.4932 KOps/s | |
| test_compile_copy_nested[tensordict-compile] | 0.4725ms | 15.1051μs | 66.2027 KOps/s | 62.9153 KOps/s | |
| test_compile_copy_nested[tensordict-eager] | 0.1016ms | 51.0739μs | 19.5795 KOps/s | 19.4632 KOps/s | |
| test_compile_copy_nested[pytree-compile] | 0.1156ms | 15.1252μs | 66.1148 KOps/s | 60.6544 KOps/s | |
| test_compile_copy_nested[pytree-eager] | 0.3630ms | 65.0964μs | 15.3618 KOps/s | 15.6225 KOps/s | |
| test_compile_add_one_flat[tensordict-compile] | 0.4295ms | 0.2402ms | 4.1633 KOps/s | 3.9094 KOps/s | |
| test_compile_add_one_flat[tensordict-eager] | 0.4312ms | 0.2778ms | 3.6003 KOps/s | 3.5561 KOps/s | |
| test_compile_add_one_flat[tensorclass-compile] | 0.2577ms | 0.1779ms | 5.6196 KOps/s | 5.2840 KOps/s | |
| test_compile_add_one_flat[tensorclass-eager] | 0.1478ms | 74.3978μs | 13.4413 KOps/s | 13.5213 KOps/s | |
| test_compile_add_one_flat[pytree-compile] | 0.4434ms | 0.1860ms | 5.3750 KOps/s | 5.1560 KOps/s | |
| test_compile_add_one_flat[pytree-eager] | 0.7953ms | 0.5103ms | 1.9595 KOps/s | 1.9552 KOps/s | |
| test_compile_add_self_flat[tensordict-eager] | 0.4176ms | 0.3240ms | 3.0869 KOps/s | 3.0391 KOps/s | |
| test_compile_add_self_flat[tensordict-compile] | 0.3856ms | 0.2421ms | 4.1313 KOps/s | 3.8574 KOps/s | |
| test_compile_add_self_flat[tensorclass-eager] | 0.1657ms | 88.9331μs | 11.2444 KOps/s | 11.2018 KOps/s | |
| test_compile_add_self_flat[tensorclass-compile] | 0.4000ms | 0.1813ms | 5.5159 KOps/s | 5.2308 KOps/s | |
| test_compile_add_self_flat[pytree-eager] | 0.6400ms | 0.4203ms | 2.3790 KOps/s | 2.3541 KOps/s | |
| test_compile_add_self_flat[pytree-compile] | 0.3283ms | 0.1868ms | 5.3532 KOps/s | 5.1868 KOps/s | |
| test_compile_copy_flat[tensordict-compile] | 70.5250μs | 18.9154μs | 52.8670 KOps/s | 52.2807 KOps/s | |
| test_compile_copy_flat[tensordict-eager] | 75.5150μs | 40.1018μs | 24.9365 KOps/s | 24.8842 KOps/s | |
| test_compile_copy_flat[pytree-compile] | 0.2713ms | 20.9984μs | 47.6227 KOps/s | 45.8088 KOps/s | |
| test_compile_copy_flat[pytree-eager] | 0.3331ms | 51.0412μs | 19.5920 KOps/s | 19.7169 KOps/s | |
| test_compile_assign_and_add[tensordict-compile] | 2.1796ms | 0.2337ms | 4.2782 KOps/s | 4.0092 KOps/s | |
| test_compile_assign_and_add[tensordict-eager] | 3.3863ms | 3.1563ms | 316.8276 Ops/s | 309.2371 Ops/s | |
| test_compile_assign_and_add[pytree-compile] | 2.1195ms | 0.1752ms | 5.7082 KOps/s | 5.5495 KOps/s | |
| test_compile_assign_and_add[pytree-eager] | 2.8781ms | 2.7114ms | 368.8136 Ops/s | 353.7132 Ops/s | |
| test_compile_indexing[tensor-tensordict-compile] | 0.2001ms | 0.1354ms | 7.3878 KOps/s | 7.4229 KOps/s | |
| test_compile_indexing[tensor-tensordict-eager] | 0.3153ms | 73.4934μs | 13.6067 KOps/s | 13.8588 KOps/s | |
| test_compile_indexing[tensor-tensorclass-compile] | 0.2478ms | 0.1142ms | 8.7565 KOps/s | 8.5988 KOps/s | |
| test_compile_indexing[tensor-tensorclass-eager] | 0.2655ms | 43.2392μs | 23.1272 KOps/s | 22.1624 KOps/s | |
| test_compile_indexing[tensor-pytree-compile] | 0.1735ms | 0.1148ms | 8.7093 KOps/s | 8.3816 KOps/s | |
| test_compile_indexing[tensor-pytree-eager] | 0.2930ms | 43.0940μs | 23.2051 KOps/s | 21.3950 KOps/s | |
| test_compile_indexing[slice-tensordict-compile] | 0.2124ms | 80.7380μs | 12.3857 KOps/s | 11.5430 KOps/s | |
| test_compile_indexing[slice-tensordict-eager] | 0.2023ms | 27.4686μs | 36.4052 KOps/s | 37.5417 KOps/s | |
| test_compile_indexing[slice-tensorclass-compile] | 0.1698ms | 68.0611μs | 14.6927 KOps/s | 14.2312 KOps/s | |
| test_compile_indexing[slice-tensorclass-eager] | 0.2326ms | 21.2366μs | 47.0885 KOps/s | 47.4482 KOps/s | |
| test_compile_indexing[slice-pytree-compile] | 0.1105ms | 68.5094μs | 14.5965 KOps/s | 14.4869 KOps/s | |
| test_compile_indexing[slice-pytree-eager] | 0.3262ms | 21.2647μs | 47.0264 KOps/s | 47.6816 KOps/s | |
| test_compile_indexing[int-tensordict-compile] | 0.1303ms | 81.6288μs | 12.2506 KOps/s | 11.8530 KOps/s | |
| test_compile_indexing[int-tensordict-eager] | 0.2892ms | 26.9838μs | 37.0593 KOps/s | 37.7645 KOps/s | |
| test_compile_indexing[int-tensorclass-compile] | 0.1154ms | 68.4273μs | 14.6141 KOps/s | 14.3765 KOps/s | |
| test_compile_indexing[int-tensorclass-eager] | 0.2438ms | 21.1081μs | 47.3752 KOps/s | 47.4326 KOps/s | |
| test_compile_indexing[int-pytree-compile] | 0.1160ms | 69.9040μs | 14.3053 KOps/s | 14.5644 KOps/s | |
| test_compile_indexing[int-pytree-eager] | 0.2335ms | 21.3236μs | 46.8965 KOps/s | 48.1483 KOps/s | |
| test_compile_replace[single-eager] | 80.2250μs | 46.2285μs | 21.6317 KOps/s | 20.6489 KOps/s | |
| test_compile_replace[single-compile] | 0.2351ms | 0.1252ms | 7.9892 KOps/s | 7.4134 KOps/s | |
| test_compile_replace[multi-eager] | 0.5990ms | 0.5545ms | 1.8035 KOps/s | 1.7047 KOps/s | |
| test_compile_replace[multi-compile] | 0.2965ms | 0.1339ms | 7.4659 KOps/s | 7.4082 KOps/s | |
| test_compile_tc_getattr_20[eager] | 0.2008ms | 0.1648ms | 6.0671 KOps/s | 6.1692 KOps/s | |
| test_compile_tc_getattr_20[compile] | 0.3730ms | 0.1426ms | 7.0119 KOps/s | 6.8545 KOps/s | |
| test_compile_clone_shallow[20-eager] | 52.7730μs | 18.7133μs | 53.4380 KOps/s | 53.5840 KOps/s | |
| test_compile_clone_shallow[20-compile] | 87.8160μs | 16.4097μs | 60.9397 KOps/s | 52.0278 KOps/s | |
| test_compile_clone_shallow[40-eager] | 61.6240μs | 33.1284μs | 30.1856 KOps/s | 30.9239 KOps/s | |
| test_compile_clone_shallow[40-compile] | 74.0040μs | 18.1328μs | 55.1487 KOps/s | 55.4028 KOps/s | |
| test_compile_clone_shallow[80-eager] | 96.9560μs | 61.2581μs | 16.3244 KOps/s | 16.4621 KOps/s | |
| test_compile_clone_shallow[80-compile] | 77.5250μs | 21.2898μs | 46.9708 KOps/s | 45.7041 KOps/s | |
| test_compile_update_inplace[eager] | 0.2084ms | 59.3408μs | 16.8518 KOps/s | 16.5617 KOps/s | |
| test_compile_update_inplace[compile] | 0.3464ms | 0.1404ms | 7.1233 KOps/s | 6.8831 KOps/s | |
| test_mod_add[eager] | 0.2694ms | 0.2209ms | 4.5259 KOps/s | 4.5339 KOps/s | |
| test_mod_add[compile] | 0.2068ms | 0.1278ms | 7.8229 KOps/s | 7.4355 KOps/s | |
| test_mod_add[compile-overhead] | 0.3966ms | 0.1738ms | 5.7535 KOps/s | 5.6496 KOps/s | |
| test_mod_wrap[eager] | 0.5363ms | 0.4557ms | 2.1944 KOps/s | 2.0948 KOps/s | |
| test_mod_wrap[compile] | 0.4702ms | 0.3830ms | 2.6107 KOps/s | 2.5318 KOps/s | |
| test_mod_wrap[compile-overhead] | 8.5223ms | 4.6518ms | 214.9718 Ops/s | 215.9954 Ops/s | |
| test_mod_wrap_and_backward[eager] | 1.8123ms | 1.6541ms | 604.5619 Ops/s | 592.6719 Ops/s | |
| test_mod_wrap_and_backward[compile] | 2.5389ms | 1.5716ms | 636.2810 Ops/s | 654.2532 Ops/s | |
| test_mod_wrap_and_backward[compile-overhead] | 1.2963ms | 0.9262ms | 1.0797 KOps/s | 969.9680 Ops/s | |
| test_seq_add[eager] | 0.6211ms | 0.5155ms | 1.9398 KOps/s | 2.0121 KOps/s | |
| test_seq_add[compile] | 0.2855ms | 0.1363ms | 7.3358 KOps/s | 6.7121 KOps/s | |
| test_seq_add[compile-overhead] | 0.3132ms | 0.1810ms | 5.5247 KOps/s | 5.4664 KOps/s | |
| test_seq_wrap[eager] | 1.4635ms | 1.3738ms | 727.9186 Ops/s | 721.8508 Ops/s | |
| test_seq_wrap[compile] | 0.4789ms | 0.4065ms | 2.4600 KOps/s | 2.4452 KOps/s | |
| test_seq_wrap[compile-overhead] | 0.3762ms | 0.2958ms | 3.3807 KOps/s | 3.3170 KOps/s | |
| test_func_call_runtime[False-eager] | 0.8559ms | 0.7642ms | 1.3086 KOps/s | 1.2697 KOps/s | |
| test_func_call_runtime[False-compile] | 1.1916ms | 0.9529ms | 1.0494 KOps/s | 1.0406 KOps/s | |
| test_func_call_runtime[False-compile-overhead] | 0.5482ms | 0.4917ms | 2.0338 KOps/s | 2.0143 KOps/s | |
| test_func_call_runtime[True-eager] | 1.1516ms | 1.0204ms | 980.0339 Ops/s | 959.4558 Ops/s | |
| test_func_call_runtime[True-compile] | 1.0725ms | 0.9672ms | 1.0339 KOps/s | 1.0247 KOps/s | |
| test_func_call_runtime[True-compile-overhead] | 0.5687ms | 0.5067ms | 1.9736 KOps/s | 1.9515 KOps/s | |
| test_func_call_cm_runtime[False-eager] | 0.8796ms | 0.7956ms | 1.2569 KOps/s | 1.2750 KOps/s | |
| test_func_call_cm_runtime[False-compile] | 1.0924ms | 0.9453ms | 1.0579 KOps/s | 1.0580 KOps/s | |
| test_func_call_cm_runtime[False-compile-overhead] | 0.6283ms | 0.4728ms | 2.1148 KOps/s | 2.1065 KOps/s | |
| test_func_call_cm_runtime[True-eager] | 1.3016ms | 1.1667ms | 857.0903 Ops/s | 839.4947 Ops/s | |
| test_func_call_cm_runtime[True-compile] | 1.1484ms | 1.0094ms | 990.6633 Ops/s | 996.7503 Ops/s | |
| test_func_call_cm_runtime[True-compile-overhead] | 0.6873ms | 0.5315ms | 1.8815 KOps/s | 1.8602 KOps/s | |
| test_vmap_func_call_cm_runtime[eager] | 2.8457ms | 2.3673ms | 422.4147 Ops/s | 415.2454 Ops/s | |
| test_vmap_func_call_cm_runtime[compile] | 3.1312ms | 2.6048ms | 383.9109 Ops/s | 383.4665 Ops/s | |
| test_vmap_func_call_cm_runtime[compile-overhead] | 3.0315ms | 2.5603ms | 390.5799 Ops/s | 383.4868 Ops/s | |
| test_distributed | 1.6019ms | 0.1571ms | 6.3661 KOps/s | 6.6817 KOps/s | |
| test_tdmodule | 0.3720ms | 0.1869ms | 5.3515 KOps/s | 5.1840 KOps/s | |
| test_tdmodule_dispatch | 0.4079ms | 0.3605ms | 2.7741 KOps/s | 2.7551 KOps/s | |
| test_tdseq | 44.6630μs | 25.7389μs | 38.8516 KOps/s | 38.3656 KOps/s | |
| test_tdseq_dispatch | 0.2514ms | 0.2111ms | 4.7365 KOps/s | 4.7353 KOps/s | |
| test_instantiation_functorch | 2.1011ms | 1.9585ms | 510.6051 Ops/s | 506.2499 Ops/s | |
| test_exec_functorch | 0.2167ms | 0.1751ms | 5.7115 KOps/s | 5.6369 KOps/s | |
| test_exec_functional_call | 0.2158ms | 0.1569ms | 6.3726 KOps/s | 6.3553 KOps/s | |
| test_exec_td_decorator | 0.4835ms | 0.2433ms | 4.1098 KOps/s | 4.0601 KOps/s | |
| test_vmap_mlp_speed_decorator[True-True] | 1.1759ms | 1.0143ms | 985.8754 Ops/s | 967.4665 Ops/s | |
| test_vmap_mlp_speed_decorator[True-False] | 1.2025ms | 1.0100ms | 990.1131 Ops/s | 974.1977 Ops/s | |
| test_vmap_mlp_speed_decorator[False-True] | 0.9364ms | 0.7208ms | 1.3873 KOps/s | 1.3324 KOps/s | |
| test_vmap_mlp_speed_decorator[False-False] | 0.9124ms | 0.7203ms | 1.3883 KOps/s | 1.3612 KOps/s | |
| test_vmap_transformer_speed_decorator[True-True] | 21.7042ms | 20.7863ms | 48.1086 Ops/s | 47.4200 Ops/s | |
| test_vmap_transformer_speed_decorator[True-False] | 21.2133ms | 20.7226ms | 48.2564 Ops/s | 47.4414 Ops/s | |
| test_vmap_transformer_speed_decorator[False-True] | 21.0579ms | 20.3601ms | 49.1157 Ops/s | 48.3616 Ops/s | |
| test_vmap_transformer_speed_decorator[False-False] | 21.0224ms | 20.3860ms | 49.0532 Ops/s | 48.2600 Ops/s | |
| test_to_module_speed[True] | 1.6834ms | 1.5770ms | 634.1172 Ops/s | 629.7749 Ops/s | |
| test_to_module_speed[False] | 1.9512ms | 1.5551ms | 643.0476 Ops/s | 635.0288 Ops/s | |
| test_to_module_plain_tensor_speed[None] | 2.1689ms | 1.6753ms | 596.9216 Ops/s | 589.3783 Ops/s | |
| test_to_module_plain_tensor_speed[False] | 1.9845ms | 1.5301ms | 653.5674 Ops/s | 650.2576 Ops/s | |
| test_to_module_plain_tensor_speed[True] | 3.3560ms | 3.2627ms | 306.4931 Ops/s | 306.7955 Ops/s | |
| test_tc_init | 76.1750μs | 41.8190μs | 23.9126 KOps/s | 24.3410 KOps/s | |
| test_tc_init_tensor_only | 80.6850μs | 8.7043μs | 114.8859 KOps/s | 116.4978 KOps/s | |
| test_tc_init_nested | 0.1237ms | 82.1726μs | 12.1695 KOps/s | 12.3623 KOps/s | |
| test_tc_init_many_fields | 54.9940μs | 14.4058μs | 69.4167 KOps/s | 71.4443 KOps/s | |
| test_tc_first_layer_tensor | 29.6810μs | 1.6935μs | 590.4989 KOps/s | 586.9932 KOps/s | |
| test_tc_first_layer_tensor_only | 1.9832μs | 0.3790μs | 2.6386 MOps/s | 2.5472 MOps/s | |
| test_tc_first_layer_tensor_set | 30.9920μs | 3.6919μs | 270.8628 KOps/s | 267.6720 KOps/s | |
| test_tc_first_layer_tensor_only_set | 25.2110μs | 3.1250μs | 319.9980 KOps/s | 322.5943 KOps/s | |
| test_tc_first_layer_nontensor | 34.9020μs | 5.8541μs | 170.8218 KOps/s | 162.6207 KOps/s | |
| test_tc_second_layer_tensor | 41.5830μs | 4.1536μs | 240.7561 KOps/s | 236.7273 KOps/s | |
| test_tc_second_layer_nontensor | 36.6420μs | 8.2118μs | 121.7765 KOps/s | 114.9566 KOps/s | |
| test_unbind | 0.2789s | 13.5686ms | 73.6998 Ops/s | 71.4464 Ops/s | |
| test_full_like | 4.5482ms | 4.3594ms | 229.3882 Ops/s | 228.0881 Ops/s | |
| test_zeros_like | 4.7957ms | 4.3371ms | 230.5698 Ops/s | 228.3874 Ops/s | |
| test_ones_like | 4.5174ms | 4.3464ms | 230.0755 Ops/s | 227.9054 Ops/s | |
| test_clone | 6.6588ms | 6.3986ms | 156.2848 Ops/s | 155.5287 Ops/s | |
| test_squeeze | 77.7750μs | 13.7690μs | 72.6267 KOps/s | 72.3235 KOps/s | |
| test_unsqueeze | 0.1808ms | 0.1129ms | 8.8586 KOps/s | 8.9626 KOps/s | |
| test_split | 0.2395ms | 0.1816ms | 5.5064 KOps/s | 5.6997 KOps/s | |
| test_permute | 0.2602ms | 0.2094ms | 4.7750 KOps/s | 4.9694 KOps/s | |
| test_stack | 18.3405ms | 18.0028ms | 55.5468 Ops/s | 19.5301 Ops/s | |
| test_cat | 18.2297ms | 17.8909ms | 55.8943 Ops/s | 19.5227 Ops/s | |
| test_sequential_tensordict | 0.9884ms | 0.5630ms | 1.7763 KOps/s | 1.7581 KOps/s | |
| test_sequential_graph_module | 0.1663ms | 0.1221ms | 8.1892 KOps/s | 8.2761 KOps/s | |
| test_nested_tensordict | 1.1944ms | 0.8115ms | 1.2322 KOps/s | 1.2583 KOps/s | |
| test_nested_graph_module | 0.5231ms | 0.1311ms | 7.6305 KOps/s | 7.7521 KOps/s |
Contributor
|
| Name | Max | Mean | Ops | Ops on Repo HEAD
|
Change |
|---|---|---|---|---|---|
| test_collector_preallocated_write[stack_out-cpu-32-2] | 0.2173ms | 0.1646ms | 6.0737 KOps/s | 6.0891 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-32-8] | 0.3403ms | 0.2847ms | 3.5125 KOps/s | 3.5149 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-32-32] | 0.8603ms | 0.7485ms | 1.3360 KOps/s | 1.3257 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-2] | 0.4615ms | 0.3423ms | 2.9218 KOps/s | 2.9788 KOps/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-8] | 1.3486ms | 1.1556ms | 865.3181 Ops/s | 914.3775 Ops/s | |
| test_collector_preallocated_write[stack_out-cpu-4096-32] | 8.1235ms | 7.3532ms | 135.9954 Ops/s | 136.4748 Ops/s | |
| test_collector_preallocated_write[stack_out-cuda-32-2] | 0.3128ms | 0.2475ms | 4.0411 KOps/s | 4.1373 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-32-8] | 0.4347ms | 0.3750ms | 2.6665 KOps/s | 2.8195 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-32-32] | 1.1271ms | 0.8021ms | 1.2467 KOps/s | 1.2737 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-2] | 0.3140ms | 0.2565ms | 3.8993 KOps/s | 4.1342 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-8] | 0.5534ms | 0.3772ms | 2.6510 KOps/s | 2.7737 KOps/s | |
| test_collector_preallocated_write[stack_out-cuda-4096-32] | 0.9919ms | 0.8621ms | 1.1600 KOps/s | 1.1783 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-2] | 0.2701ms | 0.2150ms | 4.6513 KOps/s | 4.7953 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-8] | 0.9550ms | 0.8014ms | 1.2478 KOps/s | 1.2400 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-32-32] | 3.2701ms | 3.1381ms | 318.6629 Ops/s | 311.3215 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-2] | 0.5064ms | 0.3549ms | 2.8179 KOps/s | 2.6271 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-8] | 2.0593ms | 1.6592ms | 602.7103 Ops/s | 618.8134 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cpu-4096-32] | 11.1173ms | 10.5887ms | 94.4407 Ops/s | 98.6764 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-2] | 0.5200ms | 0.3896ms | 2.5668 KOps/s | 2.6001 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-8] | 1.6122ms | 1.4293ms | 699.6674 Ops/s | 700.2586 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-32-32] | 5.8134ms | 5.6771ms | 176.1473 Ops/s | 178.1842 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-2] | 0.4507ms | 0.3856ms | 2.5932 KOps/s | 2.5951 KOps/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-8] | 1.5630ms | 1.4460ms | 691.5802 Ops/s | 688.5701 Ops/s | |
| test_collector_preallocated_write[set_at_loop-cuda-4096-32] | 5.7578ms | 5.6809ms | 176.0289 Ops/s | 176.8251 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-32-2] | 0.6262ms | 0.1684ms | 5.9368 KOps/s | 5.9272 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-32-8] | 0.7669ms | 0.6549ms | 1.5268 KOps/s | 1.5080 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-32-32] | 2.6954ms | 2.5937ms | 385.5474 Ops/s | 384.8996 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-2] | 0.4120ms | 0.3048ms | 3.2803 KOps/s | 3.1353 KOps/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-8] | 1.9943ms | 1.5303ms | 653.4705 Ops/s | 705.6741 Ops/s | |
| test_collector_preallocated_write[copy_at-cpu-4096-32] | 10.0187ms | 9.6080ms | 104.0801 Ops/s | 105.6788 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-32-2] | 0.4613ms | 0.3241ms | 3.0850 KOps/s | 3.1295 KOps/s | |
| test_collector_preallocated_write[copy_at-cuda-32-8] | 1.3041ms | 1.1909ms | 839.6684 Ops/s | 841.3932 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-32-32] | 4.8872ms | 4.6569ms | 214.7338 Ops/s | 215.4950 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-2] | 0.3978ms | 0.3216ms | 3.1095 KOps/s | 3.1300 KOps/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-8] | 1.2869ms | 1.1958ms | 836.2714 Ops/s | 837.2449 Ops/s | |
| test_collector_preallocated_write[copy_at-cuda-4096-32] | 4.9666ms | 4.7021ms | 212.6707 Ops/s | 211.7983 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] | 0.1703ms | 0.1167ms | 8.5656 KOps/s | 8.5959 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] | 0.5692ms | 0.4573ms | 2.1868 KOps/s | 2.2150 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] | 1.8837ms | 1.8209ms | 549.1816 Ops/s | 551.3775 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] | 0.6763ms | 0.2546ms | 3.9273 KOps/s | 3.8399 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] | 1.5522ms | 1.3726ms | 728.5326 Ops/s | 784.3684 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] | 9.5768ms | 8.9384ms | 111.8774 Ops/s | 117.5297 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] | 0.3873ms | 0.2935ms | 3.4070 KOps/s | 3.6683 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] | 1.1672ms | 1.0907ms | 916.8622 Ops/s | 995.8951 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] | 4.3818ms | 4.2416ms | 235.7614 Ops/s | 254.8461 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] | 0.3505ms | 0.2925ms | 3.4187 KOps/s | 3.6960 KOps/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] | 1.1479ms | 1.0922ms | 915.6003 Ops/s | 992.5475 Ops/s | |
| test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] | 4.3804ms | 4.2820ms | 233.5343 Ops/s | 250.6993 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-128-128] | 16.2824ms | 15.7297ms | 63.5740 Ops/s | 63.9903 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-128-256] | 32.1827ms | 31.9343ms | 31.3143 Ops/s | 31.7717 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-4096-128] | 88.4724ms | 88.2971ms | 11.3254 Ops/s | 12.1792 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cpu-4096-256] | 0.2241s | 0.2214s | 4.5161 Ops/s | 5.1253 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-128-128] | 13.5426ms | 13.1078ms | 76.2903 Ops/s | 75.9915 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-128-256] | 26.6612ms | 25.8462ms | 38.6904 Ops/s | 38.8752 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-4096-128] | 13.4776ms | 13.3044ms | 75.1633 Ops/s | 75.4669 Ops/s | |
| test_wide_nested_preallocated_write[stack_out-cuda-4096-256] | 26.4526ms | 26.2270ms | 38.1287 Ops/s | 38.3470 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] | 50.7068ms | 50.4332ms | 19.8282 Ops/s | 19.8087 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] | 0.1020s | 0.1017s | 9.8313 Ops/s | 9.8730 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] | 0.1494s | 0.1488s | 6.7188 Ops/s | 7.1485 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] | 0.3586s | 0.3482s | 2.8723 Ops/s | 3.2161 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] | 78.1487ms | 77.6831ms | 12.8728 Ops/s | 13.9519 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] | 0.1560s | 0.1555s | 6.4300 Ops/s | 6.9775 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] | 78.8600ms | 78.5601ms | 12.7291 Ops/s | 13.8464 Ops/s | |
| test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] | 0.1572s | 0.1565s | 6.3899 Ops/s | 6.7985 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-128-128] | 29.3635ms | 29.2577ms | 34.1790 Ops/s | 33.4482 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-128-256] | 60.1041ms | 59.9224ms | 16.6882 Ops/s | 16.6872 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-4096-128] | 0.1228s | 0.1216s | 8.2260 Ops/s | 8.6515 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cpu-4096-256] | 0.2703s | 0.2662s | 3.7564 Ops/s | 3.8370 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-128-128] | 43.0139ms | 42.6901ms | 23.4246 Ops/s | 23.5098 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-128-256] | 85.9616ms | 85.7445ms | 11.6626 Ops/s | 11.7767 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-4096-128] | 44.0875ms | 43.0789ms | 23.2132 Ops/s | 23.3897 Ops/s | |
| test_wide_nested_preallocated_write[copy_at-cuda-4096-256] | 86.4599ms | 86.2070ms | 11.6000 Ops/s | 11.6954 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] | 20.1972ms | 19.8897ms | 50.2773 Ops/s | 50.3834 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] | 40.6611ms | 40.4589ms | 24.7164 Ops/s | 24.6874 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] | 0.1073s | 0.1070s | 9.3476 Ops/s | 9.5721 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] | 0.2473s | 0.2441s | 4.0961 Ops/s | 4.2013 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] | 39.1090ms | 38.7128ms | 25.8313 Ops/s | 28.1913 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] | 78.6394ms | 78.1054ms | 12.8032 Ops/s | 14.0607 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] | 39.2059ms | 38.9447ms | 25.6775 Ops/s | 28.0743 Ops/s | |
| test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] | 77.7041ms | 77.2747ms | 12.9409 Ops/s | 14.0256 Ops/s | |
| test_plain_set_nested | 54.6510μs | 15.0282μs | 66.5418 KOps/s | 68.0032 KOps/s | |
| test_plain_set_stack_nested | 47.0210μs | 14.6058μs | 68.4658 KOps/s | 69.1399 KOps/s | |
| test_plain_set_nested_inplace | 44.8620μs | 16.1590μs | 61.8850 KOps/s | 62.3969 KOps/s | |
| test_plain_set_stack_nested_inplace | 68.3520μs | 15.9207μs | 62.8113 KOps/s | 63.0302 KOps/s | |
| test_items | 34.2610μs | 5.5222μs | 181.0886 KOps/s | 180.7767 KOps/s | |
| test_items_nested | 0.5050ms | 0.4548ms | 2.1988 KOps/s | 2.1986 KOps/s | |
| test_items_nested_locked | 0.5657ms | 0.4574ms | 2.1864 KOps/s | 2.2017 KOps/s | |
| test_items_nested_leaf | 0.1375ms | 92.5834μs | 10.8011 KOps/s | 10.9330 KOps/s | |
| test_items_stack_nested | 0.5454ms | 0.4538ms | 2.2036 KOps/s | 2.2253 KOps/s | |
| test_items_stack_nested_leaf | 0.1333ms | 92.2269μs | 10.8428 KOps/s | 10.9290 KOps/s | |
| test_items_stack_nested_locked | 0.5239ms | 0.4621ms | 2.1642 KOps/s | 2.1996 KOps/s | |
| test_keys | 25.9300μs | 4.1252μs | 242.4136 KOps/s | 241.2135 KOps/s | |
| test_keys_nested | 0.1700ms | 0.1297ms | 7.7089 KOps/s | 7.8002 KOps/s | |
| test_keys_nested_locked | 2.0508ms | 0.1362ms | 7.3414 KOps/s | 7.3117 KOps/s | |
| test_keys_nested_leaf | 0.1621ms | 0.1186ms | 8.4308 KOps/s | 8.3570 KOps/s | |
| test_keys_stack_nested | 0.1875ms | 0.1291ms | 7.7484 KOps/s | 7.8226 KOps/s | |
| test_keys_stack_nested_leaf | 0.1638ms | 0.1190ms | 8.4050 KOps/s | 8.5205 KOps/s | |
| test_keys_stack_nested_locked | 0.1863ms | 0.1368ms | 7.3109 KOps/s | 7.3341 KOps/s | |
| test_values | 6.1342μs | 1.0017μs | 998.3335 KOps/s | 1.0009 MOps/s | |
| test_values_nested | 83.6530μs | 52.3577μs | 19.0994 KOps/s | 19.6018 KOps/s | |
| test_values_nested_locked | 90.8530μs | 55.8590μs | 17.9022 KOps/s | 18.1056 KOps/s | |
| test_values_nested_leaf | 92.9930μs | 59.5913μs | 16.7810 KOps/s | 16.8157 KOps/s | |
| test_values_stack_nested | 78.9720μs | 52.0629μs | 19.2075 KOps/s | 19.3594 KOps/s | |
| test_values_stack_nested_leaf | 84.2830μs | 59.5301μs | 16.7982 KOps/s | 16.8195 KOps/s | |
| test_values_stack_nested_locked | 83.1120μs | 55.6796μs | 17.9599 KOps/s | 18.1563 KOps/s | |
| test_membership | 4.8368μs | 0.8002μs | 1.2497 MOps/s | 1.2714 MOps/s | |
| test_membership_nested | 33.0800μs | 2.8048μs | 356.5329 KOps/s | 355.8796 KOps/s | |
| test_membership_nested_leaf | 39.1910μs | 2.8121μs | 355.6011 KOps/s | 369.0687 KOps/s | |
| test_membership_stacked_nested | 29.7010μs | 2.7278μs | 366.5994 KOps/s | 371.5203 KOps/s | |
| test_membership_stacked_nested_leaf | 26.9810μs | 2.7140μs | 368.4533 KOps/s | 364.9196 KOps/s | |
| test_membership_nested_last | 41.6510μs | 4.2141μs | 237.2989 KOps/s | 240.4751 KOps/s | |
| test_membership_nested_leaf_last | 34.6110μs | 4.1942μs | 238.4254 KOps/s | 238.8474 KOps/s | |
| test_membership_stacked_nested_last | 35.4810μs | 4.0960μs | 244.1381 KOps/s | 246.3615 KOps/s | |
| test_membership_stacked_nested_leaf_last | 32.3710μs | 4.1466μs | 241.1635 KOps/s | 244.4379 KOps/s | |
| test_nested_getleaf | 48.5420μs | 20.3821μs | 49.0628 KOps/s | 48.7694 KOps/s | |
| test_nested_get | 45.2810μs | 19.5425μs | 51.1705 KOps/s | 51.4043 KOps/s | |
| test_stacked_getleaf | 45.7420μs | 20.4325μs | 48.9416 KOps/s | 49.1886 KOps/s | |
| test_stacked_get | 45.5610μs | 19.3215μs | 51.7559 KOps/s | 51.3816 KOps/s | |
| test_nested_getitemleaf | 43.6710μs | 21.0121μs | 47.5917 KOps/s | 47.3127 KOps/s | |
| test_nested_getitem | 50.0010μs | 19.9383μs | 50.1546 KOps/s | 50.3685 KOps/s | |
| test_stacked_getitemleaf | 50.7820μs | 21.1727μs | 47.2307 KOps/s | 47.5316 KOps/s | |
| test_stacked_getitem | 51.1010μs | 20.0856μs | 49.7870 KOps/s | 50.2363 KOps/s | |
| test_lock_nested | 0.5455ms | 0.4713ms | 2.1219 KOps/s | 2.0758 KOps/s | |
| test_lock_stack_nested | 0.5136ms | 0.4805ms | 2.0813 KOps/s | 2.0741 KOps/s | |
| test_unlock_nested | 0.4424ms | 0.3779ms | 2.6460 KOps/s | 2.6597 KOps/s | |
| test_unlock_stack_nested | 0.4064ms | 0.3785ms | 2.6421 KOps/s | 2.6316 KOps/s | |
| test_flatten_speed | 0.1700ms | 0.1151ms | 8.6850 KOps/s | 8.6855 KOps/s | |
| test_unflatten_speed | 0.6257ms | 0.5466ms | 1.8294 KOps/s | 1.8184 KOps/s | |
| test_common_ops | 99.9382ms | 1.9326ms | 517.4444 Ops/s | 1.3473 KOps/s | |
| test_creation | 0.1471ms | 2.9492μs | 339.0707 KOps/s | 342.1176 KOps/s | |
| test_creation_empty | 25.7900μs | 6.5306μs | 153.1253 KOps/s | 154.6366 KOps/s | |
| test_creation_nested_1 | 32.1410μs | 10.7669μs | 92.8772 KOps/s | 93.0302 KOps/s | |
| test_creation_nested_2 | 39.2410μs | 12.7205μs | 78.6135 KOps/s | 79.0801 KOps/s | |
| test_creation_many_keys[10] | 49.2610μs | 19.1875μs | 52.1172 KOps/s | 52.1290 KOps/s | |
| test_creation_many_keys[50] | 0.1358ms | 82.7155μs | 12.0896 KOps/s | 12.1531 KOps/s | |
| test_creation_many_keys[100] | 0.2159ms | 0.1637ms | 6.1091 KOps/s | 6.1879 KOps/s | |
| test_creation_nested_many_keys[10] | 70.2020μs | 41.4293μs | 24.1375 KOps/s | 23.9877 KOps/s | |
| test_creation_nested_many_keys[50] | 0.2339ms | 0.1687ms | 5.9294 KOps/s | 5.8995 KOps/s | |
| test_clone | 31.5010μs | 12.8779μs | 77.6521 KOps/s | 78.0593 KOps/s | |
| test_getitem[int] | 1.5598ms | 14.4713μs | 69.1023 KOps/s | 69.6065 KOps/s | |
| test_getitem[slice_int] | 0.1382ms | 23.7518μs | 42.1021 KOps/s | 43.3990 KOps/s | |
| test_getitem[range] | 0.1890ms | 61.6947μs | 16.2088 KOps/s | 16.2119 KOps/s | |
| test_getitem[tuple] | 0.1391ms | 23.2177μs | 43.0705 KOps/s | 43.0808 KOps/s | |
| test_getitem[list] | 0.1851ms | 57.3862μs | 17.4258 KOps/s | 17.6714 KOps/s | |
| test_setitem_dim[int] | 47.3620μs | 25.4600μs | 39.2773 KOps/s | 40.8344 KOps/s | |
| test_setitem_dim[slice_int] | 64.7420μs | 42.2630μs | 23.6614 KOps/s | 24.2398 KOps/s | |
| test_setitem_dim[range] | 0.1280ms | 95.9714μs | 10.4198 KOps/s | 10.8733 KOps/s | |
| test_setitem_dim[tuple] | 68.7320μs | 37.7898μs | 26.4621 KOps/s | 25.8582 KOps/s | |
| test_setitem | 49.5410μs | 16.8593μs | 59.3145 KOps/s | 54.6049 KOps/s | |
| test_set | 49.6910μs | 16.0153μs | 62.4404 KOps/s | 59.4197 KOps/s | |
| test_set_shared | 0.6905ms | 0.2094ms | 4.7748 KOps/s | 4.6504 KOps/s | |
| test_update | 0.3622ms | 20.8751μs | 47.9041 KOps/s | 44.2545 KOps/s | |
| test_update_nested | 68.5720μs | 32.6111μs | 30.6645 KOps/s | 28.8767 KOps/s | |
| test_update__nested | 0.5112ms | 32.0760μs | 31.1760 KOps/s | 29.8578 KOps/s | |
| test_set_nested | 53.9320μs | 18.1583μs | 55.0713 KOps/s | 51.8182 KOps/s | |
| test_set_nested_new | 69.4020μs | 22.0813μs | 45.2872 KOps/s | 41.0974 KOps/s | |
| test_select | 86.2320μs | 37.5551μs | 26.6275 KOps/s | 24.4043 KOps/s | |
| test_select_nested | 0.1035ms | 68.5971μs | 14.5779 KOps/s | 14.3120 KOps/s | |
| test_exclude_nested | 0.1440ms | 84.6729μs | 11.8102 KOps/s | 11.5513 KOps/s | |
| test_empty[True] | 0.4213ms | 0.3854ms | 2.5947 KOps/s | 2.5699 KOps/s | |
| test_empty[False] | 8.4127μs | 1.2916μs | 774.2358 KOps/s | 769.6512 KOps/s | |
| test_to | 0.1118ms | 68.8450μs | 14.5254 KOps/s | 14.0176 KOps/s | |
| test_to_nonblocking | 0.1127ms | 67.5219μs | 14.8100 KOps/s | 14.6423 KOps/s | |
| test_unbind_speed | 0.4332ms | 0.3259ms | 3.0681 KOps/s | 3.1602 KOps/s | |
| test_unbind_speed_stack0 | 0.3897ms | 0.3151ms | 3.1737 KOps/s | 3.1841 KOps/s | |
| test_unbind_speed_stack1 | 0.1158s | 0.8150ms | 1.2270 KOps/s | 1.2047 KOps/s | |
| test_split | 0.1162s | 1.2205ms | 819.3155 Ops/s | 718.5855 Ops/s | |
| test_chunk | 1.1321ms | 1.0388ms | 962.6653 Ops/s | 958.8923 Ops/s | |
| test_to_cpu_blocking | 20.0668ms | 19.8979ms | 50.2566 Ops/s | 49.7340 Ops/s | |
| test_to_cpu_global_sync | 11.9378ms | 11.7142ms | 85.3665 Ops/s | 84.9934 Ops/s | |
| test_to_cpu_event_sync | 13.1156ms | 12.7524ms | 78.4164 Ops/s | 77.9123 Ops/s | |
| test_to_cpu_default | 13.1367ms | 12.7986ms | 78.1335 Ops/s | 69.7292 Ops/s | |
| test_consolidate[False-None] | 4.0605ms | 3.9953ms | 250.2922 Ops/s | 249.4491 Ops/s | |
| test_consolidate[default-None] | 3.2986ms | 2.0549ms | 486.6449 Ops/s | 479.8286 Ops/s | |
| test_consolidate[reduce-overhead-None] | 3.5411ms | 1.9928ms | 501.7999 Ops/s | 490.9385 Ops/s | |
| test_consolidate_njt[False-None] | 8.4090ms | 8.3090ms | 120.3516 Ops/s | 121.6453 Ops/s | |
| test_to[False-False-None] | 2.2842ms | 2.1488ms | 465.3792 Ops/s | 464.7096 Ops/s | |
| test_to[True-False-None] | 2.1150ms | 1.8165ms | 550.5020 Ops/s | 533.8617 Ops/s | |
| test_to[within-False-None] | 6.2891ms | 5.9511ms | 168.0369 Ops/s | 164.1708 Ops/s | |
| test_to[True-default-None] | 10.5245ms | 10.3449ms | 96.6659 Ops/s | 94.1637 Ops/s | |
| test_to_njt[False-False-None] | 8.4086ms | 8.2894ms | 120.6365 Ops/s | 117.1680 Ops/s | |
| test_to_njt[True-False-None] | 6.8193ms | 6.5965ms | 151.5946 Ops/s | 145.9143 Ops/s | |
| test_to_njt[within-False-None] | 15.1205ms | 14.8670ms | 67.2629 Ops/s | 65.1751 Ops/s | |
| test_creation[device0] | 0.3469ms | 0.1125ms | 8.8876 KOps/s | 8.6659 KOps/s | |
| test_creation_from_tensor | 0.3575ms | 0.1101ms | 9.0860 KOps/s | 8.7179 KOps/s | |
| test_add_one[memmap_tensor0] | 0.1747ms | 6.4459μs | 155.1380 KOps/s | 150.7909 KOps/s | |
| test_contiguous[memmap_tensor0] | 27.6300μs | 0.6454μs | 1.5494 MOps/s | 2.3542 MOps/s | |
| test_stack[memmap_tensor0] | 36.7610μs | 4.1844μs | 238.9852 KOps/s | 235.3583 KOps/s | |
| test_memmaptd_index | 1.1319ms | 0.2655ms | 3.7669 KOps/s | 3.7567 KOps/s | |
| test_memmaptd_index_astensor | 0.4733ms | 0.3657ms | 2.7346 KOps/s | 2.7034 KOps/s | |
| test_memmaptd_index_op | 0.8402ms | 0.6153ms | 1.6253 KOps/s | 1.6050 KOps/s | |
| test_serialize_model | 0.1347s | 0.1331s | 7.5117 Ops/s | 7.4295 Ops/s | |
| test_serialize_model_pickle | 1.3474s | 1.2094s | 0.8269 Ops/s | 0.8234 Ops/s | |
| test_serialize_weights | 0.1350s | 0.1330s | 7.5173 Ops/s | 3.4118 Ops/s | |
| test_serialize_weights_returnearly | 0.3143s | 77.9225ms | 12.8333 Ops/s | 16.1052 Ops/s | |
| test_serialize_weights_pickle | 1.3473s | 1.2131s | 0.8243 Ops/s | 0.8230 Ops/s | |
| test_reshape_pytree | 0.1994ms | 30.9193μs | 32.3422 KOps/s | 32.0598 KOps/s | |
| test_reshape_td | 87.0030μs | 42.8383μs | 23.3436 KOps/s | 22.6084 KOps/s | |
| test_view_pytree | 0.2127ms | 30.6892μs | 32.5847 KOps/s | 32.4759 KOps/s | |
| test_view_td | 93.6730μs | 51.7464μs | 19.3250 KOps/s | 18.5355 KOps/s | |
| test_unbind_pytree | 0.2209ms | 34.7471μs | 28.7794 KOps/s | 27.9648 KOps/s | |
| test_unbind_td | 82.5320μs | 48.6663μs | 20.5481 KOps/s | 19.8966 KOps/s | |
| test_split_pytree | 0.2662ms | 40.1030μs | 24.9358 KOps/s | 23.9012 KOps/s | |
| test_split_td | 0.1082ms | 63.0592μs | 15.8581 KOps/s | 15.8639 KOps/s | |
| test_add_pytree | 0.2264ms | 39.8659μs | 25.0841 KOps/s | 24.4563 KOps/s | |
| test_add_td | 97.2430μs | 55.1971μs | 18.1169 KOps/s | 18.0737 KOps/s | |
| test_compile_add_one_nested[tensordict-compile] | 0.3310ms | 0.2318ms | 4.3137 KOps/s | 4.4715 KOps/s | |
| test_compile_add_one_nested[tensordict-eager] | 0.2847ms | 0.1960ms | 5.1023 KOps/s | 4.1648 KOps/s | |
| test_compile_add_one_nested[pytree-compile] | 0.2724ms | 0.1354ms | 7.3845 KOps/s | 7.3868 KOps/s | |
| test_compile_add_one_nested[pytree-eager] | 0.4285ms | 0.1714ms | 5.8341 KOps/s | 5.8011 KOps/s | |
| test_compile_copy_nested[tensordict-compile] | 72.0620μs | 16.2153μs | 61.6700 KOps/s | 62.4362 KOps/s | |
| test_compile_copy_nested[tensordict-eager] | 0.1387ms | 50.3314μs | 19.8683 KOps/s | 19.6588 KOps/s | |
| test_compile_copy_nested[pytree-compile] | 0.1311ms | 15.3286μs | 65.2375 KOps/s | 63.5898 KOps/s | |
| test_compile_copy_nested[pytree-eager] | 0.3618ms | 63.8417μs | 15.6638 KOps/s | 15.6125 KOps/s | |
| test_compile_add_one_flat[tensordict-compile] | 0.2857ms | 0.2406ms | 4.1556 KOps/s | 3.7323 KOps/s | |
| test_compile_add_one_flat[tensordict-eager] | 0.4065ms | 0.2766ms | 3.6153 KOps/s | 3.6145 KOps/s | |
| test_compile_add_one_flat[tensorclass-compile] | 0.2228ms | 0.1803ms | 5.5448 KOps/s | 5.2666 KOps/s | |
| test_compile_add_one_flat[tensorclass-eager] | 0.1832ms | 72.5321μs | 13.7870 KOps/s | 13.6306 KOps/s | |
| test_compile_add_one_flat[pytree-compile] | 0.4653ms | 0.1874ms | 5.3352 KOps/s | 5.1543 KOps/s | |
| test_compile_add_one_flat[pytree-eager] | 0.7929ms | 0.5045ms | 1.9820 KOps/s | 1.9308 KOps/s | |
| test_compile_add_self_flat[tensordict-eager] | 0.4556ms | 0.3225ms | 3.1009 KOps/s | 3.0848 KOps/s | |
| test_compile_add_self_flat[tensordict-compile] | 0.5966ms | 0.2459ms | 4.0660 KOps/s | 3.8654 KOps/s | |
| test_compile_add_self_flat[tensorclass-eager] | 0.1489ms | 88.3931μs | 11.3131 KOps/s | 11.2125 KOps/s | |
| test_compile_add_self_flat[tensorclass-compile] | 0.3553ms | 0.1837ms | 5.4449 KOps/s | 5.1667 KOps/s | |
| test_compile_add_self_flat[pytree-eager] | 0.6416ms | 0.4202ms | 2.3799 KOps/s | 2.3490 KOps/s | |
| test_compile_add_self_flat[pytree-compile] | 0.4479ms | 0.1890ms | 5.2908 KOps/s | 5.1116 KOps/s | |
| test_compile_copy_flat[tensordict-compile] | 64.1020μs | 19.2629μs | 51.9131 KOps/s | 48.2803 KOps/s | |
| test_compile_copy_flat[tensordict-eager] | 69.1920μs | 39.8492μs | 25.0946 KOps/s | 25.0476 KOps/s | |
| test_compile_copy_flat[pytree-compile] | 0.2215ms | 20.9571μs | 47.7165 KOps/s | 46.8703 KOps/s | |
| test_compile_copy_flat[pytree-eager] | 0.3411ms | 50.9778μs | 19.6164 KOps/s | 19.5883 KOps/s | |
| test_compile_assign_and_add[tensordict-compile] | 2.1250ms | 0.2362ms | 4.2346 KOps/s | 3.9220 KOps/s | |
| test_compile_assign_and_add[tensordict-eager] | 3.4406ms | 3.2026ms | 312.2497 Ops/s | 303.4525 Ops/s | |
| test_compile_assign_and_add[pytree-compile] | 2.0791ms | 0.1754ms | 5.7005 KOps/s | 5.5773 KOps/s | |
| test_compile_assign_and_add[pytree-eager] | 2.7958ms | 2.6611ms | 375.7820 Ops/s | 367.3775 Ops/s | |
| test_compile_indexing[tensor-tensordict-compile] | 0.2147ms | 0.1362ms | 7.3447 KOps/s | 7.5122 KOps/s | |
| test_compile_indexing[tensor-tensordict-eager] | 0.3007ms | 75.8184μs | 13.1894 KOps/s | 13.2359 KOps/s | |
| test_compile_indexing[tensor-tensorclass-compile] | 0.2185ms | 0.1149ms | 8.7032 KOps/s | 8.4774 KOps/s | |
| test_compile_indexing[tensor-tensorclass-eager] | 0.2552ms | 43.3147μs | 23.0868 KOps/s | 22.8027 KOps/s | |
| test_compile_indexing[tensor-pytree-compile] | 0.1640ms | 0.1159ms | 8.6280 KOps/s | 8.4158 KOps/s | |
| test_compile_indexing[tensor-pytree-eager] | 0.2645ms | 43.4629μs | 23.0081 KOps/s | 22.6912 KOps/s | |
| test_compile_indexing[slice-tensordict-compile] | 0.2283ms | 81.4729μs | 12.2740 KOps/s | 11.7783 KOps/s | |
| test_compile_indexing[slice-tensordict-eager] | 0.2029ms | 26.8254μs | 37.2782 KOps/s | 37.1404 KOps/s | |
| test_compile_indexing[slice-tensorclass-compile] | 0.1505ms | 66.9207μs | 14.9431 KOps/s | 14.0010 KOps/s | |
| test_compile_indexing[slice-tensorclass-eager] | 0.2426ms | 21.0455μs | 47.5160 KOps/s | 46.8484 KOps/s | |
| test_compile_indexing[slice-pytree-compile] | 0.1177ms | 69.7870μs | 14.3293 KOps/s | 13.9592 KOps/s | |
| test_compile_indexing[slice-pytree-eager] | 0.2476ms | 21.1874μs | 47.1979 KOps/s | 46.3367 KOps/s | |
| test_compile_indexing[int-tensordict-compile] | 0.1339ms | 82.2052μs | 12.1647 KOps/s | 11.5668 KOps/s | |
| test_compile_indexing[int-tensordict-eager] | 0.2791ms | 26.6527μs | 37.5197 KOps/s | 37.4222 KOps/s | |
| test_compile_indexing[int-tensorclass-compile] | 0.1081ms | 68.7223μs | 14.5513 KOps/s | 13.9922 KOps/s | |
| test_compile_indexing[int-tensorclass-eager] | 0.2437ms | 20.9263μs | 47.7868 KOps/s | 47.0251 KOps/s | |
| test_compile_indexing[int-pytree-compile] | 0.1076ms | 68.4830μs | 14.6022 KOps/s | 14.1692 KOps/s | |
| test_compile_indexing[int-pytree-eager] | 0.2931ms | 21.0271μs | 47.5578 KOps/s | 47.4310 KOps/s | |
| test_compile_replace[single-eager] | 95.2130μs | 46.5871μs | 21.4652 KOps/s | 21.2345 KOps/s | |
| test_compile_replace[single-compile] | 0.2716ms | 0.1268ms | 7.8841 KOps/s | 7.3567 KOps/s | |
| test_compile_replace[multi-eager] | 0.6543ms | 0.5498ms | 1.8187 KOps/s | 1.7472 KOps/s | |
| test_compile_replace[multi-compile] | 0.2958ms | 0.1347ms | 7.4242 KOps/s | 7.0134 KOps/s | |
| test_compile_tc_getattr_20[eager] | 0.1979ms | 0.1606ms | 6.2264 KOps/s | 6.1468 KOps/s | |
| test_compile_tc_getattr_20[compile] | 0.3942ms | 0.1427ms | 7.0070 KOps/s | 6.5527 KOps/s | |
| test_compile_clone_shallow[20-eager] | 62.2110μs | 18.5192μs | 53.9981 KOps/s | 53.9167 KOps/s | |
| test_compile_clone_shallow[20-compile] | 92.8530μs | 16.4914μs | 60.6378 KOps/s | 54.6728 KOps/s | |
| test_compile_clone_shallow[40-eager] | 62.6520μs | 32.7609μs | 30.5242 KOps/s | 31.0060 KOps/s | |
| test_compile_clone_shallow[40-compile] | 74.8420μs | 17.9614μs | 55.6751 KOps/s | 54.1039 KOps/s | |
| test_compile_clone_shallow[80-eager] | 89.6920μs | 60.6340μs | 16.4924 KOps/s | 16.5004 KOps/s | |
| test_compile_clone_shallow[80-compile] | 91.0130μs | 21.3231μs | 46.8975 KOps/s | 45.2146 KOps/s | |
| test_compile_update_inplace[eager] | 0.1004ms | 58.8136μs | 17.0029 KOps/s | 16.6588 KOps/s | |
| test_compile_update_inplace[compile] | 0.2829ms | 0.1397ms | 7.1558 KOps/s | 6.8786 KOps/s | |
| test_mod_add[eager] | 0.2713ms | 0.2183ms | 4.5805 KOps/s | 4.5314 KOps/s | |
| test_mod_add[compile] | 0.2123ms | 0.1275ms | 7.8449 KOps/s | 7.3087 KOps/s | |
| test_mod_add[compile-overhead] | 0.3866ms | 0.1736ms | 5.7598 KOps/s | 5.5804 KOps/s | |
| test_mod_wrap[eager] | 0.5334ms | 0.4555ms | 2.1953 KOps/s | 2.1547 KOps/s | |
| test_mod_wrap[compile] | 0.4660ms | 0.3830ms | 2.6108 KOps/s | 2.5410 KOps/s | |
| test_mod_wrap[compile-overhead] | 9.0447ms | 4.7667ms | 209.7870 Ops/s | 213.6846 Ops/s | |
| test_mod_wrap_and_backward[eager] | 1.8579ms | 1.6642ms | 600.8831 Ops/s | 577.8644 Ops/s | |
| test_mod_wrap_and_backward[compile] | 2.5595ms | 1.5624ms | 640.0368 Ops/s | 666.8516 Ops/s | |
| test_mod_wrap_and_backward[compile-overhead] | 1.3035ms | 0.9168ms | 1.0907 KOps/s | 1.0693 KOps/s | |
| test_seq_add[eager] | 0.5780ms | 0.4946ms | 2.0217 KOps/s | 2.0223 KOps/s | |
| test_seq_add[compile] | 0.2476ms | 0.1370ms | 7.2983 KOps/s | 7.0755 KOps/s | |
| test_seq_add[compile-overhead] | 0.2988ms | 0.1778ms | 5.6253 KOps/s | 5.4600 KOps/s | |
| test_seq_wrap[eager] | 1.4464ms | 1.3776ms | 725.8843 Ops/s | 720.9275 Ops/s | |
| test_seq_wrap[compile] | 0.5115ms | 0.3960ms | 2.5252 KOps/s | 2.4362 KOps/s | |
| test_seq_wrap[compile-overhead] | 0.3872ms | 0.2970ms | 3.3673 KOps/s | 3.2888 KOps/s | |
| test_func_call_runtime[False-eager] | 0.9033ms | 0.8248ms | 1.2125 KOps/s | 1.2316 KOps/s | |
| test_func_call_runtime[False-compile] | 1.0528ms | 0.9430ms | 1.0605 KOps/s | 1.0570 KOps/s | |
| test_func_call_runtime[False-compile-overhead] | 0.6314ms | 0.4922ms | 2.0316 KOps/s | 2.0218 KOps/s | |
| test_func_call_runtime[True-eager] | 1.1314ms | 1.0304ms | 970.4681 Ops/s | 972.1703 Ops/s | |
| test_func_call_runtime[True-compile] | 1.0568ms | 0.9534ms | 1.0489 KOps/s | 1.0393 KOps/s | |
| test_func_call_runtime[True-compile-overhead] | 0.5505ms | 0.5059ms | 1.9766 KOps/s | 1.9519 KOps/s | |
| test_func_call_cm_runtime[False-eager] | 0.9931ms | 0.7767ms | 1.2875 KOps/s | 1.2904 KOps/s | |
| test_func_call_cm_runtime[False-compile] | 1.0776ms | 0.9188ms | 1.0884 KOps/s | 1.0716 KOps/s | |
| test_func_call_cm_runtime[False-compile-overhead] | 0.6171ms | 0.4724ms | 2.1168 KOps/s | 2.0939 KOps/s | |
| test_func_call_cm_runtime[True-eager] | 1.2782ms | 1.1724ms | 852.9268 Ops/s | 841.6784 Ops/s | |
| test_func_call_cm_runtime[True-compile] | 1.0435ms | 0.9833ms | 1.0170 KOps/s | 1.0125 KOps/s | |
| test_func_call_cm_runtime[True-compile-overhead] | 0.6773ms | 0.5279ms | 1.8943 KOps/s | 1.8656 KOps/s | |
| test_vmap_func_call_cm_runtime[eager] | 2.8731ms | 2.3661ms | 422.6303 Ops/s | 419.5252 Ops/s | |
| test_vmap_func_call_cm_runtime[compile] | 3.0255ms | 2.5456ms | 392.8310 Ops/s | 389.9190 Ops/s | |
| test_vmap_func_call_cm_runtime[compile-overhead] | 3.0192ms | 2.5405ms | 393.6166 Ops/s | 390.7505 Ops/s | |
| test_distributed | 2.6350ms | 0.1689ms | 5.9218 KOps/s | 6.6744 KOps/s | |
| test_tdmodule | 0.2871ms | 0.1868ms | 5.3535 KOps/s | 5.2579 KOps/s | |
| test_tdmodule_dispatch | 0.3883ms | 0.3627ms | 2.7571 KOps/s | 2.7348 KOps/s | |
| test_tdseq | 43.8410μs | 25.7198μs | 38.8805 KOps/s | 38.6624 KOps/s | |
| test_tdseq_dispatch | 0.2380ms | 0.2099ms | 4.7637 KOps/s | 4.7625 KOps/s | |
| test_instantiation_functorch | 2.0564ms | 1.9720ms | 507.1086 Ops/s | 503.3109 Ops/s | |
| test_exec_functorch | 0.2268ms | 0.1751ms | 5.7109 KOps/s | 5.6701 KOps/s | |
| test_exec_functional_call | 0.2442ms | 0.1552ms | 6.4417 KOps/s | 6.4374 KOps/s | |
| test_exec_td_decorator | 0.4776ms | 0.2411ms | 4.1470 KOps/s | 4.1074 KOps/s | |
| test_vmap_mlp_speed_decorator[True-True] | 1.1673ms | 1.0163ms | 983.9526 Ops/s | 983.5999 Ops/s | |
| test_vmap_mlp_speed_decorator[True-False] | 1.2121ms | 1.0084ms | 991.6639 Ops/s | 982.6267 Ops/s | |
| test_vmap_mlp_speed_decorator[False-True] | 0.9538ms | 0.7205ms | 1.3879 KOps/s | 1.3632 KOps/s | |
| test_vmap_mlp_speed_decorator[False-False] | 0.9146ms | 0.7216ms | 1.3859 KOps/s | 1.3819 KOps/s | |
| test_vmap_transformer_speed_decorator[True-True] | 21.9025ms | 20.8276ms | 48.0133 Ops/s | 47.8968 Ops/s | |
| test_vmap_transformer_speed_decorator[True-False] | 20.8946ms | 20.7422ms | 48.2108 Ops/s | 47.8481 Ops/s | |
| test_vmap_transformer_speed_decorator[False-True] | 20.9816ms | 20.3512ms | 49.1371 Ops/s | 48.8855 Ops/s | |
| test_vmap_transformer_speed_decorator[False-False] | 21.1970ms | 20.3813ms | 49.0647 Ops/s | 48.8366 Ops/s | |
| test_to_module_speed[True] | 2.1371ms | 1.5696ms | 637.1083 Ops/s | 630.9796 Ops/s | |
| test_to_module_speed[False] | 2.1146ms | 1.5386ms | 649.9543 Ops/s | 639.7726 Ops/s | |
| test_to_module_plain_tensor_speed[None] | 2.1539ms | 1.6589ms | 602.8232 Ops/s | 595.1573 Ops/s | |
| test_to_module_plain_tensor_speed[False] | 1.9552ms | 1.5046ms | 664.6200 Ops/s | 656.0320 Ops/s | |
| test_to_module_plain_tensor_speed[True] | 3.3316ms | 3.2295ms | 309.6490 Ops/s | 304.2319 Ops/s | |
| test_tc_init | 95.2230μs | 41.8520μs | 23.8937 KOps/s | 24.0224 KOps/s | |
| test_tc_init_tensor_only | 31.8810μs | 8.5298μs | 117.2358 KOps/s | 117.9112 KOps/s | |
| test_tc_init_nested | 0.1319ms | 83.8342μs | 11.9283 KOps/s | 12.3236 KOps/s | |
| test_tc_init_many_fields | 38.3110μs | 14.1393μs | 70.7246 KOps/s | 70.9629 KOps/s | |
| test_tc_first_layer_tensor | 26.5810μs | 1.7419μs | 574.1020 KOps/s | 584.8148 KOps/s | |
| test_tc_first_layer_tensor_only | 3.1851μs | 0.3867μs | 2.5862 MOps/s | 2.6156 MOps/s | |
| test_tc_first_layer_tensor_set | 1.8544ms | 3.6657μs | 272.8016 KOps/s | 272.0461 KOps/s | |
| test_tc_first_layer_tensor_only_set | 19.3610μs | 3.1033μs | 322.2360 KOps/s | 324.5809 KOps/s | |
| test_tc_first_layer_nontensor | 24.0610μs | 5.7582μs | 173.6652 KOps/s | 165.7327 KOps/s | |
| test_tc_second_layer_tensor | 30.5710μs | 4.1342μs | 241.8866 KOps/s | 240.7939 KOps/s | |
| test_tc_second_layer_nontensor | 46.3720μs | 8.1810μs | 122.2347 KOps/s | 117.8033 KOps/s | |
| test_unbind | 0.2684s | 13.4556ms | 74.3185 Ops/s | 69.3976 Ops/s | |
| test_full_like | 4.5701ms | 4.3654ms | 229.0717 Ops/s | 232.8613 Ops/s | |
| test_zeros_like | 4.5314ms | 4.3391ms | 230.4649 Ops/s | 228.5197 Ops/s | |
| test_ones_like | 4.4621ms | 4.3379ms | 230.5252 Ops/s | 229.2833 Ops/s | |
| test_clone | 6.6650ms | 6.3855ms | 156.6055 Ops/s | 155.6255 Ops/s | |
| test_squeeze | 63.5420μs | 13.8000μs | 72.4637 KOps/s | 68.1887 KOps/s | |
| test_unsqueeze | 0.1624ms | 0.1067ms | 9.3723 KOps/s | 8.8306 KOps/s | |
| test_split | 0.2340ms | 0.1740ms | 5.7477 KOps/s | 5.6862 KOps/s | |
| test_permute | 0.2798ms | 0.2061ms | 4.8526 KOps/s | 4.9116 KOps/s | |
| test_stack | 18.2243ms | 18.0583ms | 55.3761 Ops/s | 28.6853 Ops/s | |
| test_cat | 28.4927ms | 27.0843ms | 36.9217 Ops/s | 28.7617 Ops/s | |
| test_sequential_tensordict | 0.6940ms | 0.5555ms | 1.8002 KOps/s | 1.7968 KOps/s | |
| test_sequential_graph_module | 0.4881ms | 0.1202ms | 8.3176 KOps/s | 8.4967 KOps/s | |
| test_nested_tensordict | 1.0211ms | 0.8029ms | 1.2455 KOps/s | 1.2638 KOps/s | |
| test_nested_graph_module | 0.5363ms | 0.1306ms | 7.6561 KOps/s | 7.7957 KOps/s |
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
Refreshes the README ahead of the 0.13 release with a more direct pitch and a faster path to understanding what TensorDict does.
Highlights:
dict[str, Tensor]that behaves like a tensor;inplace=Trueshape ops, compile fixes, robust memmap keys, andto_module(..., preserve_module_state=...).Checks
git diff --checktorchis not installed in the active Python environment