Skip to content

[Versioning] Prepare TensorDict 0.14.0 - #1764

Merged
vmoens merged 1 commit into
mainfrom
release/0.14.0
Aug 14, 2026
Merged

[Versioning] Prepare TensorDict 0.14.0#1764
vmoens merged 1 commit into
mainfrom
release/0.14.0

Conversation

@vmoens

@vmoens vmoens commented Aug 14, 2026 •
edited
Loading

Copy link
Copy Markdown
Collaborator

Summary

  • Bump the four release version sources to 0.14.0.
  • Make to_module() preserve module parameters and buffers by default, while retaining preserve_module_state=False for the previous replacement behavior.
  • Make copy_at_() use the fast tensor-only path by default, while retaining fast=False for general update semantics.
  • Keep the new memmap pickle compatibility warning for 0.14 and move the allow_pickle=False default transition to 0.15.
  • Update the associated documentation, type stubs, and regression tests.

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Aug 14, 2026
@github-actions github-actions Bot added documentation Improvements or additions to documentation CI Test tensorclass nn versioning labels Aug 14, 2026
@github-actions

github-actions Bot commented Aug 14, 2026 •
edited
Loading

Copy link
Copy Markdown
Contributor

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

Total Benchmarks: 356. Improved: $\large\color{#35bf28}24$. Worsened: $\large\color{#d91a1a}24$.

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_collector_preallocated_write[stack_out-cpu-32-2] 0.2377ms 0.1681ms 5.9497 KOps/s 5.9936 KOps/s $\color{#d91a1a}-0.73\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.3679ms 0.2950ms 3.3902 KOps/s 3.3942 KOps/s $\color{#d91a1a}-0.12\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.8432ms 0.7761ms 1.2885 KOps/s 1.2692 KOps/s $\color{#35bf28}+1.52\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.4441ms 0.3353ms 2.9828 KOps/s 3.0288 KOps/s $\color{#d91a1a}-1.52\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.3484ms 1.1899ms 840.3891 Ops/s 917.7752 Ops/s $\textbf{\color{#d91a1a}-8.43\%}$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.6974ms 7.4051ms 135.0426 Ops/s 135.7472 Ops/s $\color{#d91a1a}-0.52\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.2860ms 0.2512ms 3.9811 KOps/s 3.9909 KOps/s $\color{#d91a1a}-0.25\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.4440ms 0.3724ms 2.6851 KOps/s 2.6775 KOps/s $\color{#35bf28}+0.28\%$
test_collector_preallocated_write[stack_out-cuda-32-32] 0.9241ms 0.8297ms 1.2053 KOps/s 1.2044 KOps/s $\color{#35bf28}+0.07\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.3203ms 0.2505ms 3.9923 KOps/s 3.9278 KOps/s $\color{#35bf28}+1.64\%$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.4642ms 0.3748ms 2.6677 KOps/s 2.6572 KOps/s $\color{#35bf28}+0.40\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 0.9571ms 0.8901ms 1.1234 KOps/s 1.1247 KOps/s $\color{#d91a1a}-0.11\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2619ms 0.2114ms 4.7314 KOps/s 4.6165 KOps/s $\color{#35bf28}+2.49\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 0.9068ms 0.8181ms 1.2224 KOps/s 1.2083 KOps/s $\color{#35bf28}+1.16\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.4439ms 3.2885ms 304.0934 Ops/s 304.4133 Ops/s $\color{#d91a1a}-0.11\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.5171ms 0.3700ms 2.7027 KOps/s 2.6571 KOps/s $\color{#35bf28}+1.72\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 1.8720ms 1.7202ms 581.3221 Ops/s 587.5126 Ops/s $\color{#d91a1a}-1.05\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 10.4263ms 10.2335ms 97.7185 Ops/s 98.1217 Ops/s $\color{#d91a1a}-0.41\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.4173ms 0.3863ms 2.5890 KOps/s 2.5785 KOps/s $\color{#35bf28}+0.41\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.5675ms 1.4413ms 693.8391 Ops/s 674.2364 Ops/s $\color{#35bf28}+2.91\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 5.7758ms 5.6514ms 176.9480 Ops/s 173.2343 Ops/s $\color{#35bf28}+2.14\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.4897ms 0.3891ms 2.5700 KOps/s 2.5678 KOps/s $\color{#35bf28}+0.09\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.5402ms 1.4507ms 689.3191 Ops/s 694.7347 Ops/s $\color{#d91a1a}-0.78\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 5.9601ms 5.7012ms 175.4009 Ops/s 174.7000 Ops/s $\color{#35bf28}+0.40\%$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.2408ms 0.1738ms 5.7536 KOps/s 5.7836 KOps/s $\color{#d91a1a}-0.52\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.7408ms 0.6709ms 1.4905 KOps/s 1.4774 KOps/s $\color{#35bf28}+0.89\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 2.7754ms 2.6562ms 376.4800 Ops/s 375.6030 Ops/s $\color{#35bf28}+0.23\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.4067ms 0.3290ms 3.0397 KOps/s 2.9716 KOps/s $\color{#35bf28}+2.29\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.5611ms 1.4409ms 693.9982 Ops/s 664.2803 Ops/s $\color{#35bf28}+4.47\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 9.6812ms 9.5205ms 105.0361 Ops/s 105.4375 Ops/s $\color{#d91a1a}-0.38\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.3751ms 0.3216ms 3.1097 KOps/s 3.0809 KOps/s $\color{#35bf28}+0.93\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.2738ms 1.1924ms 838.6519 Ops/s 830.8854 Ops/s $\color{#35bf28}+0.93\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.8145ms 4.6737ms 213.9610 Ops/s 212.3413 Ops/s $\color{#35bf28}+0.76\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.3811ms 0.3219ms 3.1061 KOps/s 3.0917 KOps/s $\color{#35bf28}+0.46\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.3234ms 1.1995ms 833.6969 Ops/s 827.8627 Ops/s $\color{#35bf28}+0.70\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 5.7135ms 4.7271ms 211.5474 Ops/s 211.3981 Ops/s $\color{#35bf28}+0.07\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.1942ms 0.1203ms 8.3140 KOps/s 8.3079 KOps/s $\color{#35bf28}+0.07\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.5971ms 0.4745ms 2.1076 KOps/s 2.1126 KOps/s $\color{#d91a1a}-0.24\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 2.0017ms 1.8833ms 530.9860 Ops/s 530.4951 Ops/s $\color{#35bf28}+0.09\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.3440ms 0.2499ms 4.0019 KOps/s 3.8471 KOps/s $\color{#35bf28}+4.02\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.3712ms 1.2005ms 833.0152 Ops/s 845.4769 Ops/s $\color{#d91a1a}-1.47\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 8.6634ms 8.5242ms 117.3128 Ops/s 117.3840 Ops/s $\color{#d91a1a}-0.06\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.3337ms 0.2762ms 3.6201 KOps/s 3.6369 KOps/s $\color{#d91a1a}-0.46\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.1175ms 1.0271ms 973.5934 Ops/s 980.7900 Ops/s $\color{#d91a1a}-0.73\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 4.0955ms 4.0044ms 249.7225 Ops/s 251.1559 Ops/s $\color{#d91a1a}-0.57\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.4050ms 0.2761ms 3.6216 KOps/s 3.6453 KOps/s $\color{#d91a1a}-0.65\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.1191ms 1.0269ms 973.8136 Ops/s 978.8544 Ops/s $\color{#d91a1a}-0.51\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 5.9928ms 4.0395ms 247.5554 Ops/s 251.0031 Ops/s $\color{#d91a1a}-1.37\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.6588ms 16.4061ms 60.9530 Ops/s 61.2574 Ops/s $\color{#d91a1a}-0.50\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 33.1784ms 32.9978ms 30.3051 Ops/s 30.3699 Ops/s $\color{#d91a1a}-0.21\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 86.5295ms 84.2254ms 11.8729 Ops/s 10.9154 Ops/s $\textbf{\color{#35bf28}+8.77\%}$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.1983s 0.1974s 5.0658 Ops/s 5.0713 Ops/s $\color{#d91a1a}-0.11\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 13.9755ms 13.7695ms 72.6244 Ops/s 72.5455 Ops/s $\color{#35bf28}+0.11\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 26.9547ms 26.7308ms 37.4100 Ops/s 37.1624 Ops/s $\color{#35bf28}+0.67\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 14.0675ms 13.8630ms 72.1345 Ops/s 71.5038 Ops/s $\color{#35bf28}+0.88\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 27.4280ms 27.1849ms 36.7851 Ops/s 36.6329 Ops/s $\color{#35bf28}+0.42\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 52.4589ms 52.0020ms 19.2300 Ops/s 19.4821 Ops/s $\color{#d91a1a}-1.29\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1053s 0.1043s 9.5849 Ops/s 9.7060 Ops/s $\color{#d91a1a}-1.25\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1476s 0.1446s 6.9141 Ops/s 6.9492 Ops/s $\color{#d91a1a}-0.50\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3204s 0.3174s 3.1504 Ops/s 3.1602 Ops/s $\color{#d91a1a}-0.31\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 73.5274ms 72.4272ms 13.8070 Ops/s 13.2862 Ops/s $\color{#35bf28}+3.92\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1461s 0.1452s 6.8866 Ops/s 6.6952 Ops/s $\color{#35bf28}+2.86\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 74.2909ms 73.2015ms 13.6609 Ops/s 13.4064 Ops/s $\color{#35bf28}+1.90\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1468s 0.1461s 6.8450 Ops/s 6.7149 Ops/s $\color{#35bf28}+1.94\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 30.6862ms 30.3202ms 32.9813 Ops/s 32.9023 Ops/s $\color{#35bf28}+0.24\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 62.0043ms 61.6943ms 16.2089 Ops/s 16.3621 Ops/s $\color{#d91a1a}-0.94\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1205s 0.1200s 8.3306 Ops/s 8.3692 Ops/s $\color{#d91a1a}-0.46\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2649s 0.2632s 3.7992 Ops/s 3.6022 Ops/s $\textbf{\color{#35bf28}+5.47\%}$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 43.2766ms 43.1270ms 23.1873 Ops/s 23.1773 Ops/s $\color{#35bf28}+0.04\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 85.9642ms 85.7673ms 11.6595 Ops/s 11.5653 Ops/s $\color{#35bf28}+0.81\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 44.2744ms 43.2353ms 23.1293 Ops/s 22.9971 Ops/s $\color{#35bf28}+0.57\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 86.6000ms 86.3786ms 11.5769 Ops/s 11.5252 Ops/s $\color{#35bf28}+0.45\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 21.0192ms 20.7084ms 48.2897 Ops/s 48.3676 Ops/s $\color{#d91a1a}-0.16\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 42.3964ms 42.0169ms 23.7999 Ops/s 23.8222 Ops/s $\color{#d91a1a}-0.09\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1100s 0.1093s 9.1527 Ops/s 9.2877 Ops/s $\color{#d91a1a}-1.45\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2399s 0.2392s 4.1812 Ops/s 3.9531 Ops/s $\textbf{\color{#35bf28}+5.77\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 37.1368ms 36.8384ms 27.1456 Ops/s 27.5235 Ops/s $\color{#d91a1a}-1.37\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 73.7285ms 73.3962ms 13.6247 Ops/s 13.7618 Ops/s $\color{#d91a1a}-1.00\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 37.3964ms 37.0216ms 27.0113 Ops/s 27.5379 Ops/s $\color{#d91a1a}-1.91\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 74.1613ms 73.5077ms 13.6040 Ops/s 13.8147 Ops/s $\color{#d91a1a}-1.52\%$
test_plain_set_nested 41.1210μs 15.2516μs 65.5670 KOps/s 65.2651 KOps/s $\color{#35bf28}+0.46\%$
test_plain_set_stack_nested 40.0410μs 15.1421μs 66.0411 KOps/s 65.9527 KOps/s $\color{#35bf28}+0.13\%$
test_plain_set_nested_inplace 44.6310μs 16.6863μs 59.9294 KOps/s 59.8345 KOps/s $\color{#35bf28}+0.16\%$
test_plain_set_stack_nested_inplace 64.0610μs 16.5105μs 60.5676 KOps/s 60.1814 KOps/s $\color{#35bf28}+0.64\%$
test_items 30.3710μs 6.0748μs 164.6148 KOps/s 159.4819 KOps/s $\color{#35bf28}+3.22\%$
test_items_nested 0.5467ms 0.4612ms 2.1681 KOps/s 2.0977 KOps/s $\color{#35bf28}+3.35\%$
test_items_nested_locked 0.5633ms 0.4635ms 2.1575 KOps/s 2.0937 KOps/s $\color{#35bf28}+3.05\%$
test_items_nested_leaf 0.1306ms 98.3211μs 10.1708 KOps/s 10.3004 KOps/s $\color{#d91a1a}-1.26\%$
test_items_stack_nested 0.5265ms 0.4649ms 2.1511 KOps/s 2.1221 KOps/s $\color{#35bf28}+1.36\%$
test_items_stack_nested_leaf 0.1803ms 96.8651μs 10.3236 KOps/s 10.0826 KOps/s $\color{#35bf28}+2.39\%$
test_items_stack_nested_locked 0.5407ms 0.4660ms 2.1459 KOps/s 2.0845 KOps/s $\color{#35bf28}+2.94\%$
test_keys 27.1410μs 4.2051μs 237.8091 KOps/s 236.4964 KOps/s $\color{#35bf28}+0.56\%$
test_keys_nested 0.1674ms 0.1315ms 7.6035 KOps/s 7.4523 KOps/s $\color{#35bf28}+2.03\%$
test_keys_nested_locked 0.8100ms 0.1409ms 7.0955 KOps/s 7.0469 KOps/s $\color{#35bf28}+0.69\%$
test_keys_nested_leaf 0.1572ms 0.1209ms 8.2704 KOps/s 8.1849 KOps/s $\color{#35bf28}+1.04\%$
test_keys_stack_nested 0.1868ms 0.1318ms 7.5892 KOps/s 7.7746 KOps/s $\color{#d91a1a}-2.38\%$
test_keys_stack_nested_leaf 0.1640ms 0.1206ms 8.2936 KOps/s 8.3430 KOps/s $\color{#d91a1a}-0.59\%$
test_keys_stack_nested_locked 0.1922ms 0.1385ms 7.2214 KOps/s 7.1181 KOps/s $\color{#35bf28}+1.45\%$
test_values 6.4082μs 1.0233μs 977.2436 KOps/s 985.6988 KOps/s $\color{#d91a1a}-0.86\%$
test_values_nested 0.1018ms 52.8959μs 18.9051 KOps/s 18.8887 KOps/s $\color{#35bf28}+0.09\%$
test_values_nested_locked 99.4820μs 55.4191μs 18.0443 KOps/s 17.9472 KOps/s $\color{#35bf28}+0.54\%$
test_values_nested_leaf 96.7520μs 60.5162μs 16.5245 KOps/s 16.4344 KOps/s $\color{#35bf28}+0.55\%$
test_values_stack_nested 90.0720μs 52.1020μs 19.1931 KOps/s 19.1552 KOps/s $\color{#35bf28}+0.20\%$
test_values_stack_nested_leaf 99.4130μs 59.9878μs 16.6700 KOps/s 16.4846 KOps/s $\color{#35bf28}+1.12\%$
test_values_stack_nested_locked 90.1220μs 55.2949μs 18.0849 KOps/s 17.9947 KOps/s $\color{#35bf28}+0.50\%$
test_membership 5.3168μs 0.8377μs 1.1937 MOps/s 1.2062 MOps/s $\color{#d91a1a}-1.04\%$
test_membership_nested 67.9720μs 2.9239μs 342.0084 KOps/s 331.0230 KOps/s $\color{#35bf28}+3.32\%$
test_membership_nested_leaf 34.2510μs 2.9685μs 336.8687 KOps/s 344.8723 KOps/s $\color{#d91a1a}-2.32\%$
test_membership_stacked_nested 25.6610μs 2.8941μs 345.5298 KOps/s 342.0145 KOps/s $\color{#35bf28}+1.03\%$
test_membership_stacked_nested_leaf 27.7910μs 2.8752μs 347.8015 KOps/s 340.9511 KOps/s $\color{#35bf28}+2.01\%$
test_membership_nested_last 33.4310μs 4.4731μs 223.5606 KOps/s 223.0350 KOps/s $\color{#35bf28}+0.24\%$
test_membership_nested_leaf_last 35.2910μs 4.4466μs 224.8932 KOps/s 232.2777 KOps/s $\color{#d91a1a}-3.18\%$
test_membership_stacked_nested_last 28.0810μs 4.3195μs 231.5065 KOps/s 232.4966 KOps/s $\color{#d91a1a}-0.43\%$
test_membership_stacked_nested_leaf_last 30.4210μs 4.2823μs 233.5193 KOps/s 232.0145 KOps/s $\color{#35bf28}+0.65\%$
test_nested_getleaf 50.9410μs 21.6519μs 46.1854 KOps/s 45.4909 KOps/s $\color{#35bf28}+1.53\%$
test_nested_get 49.7810μs 20.7216μs 48.2588 KOps/s 48.4893 KOps/s $\color{#d91a1a}-0.48\%$
test_stacked_getleaf 54.6410μs 21.5314μs 46.4437 KOps/s 46.0326 KOps/s $\color{#35bf28}+0.89\%$
test_stacked_get 46.7910μs 20.6544μs 48.4159 KOps/s 48.4458 KOps/s $\color{#d91a1a}-0.06\%$
test_nested_getitemleaf 52.7610μs 22.3386μs 44.7656 KOps/s 44.5892 KOps/s $\color{#35bf28}+0.40\%$
test_nested_getitem 43.7410μs 21.1653μs 47.2470 KOps/s 47.0885 KOps/s $\color{#35bf28}+0.34\%$
test_stacked_getitemleaf 51.3520μs 21.9308μs 45.5981 KOps/s 45.2628 KOps/s $\color{#35bf28}+0.74\%$
test_stacked_getitem 46.8410μs 20.7058μs 48.2956 KOps/s 47.9410 KOps/s $\color{#35bf28}+0.74\%$
test_lock_nested 8.2239ms 0.5110ms 1.9571 KOps/s 1.9733 KOps/s $\color{#d91a1a}-0.82\%$
test_lock_stack_nested 0.5962ms 0.5103ms 1.9594 KOps/s 1.9683 KOps/s $\color{#d91a1a}-0.45\%$
test_unlock_nested 0.4975ms 0.3982ms 2.5111 KOps/s 2.5130 KOps/s $\color{#d91a1a}-0.08\%$
test_unlock_stack_nested 0.4722ms 0.4005ms 2.4970 KOps/s 2.5061 KOps/s $\color{#d91a1a}-0.36\%$
test_flatten_speed 0.1657ms 0.1230ms 8.1314 KOps/s 8.0351 KOps/s $\color{#35bf28}+1.20\%$
test_unflatten_speed 0.6235ms 0.5574ms 1.7939 KOps/s 1.7854 KOps/s $\color{#35bf28}+0.48\%$
test_common_ops 0.9115ms 0.7019ms 1.4248 KOps/s 1.4351 KOps/s $\color{#d91a1a}-0.72\%$
test_creation 0.1311ms 3.1828μs 314.1897 KOps/s 312.9331 KOps/s $\color{#35bf28}+0.40\%$
test_creation_empty 37.9910μs 6.9748μs 143.3739 KOps/s 141.8528 KOps/s $\color{#35bf28}+1.07\%$
test_creation_nested_1 43.5810μs 11.5092μs 86.8870 KOps/s 86.9223 KOps/s $\color{#d91a1a}-0.04\%$
test_creation_nested_2 40.2510μs 13.5565μs 73.7655 KOps/s 73.0167 KOps/s $\color{#35bf28}+1.03\%$
test_creation_many_keys[10] 50.5110μs 21.0061μs 47.6052 KOps/s 47.4873 KOps/s $\color{#35bf28}+0.25\%$
test_creation_many_keys[50] 0.1232ms 89.1504μs 11.2170 KOps/s 11.0227 KOps/s $\color{#35bf28}+1.76\%$
test_creation_many_keys[100] 0.2465ms 0.1754ms 5.7008 KOps/s 5.6737 KOps/s $\color{#35bf28}+0.48\%$
test_creation_nested_many_keys[10] 75.9020μs 45.0588μs 22.1932 KOps/s 22.0737 KOps/s $\color{#35bf28}+0.54\%$
test_creation_nested_many_keys[50] 0.2411ms 0.1824ms 5.4813 KOps/s 5.4646 KOps/s $\color{#35bf28}+0.31\%$
test_clone 44.0410μs 12.8662μs 77.7228 KOps/s 77.5512 KOps/s $\color{#35bf28}+0.22\%$
test_getitem[int] 1.5423ms 15.7669μs 63.4241 KOps/s 63.8492 KOps/s $\color{#d91a1a}-0.67\%$
test_getitem[slice_int] 0.1605ms 24.6125μs 40.6298 KOps/s 40.1212 KOps/s $\color{#35bf28}+1.27\%$
test_getitem[range] 0.2033ms 63.8950μs 15.6507 KOps/s 16.0338 KOps/s $\color{#d91a1a}-2.39\%$
test_getitem[tuple] 0.1519ms 24.6857μs 40.5093 KOps/s 40.6367 KOps/s $\color{#d91a1a}-0.31\%$
test_getitem[list] 0.1906ms 58.8555μs 16.9908 KOps/s 17.6419 KOps/s $\color{#d91a1a}-3.69\%$
test_setitem_dim[int] 45.5810μs 26.1940μs 38.1766 KOps/s 39.6386 KOps/s $\color{#d91a1a}-3.69\%$
test_setitem_dim[slice_int] 65.0320μs 42.7893μs 23.3703 KOps/s 23.9443 KOps/s $\color{#d91a1a}-2.40\%$
test_setitem_dim[range] 0.1359ms 96.2109μs 10.3938 KOps/s 10.8293 KOps/s $\color{#d91a1a}-4.02\%$
test_setitem_dim[tuple] 68.5320μs 39.4778μs 25.3307 KOps/s 25.5991 KOps/s $\color{#d91a1a}-1.05\%$
test_setitem 57.8820μs 17.5230μs 57.0677 KOps/s 57.4566 KOps/s $\color{#d91a1a}-0.68\%$
test_set 69.0520μs 16.6398μs 60.0968 KOps/s 60.1541 KOps/s $\color{#d91a1a}-0.10\%$
test_set_shared 0.6131ms 0.2099ms 4.7631 KOps/s 4.8605 KOps/s $\color{#d91a1a}-2.00\%$
test_update 74.3120μs 20.4693μs 48.8535 KOps/s 47.1604 KOps/s $\color{#35bf28}+3.59\%$
test_update_nested 69.6020μs 32.9567μs 30.3428 KOps/s 30.7997 KOps/s $\color{#d91a1a}-1.48\%$
test_update__nested 0.5471ms 32.6699μs 30.6092 KOps/s 29.9791 KOps/s $\color{#35bf28}+2.10\%$
test_set_nested 53.0720μs 18.1087μs 55.2222 KOps/s 54.5229 KOps/s $\color{#35bf28}+1.28\%$
test_set_nested_new 90.5420μs 22.8763μs 43.7134 KOps/s 43.5856 KOps/s $\color{#35bf28}+0.29\%$
test_select 82.4420μs 39.1581μs 25.5375 KOps/s 25.0015 KOps/s $\color{#35bf28}+2.14\%$
test_select_nested 0.1222ms 73.1574μs 13.6692 KOps/s 13.2747 KOps/s $\color{#35bf28}+2.97\%$
test_exclude_nested 0.1548ms 91.8526μs 10.8870 KOps/s 10.8598 KOps/s $\color{#35bf28}+0.25\%$
test_empty[True] 0.4718ms 0.3992ms 2.5050 KOps/s 2.4611 KOps/s $\color{#35bf28}+1.79\%$
test_empty[False] 8.4153μs 1.3781μs 725.6604 KOps/s 704.9295 KOps/s $\color{#35bf28}+2.94\%$
test_to 0.1079ms 72.1178μs 13.8662 KOps/s 13.9753 KOps/s $\color{#d91a1a}-0.78\%$
test_to_nonblocking 0.1107ms 71.1329μs 14.0582 KOps/s 13.7830 KOps/s $\color{#35bf28}+2.00\%$
test_unbind_speed 0.3766ms 0.3391ms 2.9489 KOps/s 2.9573 KOps/s $\color{#d91a1a}-0.29\%$
test_unbind_speed_stack0 0.4890ms 0.3291ms 3.0386 KOps/s 3.0690 KOps/s $\color{#d91a1a}-0.99\%$
test_unbind_speed_stack1 0.1257s 0.8687ms 1.1511 KOps/s 1.1389 KOps/s $\color{#35bf28}+1.07\%$
test_split 0.1253s 1.4647ms 682.7190 Ops/s 781.1503 Ops/s $\textbf{\color{#d91a1a}-12.60\%}$
test_chunk 1.3637ms 1.2311ms 812.2508 Ops/s 811.1223 Ops/s $\color{#35bf28}+0.14\%$
test_to_cpu_blocking 20.1371ms 19.9575ms 50.1066 Ops/s 44.1906 Ops/s $\textbf{\color{#35bf28}+13.39\%}$
test_to_cpu_global_sync 0.1372s 13.3382ms 74.9725 Ops/s 84.4717 Ops/s $\textbf{\color{#d91a1a}-11.25\%}$
test_to_cpu_event_sync 13.2653ms 12.8862ms 77.6024 Ops/s 77.6023 Ops/s $+0.00\%$
test_to_cpu_default 13.2228ms 12.8685ms 77.7094 Ops/s 77.7333 Ops/s $\color{#d91a1a}-0.03\%$
test_consolidate[False-None] 4.3724ms 4.1988ms 238.1651 Ops/s 237.0478 Ops/s $\color{#35bf28}+0.47\%$
test_consolidate[default-None] 3.3635ms 2.3056ms 433.7180 Ops/s 430.8110 Ops/s $\color{#35bf28}+0.67\%$
test_consolidate[reduce-overhead-None] 3.1820ms 2.1890ms 456.8336 Ops/s 445.8314 Ops/s $\color{#35bf28}+2.47\%$
test_consolidate_njt[False-None] 5.6833ms 5.1761ms 193.1957 Ops/s 191.6732 Ops/s $\color{#35bf28}+0.79\%$
test_to[False-False-None] 2.3441ms 2.2428ms 445.8705 Ops/s 444.8855 Ops/s $\color{#35bf28}+0.22\%$
test_to[True-False-None] 2.2447ms 1.9922ms 501.9625 Ops/s 507.3881 Ops/s $\color{#d91a1a}-1.07\%$
test_to[within-False-None] 6.6305ms 6.2994ms 158.7451 Ops/s 157.9006 Ops/s $\color{#35bf28}+0.53\%$
test_to[True-default-None] 10.7306ms 10.3705ms 96.4271 Ops/s 94.2468 Ops/s $\color{#35bf28}+2.31\%$
test_to_njt[False-False-None] 7.2311ms 7.0967ms 140.9102 Ops/s 141.5805 Ops/s $\color{#d91a1a}-0.47\%$
test_to_njt[True-False-None] 4.1301ms 3.9644ms 252.2473 Ops/s 256.3710 Ops/s $\color{#d91a1a}-1.61\%$
test_to_njt[within-False-None] 9.3702ms 9.1965ms 108.7371 Ops/s 110.2344 Ops/s $\color{#d91a1a}-1.36\%$
test_creation[device0] 0.5225ms 0.1248ms 8.0145 KOps/s 8.4175 KOps/s $\color{#d91a1a}-4.79\%$
test_creation_from_tensor 0.6406ms 0.1191ms 8.3989 KOps/s 8.4650 KOps/s $\color{#d91a1a}-0.78\%$
test_add_one[memmap_tensor0] 0.3817ms 6.8754μs 145.4470 KOps/s 159.8208 KOps/s $\textbf{\color{#d91a1a}-8.99\%}$
test_contiguous[memmap_tensor0] 14.4500μs 0.7306μs 1.3687 MOps/s 1.9387 MOps/s $\textbf{\color{#d91a1a}-29.40\%}$
test_stack[memmap_tensor0] 86.3720μs 4.2379μs 235.9680 KOps/s 235.7786 KOps/s $\color{#35bf28}+0.08\%$
test_memmaptd_index 1.4601ms 0.2882ms 3.4693 KOps/s 3.5225 KOps/s $\color{#d91a1a}-1.51\%$
test_memmaptd_index_astensor 0.5453ms 0.3917ms 2.5529 KOps/s 2.6086 KOps/s $\color{#d91a1a}-2.14\%$
test_memmaptd_index_op 0.7920ms 0.6362ms 1.5720 KOps/s 1.5900 KOps/s $\color{#d91a1a}-1.13\%$
test_serialize_model 0.1420s 0.1409s 7.0984 Ops/s 7.1193 Ops/s $\color{#d91a1a}-0.29\%$
test_serialize_model_archive 1.3525s 1.2449s 0.8033 Ops/s 0.8044 Ops/s $\color{#d91a1a}-0.14\%$
test_load_archive 18.3999ms 18.0936ms 55.2681 Ops/s 55.0124 Ops/s $\color{#35bf28}+0.46\%$
test_serialize_model_zarr 2.5126s 2.5095s 0.3985 Ops/s 0.4000 Ops/s $\color{#d91a1a}-0.38\%$
test_load_zarr 0.3972s 0.3784s 2.6427 Ops/s 2.4954 Ops/s $\textbf{\color{#35bf28}+5.90\%}$
test_serialize_model_pickle 1.6106s 1.2537s 0.7976 Ops/s 0.6959 Ops/s $\textbf{\color{#35bf28}+14.62\%}$
test_serialize_weights 0.3411s 0.1690s 5.9170 Ops/s 7.1380 Ops/s $\textbf{\color{#d91a1a}-17.11\%}$
test_serialize_weights_returnearly 67.7425ms 53.0298ms 18.8573 Ops/s 18.6165 Ops/s $\color{#35bf28}+1.29\%$
test_serialize_weights_pickle 1.3870s 1.2345s 0.8101 Ops/s 0.8263 Ops/s $\color{#d91a1a}-1.97\%$
test_consolidate[few_large-0] 60.2618ms 7.1843ms 139.1934 Ops/s 125.4002 Ops/s $\textbf{\color{#35bf28}+11.00\%}$
test_consolidate[few_large-8] 53.4932ms 7.4375ms 134.4531 Ops/s 131.2638 Ops/s $\color{#35bf28}+2.43\%$
test_consolidate[many_small-0] 0.1237s 0.1202s 8.3197 Ops/s 6.7749 Ops/s $\textbf{\color{#35bf28}+22.80\%}$
test_consolidate[many_small-8] 0.2444s 48.8752ms 20.4603 Ops/s 24.6281 Ops/s $\textbf{\color{#d91a1a}-16.92\%}$
test_consolidate_to_file[few_large-0] 0.1384s 22.1372ms 45.1729 Ops/s 46.3825 Ops/s $\color{#d91a1a}-2.61\%$
test_consolidate_to_file[few_large-8] 0.1416s 26.5074ms 37.7253 Ops/s 31.4635 Ops/s $\textbf{\color{#35bf28}+19.90\%}$
test_consolidate_to_file[many_small-0] 0.3832s 0.2416s 4.1385 Ops/s 2.1546 Ops/s $\textbf{\color{#35bf28}+92.07\%}$
test_consolidate_to_file[many_small-8] 0.2637s 0.2369s 4.2206 Ops/s 2.0586 Ops/s $\textbf{\color{#35bf28}+105.02\%}$
test_reshape_pytree 13.3530ms 33.0094μs 30.2944 KOps/s 31.1283 KOps/s $\color{#d91a1a}-2.68\%$
test_reshape_td 95.5330μs 45.7528μs 21.8566 KOps/s 21.8898 KOps/s $\color{#d91a1a}-0.15\%$
test_view_pytree 0.2349ms 31.7469μs 31.4991 KOps/s 31.6580 KOps/s $\color{#d91a1a}-0.50\%$
test_view_td 90.7520μs 53.8407μs 18.5733 KOps/s 18.7829 KOps/s $\color{#d91a1a}-1.12\%$
test_unbind_pytree 0.2509ms 36.0711μs 27.7231 KOps/s 27.7552 KOps/s $\color{#d91a1a}-0.12\%$
test_unbind_td 0.1060ms 50.1068μs 19.9574 KOps/s 20.2091 KOps/s $\color{#d91a1a}-1.25\%$
test_split_pytree 0.2444ms 41.2411μs 24.2477 KOps/s 24.5470 KOps/s $\color{#d91a1a}-1.22\%$
test_split_td 0.1125ms 72.7362μs 13.7483 KOps/s 13.6244 KOps/s $\color{#35bf28}+0.91\%$
test_add_pytree 0.2268ms 39.3277μs 25.4274 KOps/s 25.3787 KOps/s $\color{#35bf28}+0.19\%$
test_add_td 92.9430μs 52.8498μs 18.9215 KOps/s 19.0449 KOps/s $\color{#d91a1a}-0.65\%$
test_compile_add_one_nested[tensordict-compile] 0.5645ms 0.2779ms 3.5988 KOps/s 3.5522 KOps/s $\color{#35bf28}+1.31\%$
test_compile_add_one_nested[tensordict-eager] 0.3035ms 0.2056ms 4.8642 KOps/s 4.9041 KOps/s $\color{#d91a1a}-0.81\%$
test_compile_add_one_nested[pytree-compile] 0.4411ms 0.1585ms 6.3074 KOps/s 6.1538 KOps/s $\color{#35bf28}+2.50\%$
test_compile_add_one_nested[pytree-eager] 0.5281ms 0.1861ms 5.3748 KOps/s 5.5980 KOps/s $\color{#d91a1a}-3.99\%$
test_compile_copy_nested[tensordict-compile] 89.3220μs 23.0371μs 43.4083 KOps/s 46.5298 KOps/s $\textbf{\color{#d91a1a}-6.71\%}$
test_compile_copy_nested[tensordict-eager] 82.9720μs 53.9088μs 18.5499 KOps/s 19.0173 KOps/s $\color{#d91a1a}-2.46\%$
test_compile_copy_nested[pytree-compile] 0.1863ms 19.4416μs 51.4360 KOps/s 53.4140 KOps/s $\color{#d91a1a}-3.70\%$
test_compile_copy_nested[pytree-eager] 0.3840ms 66.7001μs 14.9925 KOps/s 15.0382 KOps/s $\color{#d91a1a}-0.30\%$
test_compile_add_one_flat[tensordict-compile] 0.3623ms 0.3097ms 3.2284 KOps/s 3.0738 KOps/s $\textbf{\color{#35bf28}+5.03\%}$
test_compile_add_one_flat[tensordict-eager] 0.4196ms 0.2943ms 3.3979 KOps/s 3.4553 KOps/s $\color{#d91a1a}-1.66\%$
test_compile_add_one_flat[tensorclass-compile] 0.3654ms 0.2137ms 4.6784 KOps/s 4.3789 KOps/s $\textbf{\color{#35bf28}+6.84\%}$
test_compile_add_one_flat[tensorclass-eager] 0.1246ms 77.5976μs 12.8870 KOps/s 12.9714 KOps/s $\color{#d91a1a}-0.65\%$
test_compile_add_one_flat[pytree-compile] 0.5870ms 0.2439ms 4.0999 KOps/s 3.9804 KOps/s $\color{#35bf28}+3.00\%$
test_compile_add_one_flat[pytree-eager] 0.8219ms 0.5369ms 1.8627 KOps/s 1.8539 KOps/s $\color{#35bf28}+0.47\%$
test_compile_add_self_flat[tensordict-eager] 0.6879ms 0.3422ms 2.9222 KOps/s 2.9426 KOps/s $\color{#d91a1a}-0.69\%$
test_compile_add_self_flat[tensordict-compile] 0.7391ms 0.3127ms 3.1983 KOps/s 3.0869 KOps/s $\color{#35bf28}+3.61\%$
test_compile_add_self_flat[tensorclass-eager] 0.1413ms 94.6653μs 10.5635 KOps/s 10.7590 KOps/s $\color{#d91a1a}-1.82\%$
test_compile_add_self_flat[tensorclass-compile] 0.3481ms 0.2120ms 4.7170 KOps/s 4.4863 KOps/s $\textbf{\color{#35bf28}+5.14\%}$
test_compile_add_self_flat[pytree-eager] 0.7328ms 0.4509ms 2.2178 KOps/s 2.2382 KOps/s $\color{#d91a1a}-0.91\%$
test_compile_add_self_flat[pytree-compile] 0.3379ms 0.2436ms 4.1057 KOps/s 3.9928 KOps/s $\color{#35bf28}+2.83\%$
test_compile_copy_flat[tensordict-compile] 91.2520μs 27.2984μs 36.6322 KOps/s 36.2644 KOps/s $\color{#35bf28}+1.01\%$
test_compile_copy_flat[tensordict-eager] 78.4020μs 40.8570μs 24.4756 KOps/s 24.7389 KOps/s $\color{#d91a1a}-1.06\%$
test_compile_copy_flat[pytree-compile] 0.1653ms 25.3555μs 39.4392 KOps/s 39.8430 KOps/s $\color{#d91a1a}-1.01\%$
test_compile_copy_flat[pytree-eager] 0.3460ms 51.9535μs 19.2480 KOps/s 19.4088 KOps/s $\color{#d91a1a}-0.83\%$
test_compile_assign_and_add[tensordict-compile] 2.1235ms 0.2589ms 3.8632 KOps/s 3.6322 KOps/s $\textbf{\color{#35bf28}+6.36\%}$
test_compile_assign_and_add[tensordict-eager] 3.5909ms 3.3415ms 299.2689 Ops/s 297.0846 Ops/s $\color{#35bf28}+0.74\%$
test_compile_assign_and_add[pytree-compile] 1.9855ms 0.1940ms 5.1555 KOps/s 4.9665 KOps/s $\color{#35bf28}+3.80\%$
test_compile_assign_and_add[pytree-eager] 3.0836ms 2.8117ms 355.6552 Ops/s 350.9099 Ops/s $\color{#35bf28}+1.35\%$
test_compile_indexing[tensor-tensordict-compile] 0.3118ms 0.1577ms 6.3401 KOps/s 6.1036 KOps/s $\color{#35bf28}+3.88\%$
test_compile_indexing[tensor-tensordict-eager] 0.3765ms 74.4018μs 13.4405 KOps/s 12.6829 KOps/s $\textbf{\color{#35bf28}+5.97\%}$
test_compile_indexing[tensor-tensorclass-compile] 0.2752ms 0.1335ms 7.4919 KOps/s 7.1619 KOps/s $\color{#35bf28}+4.61\%$
test_compile_indexing[tensor-tensorclass-eager] 0.3193ms 44.5405μs 22.4515 KOps/s 21.0120 KOps/s $\textbf{\color{#35bf28}+6.85\%}$
test_compile_indexing[tensor-pytree-compile] 0.1853ms 0.1345ms 7.4371 KOps/s 7.1916 KOps/s $\color{#35bf28}+3.41\%$
test_compile_indexing[tensor-pytree-eager] 0.2866ms 46.4389μs 21.5337 KOps/s 21.1326 KOps/s $\color{#35bf28}+1.90\%$
test_compile_indexing[slice-tensordict-compile] 0.2429ms 93.5609μs 10.6882 KOps/s 10.4389 KOps/s $\color{#35bf28}+2.39\%$
test_compile_indexing[slice-tensordict-eager] 0.2266ms 28.5757μs 34.9948 KOps/s 36.0478 KOps/s $\color{#d91a1a}-2.92\%$
test_compile_indexing[slice-tensorclass-compile] 0.1174ms 72.8450μs 13.7278 KOps/s 13.6790 KOps/s $\color{#35bf28}+0.36\%$
test_compile_indexing[slice-tensorclass-eager] 0.2554ms 22.4695μs 44.5049 KOps/s 45.0683 KOps/s $\color{#d91a1a}-1.25\%$
test_compile_indexing[slice-pytree-compile] 0.1217ms 74.7125μs 13.3846 KOps/s 13.5407 KOps/s $\color{#d91a1a}-1.15\%$
test_compile_indexing[slice-pytree-eager] 0.2485ms 22.3943μs 44.6543 KOps/s 45.4716 KOps/s $\color{#d91a1a}-1.80\%$
test_compile_indexing[int-tensordict-compile] 0.1401ms 96.7522μs 10.3357 KOps/s 10.1933 KOps/s $\color{#35bf28}+1.40\%$
test_compile_indexing[int-tensordict-eager] 0.2789ms 28.1629μs 35.5077 KOps/s 36.0798 KOps/s $\color{#d91a1a}-1.59\%$
test_compile_indexing[int-tensorclass-compile] 0.1196ms 74.4793μs 13.4266 KOps/s 13.6172 KOps/s $\color{#d91a1a}-1.40\%$
test_compile_indexing[int-tensorclass-eager] 0.2677ms 22.2331μs 44.9780 KOps/s 45.4343 KOps/s $\color{#d91a1a}-1.00\%$
test_compile_indexing[int-pytree-compile] 0.1134ms 73.2042μs 13.6604 KOps/s 13.4094 KOps/s $\color{#35bf28}+1.87\%$
test_compile_indexing[int-pytree-eager] 0.2673ms 21.7308μs 46.0176 KOps/s 45.5221 KOps/s $\color{#35bf28}+1.09\%$
test_compile_replace[single-eager] 0.1489ms 46.6923μs 21.4168 KOps/s 20.1623 KOps/s $\textbf{\color{#35bf28}+6.22\%}$
test_compile_replace[single-compile] 0.2735ms 0.1466ms 6.8222 KOps/s 6.4472 KOps/s $\textbf{\color{#35bf28}+5.82\%}$
test_compile_replace[multi-eager] 0.6929ms 0.5910ms 1.6920 KOps/s 1.6929 KOps/s $\color{#d91a1a}-0.05\%$
test_compile_replace[multi-compile] 0.2118ms 0.1583ms 6.3179 KOps/s 5.8979 KOps/s $\textbf{\color{#35bf28}+7.12\%}$
test_compile_tc_getattr_20[eager] 0.2585ms 0.1758ms 5.6871 KOps/s 5.6126 KOps/s $\color{#35bf28}+1.33\%$
test_compile_tc_getattr_20[compile] 0.3906ms 0.1770ms 5.6500 KOps/s 5.4456 KOps/s $\color{#35bf28}+3.75\%$
test_compile_clone_shallow[20-eager] 48.7810μs 19.0774μs 52.4180 KOps/s 52.4996 KOps/s $\color{#d91a1a}-0.16\%$
test_compile_clone_shallow[20-compile] 0.1029ms 22.6068μs 44.2344 KOps/s 43.5050 KOps/s $\color{#35bf28}+1.68\%$
test_compile_clone_shallow[40-eager] 0.1094ms 33.5485μs 29.8076 KOps/s 30.2553 KOps/s $\color{#d91a1a}-1.48\%$
test_compile_clone_shallow[40-compile] 0.1207ms 24.0900μs 41.5111 KOps/s 38.6246 KOps/s $\textbf{\color{#35bf28}+7.47\%}$
test_compile_clone_shallow[80-eager] 0.1029ms 62.2550μs 16.0630 KOps/s 15.9880 KOps/s $\color{#35bf28}+0.47\%$
test_compile_clone_shallow[80-compile] 87.1120μs 28.6032μs 34.9612 KOps/s 34.7532 KOps/s $\color{#35bf28}+0.60\%$
test_compile_update_inplace[eager] 0.1578ms 61.3860μs 16.2904 KOps/s 16.0959 KOps/s $\color{#35bf28}+1.21\%$
test_compile_update_inplace[compile] 0.2037ms 0.1290ms 7.7523 KOps/s 8.0116 KOps/s $\color{#d91a1a}-3.24\%$
test_mod_add[eager] 0.2874ms 0.2273ms 4.3988 KOps/s 4.4158 KOps/s $\color{#d91a1a}-0.38\%$
test_mod_add[compile] 0.3761ms 0.1220ms 8.2000 KOps/s 7.7679 KOps/s $\textbf{\color{#35bf28}+5.56\%}$
test_mod_add[compile-overhead] 0.4283ms 0.1976ms 5.0611 KOps/s 4.9765 KOps/s $\color{#35bf28}+1.70\%$
test_mod_wrap[eager] 0.5553ms 0.4746ms 2.1070 KOps/s 2.1364 KOps/s $\color{#d91a1a}-1.37\%$
test_mod_wrap[compile] 0.4391ms 0.3631ms 2.7541 KOps/s 2.7061 KOps/s $\color{#35bf28}+1.77\%$
test_mod_wrap[compile-overhead] 7.1692ms 4.0105ms 249.3446 Ops/s 260.3410 Ops/s $\color{#d91a1a}-4.22\%$
test_mod_wrap_and_backward[eager] 1.8687ms 1.7247ms 579.8077 Ops/s 598.6053 Ops/s $\color{#d91a1a}-3.14\%$
test_mod_wrap_and_backward[compile] 1.5171ms 1.3667ms 731.7057 Ops/s 745.1069 Ops/s $\color{#d91a1a}-1.80\%$
test_mod_wrap_and_backward[compile-overhead] 1.5841ms 0.9990ms 1.0010 KOps/s 1.0279 KOps/s $\color{#d91a1a}-2.61\%$
test_seq_add[eager] 0.6099ms 0.5265ms 1.8993 KOps/s 1.9768 KOps/s $\color{#d91a1a}-3.92\%$
test_seq_add[compile] 0.2103ms 0.1440ms 6.9424 KOps/s 6.8506 KOps/s $\color{#35bf28}+1.34\%$
test_seq_add[compile-overhead] 0.3387ms 0.2115ms 4.7283 KOps/s 4.4680 KOps/s $\textbf{\color{#35bf28}+5.83\%}$
test_seq_wrap[eager] 1.5506ms 1.4201ms 704.1563 Ops/s 706.1244 Ops/s $\color{#d91a1a}-0.28\%$
test_seq_wrap[compile] 0.4942ms 0.3781ms 2.6448 KOps/s 2.6653 KOps/s $\color{#d91a1a}-0.77\%$
test_seq_wrap[compile-overhead] 0.4889ms 0.3342ms 2.9920 KOps/s 3.0280 KOps/s $\color{#d91a1a}-1.19\%$
test_func_call_runtime[False-eager] 0.9326ms 0.8418ms 1.1879 KOps/s 1.2550 KOps/s $\textbf{\color{#d91a1a}-5.34\%}$
test_func_call_runtime[False-compile] 0.8706ms 0.7616ms 1.3130 KOps/s 1.3104 KOps/s $\color{#35bf28}+0.20\%$
test_func_call_runtime[False-compile-overhead] 0.6742ms 0.5149ms 1.9423 KOps/s 1.9380 KOps/s $\color{#35bf28}+0.22\%$
test_func_call_runtime[True-eager] 1.3783ms 1.3058ms 765.7982 Ops/s 933.3875 Ops/s $\textbf{\color{#d91a1a}-17.95\%}$
test_func_call_runtime[True-compile] 0.8538ms 0.7850ms 1.2739 KOps/s 1.2696 KOps/s $\color{#35bf28}+0.34\%$
test_func_call_runtime[True-compile-overhead] 0.7128ms 0.5519ms 1.8118 KOps/s 1.8603 KOps/s $\color{#d91a1a}-2.61\%$
test_func_call_cm_runtime[False-eager] 0.8816ms 0.7987ms 1.2520 KOps/s 1.2047 KOps/s $\color{#35bf28}+3.92\%$
test_func_call_cm_runtime[False-compile] 0.8222ms 0.7510ms 1.3315 KOps/s 1.3292 KOps/s $\color{#35bf28}+0.17\%$
test_func_call_cm_runtime[False-compile-overhead] 0.5938ms 0.4963ms 2.0148 KOps/s 1.9983 KOps/s $\color{#35bf28}+0.83\%$
test_func_call_cm_runtime[True-eager] 1.5912ms 1.4557ms 686.9422 Ops/s 832.5190 Ops/s $\textbf{\color{#d91a1a}-17.49\%}$
test_func_call_cm_runtime[True-compile] 0.9634ms 0.8647ms 1.1565 KOps/s 1.1576 KOps/s $\color{#d91a1a}-0.10\%$
test_func_call_cm_runtime[True-compile-overhead] 1.2524ms 0.8794ms 1.1371 KOps/s 1.1249 KOps/s $\color{#35bf28}+1.08\%$
test_vmap_func_call_cm_runtime[eager] 3.2556ms 2.6951ms 371.0412 Ops/s 374.0838 Ops/s $\color{#d91a1a}-0.81\%$
test_vmap_func_call_cm_runtime[compile] 4.3140ms 3.7867ms 264.0821 Ops/s 269.3809 Ops/s $\color{#d91a1a}-1.97\%$
test_vmap_func_call_cm_runtime[compile-overhead] 4.3097ms 3.7778ms 264.7033 Ops/s 269.3339 Ops/s $\color{#d91a1a}-1.72\%$
test_distributed 1.9059ms 0.1586ms 6.3035 KOps/s 6.4943 KOps/s $\color{#d91a1a}-2.94\%$
test_tdmodule 0.4629ms 0.1942ms 5.1496 KOps/s 5.1568 KOps/s $\color{#d91a1a}-0.14\%$
test_tdmodule_dispatch 0.4439ms 0.3747ms 2.6691 KOps/s 2.6665 KOps/s $\color{#35bf28}+0.10\%$
test_tdseq 45.9610μs 26.3728μs 37.9179 KOps/s 37.7024 KOps/s $\color{#35bf28}+0.57\%$
test_tdseq_dispatch 0.2624ms 0.2210ms 4.5249 KOps/s 4.5848 KOps/s $\color{#d91a1a}-1.31\%$
test_instantiation_functorch 2.8756ms 2.7168ms 368.0867 Ops/s 368.9353 Ops/s $\color{#d91a1a}-0.23\%$
test_exec_functorch 0.3114ms 0.1977ms 5.0589 KOps/s 5.0182 KOps/s $\color{#35bf28}+0.81\%$
test_exec_functional_call 0.2000ms 0.1547ms 6.4631 KOps/s 6.4589 KOps/s $\color{#35bf28}+0.07\%$
test_exec_td_decorator 0.7304ms 0.3697ms 2.7049 KOps/s 4.1151 KOps/s $\textbf{\color{#d91a1a}-34.27\%}$
test_vmap_mlp_speed_decorator[True-True] 1.4495ms 1.2029ms 831.3411 Ops/s 889.3750 Ops/s $\textbf{\color{#d91a1a}-6.53\%}$
test_vmap_mlp_speed_decorator[True-False] 1.4123ms 1.2014ms 832.3431 Ops/s 895.9720 Ops/s $\textbf{\color{#d91a1a}-7.10\%}$
test_vmap_mlp_speed_decorator[False-True] 1.1018ms 0.9046ms 1.1055 KOps/s 1.2148 KOps/s $\textbf{\color{#d91a1a}-8.99\%}$
test_vmap_mlp_speed_decorator[False-False] 1.0850ms 0.9036ms 1.1067 KOps/s 1.2244 KOps/s $\textbf{\color{#d91a1a}-9.61\%}$
test_vmap_transformer_speed_decorator[True-True] 27.0378ms 26.8515ms 37.2419 Ops/s 40.0224 Ops/s $\textbf{\color{#d91a1a}-6.95\%}$
test_vmap_transformer_speed_decorator[True-False] 27.0543ms 26.7995ms 37.3141 Ops/s 40.1824 Ops/s $\textbf{\color{#d91a1a}-7.14\%}$
test_vmap_transformer_speed_decorator[False-True] 26.6592ms 26.4568ms 37.7975 Ops/s 40.7703 Ops/s $\textbf{\color{#d91a1a}-7.29\%}$
test_vmap_transformer_speed_decorator[False-False] 27.4851ms 26.6016ms 37.5917 Ops/s 40.9388 Ops/s $\textbf{\color{#d91a1a}-8.18\%}$
test_to_module_speed[True] 3.5370ms 3.4115ms 293.1243 Ops/s 612.3675 Ops/s $\textbf{\color{#d91a1a}-52.13\%}$
test_to_module_speed[False] 3.5167ms 3.4032ms 293.8416 Ops/s 624.1709 Ops/s $\textbf{\color{#d91a1a}-52.92\%}$
test_to_module_plain_tensor_speed[None] 4.8320ms 3.4076ms 293.4643 Ops/s 579.9428 Ops/s $\textbf{\color{#d91a1a}-49.40\%}$
test_to_module_plain_tensor_speed[False] 1.7465ms 1.5708ms 636.6217 Ops/s 640.6027 Ops/s $\color{#d91a1a}-0.62\%$
test_to_module_plain_tensor_speed[True] 3.5676ms 3.4169ms 292.6633 Ops/s 293.6150 Ops/s $\color{#d91a1a}-0.32\%$
test_tc_init 83.3720μs 44.4503μs 22.4970 KOps/s 22.5203 KOps/s $\color{#d91a1a}-0.10\%$
test_tc_init_tensor_only 38.8010μs 9.3268μs 107.2184 KOps/s 107.1674 KOps/s $\color{#35bf28}+0.05\%$
test_tc_init_nested 0.1247ms 86.4688μs 11.5649 KOps/s 11.5033 KOps/s $\color{#35bf28}+0.54\%$
test_tc_init_many_fields 60.0320μs 16.6477μs 60.0685 KOps/s 59.0841 KOps/s $\color{#35bf28}+1.67\%$
test_tc_first_layer_tensor 29.5610μs 1.7713μs 564.5444 KOps/s 574.1577 KOps/s $\color{#d91a1a}-1.67\%$
test_tc_first_layer_tensor_only 1.9571μs 0.3927μs 2.5463 MOps/s 2.4980 MOps/s $\color{#35bf28}+1.93\%$
test_tc_first_layer_tensor_set 37.8710μs 4.1499μs 240.9687 KOps/s 241.0986 KOps/s $\color{#d91a1a}-0.05\%$
test_tc_first_layer_tensor_only_set 31.9210μs 3.4461μs 290.1871 KOps/s 288.4371 KOps/s $\color{#35bf28}+0.61\%$
test_tc_first_layer_nontensor 33.2910μs 6.0863μs 164.3031 KOps/s 162.6627 KOps/s $\color{#35bf28}+1.01\%$
test_tc_second_layer_tensor 46.1310μs 4.2365μs 236.0429 KOps/s 229.1151 KOps/s $\color{#35bf28}+3.02\%$
test_tc_second_layer_nontensor 29.1210μs 8.5253μs 117.2979 KOps/s 116.1173 KOps/s $\color{#35bf28}+1.02\%$
test_unbind 0.2989s 14.3863ms 69.5104 Ops/s 64.6923 Ops/s $\textbf{\color{#35bf28}+7.45\%}$
test_full_like 4.8027ms 4.6546ms 214.8410 Ops/s 215.6702 Ops/s $\color{#d91a1a}-0.38\%$
test_zeros_like 4.9266ms 4.6471ms 215.1863 Ops/s 216.8244 Ops/s $\color{#d91a1a}-0.76\%$
test_ones_like 4.8383ms 4.5871ms 218.0050 Ops/s 215.6340 Ops/s $\color{#35bf28}+1.10\%$
test_clone 7.3417ms 6.8031ms 146.9923 Ops/s 147.1019 Ops/s $\color{#d91a1a}-0.07\%$
test_squeeze 0.1817ms 14.3456μs 69.7079 KOps/s 67.6226 KOps/s $\color{#35bf28}+3.08\%$
test_unsqueeze 0.1755ms 0.1150ms 8.6993 KOps/s 8.5956 KOps/s $\color{#35bf28}+1.21\%$
test_split 0.4124ms 0.1938ms 5.1608 KOps/s 5.1584 KOps/s $\color{#35bf28}+0.05\%$
test_permute 0.3324ms 0.1969ms 5.0799 KOps/s 4.9854 KOps/s $\color{#35bf28}+1.90\%$
test_stack 19.6706ms 19.1057ms 52.3405 Ops/s 52.5946 Ops/s $\color{#d91a1a}-0.48\%$
test_cat 19.8579ms 19.1332ms 52.2651 Ops/s 52.2215 Ops/s $\color{#35bf28}+0.08\%$
test_sequential_tensordict 1.3410ms 0.9020ms 1.1087 KOps/s 1.1171 KOps/s $\color{#d91a1a}-0.75\%$
test_sequential_graph_module 0.5827ms 0.4303ms 2.3241 KOps/s 2.3625 KOps/s $\color{#d91a1a}-1.63\%$
test_nested_tensordict 2.6819ms 1.2992ms 769.7260 Ops/s 807.2930 Ops/s $\color{#d91a1a}-4.65\%$
test_nested_graph_module 0.6376ms 0.5610ms 1.7826 KOps/s 1.8867 KOps/s $\textbf{\color{#d91a1a}-5.52\%}$

@github-actions

github-actions Bot commented Aug 14, 2026 •
edited
Loading

Copy link
Copy Markdown
Contributor

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

Total Benchmarks: 356. Improved: $\large\color{#35bf28}23$. Worsened: $\large\color{#d91a1a}27$.

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_collector_preallocated_write[stack_out-cpu-32-2] 0.2415ms 0.1667ms 5.9970 KOps/s 6.0450 KOps/s $\color{#d91a1a}-0.79\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.3542ms 0.2908ms 3.4389 KOps/s 3.3731 KOps/s $\color{#35bf28}+1.95\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.8719ms 0.7678ms 1.3023 KOps/s 1.2946 KOps/s $\color{#35bf28}+0.60\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.4050ms 0.3231ms 3.0952 KOps/s 3.0338 KOps/s $\color{#35bf28}+2.02\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.3262ms 1.0967ms 911.8538 Ops/s 905.7955 Ops/s $\color{#35bf28}+0.67\%$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.3872ms 7.2706ms 137.5408 Ops/s 136.2114 Ops/s $\color{#35bf28}+0.98\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.3114ms 0.2447ms 4.0865 KOps/s 4.0134 KOps/s $\color{#35bf28}+1.82\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.4476ms 0.3648ms 2.7409 KOps/s 2.6531 KOps/s $\color{#35bf28}+3.31\%$
test_collector_preallocated_write[stack_out-cuda-32-32] 0.8800ms 0.8100ms 1.2345 KOps/s 1.2285 KOps/s $\color{#35bf28}+0.49\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.2922ms 0.2434ms 4.1084 KOps/s 3.8473 KOps/s $\textbf{\color{#35bf28}+6.79\%}$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.4166ms 0.3656ms 2.7350 KOps/s 2.6353 KOps/s $\color{#35bf28}+3.79\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 0.9407ms 0.8729ms 1.1456 KOps/s 1.1429 KOps/s $\color{#35bf28}+0.24\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2954ms 0.2148ms 4.6557 KOps/s 4.6268 KOps/s $\color{#35bf28}+0.62\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 0.9378ms 0.7976ms 1.2537 KOps/s 1.2125 KOps/s $\color{#35bf28}+3.40\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.2622ms 3.1715ms 315.3127 Ops/s 306.1969 Ops/s $\color{#35bf28}+2.98\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.4542ms 0.3615ms 2.7661 KOps/s 2.7145 KOps/s $\color{#35bf28}+1.90\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 1.8082ms 1.6580ms 603.1353 Ops/s 605.6802 Ops/s $\color{#d91a1a}-0.42\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 10.0866ms 9.9702ms 100.2989 Ops/s 98.8150 Ops/s $\color{#35bf28}+1.50\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.4231ms 0.3801ms 2.6307 KOps/s 2.5637 KOps/s $\color{#35bf28}+2.62\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.6176ms 1.5078ms 663.2185 Ops/s 682.3458 Ops/s $\color{#d91a1a}-2.80\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 6.0474ms 5.9476ms 168.1358 Ops/s 163.2841 Ops/s $\color{#35bf28}+2.97\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.4615ms 0.4053ms 2.4676 KOps/s 2.4464 KOps/s $\color{#35bf28}+0.86\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.6043ms 1.5034ms 665.1434 Ops/s 678.0505 Ops/s $\color{#d91a1a}-1.90\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 5.7419ms 5.5571ms 179.9502 Ops/s 172.4086 Ops/s $\color{#35bf28}+4.37\%$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.2073ms 0.1690ms 5.9156 KOps/s 5.8002 KOps/s $\color{#35bf28}+1.99\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.7587ms 0.6585ms 1.5187 KOps/s 1.5006 KOps/s $\color{#35bf28}+1.20\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 2.7072ms 2.6287ms 380.4142 Ops/s 377.5854 Ops/s $\color{#35bf28}+0.75\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.3835ms 0.3254ms 3.0734 KOps/s 3.0798 KOps/s $\color{#d91a1a}-0.21\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.6449ms 1.4853ms 673.2769 Ops/s 667.5943 Ops/s $\color{#35bf28}+0.85\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 9.5872ms 9.3939ms 106.4520 Ops/s 105.8789 Ops/s $\color{#35bf28}+0.54\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.4097ms 0.3170ms 3.1547 KOps/s 3.1091 KOps/s $\color{#35bf28}+1.47\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.2937ms 1.1896ms 840.5864 Ops/s 798.1269 Ops/s $\textbf{\color{#35bf28}+5.32\%}$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.7408ms 4.6419ms 215.4281 Ops/s 207.6944 Ops/s $\color{#35bf28}+3.72\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.4380ms 0.3195ms 3.1295 KOps/s 3.0692 KOps/s $\color{#35bf28}+1.96\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.2618ms 1.1925ms 838.5892 Ops/s 820.3827 Ops/s $\color{#35bf28}+2.22\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 4.7299ms 4.6720ms 214.0391 Ops/s 209.6875 Ops/s $\color{#35bf28}+2.08\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.1670ms 0.1147ms 8.7187 KOps/s 8.6651 KOps/s $\color{#35bf28}+0.62\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.5259ms 0.4509ms 2.2177 KOps/s 2.2096 KOps/s $\color{#35bf28}+0.37\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 1.8878ms 1.8067ms 553.5022 Ops/s 558.3819 Ops/s $\color{#d91a1a}-0.87\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.3241ms 0.2570ms 3.8914 KOps/s 3.9352 KOps/s $\color{#d91a1a}-1.11\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.4246ms 1.2294ms 813.3955 Ops/s 823.9258 Ops/s $\color{#d91a1a}-1.28\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 8.4194ms 8.3319ms 120.0204 Ops/s 119.4645 Ops/s $\color{#35bf28}+0.47\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.3315ms 0.2696ms 3.7098 KOps/s 3.6621 KOps/s $\color{#35bf28}+1.30\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.0819ms 0.9938ms 1.0063 KOps/s 983.9431 Ops/s $\color{#35bf28}+2.27\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 3.9735ms 3.9042ms 256.1347 Ops/s 253.4893 Ops/s $\color{#35bf28}+1.04\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.3282ms 0.2694ms 3.7114 KOps/s 3.6710 KOps/s $\color{#35bf28}+1.10\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.0997ms 0.9995ms 1.0005 KOps/s 981.7552 Ops/s $\color{#35bf28}+1.91\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 4.2753ms 4.0211ms 248.6893 Ops/s 252.1433 Ops/s $\color{#d91a1a}-1.37\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.3077ms 16.0327ms 62.3726 Ops/s 61.8033 Ops/s $\color{#35bf28}+0.92\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 32.4512ms 32.3134ms 30.9469 Ops/s 30.8117 Ops/s $\color{#35bf28}+0.44\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 83.1902ms 82.7531ms 12.0841 Ops/s 11.3405 Ops/s $\textbf{\color{#35bf28}+6.56\%}$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.1916s 0.1908s 5.2401 Ops/s 4.9605 Ops/s $\textbf{\color{#35bf28}+5.64\%}$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 13.6159ms 13.4449ms 74.3777 Ops/s 74.1305 Ops/s $\color{#35bf28}+0.33\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 28.2498ms 26.3719ms 37.9191 Ops/s 38.0300 Ops/s $\color{#d91a1a}-0.29\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 13.7157ms 13.6505ms 73.2576 Ops/s 73.0341 Ops/s $\color{#35bf28}+0.31\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 26.8507ms 26.6804ms 37.4807 Ops/s 37.4736 Ops/s $\color{#35bf28}+0.02\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 51.4820ms 50.9541ms 19.6255 Ops/s 19.4200 Ops/s $\color{#35bf28}+1.06\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1039s 0.1031s 9.7001 Ops/s 9.6162 Ops/s $\color{#35bf28}+0.87\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1467s 0.1440s 6.9440 Ops/s 6.6596 Ops/s $\color{#35bf28}+4.27\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3113s 0.3109s 3.2161 Ops/s 3.0823 Ops/s $\color{#35bf28}+4.34\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 76.7381ms 75.8973ms 13.1757 Ops/s 13.3715 Ops/s $\color{#d91a1a}-1.46\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1538s 0.1505s 6.6452 Ops/s 6.8573 Ops/s $\color{#d91a1a}-3.09\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 72.2793ms 71.5353ms 13.9791 Ops/s 13.7992 Ops/s $\color{#35bf28}+1.30\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1542s 0.1521s 6.5726 Ops/s 6.8704 Ops/s $\color{#d91a1a}-4.34\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 30.3007ms 30.1722ms 33.1431 Ops/s 32.9423 Ops/s $\color{#35bf28}+0.61\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 61.0388ms 60.9060ms 16.4187 Ops/s 16.3839 Ops/s $\color{#35bf28}+0.21\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1184s 0.1178s 8.4860 Ops/s 7.8032 Ops/s $\textbf{\color{#35bf28}+8.75\%}$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2645s 0.2615s 3.8239 Ops/s 3.6721 Ops/s $\color{#35bf28}+4.14\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 42.9947ms 42.8180ms 23.3547 Ops/s 23.2925 Ops/s $\color{#35bf28}+0.27\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 85.5492ms 85.3053ms 11.7226 Ops/s 11.6629 Ops/s $\color{#35bf28}+0.51\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 43.2449ms 42.9847ms 23.2641 Ops/s 23.1329 Ops/s $\color{#35bf28}+0.57\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 86.3209ms 86.0217ms 11.6250 Ops/s 11.5352 Ops/s $\color{#35bf28}+0.78\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 19.9802ms 19.7583ms 50.6116 Ops/s 50.3643 Ops/s $\color{#35bf28}+0.49\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 40.0969ms 39.9664ms 25.0210 Ops/s 24.7086 Ops/s $\color{#35bf28}+1.26\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1052s 0.1047s 9.5480 Ops/s 8.9588 Ops/s $\textbf{\color{#35bf28}+6.58\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2407s 0.2382s 4.1983 Ops/s 4.0738 Ops/s $\color{#35bf28}+3.06\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 37.5280ms 36.5542ms 27.3566 Ops/s 27.8245 Ops/s $\color{#d91a1a}-1.68\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 71.4845ms 71.2569ms 14.0337 Ops/s 13.9085 Ops/s $\color{#35bf28}+0.90\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 38.0871ms 36.2183ms 27.6104 Ops/s 27.7049 Ops/s $\color{#d91a1a}-0.34\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 74.9816ms 73.3858ms 13.6266 Ops/s 13.8914 Ops/s $\color{#d91a1a}-1.91\%$
test_plain_set_nested 51.0310μs 15.1060μs 66.1987 KOps/s 67.0633 KOps/s $\color{#d91a1a}-1.29\%$
test_plain_set_stack_nested 43.7100μs 14.8190μs 67.4810 KOps/s 68.4190 KOps/s $\color{#d91a1a}-1.37\%$
test_plain_set_nested_inplace 43.9000μs 16.2997μs 61.3510 KOps/s 60.7324 KOps/s $\color{#35bf28}+1.02\%$
test_plain_set_stack_nested_inplace 52.1810μs 16.1018μs 62.1050 KOps/s 61.6260 KOps/s $\color{#35bf28}+0.78\%$
test_items 42.1600μs 5.7197μs 174.8351 KOps/s 179.6962 KOps/s $\color{#d91a1a}-2.71\%$
test_items_nested 0.5365ms 0.4480ms 2.2323 KOps/s 2.1746 KOps/s $\color{#35bf28}+2.65\%$
test_items_nested_locked 0.5605ms 0.4474ms 2.2351 KOps/s 2.1703 KOps/s $\color{#35bf28}+2.99\%$
test_items_nested_leaf 0.1389ms 94.4929μs 10.5828 KOps/s 10.3389 KOps/s $\color{#35bf28}+2.36\%$
test_items_stack_nested 0.5293ms 0.4511ms 2.2167 KOps/s 2.2140 KOps/s $\color{#35bf28}+0.12\%$
test_items_stack_nested_leaf 0.1586ms 94.5313μs 10.5785 KOps/s 10.3936 KOps/s $\color{#35bf28}+1.78\%$
test_items_stack_nested_locked 0.5077ms 0.4521ms 2.2117 KOps/s 2.1637 KOps/s $\color{#35bf28}+2.22\%$
test_keys 33.5100μs 4.1223μs 242.5859 KOps/s 241.4769 KOps/s $\color{#35bf28}+0.46\%$
test_keys_nested 0.1862ms 0.1302ms 7.6813 KOps/s 7.6883 KOps/s $\color{#d91a1a}-0.09\%$
test_keys_nested_locked 0.8140ms 0.1386ms 7.2131 KOps/s 7.1930 KOps/s $\color{#35bf28}+0.28\%$
test_keys_nested_leaf 0.1677ms 0.1201ms 8.3249 KOps/s 8.2485 KOps/s $\color{#35bf28}+0.93\%$
test_keys_stack_nested 0.1791ms 0.1295ms 7.7218 KOps/s 7.7312 KOps/s $\color{#d91a1a}-0.12\%$
test_keys_stack_nested_leaf 0.1773ms 0.1196ms 8.3596 KOps/s 8.3517 KOps/s $\color{#35bf28}+0.09\%$
test_keys_stack_nested_locked 0.1926ms 0.1386ms 7.2134 KOps/s 7.2646 KOps/s $\color{#d91a1a}-0.70\%$
test_values 11.7842μs 0.9990μs 1.0010 MOps/s 995.6590 KOps/s $\color{#35bf28}+0.54\%$
test_values_nested 0.1134ms 51.8089μs 19.3017 KOps/s 19.0075 KOps/s $\color{#35bf28}+1.55\%$
test_values_nested_locked 0.1450ms 54.4554μs 18.3637 KOps/s 18.1895 KOps/s $\color{#35bf28}+0.96\%$
test_values_nested_leaf 93.5320μs 59.2361μs 16.8816 KOps/s 16.6916 KOps/s $\color{#35bf28}+1.14\%$
test_values_stack_nested 91.3410μs 51.8344μs 19.2922 KOps/s 19.0852 KOps/s $\color{#35bf28}+1.08\%$
test_values_stack_nested_leaf 97.5810μs 59.4950μs 16.8081 KOps/s 16.8041 KOps/s $\color{#35bf28}+0.02\%$
test_values_stack_nested_locked 0.1146ms 53.7177μs 18.6158 KOps/s 18.3334 KOps/s $\color{#35bf28}+1.54\%$
test_membership 5.0000μs 0.8037μs 1.2442 MOps/s 1.2285 MOps/s $\color{#35bf28}+1.28\%$
test_membership_nested 30.0410μs 2.8129μs 355.5035 KOps/s 344.3458 KOps/s $\color{#35bf28}+3.24\%$
test_membership_nested_leaf 31.6600μs 2.8497μs 350.9180 KOps/s 343.8415 KOps/s $\color{#35bf28}+2.06\%$
test_membership_stacked_nested 37.0710μs 2.7455μs 364.2270 KOps/s 353.8133 KOps/s $\color{#35bf28}+2.94\%$
test_membership_stacked_nested_leaf 34.9700μs 2.7295μs 366.3612 KOps/s 357.5845 KOps/s $\color{#35bf28}+2.45\%$
test_membership_nested_last 33.8200μs 4.2524μs 235.1597 KOps/s 231.4218 KOps/s $\color{#35bf28}+1.62\%$
test_membership_nested_leaf_last 69.2510μs 4.1538μs 240.7456 KOps/s 232.0736 KOps/s $\color{#35bf28}+3.74\%$
test_membership_stacked_nested_last 30.1500μs 4.0987μs 243.9813 KOps/s 239.3937 KOps/s $\color{#35bf28}+1.92\%$
test_membership_stacked_nested_leaf_last 33.1600μs 4.1163μs 242.9360 KOps/s 240.0925 KOps/s $\color{#35bf28}+1.18\%$
test_nested_getleaf 57.1000μs 21.0882μs 47.4198 KOps/s 47.8814 KOps/s $\color{#d91a1a}-0.96\%$
test_nested_get 54.4210μs 20.1892μs 49.5314 KOps/s 49.2472 KOps/s $\color{#35bf28}+0.58\%$
test_stacked_getleaf 63.6810μs 20.4342μs 48.9375 KOps/s 48.2007 KOps/s $\color{#35bf28}+1.53\%$
test_stacked_get 52.8410μs 19.7173μs 50.7168 KOps/s 49.8970 KOps/s $\color{#35bf28}+1.64\%$
test_nested_getitemleaf 55.4810μs 21.4635μs 46.5907 KOps/s 46.4083 KOps/s $\color{#35bf28}+0.39\%$
test_nested_getitem 66.6210μs 20.0373μs 49.9070 KOps/s 48.8659 KOps/s $\color{#35bf28}+2.13\%$
test_stacked_getitemleaf 53.0110μs 21.2902μs 46.9699 KOps/s 47.0114 KOps/s $\color{#d91a1a}-0.09\%$
test_stacked_getitem 57.4810μs 20.3304μs 49.1875 KOps/s 49.4746 KOps/s $\color{#d91a1a}-0.58\%$
test_lock_nested 8.6713ms 0.5022ms 1.9914 KOps/s 2.0154 KOps/s $\color{#d91a1a}-1.19\%$
test_lock_stack_nested 0.5517ms 0.4968ms 2.0129 KOps/s 2.0030 KOps/s $\color{#35bf28}+0.49\%$
test_unlock_nested 0.4742ms 0.3937ms 2.5402 KOps/s 2.5519 KOps/s $\color{#d91a1a}-0.46\%$
test_unlock_stack_nested 0.4528ms 0.3936ms 2.5408 KOps/s 2.5340 KOps/s $\color{#35bf28}+0.27\%$
test_flatten_speed 0.1588ms 0.1227ms 8.1524 KOps/s 8.1649 KOps/s $\color{#d91a1a}-0.15\%$
test_unflatten_speed 0.5991ms 0.5577ms 1.7930 KOps/s 1.7915 KOps/s $\color{#35bf28}+0.08\%$
test_common_ops 0.8876ms 0.7131ms 1.4022 KOps/s 1.4341 KOps/s $\color{#d91a1a}-2.22\%$
test_creation 0.1176ms 3.0515μs 327.7094 KOps/s 326.4603 KOps/s $\color{#35bf28}+0.38\%$
test_creation_empty 38.3810μs 6.7777μs 147.5427 KOps/s 147.8742 KOps/s $\color{#d91a1a}-0.22\%$
test_creation_nested_1 34.4100μs 11.2643μs 88.7757 KOps/s 88.6745 KOps/s $\color{#35bf28}+0.11\%$
test_creation_nested_2 41.9110μs 13.3377μs 74.9752 KOps/s 76.3090 KOps/s $\color{#d91a1a}-1.75\%$
test_creation_many_keys[10] 53.2210μs 20.2649μs 49.3465 KOps/s 49.0830 KOps/s $\color{#35bf28}+0.54\%$
test_creation_many_keys[50] 0.1191ms 86.8317μs 11.5165 KOps/s 11.3853 KOps/s $\color{#35bf28}+1.15\%$
test_creation_many_keys[100] 0.2102ms 0.1695ms 5.8992 KOps/s 5.8222 KOps/s $\color{#35bf28}+1.32\%$
test_creation_nested_many_keys[10] 75.3810μs 43.9695μs 22.7430 KOps/s 22.9268 KOps/s $\color{#d91a1a}-0.80\%$
test_creation_nested_many_keys[50] 0.2204ms 0.1764ms 5.6674 KOps/s 5.7137 KOps/s $\color{#d91a1a}-0.81\%$
test_clone 32.8200μs 12.5096μs 79.9388 KOps/s 78.7456 KOps/s $\color{#35bf28}+1.52\%$
test_getitem[int] 1.5715ms 15.2839μs 65.4282 KOps/s 65.2761 KOps/s $\color{#35bf28}+0.23\%$
test_getitem[slice_int] 0.1373ms 24.9581μs 40.0672 KOps/s 40.4776 KOps/s $\color{#d91a1a}-1.01\%$
test_getitem[range] 0.1846ms 62.5249μs 15.9936 KOps/s 16.3032 KOps/s $\color{#d91a1a}-1.90\%$
test_getitem[tuple] 0.1444ms 23.8512μs 41.9267 KOps/s 41.3044 KOps/s $\color{#35bf28}+1.51\%$
test_getitem[list] 0.1909ms 59.1644μs 16.9021 KOps/s 17.9380 KOps/s $\textbf{\color{#d91a1a}-5.77\%}$
test_setitem_dim[int] 48.3110μs 26.8099μs 37.2996 KOps/s 40.0035 KOps/s $\textbf{\color{#d91a1a}-6.76\%}$
test_setitem_dim[slice_int] 77.1410μs 42.8095μs 23.3593 KOps/s 23.7580 KOps/s $\color{#d91a1a}-1.68\%$
test_setitem_dim[range] 0.1353ms 96.6560μs 10.3460 KOps/s 10.8737 KOps/s $\color{#d91a1a}-4.85\%$
test_setitem_dim[tuple] 75.0910μs 39.2931μs 25.4498 KOps/s 25.4503 KOps/s $-0.00\%$
test_setitem 50.9110μs 16.6412μs 60.0919 KOps/s 58.6964 KOps/s $\color{#35bf28}+2.38\%$
test_set 48.7510μs 15.9577μs 62.6657 KOps/s 60.8282 KOps/s $\color{#35bf28}+3.02\%$
test_set_shared 0.6589ms 0.2095ms 4.7736 KOps/s 4.9161 KOps/s $\color{#d91a1a}-2.90\%$
test_update 0.3803ms 20.6150μs 48.5083 KOps/s 47.2194 KOps/s $\color{#35bf28}+2.73\%$
test_update_nested 0.2000ms 30.5643μs 32.7179 KOps/s 31.0458 KOps/s $\textbf{\color{#35bf28}+5.39\%}$
test_update__nested 0.6319ms 32.4406μs 30.8256 KOps/s 30.9870 KOps/s $\color{#d91a1a}-0.52\%$
test_set_nested 53.5010μs 18.2348μs 54.8401 KOps/s 55.8706 KOps/s $\color{#d91a1a}-1.84\%$
test_set_nested_new 61.0210μs 22.2762μs 44.8910 KOps/s 44.7919 KOps/s $\color{#35bf28}+0.22\%$
test_select 80.3220μs 37.3869μs 26.7473 KOps/s 25.8466 KOps/s $\color{#35bf28}+3.48\%$
test_select_nested 0.1102ms 72.0898μs 13.8716 KOps/s 13.9006 KOps/s $\color{#d91a1a}-0.21\%$
test_exclude_nested 0.1304ms 88.2906μs 11.3262 KOps/s 11.4360 KOps/s $\color{#d91a1a}-0.96\%$
test_empty[True] 0.8131ms 0.3853ms 2.5955 KOps/s 2.5564 KOps/s $\color{#35bf28}+1.53\%$
test_empty[False] 8.3102μs 1.3682μs 730.9059 KOps/s 750.9571 KOps/s $\color{#d91a1a}-2.67\%$
test_to 0.1161ms 70.5232μs 14.1797 KOps/s 14.3044 KOps/s $\color{#d91a1a}-0.87\%$
test_to_nonblocking 0.1172ms 70.3667μs 14.2113 KOps/s 14.1423 KOps/s $\color{#35bf28}+0.49\%$
test_unbind_speed 0.4033ms 0.3305ms 3.0260 KOps/s 2.9918 KOps/s $\color{#35bf28}+1.14\%$
test_unbind_speed_stack0 0.4049ms 0.3291ms 3.0389 KOps/s 3.0492 KOps/s $\color{#d91a1a}-0.34\%$
test_unbind_speed_stack1 0.1262s 0.8600ms 1.1627 KOps/s 1.3210 KOps/s $\textbf{\color{#d91a1a}-11.98\%}$
test_split 0.1259s 1.4382ms 695.3031 Ops/s 643.2391 Ops/s $\textbf{\color{#35bf28}+8.09\%}$
test_chunk 1.3310ms 1.2169ms 821.7662 Ops/s 827.3730 Ops/s $\color{#d91a1a}-0.68\%$
test_to_cpu_blocking 19.9590ms 19.8557ms 50.3633 Ops/s 49.8176 Ops/s $\color{#35bf28}+1.10\%$
test_to_cpu_global_sync 0.1375s 13.1485ms 76.0542 Ops/s 85.1767 Ops/s $\textbf{\color{#d91a1a}-10.71\%}$
test_to_cpu_event_sync 13.0763ms 12.7022ms 78.7265 Ops/s 78.3667 Ops/s $\color{#35bf28}+0.46\%$
test_to_cpu_default 13.0476ms 12.6859ms 78.8276 Ops/s 69.1935 Ops/s $\textbf{\color{#35bf28}+13.92\%}$
test_consolidate[False-None] 4.5584ms 4.1075ms 243.4562 Ops/s 240.8660 Ops/s $\color{#35bf28}+1.08\%$
test_consolidate[default-None] 3.1793ms 2.2021ms 454.1138 Ops/s 445.5605 Ops/s $\color{#35bf28}+1.92\%$
test_consolidate[reduce-overhead-None] 3.1160ms 2.1058ms 474.8851 Ops/s 463.0826 Ops/s $\color{#35bf28}+2.55\%$
test_consolidate_njt[False-None] 5.9652ms 5.0959ms 196.2349 Ops/s 195.8799 Ops/s $\color{#35bf28}+0.18\%$
test_to[False-False-None] 2.3133ms 2.1743ms 459.9242 Ops/s 452.9665 Ops/s $\color{#35bf28}+1.54\%$
test_to[True-False-None] 2.1830ms 1.9031ms 525.4604 Ops/s 507.7561 Ops/s $\color{#35bf28}+3.49\%$
test_to[within-False-None] 6.4718ms 6.1556ms 162.4547 Ops/s 160.7799 Ops/s $\color{#35bf28}+1.04\%$
test_to[True-default-None] 10.9241ms 10.4806ms 95.4141 Ops/s 95.1311 Ops/s $\color{#35bf28}+0.30\%$
test_to_njt[False-False-None] 7.2592ms 7.0909ms 141.0254 Ops/s 141.5598 Ops/s $\color{#d91a1a}-0.38\%$
test_to_njt[True-False-None] 4.0220ms 3.8611ms 258.9904 Ops/s 253.4748 Ops/s $\color{#35bf28}+2.18\%$
test_to_njt[within-False-None] 9.1977ms 9.0057ms 111.0406 Ops/s 108.8054 Ops/s $\color{#35bf28}+2.05\%$
test_creation[device0] 0.3926ms 0.1180ms 8.4764 KOps/s 8.5520 KOps/s $\color{#d91a1a}-0.88\%$
test_creation_from_tensor 0.4917ms 0.1167ms 8.5676 KOps/s 8.7028 KOps/s $\color{#d91a1a}-1.55\%$
test_add_one[memmap_tensor0] 0.3634ms 6.0858μs 164.3175 KOps/s 162.1307 KOps/s $\color{#35bf28}+1.35\%$
test_contiguous[memmap_tensor0] 13.3310μs 0.6403μs 1.5618 MOps/s 2.1796 MOps/s $\textbf{\color{#d91a1a}-28.34\%}$
test_stack[memmap_tensor0] 90.9320μs 4.0828μs 244.9296 KOps/s 235.3516 KOps/s $\color{#35bf28}+4.07\%$
test_memmaptd_index 1.4131ms 0.2715ms 3.6828 KOps/s 3.5739 KOps/s $\color{#35bf28}+3.05\%$
test_memmaptd_index_astensor 0.5365ms 0.3745ms 2.6706 KOps/s 2.6107 KOps/s $\color{#35bf28}+2.29\%$
test_memmaptd_index_op 1.0550ms 0.6121ms 1.6336 KOps/s 1.5944 KOps/s $\color{#35bf28}+2.46\%$
test_serialize_model 0.1425s 0.1409s 7.0969 Ops/s 7.0105 Ops/s $\color{#35bf28}+1.23\%$
test_serialize_model_archive 1.3870s 1.2520s 0.7987 Ops/s 0.8040 Ops/s $\color{#d91a1a}-0.66\%$
test_load_archive 18.2888ms 17.7154ms 56.4482 Ops/s 55.8352 Ops/s $\color{#35bf28}+1.10\%$
test_serialize_model_zarr 2.4976s 2.4873s 0.4020 Ops/s 0.4004 Ops/s $\color{#35bf28}+0.41\%$
test_load_zarr 0.5968s 0.4171s 2.3978 Ops/s 2.5062 Ops/s $\color{#d91a1a}-4.32\%$
test_serialize_model_pickle 1.4774s 1.2328s 0.8112 Ops/s 0.7582 Ops/s $\textbf{\color{#35bf28}+6.99\%}$
test_serialize_weights 0.1423s 0.1404s 7.1227 Ops/s 7.1448 Ops/s $\color{#d91a1a}-0.31\%$
test_serialize_weights_returnearly 58.5923ms 49.7336ms 20.1071 Ops/s 18.5428 Ops/s $\textbf{\color{#35bf28}+8.44\%}$
test_serialize_weights_pickle 1.3624s 1.2315s 0.8120 Ops/s 0.8247 Ops/s $\color{#d91a1a}-1.54\%$
test_consolidate[few_large-0] 60.4306ms 7.3993ms 135.1473 Ops/s 139.9427 Ops/s $\color{#d91a1a}-3.43\%$
test_consolidate[few_large-8] 53.4534ms 7.2280ms 138.3515 Ops/s 145.3491 Ops/s $\color{#d91a1a}-4.81\%$
test_consolidate[many_small-0] 0.1249s 0.1217s 8.2151 Ops/s 3.5742 Ops/s $\textbf{\color{#35bf28}+129.85\%}$
test_consolidate[many_small-8] 42.8553ms 39.7468ms 25.1593 Ops/s 24.4587 Ops/s $\color{#35bf28}+2.86\%$
test_consolidate_to_file[few_large-0] 0.1307s 24.2751ms 41.1945 Ops/s 42.0687 Ops/s $\color{#d91a1a}-2.08\%$
test_consolidate_to_file[few_large-8] 0.1435s 24.3418ms 41.0816 Ops/s 42.8977 Ops/s $\color{#d91a1a}-4.23\%$
test_consolidate_to_file[many_small-0] 0.1964s 0.1862s 5.3700 Ops/s 5.2470 Ops/s $\color{#35bf28}+2.35\%$
test_consolidate_to_file[many_small-8] 0.3488s 0.2732s 3.6604 Ops/s 4.7308 Ops/s $\textbf{\color{#d91a1a}-22.63\%}$
test_reshape_pytree 7.0476ms 31.8046μs 31.4420 KOps/s 31.1097 KOps/s $\color{#35bf28}+1.07\%$
test_reshape_td 84.6910μs 43.9740μs 22.7407 KOps/s 22.6306 KOps/s $\color{#35bf28}+0.49\%$
test_view_pytree 0.2283ms 30.9303μs 32.3307 KOps/s 32.0714 KOps/s $\color{#35bf28}+0.81\%$
test_view_td 92.7110μs 51.7878μs 19.3096 KOps/s 18.7859 KOps/s $\color{#35bf28}+2.79\%$
test_unbind_pytree 0.2374ms 35.1618μs 28.4399 KOps/s 28.0409 KOps/s $\color{#35bf28}+1.42\%$
test_unbind_td 0.2067ms 48.7503μs 20.5127 KOps/s 19.8507 KOps/s $\color{#35bf28}+3.33\%$
test_split_pytree 0.2116ms 40.4180μs 24.7414 KOps/s 24.7852 KOps/s $\color{#d91a1a}-0.18\%$
test_split_td 0.1876ms 70.1526μs 14.2546 KOps/s 14.0290 KOps/s $\color{#35bf28}+1.61\%$
test_add_pytree 0.2052ms 38.5478μs 25.9418 KOps/s 25.5842 KOps/s $\color{#35bf28}+1.40\%$
test_add_td 89.2110μs 50.9959μs 19.6094 KOps/s 18.5703 KOps/s $\textbf{\color{#35bf28}+5.60\%}$
test_compile_add_one_nested[tensordict-compile] 0.3734ms 0.2782ms 3.5947 KOps/s 3.5581 KOps/s $\color{#35bf28}+1.03\%$
test_compile_add_one_nested[tensordict-eager] 0.2840ms 0.2011ms 4.9723 KOps/s 4.8896 KOps/s $\color{#35bf28}+1.69\%$
test_compile_add_one_nested[pytree-compile] 0.3269ms 0.1525ms 6.5593 KOps/s 6.1658 KOps/s $\textbf{\color{#35bf28}+6.38\%}$
test_compile_add_one_nested[pytree-eager] 0.5042ms 0.1757ms 5.6922 KOps/s 5.5639 KOps/s $\color{#35bf28}+2.30\%$
test_compile_copy_nested[tensordict-compile] 0.1059ms 21.0028μs 47.6128 KOps/s 47.7952 KOps/s $\color{#d91a1a}-0.38\%$
test_compile_copy_nested[tensordict-eager] 0.2886ms 51.7267μs 19.3324 KOps/s 19.3962 KOps/s $\color{#d91a1a}-0.33\%$
test_compile_copy_nested[pytree-compile] 0.1434ms 18.5900μs 53.7923 KOps/s 51.9172 KOps/s $\color{#35bf28}+3.61\%$
test_compile_copy_nested[pytree-eager] 0.2323s 80.2183μs 12.4660 KOps/s 15.4924 KOps/s $\textbf{\color{#d91a1a}-19.53\%}$
test_compile_add_one_flat[tensordict-compile] 0.4041ms 0.3085ms 3.2417 KOps/s 3.0668 KOps/s $\textbf{\color{#35bf28}+5.70\%}$
test_compile_add_one_flat[tensordict-eager] 0.4300ms 0.2912ms 3.4344 KOps/s 3.4394 KOps/s $\color{#d91a1a}-0.15\%$
test_compile_add_one_flat[tensorclass-compile] 0.2992ms 0.2106ms 4.7480 KOps/s 4.4661 KOps/s $\textbf{\color{#35bf28}+6.31\%}$
test_compile_add_one_flat[tensorclass-eager] 0.1214ms 76.4953μs 13.0727 KOps/s 13.0604 KOps/s $\color{#35bf28}+0.09\%$
test_compile_add_one_flat[pytree-compile] 0.3354ms 0.2392ms 4.1804 KOps/s 4.0315 KOps/s $\color{#35bf28}+3.69\%$
test_compile_add_one_flat[pytree-eager] 0.8465ms 0.5662ms 1.7663 KOps/s 1.8875 KOps/s $\textbf{\color{#d91a1a}-6.42\%}$
test_compile_add_self_flat[tensordict-eager] 0.4014ms 0.3368ms 2.9690 KOps/s 2.9762 KOps/s $\color{#d91a1a}-0.24\%$
test_compile_add_self_flat[tensordict-compile] 0.3697ms 0.3054ms 3.2741 KOps/s 3.1033 KOps/s $\textbf{\color{#35bf28}+5.51\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1457ms 92.4455μs 10.8172 KOps/s 10.6570 KOps/s $\color{#35bf28}+1.50\%$
test_compile_add_self_flat[tensorclass-compile] 0.3599ms 0.2092ms 4.7811 KOps/s 4.6374 KOps/s $\color{#35bf28}+3.10\%$
test_compile_add_self_flat[pytree-eager] 0.7028ms 0.4509ms 2.2178 KOps/s 2.2551 KOps/s $\color{#d91a1a}-1.65\%$
test_compile_add_self_flat[pytree-compile] 0.5343ms 0.2391ms 4.1822 KOps/s 4.0091 KOps/s $\color{#35bf28}+4.32\%$
test_compile_copy_flat[tensordict-compile] 83.0910μs 25.6575μs 38.9749 KOps/s 38.7340 KOps/s $\color{#35bf28}+0.62\%$
test_compile_copy_flat[tensordict-eager] 78.3020μs 41.0945μs 24.3342 KOps/s 24.6964 KOps/s $\color{#d91a1a}-1.47\%$
test_compile_copy_flat[pytree-compile] 0.1759ms 25.0290μs 39.9536 KOps/s 38.6375 KOps/s $\color{#35bf28}+3.41\%$
test_compile_copy_flat[pytree-eager] 0.3396ms 51.8229μs 19.2965 KOps/s 19.4122 KOps/s $\color{#d91a1a}-0.60\%$
test_compile_assign_and_add[tensordict-compile] 2.0788ms 0.2641ms 3.7867 KOps/s 3.6668 KOps/s $\color{#35bf28}+3.27\%$
test_compile_assign_and_add[tensordict-eager] 3.6043ms 3.3649ms 297.1844 Ops/s 292.4222 Ops/s $\color{#35bf28}+1.63\%$
test_compile_assign_and_add[pytree-compile] 1.9520ms 0.1943ms 5.1471 KOps/s 5.0172 KOps/s $\color{#35bf28}+2.59\%$
test_compile_assign_and_add[pytree-eager] 3.1658ms 2.8174ms 354.9373 Ops/s 349.8956 Ops/s $\color{#35bf28}+1.44\%$
test_compile_indexing[tensor-tensordict-compile] 0.3009ms 0.1570ms 6.3684 KOps/s 6.2775 KOps/s $\color{#35bf28}+1.45\%$
test_compile_indexing[tensor-tensordict-eager] 0.3780ms 75.0447μs 13.3254 KOps/s 9.4743 KOps/s $\textbf{\color{#35bf28}+40.65\%}$
test_compile_indexing[tensor-tensorclass-compile] 0.3115ms 0.1326ms 7.5421 KOps/s 7.4204 KOps/s $\color{#35bf28}+1.64\%$
test_compile_indexing[tensor-tensorclass-eager] 0.3304ms 44.0274μs 22.7131 KOps/s 22.1975 KOps/s $\color{#35bf28}+2.32\%$
test_compile_indexing[tensor-pytree-compile] 0.1867ms 0.1332ms 7.5083 KOps/s 7.2407 KOps/s $\color{#35bf28}+3.70\%$
test_compile_indexing[tensor-pytree-eager] 0.2598ms 43.9976μs 22.7285 KOps/s 22.4385 KOps/s $\color{#35bf28}+1.29\%$
test_compile_indexing[slice-tensordict-compile] 0.2233ms 0.1021ms 9.7909 KOps/s 10.3933 KOps/s $\textbf{\color{#d91a1a}-5.80\%}$
test_compile_indexing[slice-tensordict-eager] 0.2396ms 27.1521μs 36.8296 KOps/s 35.6551 KOps/s $\color{#35bf28}+3.29\%$
test_compile_indexing[slice-tensorclass-compile] 0.1971ms 71.8476μs 13.9183 KOps/s 13.8749 KOps/s $\color{#35bf28}+0.31\%$
test_compile_indexing[slice-tensorclass-eager] 0.2727ms 21.4910μs 46.5312 KOps/s 46.2818 KOps/s $\color{#35bf28}+0.54\%$
test_compile_indexing[slice-pytree-compile] 0.1297ms 73.9366μs 13.5251 KOps/s 14.0048 KOps/s $\color{#d91a1a}-3.43\%$
test_compile_indexing[slice-pytree-eager] 0.2568ms 21.4414μs 46.6388 KOps/s 46.2006 KOps/s $\color{#35bf28}+0.95\%$
test_compile_indexing[int-tensordict-compile] 0.1749ms 92.5906μs 10.8002 KOps/s 10.3462 KOps/s $\color{#35bf28}+4.39\%$
test_compile_indexing[int-tensordict-eager] 0.2326ms 27.0586μs 36.9569 KOps/s 34.9682 KOps/s $\textbf{\color{#35bf28}+5.69\%}$
test_compile_indexing[int-tensorclass-compile] 0.1148ms 71.3746μs 14.0106 KOps/s 13.6401 KOps/s $\color{#35bf28}+2.72\%$
test_compile_indexing[int-tensorclass-eager] 0.2692ms 21.2838μs 46.9840 KOps/s 46.1265 KOps/s $\color{#35bf28}+1.86\%$
test_compile_indexing[int-pytree-compile] 0.1188ms 73.9510μs 13.5225 KOps/s 13.7354 KOps/s $\color{#d91a1a}-1.55\%$
test_compile_indexing[int-pytree-eager] 0.3099ms 21.4335μs 46.6559 KOps/s 46.4814 KOps/s $\color{#35bf28}+0.38\%$
test_compile_replace[single-eager] 96.5710μs 49.8861μs 20.0456 KOps/s 21.3956 KOps/s $\textbf{\color{#d91a1a}-6.31\%}$
test_compile_replace[single-compile] 0.2755ms 0.1522ms 6.5682 KOps/s 6.5384 KOps/s $\color{#35bf28}+0.46\%$
test_compile_replace[multi-eager] 0.7171ms 0.5723ms 1.7474 KOps/s 1.7431 KOps/s $\color{#35bf28}+0.25\%$
test_compile_replace[multi-compile] 0.3674ms 0.1621ms 6.1693 KOps/s 5.9959 KOps/s $\color{#35bf28}+2.89\%$
test_compile_tc_getattr_20[eager] 0.2365ms 0.1739ms 5.7498 KOps/s 5.7435 KOps/s $\color{#35bf28}+0.11\%$
test_compile_tc_getattr_20[compile] 0.3706ms 0.1781ms 5.6162 KOps/s 5.5904 KOps/s $\color{#35bf28}+0.46\%$
test_compile_clone_shallow[20-eager] 55.3200μs 19.1176μs 52.3078 KOps/s 53.4595 KOps/s $\color{#d91a1a}-2.15\%$
test_compile_clone_shallow[20-compile] 97.8710μs 21.4816μs 46.5515 KOps/s 43.1752 KOps/s $\textbf{\color{#35bf28}+7.82\%}$
test_compile_clone_shallow[40-eager] 72.2010μs 33.7648μs 29.6166 KOps/s 29.9986 KOps/s $\color{#d91a1a}-1.27\%$
test_compile_clone_shallow[40-compile] 0.1301ms 24.1360μs 41.4318 KOps/s 38.2996 KOps/s $\textbf{\color{#35bf28}+8.18\%}$
test_compile_clone_shallow[80-eager] 92.4310μs 62.5334μs 15.9914 KOps/s 16.3181 KOps/s $\color{#d91a1a}-2.00\%$
test_compile_clone_shallow[80-compile] 87.4710μs 29.7523μs 33.6109 KOps/s 35.1640 KOps/s $\color{#d91a1a}-4.42\%$
test_compile_update_inplace[eager] 0.2490ms 63.0828μs 15.8522 KOps/s 16.4116 KOps/s $\color{#d91a1a}-3.41\%$
test_compile_update_inplace[compile] 0.5783ms 0.1227ms 8.1525 KOps/s 7.8415 KOps/s $\color{#35bf28}+3.97\%$
test_mod_add[eager] 0.2814ms 0.2219ms 4.5071 KOps/s 4.4409 KOps/s $\color{#35bf28}+1.49\%$
test_mod_add[compile] 0.3423ms 0.1233ms 8.1102 KOps/s 7.8368 KOps/s $\color{#35bf28}+3.49\%$
test_mod_add[compile-overhead] 0.4036ms 0.1994ms 5.0152 KOps/s 4.8693 KOps/s $\color{#35bf28}+3.00\%$
test_mod_wrap[eager] 0.6196ms 0.4746ms 2.1070 KOps/s 2.0703 KOps/s $\color{#35bf28}+1.77\%$
test_mod_wrap[compile] 0.4420ms 0.3482ms 2.8723 KOps/s 2.9023 KOps/s $\color{#d91a1a}-1.03\%$
test_mod_wrap[compile-overhead] 7.2232ms 4.0225ms 248.6015 Ops/s 250.3652 Ops/s $\color{#d91a1a}-0.70\%$
test_mod_wrap_and_backward[eager] 1.9289ms 1.7359ms 576.0710 Ops/s 605.6563 Ops/s $\color{#d91a1a}-4.88\%$
test_mod_wrap_and_backward[compile] 1.8111ms 1.4497ms 689.8165 Ops/s 745.8932 Ops/s $\textbf{\color{#d91a1a}-7.52\%}$
test_mod_wrap_and_backward[compile-overhead] 1.5867ms 0.9528ms 1.0496 KOps/s 1.0404 KOps/s $\color{#35bf28}+0.88\%$
test_seq_add[eager] 0.5872ms 0.4977ms 2.0092 KOps/s 2.0036 KOps/s $\color{#35bf28}+0.28\%$
test_seq_add[compile] 0.3075ms 0.1418ms 7.0516 KOps/s 7.3141 KOps/s $\color{#d91a1a}-3.59\%$
test_seq_add[compile-overhead] 0.3054ms 0.2055ms 4.8657 KOps/s 4.7641 KOps/s $\color{#35bf28}+2.13\%$
test_seq_wrap[eager] 1.4989ms 1.3837ms 722.7058 Ops/s 718.7182 Ops/s $\color{#35bf28}+0.55\%$
test_seq_wrap[compile] 0.4407ms 0.3694ms 2.7069 KOps/s 2.6984 KOps/s $\color{#35bf28}+0.32\%$
test_seq_wrap[compile-overhead] 0.4538ms 0.3266ms 3.0617 KOps/s 3.0442 KOps/s $\color{#35bf28}+0.57\%$
test_func_call_runtime[False-eager] 0.9252ms 0.8326ms 1.2011 KOps/s 1.2578 KOps/s $\color{#d91a1a}-4.50\%$
test_func_call_runtime[False-compile] 0.8472ms 0.7561ms 1.3226 KOps/s 1.3318 KOps/s $\color{#d91a1a}-0.69\%$
test_func_call_runtime[False-compile-overhead] 0.6812ms 0.5026ms 1.9897 KOps/s 1.9729 KOps/s $\color{#35bf28}+0.85\%$
test_func_call_runtime[True-eager] 1.3577ms 1.2669ms 789.3422 Ops/s 956.0196 Ops/s $\textbf{\color{#d91a1a}-17.43\%}$
test_func_call_runtime[True-compile] 0.8733ms 0.7618ms 1.3127 KOps/s 1.3002 KOps/s $\color{#35bf28}+0.96\%$
test_func_call_runtime[True-compile-overhead] 0.6965ms 0.5188ms 1.9276 KOps/s 1.8946 KOps/s $\color{#35bf28}+1.74\%$
test_func_call_cm_runtime[False-eager] 0.8970ms 0.7924ms 1.2619 KOps/s 1.2454 KOps/s $\color{#35bf28}+1.33\%$
test_func_call_cm_runtime[False-compile] 0.8436ms 0.7313ms 1.3674 KOps/s 1.3111 KOps/s $\color{#35bf28}+4.30\%$
test_func_call_cm_runtime[False-compile-overhead] 0.5480ms 0.4822ms 2.0738 KOps/s 2.0231 KOps/s $\color{#35bf28}+2.51\%$
test_func_call_cm_runtime[True-eager] 1.5164ms 1.4171ms 705.6798 Ops/s 832.9526 Ops/s $\textbf{\color{#d91a1a}-15.28\%}$
test_func_call_cm_runtime[True-compile] 0.9380ms 0.8386ms 1.1925 KOps/s 1.1665 KOps/s $\color{#35bf28}+2.23\%$
test_func_call_cm_runtime[True-compile-overhead] 1.0910ms 0.8723ms 1.1463 KOps/s 1.1334 KOps/s $\color{#35bf28}+1.14\%$
test_vmap_func_call_cm_runtime[eager] 3.2106ms 2.6428ms 378.3834 Ops/s 376.9825 Ops/s $\color{#35bf28}+0.37\%$
test_vmap_func_call_cm_runtime[compile] 4.2370ms 3.6928ms 270.7956 Ops/s 274.5264 Ops/s $\color{#d91a1a}-1.36\%$
test_vmap_func_call_cm_runtime[compile-overhead] 4.2610ms 3.7003ms 270.2472 Ops/s 272.9506 Ops/s $\color{#d91a1a}-0.99\%$
test_distributed 0.6764ms 0.1519ms 6.5838 KOps/s 6.3243 KOps/s $\color{#35bf28}+4.10\%$
test_tdmodule 0.5521ms 0.1895ms 5.2768 KOps/s 5.2987 KOps/s $\color{#d91a1a}-0.41\%$
test_tdmodule_dispatch 0.4274ms 0.3660ms 2.7324 KOps/s 2.7233 KOps/s $\color{#35bf28}+0.33\%$
test_tdseq 45.1600μs 25.6830μs 38.9363 KOps/s 36.9275 KOps/s $\textbf{\color{#35bf28}+5.44\%}$
test_tdseq_dispatch 0.2528ms 0.2126ms 4.7031 KOps/s 4.7113 KOps/s $\color{#d91a1a}-0.17\%$
test_instantiation_functorch 2.7380ms 2.6145ms 382.4788 Ops/s 379.6065 Ops/s $\color{#35bf28}+0.76\%$
test_exec_functorch 0.2356ms 0.1920ms 5.2074 KOps/s 5.1141 KOps/s $\color{#35bf28}+1.83\%$
test_exec_functional_call 0.4263ms 0.1485ms 6.7356 KOps/s 6.6166 KOps/s $\color{#35bf28}+1.80\%$
test_exec_td_decorator 0.6292ms 0.3573ms 2.7984 KOps/s 4.1803 KOps/s $\textbf{\color{#d91a1a}-33.06\%}$
test_vmap_mlp_speed_decorator[True-True] 1.3911ms 1.1764ms 850.0747 Ops/s 891.6112 Ops/s $\color{#d91a1a}-4.66\%$
test_vmap_mlp_speed_decorator[True-False] 1.3822ms 1.1746ms 851.3707 Ops/s 895.2362 Ops/s $\color{#d91a1a}-4.90\%$
test_vmap_mlp_speed_decorator[False-True] 1.1271ms 0.8850ms 1.1300 KOps/s 1.1960 KOps/s $\textbf{\color{#d91a1a}-5.52\%}$
test_vmap_mlp_speed_decorator[False-False] 1.1017ms 0.8836ms 1.1317 KOps/s 1.2061 KOps/s $\textbf{\color{#d91a1a}-6.17\%}$
test_vmap_transformer_speed_decorator[True-True] 28.4054ms 26.4587ms 37.7948 Ops/s 40.6459 Ops/s $\textbf{\color{#d91a1a}-7.01\%}$
test_vmap_transformer_speed_decorator[True-False] 27.0730ms 26.3143ms 38.0022 Ops/s 40.8767 Ops/s $\textbf{\color{#d91a1a}-7.03\%}$
test_vmap_transformer_speed_decorator[False-True] 26.7368ms 25.9539ms 38.5299 Ops/s 41.4483 Ops/s $\textbf{\color{#d91a1a}-7.04\%}$
test_vmap_transformer_speed_decorator[False-False] 26.7291ms 25.8419ms 38.6969 Ops/s 41.2792 Ops/s $\textbf{\color{#d91a1a}-6.26\%}$
test_to_module_speed[True] 3.4763ms 3.3386ms 299.5288 Ops/s 623.0403 Ops/s $\textbf{\color{#d91a1a}-51.92\%}$
test_to_module_speed[False] 3.4291ms 3.3173ms 301.4483 Ops/s 636.4996 Ops/s $\textbf{\color{#d91a1a}-52.64\%}$
test_to_module_plain_tensor_speed[None] 3.4194ms 3.3188ms 301.3147 Ops/s 594.5665 Ops/s $\textbf{\color{#d91a1a}-49.32\%}$
test_to_module_plain_tensor_speed[False] 3.1396ms 1.5360ms 651.0559 Ops/s 653.8719 Ops/s $\color{#d91a1a}-0.43\%$
test_to_module_plain_tensor_speed[True] 4.0194ms 3.3187ms 301.3196 Ops/s 303.1213 Ops/s $\color{#d91a1a}-0.59\%$
test_tc_init 88.6610μs 43.5216μs 22.9771 KOps/s 22.7236 KOps/s $\color{#35bf28}+1.12\%$
test_tc_init_tensor_only 31.5400μs 9.2349μs 108.2847 KOps/s 109.3235 KOps/s $\color{#d91a1a}-0.95\%$
test_tc_init_nested 0.1299ms 85.9091μs 11.6402 KOps/s 11.7778 KOps/s $\color{#d91a1a}-1.17\%$
test_tc_init_many_fields 42.0410μs 16.5788μs 60.3179 KOps/s 59.6740 KOps/s $\color{#35bf28}+1.08\%$
test_tc_first_layer_tensor 24.9500μs 1.7203μs 581.3053 KOps/s 581.1000 KOps/s $\color{#35bf28}+0.04\%$
test_tc_first_layer_tensor_only 2.5884μs 0.3914μs 2.5546 MOps/s 2.5863 MOps/s $\color{#d91a1a}-1.22\%$
test_tc_first_layer_tensor_set 24.4310μs 3.9962μs 250.2400 KOps/s 244.5561 KOps/s $\color{#35bf28}+2.32\%$
test_tc_first_layer_tensor_only_set 21.5400μs 3.3136μs 301.7851 KOps/s 296.4636 KOps/s $\color{#35bf28}+1.79\%$
test_tc_first_layer_nontensor 25.1100μs 5.9690μs 167.5322 KOps/s 166.8538 KOps/s $\color{#35bf28}+0.41\%$
test_tc_second_layer_tensor 36.3910μs 4.1517μs 240.8661 KOps/s 236.9650 KOps/s $\color{#35bf28}+1.65\%$
test_tc_second_layer_nontensor 31.6200μs 8.3977μs 119.0799 KOps/s 120.1348 KOps/s $\color{#d91a1a}-0.88\%$
test_unbind 0.2993s 14.4617ms 69.1480 Ops/s 64.5107 Ops/s $\textbf{\color{#35bf28}+7.19\%}$
test_full_like 4.7677ms 4.6374ms 215.6365 Ops/s 215.8005 Ops/s $\color{#d91a1a}-0.08\%$
test_zeros_like 4.7216ms 4.5965ms 217.5564 Ops/s 217.7433 Ops/s $\color{#d91a1a}-0.09\%$
test_ones_like 4.7634ms 4.6175ms 216.5684 Ops/s 216.0236 Ops/s $\color{#35bf28}+0.25\%$
test_clone 6.9355ms 6.7742ms 147.6198 Ops/s 148.0112 Ops/s $\color{#d91a1a}-0.26\%$
test_squeeze 0.1844ms 14.5336μs 68.8061 KOps/s 71.4525 KOps/s $\color{#d91a1a}-3.70\%$
test_unsqueeze 0.1771ms 0.1196ms 8.3631 KOps/s 8.8529 KOps/s $\textbf{\color{#d91a1a}-5.53\%}$
test_split 0.2834ms 0.1958ms 5.1062 KOps/s 5.2511 KOps/s $\color{#d91a1a}-2.76\%$
test_permute 0.2681ms 0.2055ms 4.8674 KOps/s 5.2008 KOps/s $\textbf{\color{#d91a1a}-6.41\%}$
test_stack 55.1660ms 54.7905ms 18.2513 Ops/s 52.6040 Ops/s $\textbf{\color{#d91a1a}-65.30\%}$
test_cat 55.3056ms 54.9970ms 18.1828 Ops/s 52.7542 Ops/s $\textbf{\color{#d91a1a}-65.53\%}$
test_sequential_tensordict 0.9959ms 0.9226ms 1.0839 KOps/s 1.1396 KOps/s $\color{#d91a1a}-4.89\%$
test_sequential_graph_module 0.5299ms 0.4341ms 2.3038 KOps/s 2.3960 KOps/s $\color{#d91a1a}-3.85\%$
test_nested_tensordict 1.3754ms 1.2736ms 785.2025 Ops/s 820.2784 Ops/s $\color{#d91a1a}-4.28\%$
test_nested_graph_module 0.6635ms 0.5488ms 1.8223 KOps/s 1.8784 KOps/s $\color{#d91a1a}-2.98\%$

@vmoens
vmoens marked this pull request as ready for review August 14, 2026 14:41
@vmoens
vmoens merged commit c0c4859 into main Aug 14, 2026
91 of 93 checks passed
@vmoens vmoens mentioned this pull request Aug 14, 2026
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

CI CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. documentation Improvements or additions to documentation nn tensorclass Test versioning

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@vmoens