Skip to content

[Doc] Refresh TensorDict README for 0.13 - #1721

Merged
vmoens merged 1 commit into
mainfrom
codex/readme-refresh-0.13
Jun 4, 2026
Merged

[Doc] Refresh TensorDict README for 0.13#1721
vmoens merged 1 commit into
mainfrom
codex/readme-refresh-0.13

Conversation

@vmoens

@vmoens vmoens commented Jun 4, 2026

Copy link
Copy Markdown
Collaborator

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:

  • lead with TensorDict as a batched, nested dict[str, Tensor] that behaves like a tensor;
  • add a short copy-pastable demo covering slicing, device transfer, arithmetic, and stacking;
  • explain why TensorDict is preferable to a plain dict for structured tensor batches;
  • call out performance-sensitive paths such as TensorDict arithmetic, D2H/H2D transfer, foreach-backed operations, low-allocation workflows, and compile-aware internals;
  • add a concise 0.13 teaser covering tabular import/export, broader inplace=True shape ops, compile fixes, robust memmap keys, and to_module(..., preserve_module_state=...).

Checks

  • git diff --check
  • README internal/ref link sanity script
  • README code fence balance script
  • Demo execution skipped locally because torch is not installed in the active Python environment

@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 Jun 4, 2026
@github-actions github-actions Bot added the documentation Improvements or additions to documentation label Jun 4, 2026
@vmoens
vmoens force-pushed the codex/readme-refresh-0.13 branch from 4d13ac2 to 5506b2a Compare June 4, 2026 14:05
@vmoens
vmoens merged commit c41270e into main Jun 4, 2026
54 of 60 checks passed
@vmoens
vmoens deleted the codex/readme-refresh-0.13 branch June 4, 2026 14:08
@github-actions

github-actions Bot commented Jun 4, 2026

Copy link
Copy Markdown
Contributor

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

Total Benchmarks: 344. Improved: $\large\color{#35bf28}33$. Worsened: $\large\color{#d91a1a}4$.

Expand to view detailed results
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 $\color{#35bf28}+2.68\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.3395ms 0.2859ms 3.4978 KOps/s 3.4459 KOps/s $\color{#35bf28}+1.51\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.8219ms 0.7463ms 1.3400 KOps/s 1.3198 KOps/s $\color{#35bf28}+1.53\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.4740ms 0.3346ms 2.9885 KOps/s 2.9028 KOps/s $\color{#35bf28}+2.95\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.2587ms 1.1388ms 878.1497 Ops/s 924.7807 Ops/s $\textbf{\color{#d91a1a}-5.04\%}$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.5220ms 7.2568ms 137.8020 Ops/s 136.3594 Ops/s $\color{#35bf28}+1.06\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.2798ms 0.2398ms 4.1706 KOps/s 3.9089 KOps/s $\textbf{\color{#35bf28}+6.70\%}$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.4549ms 0.3656ms 2.7354 KOps/s 2.7136 KOps/s $\color{#35bf28}+0.80\%$
test_collector_preallocated_write[stack_out-cuda-32-32] 0.8811ms 0.7943ms 1.2590 KOps/s 1.2541 KOps/s $\color{#35bf28}+0.39\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.3795ms 0.2549ms 3.9232 KOps/s 3.9123 KOps/s $\color{#35bf28}+0.28\%$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.4527ms 0.3716ms 2.6912 KOps/s 2.7444 KOps/s $\color{#d91a1a}-1.94\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 0.9440ms 0.8569ms 1.1670 KOps/s 1.1557 KOps/s $\color{#35bf28}+0.98\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2771ms 0.2167ms 4.6148 KOps/s 4.7030 KOps/s $\color{#d91a1a}-1.88\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 0.9834ms 0.8191ms 1.2208 KOps/s 1.1777 KOps/s $\color{#35bf28}+3.66\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.3627ms 3.2244ms 310.1357 Ops/s 309.9366 Ops/s $\color{#35bf28}+0.06\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.4724ms 0.3658ms 2.7340 KOps/s 2.6745 KOps/s $\color{#35bf28}+2.23\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 1.9219ms 1.7000ms 588.2448 Ops/s 610.0313 Ops/s $\color{#d91a1a}-3.57\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 10.2419ms 10.0494ms 99.5081 Ops/s 98.3733 Ops/s $\color{#35bf28}+1.15\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.4416ms 0.3860ms 2.5906 KOps/s 2.5229 KOps/s $\color{#35bf28}+2.68\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.6021ms 1.4429ms 693.0285 Ops/s 685.6119 Ops/s $\color{#35bf28}+1.08\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 5.7956ms 5.6478ms 177.0607 Ops/s 174.6005 Ops/s $\color{#35bf28}+1.41\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.4696ms 0.3873ms 2.5819 KOps/s 2.5501 KOps/s $\color{#35bf28}+1.25\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.6539ms 1.4505ms 689.4317 Ops/s 682.6471 Ops/s $\color{#35bf28}+0.99\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 5.8813ms 5.6867ms 175.8504 Ops/s 174.5376 Ops/s $\color{#35bf28}+0.75\%$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.6011ms 0.1684ms 5.9383 KOps/s 5.8638 KOps/s $\color{#35bf28}+1.27\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.7303ms 0.6512ms 1.5357 KOps/s 1.5005 KOps/s $\color{#35bf28}+2.34\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 2.7319ms 2.6022ms 384.2858 Ops/s 379.9325 Ops/s $\color{#35bf28}+1.15\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.3983ms 0.3072ms 3.2548 KOps/s 3.2172 KOps/s $\color{#35bf28}+1.17\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.9478ms 1.5034ms 665.1448 Ops/s 681.5286 Ops/s $\color{#d91a1a}-2.40\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 9.7880ms 9.4761ms 105.5284 Ops/s 106.7550 Ops/s $\color{#d91a1a}-1.15\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.4041ms 0.3185ms 3.1397 KOps/s 3.1033 KOps/s $\color{#35bf28}+1.17\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.2927ms 1.1888ms 841.2110 Ops/s 831.6142 Ops/s $\color{#35bf28}+1.15\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.8197ms 4.6564ms 214.7583 Ops/s 211.7915 Ops/s $\color{#35bf28}+1.40\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.4031ms 0.3205ms 3.1197 KOps/s 3.0816 KOps/s $\color{#35bf28}+1.24\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.2938ms 1.1928ms 838.3303 Ops/s 826.2019 Ops/s $\color{#35bf28}+1.47\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 4.8427ms 4.6792ms 213.7130 Ops/s 211.0837 Ops/s $\color{#35bf28}+1.25\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.1483ms 0.1154ms 8.6678 KOps/s 8.6127 KOps/s $\color{#35bf28}+0.64\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.5252ms 0.4520ms 2.2122 KOps/s 2.1907 KOps/s $\color{#35bf28}+0.98\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 1.9097ms 1.7997ms 555.6567 Ops/s 551.4438 Ops/s $\color{#35bf28}+0.76\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.3850ms 0.2435ms 4.1071 KOps/s 4.0181 KOps/s $\color{#35bf28}+2.21\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.6767ms 1.2264ms 815.3690 Ops/s 830.7674 Ops/s $\color{#d91a1a}-1.85\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 8.7775ms 8.5197ms 117.3748 Ops/s 117.4982 Ops/s $\color{#d91a1a}-0.10\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.3359ms 0.2719ms 3.6778 KOps/s 3.6499 KOps/s $\color{#35bf28}+0.77\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.1047ms 1.0129ms 987.3105 Ops/s 978.8340 Ops/s $\color{#35bf28}+0.87\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 4.1183ms 3.9652ms 252.1971 Ops/s 249.6830 Ops/s $\color{#35bf28}+1.01\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.3498ms 0.2715ms 3.6826 KOps/s 3.6427 KOps/s $\color{#35bf28}+1.09\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.0991ms 1.0114ms 988.7048 Ops/s 970.1529 Ops/s $\color{#35bf28}+1.91\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 4.2038ms 3.9752ms 251.5614 Ops/s 249.5549 Ops/s $\color{#35bf28}+0.80\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.2219ms 15.8647ms 63.0330 Ops/s 63.0464 Ops/s $\color{#d91a1a}-0.02\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 31.8754ms 31.7285ms 31.5174 Ops/s 31.3896 Ops/s $\color{#35bf28}+0.41\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 83.4315ms 83.0988ms 12.0339 Ops/s 12.0338 Ops/s $+0.00\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.1986s 0.1984s 5.0403 Ops/s 5.1196 Ops/s $\color{#d91a1a}-1.55\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 13.3360ms 13.1196ms 76.2220 Ops/s 75.6014 Ops/s $\color{#35bf28}+0.82\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 26.1283ms 25.8981ms 38.6129 Ops/s 38.8084 Ops/s $\color{#d91a1a}-0.50\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 13.5276ms 13.3122ms 75.1190 Ops/s 74.6967 Ops/s $\color{#35bf28}+0.57\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 26.5005ms 26.2683ms 38.0687 Ops/s 38.0982 Ops/s $\color{#d91a1a}-0.08\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 51.4995ms 50.9128ms 19.6414 Ops/s 18.4902 Ops/s $\textbf{\color{#35bf28}+6.23\%}$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1030s 0.1022s 9.7885 Ops/s 9.1851 Ops/s $\textbf{\color{#35bf28}+6.57\%}$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1462s 0.1450s 6.8945 Ops/s 7.0536 Ops/s $\color{#d91a1a}-2.26\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3275s 0.3261s 3.0667 Ops/s 3.1685 Ops/s $\color{#d91a1a}-3.21\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 73.4315ms 72.7659ms 13.7427 Ops/s 12.9201 Ops/s $\textbf{\color{#35bf28}+6.37\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1469s 0.1460s 6.8501 Ops/s 6.4340 Ops/s $\textbf{\color{#35bf28}+6.47\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 74.5429ms 73.7783ms 13.5541 Ops/s 12.5990 Ops/s $\textbf{\color{#35bf28}+7.58\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1479s 0.1473s 6.7867 Ops/s 6.4121 Ops/s $\textbf{\color{#35bf28}+5.84\%}$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 30.1226ms 29.7204ms 33.6469 Ops/s 33.5192 Ops/s $\color{#35bf28}+0.38\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 60.7460ms 60.4367ms 16.5462 Ops/s 16.6681 Ops/s $\color{#d91a1a}-0.73\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1202s 0.1198s 8.3466 Ops/s 8.5761 Ops/s $\color{#d91a1a}-2.68\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2640s 0.2633s 3.7976 Ops/s 3.8356 Ops/s $\color{#d91a1a}-0.99\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 42.5801ms 42.4331ms 23.5665 Ops/s 23.2944 Ops/s $\color{#35bf28}+1.17\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 85.8638ms 85.5223ms 11.6929 Ops/s 11.6720 Ops/s $\color{#35bf28}+0.18\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 42.9909ms 42.6562ms 23.4432 Ops/s 23.1630 Ops/s $\color{#35bf28}+1.21\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 86.0841ms 85.8620ms 11.6466 Ops/s 11.5928 Ops/s $\color{#35bf28}+0.46\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 20.0511ms 19.9000ms 50.2512 Ops/s 50.5648 Ops/s $\color{#d91a1a}-0.62\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 41.0881ms 40.7522ms 24.5385 Ops/s 24.6906 Ops/s $\color{#d91a1a}-0.62\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1076s 0.1073s 9.3167 Ops/s 9.5220 Ops/s $\color{#d91a1a}-2.16\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2408s 0.2401s 4.1656 Ops/s 4.2253 Ops/s $\color{#d91a1a}-1.41\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 38.6911ms 37.1047ms 26.9508 Ops/s 27.2872 Ops/s $\color{#d91a1a}-1.23\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 77.5143ms 74.1582ms 13.4847 Ops/s 13.6799 Ops/s $\color{#d91a1a}-1.43\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 38.7022ms 37.0715ms 26.9749 Ops/s 27.1522 Ops/s $\color{#d91a1a}-0.65\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 77.4455ms 75.4323ms 13.2569 Ops/s 13.6536 Ops/s $\color{#d91a1a}-2.91\%$
test_plain_set_nested 46.9130μs 15.0161μs 66.5952 KOps/s 66.7874 KOps/s $\color{#d91a1a}-0.29\%$
test_plain_set_stack_nested 40.2120μs 14.8082μs 67.5304 KOps/s 67.9997 KOps/s $\color{#d91a1a}-0.69\%$
test_plain_set_nested_inplace 45.6920μs 16.3394μs 61.2019 KOps/s 59.9999 KOps/s $\color{#35bf28}+2.00\%$
test_plain_set_stack_nested_inplace 40.6320μs 16.0512μs 62.3007 KOps/s 61.2068 KOps/s $\color{#35bf28}+1.79\%$
test_items 27.0420μs 5.5308μs 180.8047 KOps/s 181.6392 KOps/s $\color{#d91a1a}-0.46\%$
test_items_nested 0.5174ms 0.4539ms 2.2031 KOps/s 2.2144 KOps/s $\color{#d91a1a}-0.51\%$
test_items_nested_locked 0.5219ms 0.4523ms 2.2112 KOps/s 2.2099 KOps/s $\color{#35bf28}+0.06\%$
test_items_nested_leaf 0.1516ms 92.0373μs 10.8652 KOps/s 10.8167 KOps/s $\color{#35bf28}+0.45\%$
test_items_stack_nested 0.5473ms 0.4523ms 2.2110 KOps/s 2.2299 KOps/s $\color{#d91a1a}-0.85\%$
test_items_stack_nested_leaf 0.1492ms 90.8555μs 11.0065 KOps/s 10.8142 KOps/s $\color{#35bf28}+1.78\%$
test_items_stack_nested_locked 0.5750ms 0.4597ms 2.1752 KOps/s 2.2162 KOps/s $\color{#d91a1a}-1.85\%$
test_keys 27.2710μs 4.1262μs 242.3515 KOps/s 240.6518 KOps/s $\color{#35bf28}+0.71\%$
test_keys_nested 0.1723ms 0.1308ms 7.6476 KOps/s 7.6599 KOps/s $\color{#d91a1a}-0.16\%$
test_keys_nested_locked 2.1260ms 0.1395ms 7.1662 KOps/s 7.2723 KOps/s $\color{#d91a1a}-1.46\%$
test_keys_nested_leaf 0.1643ms 0.1216ms 8.2212 KOps/s 8.3180 KOps/s $\color{#d91a1a}-1.16\%$
test_keys_stack_nested 0.1637ms 0.1301ms 7.6870 KOps/s 7.6689 KOps/s $\color{#35bf28}+0.24\%$
test_keys_stack_nested_leaf 0.1681ms 0.1212ms 8.2496 KOps/s 8.3850 KOps/s $\color{#d91a1a}-1.61\%$
test_keys_stack_nested_locked 0.1870ms 0.1383ms 7.2327 KOps/s 7.2340 KOps/s $\color{#d91a1a}-0.02\%$
test_values 7.0064μs 0.9967μs 1.0033 MOps/s 996.5500 KOps/s $\color{#35bf28}+0.68\%$
test_values_nested 80.8850μs 53.4838μs 18.6973 KOps/s 19.1564 KOps/s $\color{#d91a1a}-2.40\%$
test_values_nested_locked 82.8150μs 56.3874μs 17.7345 KOps/s 18.0643 KOps/s $\color{#d91a1a}-1.83\%$
test_values_nested_leaf 89.4860μs 59.9954μs 16.6679 KOps/s 16.7445 KOps/s $\color{#d91a1a}-0.46\%$
test_values_stack_nested 79.2950μs 53.1679μs 18.8083 KOps/s 19.2114 KOps/s $\color{#d91a1a}-2.10\%$
test_values_stack_nested_leaf 88.7050μs 60.1194μs 16.6336 KOps/s 16.7209 KOps/s $\color{#d91a1a}-0.52\%$
test_values_stack_nested_locked 81.3750μs 56.1988μs 17.7940 KOps/s 18.0513 KOps/s $\color{#d91a1a}-1.43\%$
test_membership 4.6337μs 0.7943μs 1.2590 MOps/s 1.2738 MOps/s $\color{#d91a1a}-1.16\%$
test_membership_nested 25.3820μs 2.8082μs 356.1042 KOps/s 355.9579 KOps/s $\color{#35bf28}+0.04\%$
test_membership_nested_leaf 28.6920μs 2.8155μs 355.1754 KOps/s 354.6083 KOps/s $\color{#35bf28}+0.16\%$
test_membership_stacked_nested 34.2420μs 2.7186μs 367.8312 KOps/s 360.2936 KOps/s $\color{#35bf28}+2.09\%$
test_membership_stacked_nested_leaf 36.5530μs 2.7039μs 369.8313 KOps/s 366.7718 KOps/s $\color{#35bf28}+0.83\%$
test_membership_nested_last 20.7620μs 4.2291μs 236.4542 KOps/s 236.9663 KOps/s $\color{#d91a1a}-0.22\%$
test_membership_nested_leaf_last 30.1420μs 4.2292μs 236.4531 KOps/s 238.7855 KOps/s $\color{#d91a1a}-0.98\%$
test_membership_stacked_nested_last 38.0820μs 4.1671μs 239.9770 KOps/s 242.5931 KOps/s $\color{#d91a1a}-1.08\%$
test_membership_stacked_nested_leaf_last 34.1420μs 4.1337μs 241.9141 KOps/s 244.1975 KOps/s $\color{#d91a1a}-0.94\%$
test_nested_getleaf 62.8240μs 20.6284μs 48.4768 KOps/s 48.2951 KOps/s $\color{#35bf28}+0.38\%$
test_nested_get 49.6730μs 19.6491μs 50.8928 KOps/s 51.7701 KOps/s $\color{#d91a1a}-1.69\%$
test_stacked_getleaf 0.1005ms 20.4941μs 48.7946 KOps/s 48.2918 KOps/s $\color{#35bf28}+1.04\%$
test_stacked_get 37.6120μs 19.7017μs 50.7570 KOps/s 50.6074 KOps/s $\color{#35bf28}+0.30\%$
test_nested_getitemleaf 47.8330μs 21.2204μs 47.1245 KOps/s 46.8872 KOps/s $\color{#35bf28}+0.51\%$
test_nested_getitem 56.3040μs 20.2838μs 49.3004 KOps/s 49.3139 KOps/s $\color{#d91a1a}-0.03\%$
test_stacked_getitemleaf 48.1130μs 21.2109μs 47.1455 KOps/s 47.3659 KOps/s $\color{#d91a1a}-0.47\%$
test_stacked_getitem 46.7030μs 20.2079μs 49.4857 KOps/s 49.1738 KOps/s $\color{#35bf28}+0.63\%$
test_lock_nested 0.5553ms 0.4777ms 2.0933 KOps/s 2.0507 KOps/s $\color{#35bf28}+2.08\%$
test_lock_stack_nested 0.5776ms 0.4844ms 2.0646 KOps/s 2.0623 KOps/s $\color{#35bf28}+0.11\%$
test_unlock_nested 0.4991ms 0.3802ms 2.6302 KOps/s 2.6232 KOps/s $\color{#35bf28}+0.27\%$
test_unlock_stack_nested 0.4682ms 0.3820ms 2.6176 KOps/s 2.5977 KOps/s $\color{#35bf28}+0.77\%$
test_flatten_speed 0.1905ms 0.1155ms 8.6573 KOps/s 8.6271 KOps/s $\color{#35bf28}+0.35\%$
test_unflatten_speed 0.6187ms 0.5500ms 1.8180 KOps/s 1.8078 KOps/s $\color{#35bf28}+0.56\%$
test_common_ops 0.1000s 1.6274ms 614.4859 Ops/s 263.2751 Ops/s $\textbf{\color{#35bf28}+133.40\%}$
test_creation 0.1261ms 2.9968μs 333.6887 KOps/s 336.8858 KOps/s $\color{#d91a1a}-0.95\%$
test_creation_empty 34.6330μs 6.5858μs 151.8426 KOps/s 151.3372 KOps/s $\color{#35bf28}+0.33\%$
test_creation_nested_1 30.1310μs 10.8575μs 92.1026 KOps/s 91.3834 KOps/s $\color{#35bf28}+0.79\%$
test_creation_nested_2 38.5620μs 12.9027μs 77.5030 KOps/s 77.2844 KOps/s $\color{#35bf28}+0.28\%$
test_creation_many_keys[10] 48.5730μs 19.6618μs 50.8601 KOps/s 50.9966 KOps/s $\color{#d91a1a}-0.27\%$
test_creation_many_keys[50] 0.1133ms 84.9659μs 11.7694 KOps/s 11.7746 KOps/s $\color{#d91a1a}-0.04\%$
test_creation_many_keys[100] 0.2136ms 0.1658ms 6.0330 KOps/s 6.0368 KOps/s $\color{#d91a1a}-0.06\%$
test_creation_nested_many_keys[10] 65.0040μs 42.5515μs 23.5009 KOps/s 23.5484 KOps/s $\color{#d91a1a}-0.20\%$
test_creation_nested_many_keys[50] 0.2205ms 0.1732ms 5.7743 KOps/s 5.7556 KOps/s $\color{#35bf28}+0.32\%$
test_clone 40.5430μs 12.9003μs 77.5179 KOps/s 77.4721 KOps/s $\color{#35bf28}+0.06\%$
test_getitem[int] 1.4976ms 14.5046μs 68.9437 KOps/s 68.6267 KOps/s $\color{#35bf28}+0.46\%$
test_getitem[slice_int] 0.1343ms 24.0929μs 41.5060 KOps/s 41.9667 KOps/s $\color{#d91a1a}-1.10\%$
test_getitem[range] 0.1651ms 61.1990μs 16.3401 KOps/s 15.9732 KOps/s $\color{#35bf28}+2.30\%$
test_getitem[tuple] 0.1458ms 23.2254μs 43.0563 KOps/s 41.9413 KOps/s $\color{#35bf28}+2.66\%$
test_getitem[list] 0.1769ms 57.8320μs 17.2915 KOps/s 17.2905 KOps/s $+0.01\%$
test_setitem_dim[int] 44.1830μs 24.5020μs 40.8131 KOps/s 39.2894 KOps/s $\color{#35bf28}+3.88\%$
test_setitem_dim[slice_int] 60.5430μs 40.8677μs 24.4692 KOps/s 23.9854 KOps/s $\color{#35bf28}+2.02\%$
test_setitem_dim[range] 0.1168ms 91.7437μs 10.8999 KOps/s 10.7671 KOps/s $\color{#35bf28}+1.23\%$
test_setitem_dim[tuple] 61.2840μs 38.6052μs 25.9032 KOps/s 25.1706 KOps/s $\color{#35bf28}+2.91\%$
test_setitem 46.3330μs 17.2590μs 57.9408 KOps/s 57.7035 KOps/s $\color{#35bf28}+0.41\%$
test_set 50.3030μs 16.6155μs 60.1846 KOps/s 60.4581 KOps/s $\color{#d91a1a}-0.45\%$
test_set_shared 0.6354ms 0.2069ms 4.8339 KOps/s 4.8311 KOps/s $\color{#35bf28}+0.06\%$
test_update 63.5140μs 21.1954μs 47.1799 KOps/s 46.7612 KOps/s $\color{#35bf28}+0.90\%$
test_update_nested 65.2140μs 32.5921μs 30.6823 KOps/s 30.1190 KOps/s $\color{#35bf28}+1.87\%$
test_update__nested 0.4563ms 32.3190μs 30.9415 KOps/s 30.8616 KOps/s $\color{#35bf28}+0.26\%$
test_set_nested 49.2530μs 18.4127μs 54.3105 KOps/s 53.7419 KOps/s $\color{#35bf28}+1.06\%$
test_set_nested_new 49.7830μs 23.1736μs 43.1526 KOps/s 42.5983 KOps/s $\color{#35bf28}+1.30\%$
test_select 70.9550μs 39.0347μs 25.6182 KOps/s 25.2217 KOps/s $\color{#35bf28}+1.57\%$
test_select_nested 92.8660μs 70.9945μs 14.0856 KOps/s 14.1278 KOps/s $\color{#d91a1a}-0.30\%$
test_exclude_nested 0.1239ms 87.3740μs 11.4451 KOps/s 11.5829 KOps/s $\color{#d91a1a}-1.19\%$
test_empty[True] 0.5243ms 0.3905ms 2.5609 KOps/s 2.5840 KOps/s $\color{#d91a1a}-0.89\%$
test_empty[False] 6.7978μs 1.3148μs 760.5708 KOps/s 759.0103 KOps/s $\color{#35bf28}+0.21\%$
test_to 0.1334ms 67.6549μs 14.7809 KOps/s 14.8064 KOps/s $\color{#d91a1a}-0.17\%$
test_to_nonblocking 0.1063ms 67.1710μs 14.8874 KOps/s 14.5779 KOps/s $\color{#35bf28}+2.12\%$
test_unbind_speed 0.3602ms 0.3227ms 3.0985 KOps/s 3.0859 KOps/s $\color{#35bf28}+0.41\%$
test_unbind_speed_stack0 0.3648ms 0.3217ms 3.1082 KOps/s 3.1524 KOps/s $\color{#d91a1a}-1.40\%$
test_unbind_speed_stack1 0.1157s 0.8278ms 1.2080 KOps/s 938.4833 Ops/s $\textbf{\color{#35bf28}+28.72\%}$
test_split 0.1150s 1.2251ms 816.2918 Ops/s 919.5146 Ops/s $\textbf{\color{#d91a1a}-11.23\%}$
test_chunk 1.1036ms 1.0491ms 953.2223 Ops/s 953.2012 Ops/s $+0.00\%$
test_to_cpu_blocking 20.0983ms 19.9823ms 50.0442 Ops/s 49.9143 Ops/s $\color{#35bf28}+0.26\%$
test_to_cpu_global_sync 11.8096ms 11.6700ms 85.6899 Ops/s 75.5154 Ops/s $\textbf{\color{#35bf28}+13.47\%}$
test_to_cpu_event_sync 12.9321ms 12.6511ms 79.0442 Ops/s 78.2504 Ops/s $\color{#35bf28}+1.01\%$
test_to_cpu_default 12.9124ms 12.6404ms 79.1117 Ops/s 78.3228 Ops/s $\color{#35bf28}+1.01\%$
test_consolidate[False-None] 4.2680ms 4.0314ms 248.0547 Ops/s 246.1325 Ops/s $\color{#35bf28}+0.78\%$
test_consolidate[default-None] 3.4874ms 2.0902ms 478.4278 Ops/s 466.7460 Ops/s $\color{#35bf28}+2.50\%$
test_consolidate[reduce-overhead-None] 3.2942ms 1.9993ms 500.1875 Ops/s 479.1723 Ops/s $\color{#35bf28}+4.39\%$
test_consolidate_njt[False-None] 9.1928ms 8.3443ms 119.8424 Ops/s 119.6906 Ops/s $\color{#35bf28}+0.13\%$
test_to[False-False-None] 2.1852ms 2.1335ms 468.7061 Ops/s 465.5011 Ops/s $\color{#35bf28}+0.69\%$
test_to[True-False-None] 2.1144ms 1.8334ms 545.4389 Ops/s 531.5536 Ops/s $\color{#35bf28}+2.61\%$
test_to[within-False-None] 6.0907ms 5.9704ms 167.4922 Ops/s 164.7493 Ops/s $\color{#35bf28}+1.66\%$
test_to[True-default-None] 10.8504ms 10.4951ms 95.2829 Ops/s 94.2831 Ops/s $\color{#35bf28}+1.06\%$
test_to_njt[False-False-None] 8.7352ms 8.4070ms 118.9488 Ops/s 117.6245 Ops/s $\color{#35bf28}+1.13\%$
test_to_njt[True-False-None] 7.0226ms 6.7877ms 147.3244 Ops/s 148.8487 Ops/s $\color{#d91a1a}-1.02\%$
test_to_njt[within-False-None] 15.4016ms 14.9790ms 66.7600 Ops/s 62.3933 Ops/s $\textbf{\color{#35bf28}+7.00\%}$
test_creation[device0] 0.4016ms 0.1196ms 8.3625 KOps/s 8.2628 KOps/s $\color{#35bf28}+1.21\%$
test_creation_from_tensor 0.5239ms 0.1166ms 8.5766 KOps/s 8.4812 KOps/s $\color{#35bf28}+1.12\%$
test_add_one[memmap_tensor0] 83.2050μs 6.4903μs 154.0768 KOps/s 152.3362 KOps/s $\color{#35bf28}+1.14\%$
test_contiguous[memmap_tensor0] 25.9820μs 0.5928μs 1.6870 MOps/s 2.2991 MOps/s $\textbf{\color{#d91a1a}-26.62\%}$
test_stack[memmap_tensor0] 26.9920μs 3.9972μs 250.1767 KOps/s 246.1269 KOps/s $\color{#35bf28}+1.65\%$
test_memmaptd_index 1.1431ms 0.2655ms 3.7669 KOps/s 3.7096 KOps/s $\color{#35bf28}+1.55\%$
test_memmaptd_index_astensor 0.5185ms 0.3654ms 2.7369 KOps/s 2.7173 KOps/s $\color{#35bf28}+0.72\%$
test_memmaptd_index_op 0.7933ms 0.6140ms 1.6286 KOps/s 1.5280 KOps/s $\textbf{\color{#35bf28}+6.58\%}$
test_serialize_model 0.1344s 0.1328s 7.5321 Ops/s 7.3876 Ops/s $\color{#35bf28}+1.96\%$
test_serialize_model_pickle 1.3471s 1.2097s 0.8267 Ops/s 0.8221 Ops/s $\color{#35bf28}+0.56\%$
test_serialize_weights 0.3168s 0.1591s 6.2835 Ops/s 7.4592 Ops/s $\textbf{\color{#d91a1a}-15.76\%}$
test_serialize_weights_returnearly 84.6801ms 63.7906ms 15.6763 Ops/s 11.0849 Ops/s $\textbf{\color{#35bf28}+41.42\%}$
test_serialize_weights_pickle 1.3693s 1.2131s 0.8243 Ops/s 0.8244 Ops/s $-0.01\%$
test_reshape_pytree 0.2027ms 31.2605μs 31.9893 KOps/s 31.9769 KOps/s $\color{#35bf28}+0.04\%$
test_reshape_td 87.4460μs 43.5612μs 22.9562 KOps/s 22.9969 KOps/s $\color{#d91a1a}-0.18\%$
test_view_pytree 0.2234ms 30.7498μs 32.5205 KOps/s 32.6981 KOps/s $\color{#d91a1a}-0.54\%$
test_view_td 94.5250μs 51.4983μs 19.4181 KOps/s 19.8461 KOps/s $\color{#d91a1a}-2.16\%$
test_unbind_pytree 0.2516ms 36.0494μs 27.7397 KOps/s 28.4268 KOps/s $\color{#d91a1a}-2.42\%$
test_unbind_td 0.1453ms 49.0661μs 20.3807 KOps/s 20.6126 KOps/s $\color{#d91a1a}-1.13\%$
test_split_pytree 0.2442ms 40.8931μs 24.4540 KOps/s 24.4135 KOps/s $\color{#35bf28}+0.17\%$
test_split_td 0.1051ms 61.8975μs 16.1557 KOps/s 16.4622 KOps/s $\color{#d91a1a}-1.86\%$
test_add_pytree 0.2279ms 42.1837μs 23.7059 KOps/s 24.7350 KOps/s $\color{#d91a1a}-4.16\%$
test_add_td 96.0460μs 53.6143μs 18.6517 KOps/s 18.7667 KOps/s $\color{#d91a1a}-0.61\%$
test_compile_add_one_nested[tensordict-compile] 0.5120ms 0.2177ms 4.5940 KOps/s 4.4415 KOps/s $\color{#35bf28}+3.43\%$
test_compile_add_one_nested[tensordict-eager] 20.2780ms 0.2038ms 4.9070 KOps/s 5.1507 KOps/s $\color{#d91a1a}-4.73\%$
test_compile_add_one_nested[pytree-compile] 0.3634ms 0.1299ms 7.7002 KOps/s 7.4694 KOps/s $\color{#35bf28}+3.09\%$
test_compile_add_one_nested[pytree-eager] 0.4010ms 0.1724ms 5.8016 KOps/s 5.4932 KOps/s $\textbf{\color{#35bf28}+5.61\%}$
test_compile_copy_nested[tensordict-compile] 0.4725ms 15.1051μs 66.2027 KOps/s 62.9153 KOps/s $\textbf{\color{#35bf28}+5.23\%}$
test_compile_copy_nested[tensordict-eager] 0.1016ms 51.0739μs 19.5795 KOps/s 19.4632 KOps/s $\color{#35bf28}+0.60\%$
test_compile_copy_nested[pytree-compile] 0.1156ms 15.1252μs 66.1148 KOps/s 60.6544 KOps/s $\textbf{\color{#35bf28}+9.00\%}$
test_compile_copy_nested[pytree-eager] 0.3630ms 65.0964μs 15.3618 KOps/s 15.6225 KOps/s $\color{#d91a1a}-1.67\%$
test_compile_add_one_flat[tensordict-compile] 0.4295ms 0.2402ms 4.1633 KOps/s 3.9094 KOps/s $\textbf{\color{#35bf28}+6.49\%}$
test_compile_add_one_flat[tensordict-eager] 0.4312ms 0.2778ms 3.6003 KOps/s 3.5561 KOps/s $\color{#35bf28}+1.24\%$
test_compile_add_one_flat[tensorclass-compile] 0.2577ms 0.1779ms 5.6196 KOps/s 5.2840 KOps/s $\textbf{\color{#35bf28}+6.35\%}$
test_compile_add_one_flat[tensorclass-eager] 0.1478ms 74.3978μs 13.4413 KOps/s 13.5213 KOps/s $\color{#d91a1a}-0.59\%$
test_compile_add_one_flat[pytree-compile] 0.4434ms 0.1860ms 5.3750 KOps/s 5.1560 KOps/s $\color{#35bf28}+4.25\%$
test_compile_add_one_flat[pytree-eager] 0.7953ms 0.5103ms 1.9595 KOps/s 1.9552 KOps/s $\color{#35bf28}+0.22\%$
test_compile_add_self_flat[tensordict-eager] 0.4176ms 0.3240ms 3.0869 KOps/s 3.0391 KOps/s $\color{#35bf28}+1.57\%$
test_compile_add_self_flat[tensordict-compile] 0.3856ms 0.2421ms 4.1313 KOps/s 3.8574 KOps/s $\textbf{\color{#35bf28}+7.10\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1657ms 88.9331μs 11.2444 KOps/s 11.2018 KOps/s $\color{#35bf28}+0.38\%$
test_compile_add_self_flat[tensorclass-compile] 0.4000ms 0.1813ms 5.5159 KOps/s 5.2308 KOps/s $\textbf{\color{#35bf28}+5.45\%}$
test_compile_add_self_flat[pytree-eager] 0.6400ms 0.4203ms 2.3790 KOps/s 2.3541 KOps/s $\color{#35bf28}+1.06\%$
test_compile_add_self_flat[pytree-compile] 0.3283ms 0.1868ms 5.3532 KOps/s 5.1868 KOps/s $\color{#35bf28}+3.21\%$
test_compile_copy_flat[tensordict-compile] 70.5250μs 18.9154μs 52.8670 KOps/s 52.2807 KOps/s $\color{#35bf28}+1.12\%$
test_compile_copy_flat[tensordict-eager] 75.5150μs 40.1018μs 24.9365 KOps/s 24.8842 KOps/s $\color{#35bf28}+0.21\%$
test_compile_copy_flat[pytree-compile] 0.2713ms 20.9984μs 47.6227 KOps/s 45.8088 KOps/s $\color{#35bf28}+3.96\%$
test_compile_copy_flat[pytree-eager] 0.3331ms 51.0412μs 19.5920 KOps/s 19.7169 KOps/s $\color{#d91a1a}-0.63\%$
test_compile_assign_and_add[tensordict-compile] 2.1796ms 0.2337ms 4.2782 KOps/s 4.0092 KOps/s $\textbf{\color{#35bf28}+6.71\%}$
test_compile_assign_and_add[tensordict-eager] 3.3863ms 3.1563ms 316.8276 Ops/s 309.2371 Ops/s $\color{#35bf28}+2.45\%$
test_compile_assign_and_add[pytree-compile] 2.1195ms 0.1752ms 5.7082 KOps/s 5.5495 KOps/s $\color{#35bf28}+2.86\%$
test_compile_assign_and_add[pytree-eager] 2.8781ms 2.7114ms 368.8136 Ops/s 353.7132 Ops/s $\color{#35bf28}+4.27\%$
test_compile_indexing[tensor-tensordict-compile] 0.2001ms 0.1354ms 7.3878 KOps/s 7.4229 KOps/s $\color{#d91a1a}-0.47\%$
test_compile_indexing[tensor-tensordict-eager] 0.3153ms 73.4934μs 13.6067 KOps/s 13.8588 KOps/s $\color{#d91a1a}-1.82\%$
test_compile_indexing[tensor-tensorclass-compile] 0.2478ms 0.1142ms 8.7565 KOps/s 8.5988 KOps/s $\color{#35bf28}+1.83\%$
test_compile_indexing[tensor-tensorclass-eager] 0.2655ms 43.2392μs 23.1272 KOps/s 22.1624 KOps/s $\color{#35bf28}+4.35\%$
test_compile_indexing[tensor-pytree-compile] 0.1735ms 0.1148ms 8.7093 KOps/s 8.3816 KOps/s $\color{#35bf28}+3.91\%$
test_compile_indexing[tensor-pytree-eager] 0.2930ms 43.0940μs 23.2051 KOps/s 21.3950 KOps/s $\textbf{\color{#35bf28}+8.46\%}$
test_compile_indexing[slice-tensordict-compile] 0.2124ms 80.7380μs 12.3857 KOps/s 11.5430 KOps/s $\textbf{\color{#35bf28}+7.30\%}$
test_compile_indexing[slice-tensordict-eager] 0.2023ms 27.4686μs 36.4052 KOps/s 37.5417 KOps/s $\color{#d91a1a}-3.03\%$
test_compile_indexing[slice-tensorclass-compile] 0.1698ms 68.0611μs 14.6927 KOps/s 14.2312 KOps/s $\color{#35bf28}+3.24\%$
test_compile_indexing[slice-tensorclass-eager] 0.2326ms 21.2366μs 47.0885 KOps/s 47.4482 KOps/s $\color{#d91a1a}-0.76\%$
test_compile_indexing[slice-pytree-compile] 0.1105ms 68.5094μs 14.5965 KOps/s 14.4869 KOps/s $\color{#35bf28}+0.76\%$
test_compile_indexing[slice-pytree-eager] 0.3262ms 21.2647μs 47.0264 KOps/s 47.6816 KOps/s $\color{#d91a1a}-1.37\%$
test_compile_indexing[int-tensordict-compile] 0.1303ms 81.6288μs 12.2506 KOps/s 11.8530 KOps/s $\color{#35bf28}+3.35\%$
test_compile_indexing[int-tensordict-eager] 0.2892ms 26.9838μs 37.0593 KOps/s 37.7645 KOps/s $\color{#d91a1a}-1.87\%$
test_compile_indexing[int-tensorclass-compile] 0.1154ms 68.4273μs 14.6141 KOps/s 14.3765 KOps/s $\color{#35bf28}+1.65\%$
test_compile_indexing[int-tensorclass-eager] 0.2438ms 21.1081μs 47.3752 KOps/s 47.4326 KOps/s $\color{#d91a1a}-0.12\%$
test_compile_indexing[int-pytree-compile] 0.1160ms 69.9040μs 14.3053 KOps/s 14.5644 KOps/s $\color{#d91a1a}-1.78\%$
test_compile_indexing[int-pytree-eager] 0.2335ms 21.3236μs 46.8965 KOps/s 48.1483 KOps/s $\color{#d91a1a}-2.60\%$
test_compile_replace[single-eager] 80.2250μs 46.2285μs 21.6317 KOps/s 20.6489 KOps/s $\color{#35bf28}+4.76\%$
test_compile_replace[single-compile] 0.2351ms 0.1252ms 7.9892 KOps/s 7.4134 KOps/s $\textbf{\color{#35bf28}+7.77\%}$
test_compile_replace[multi-eager] 0.5990ms 0.5545ms 1.8035 KOps/s 1.7047 KOps/s $\textbf{\color{#35bf28}+5.80\%}$
test_compile_replace[multi-compile] 0.2965ms 0.1339ms 7.4659 KOps/s 7.4082 KOps/s $\color{#35bf28}+0.78\%$
test_compile_tc_getattr_20[eager] 0.2008ms 0.1648ms 6.0671 KOps/s 6.1692 KOps/s $\color{#d91a1a}-1.66\%$
test_compile_tc_getattr_20[compile] 0.3730ms 0.1426ms 7.0119 KOps/s 6.8545 KOps/s $\color{#35bf28}+2.30\%$
test_compile_clone_shallow[20-eager] 52.7730μs 18.7133μs 53.4380 KOps/s 53.5840 KOps/s $\color{#d91a1a}-0.27\%$
test_compile_clone_shallow[20-compile] 87.8160μs 16.4097μs 60.9397 KOps/s 52.0278 KOps/s $\textbf{\color{#35bf28}+17.13\%}$
test_compile_clone_shallow[40-eager] 61.6240μs 33.1284μs 30.1856 KOps/s 30.9239 KOps/s $\color{#d91a1a}-2.39\%$
test_compile_clone_shallow[40-compile] 74.0040μs 18.1328μs 55.1487 KOps/s 55.4028 KOps/s $\color{#d91a1a}-0.46\%$
test_compile_clone_shallow[80-eager] 96.9560μs 61.2581μs 16.3244 KOps/s 16.4621 KOps/s $\color{#d91a1a}-0.84\%$
test_compile_clone_shallow[80-compile] 77.5250μs 21.2898μs 46.9708 KOps/s 45.7041 KOps/s $\color{#35bf28}+2.77\%$
test_compile_update_inplace[eager] 0.2084ms 59.3408μs 16.8518 KOps/s 16.5617 KOps/s $\color{#35bf28}+1.75\%$
test_compile_update_inplace[compile] 0.3464ms 0.1404ms 7.1233 KOps/s 6.8831 KOps/s $\color{#35bf28}+3.49\%$
test_mod_add[eager] 0.2694ms 0.2209ms 4.5259 KOps/s 4.5339 KOps/s $\color{#d91a1a}-0.18\%$
test_mod_add[compile] 0.2068ms 0.1278ms 7.8229 KOps/s 7.4355 KOps/s $\textbf{\color{#35bf28}+5.21\%}$
test_mod_add[compile-overhead] 0.3966ms 0.1738ms 5.7535 KOps/s 5.6496 KOps/s $\color{#35bf28}+1.84\%$
test_mod_wrap[eager] 0.5363ms 0.4557ms 2.1944 KOps/s 2.0948 KOps/s $\color{#35bf28}+4.75\%$
test_mod_wrap[compile] 0.4702ms 0.3830ms 2.6107 KOps/s 2.5318 KOps/s $\color{#35bf28}+3.12\%$
test_mod_wrap[compile-overhead] 8.5223ms 4.6518ms 214.9718 Ops/s 215.9954 Ops/s $\color{#d91a1a}-0.47\%$
test_mod_wrap_and_backward[eager] 1.8123ms 1.6541ms 604.5619 Ops/s 592.6719 Ops/s $\color{#35bf28}+2.01\%$
test_mod_wrap_and_backward[compile] 2.5389ms 1.5716ms 636.2810 Ops/s 654.2532 Ops/s $\color{#d91a1a}-2.75\%$
test_mod_wrap_and_backward[compile-overhead] 1.2963ms 0.9262ms 1.0797 KOps/s 969.9680 Ops/s $\textbf{\color{#35bf28}+11.31\%}$
test_seq_add[eager] 0.6211ms 0.5155ms 1.9398 KOps/s 2.0121 KOps/s $\color{#d91a1a}-3.59\%$
test_seq_add[compile] 0.2855ms 0.1363ms 7.3358 KOps/s 6.7121 KOps/s $\textbf{\color{#35bf28}+9.29\%}$
test_seq_add[compile-overhead] 0.3132ms 0.1810ms 5.5247 KOps/s 5.4664 KOps/s $\color{#35bf28}+1.07\%$
test_seq_wrap[eager] 1.4635ms 1.3738ms 727.9186 Ops/s 721.8508 Ops/s $\color{#35bf28}+0.84\%$
test_seq_wrap[compile] 0.4789ms 0.4065ms 2.4600 KOps/s 2.4452 KOps/s $\color{#35bf28}+0.60\%$
test_seq_wrap[compile-overhead] 0.3762ms 0.2958ms 3.3807 KOps/s 3.3170 KOps/s $\color{#35bf28}+1.92\%$
test_func_call_runtime[False-eager] 0.8559ms 0.7642ms 1.3086 KOps/s 1.2697 KOps/s $\color{#35bf28}+3.07\%$
test_func_call_runtime[False-compile] 1.1916ms 0.9529ms 1.0494 KOps/s 1.0406 KOps/s $\color{#35bf28}+0.84\%$
test_func_call_runtime[False-compile-overhead] 0.5482ms 0.4917ms 2.0338 KOps/s 2.0143 KOps/s $\color{#35bf28}+0.97\%$
test_func_call_runtime[True-eager] 1.1516ms 1.0204ms 980.0339 Ops/s 959.4558 Ops/s $\color{#35bf28}+2.14\%$
test_func_call_runtime[True-compile] 1.0725ms 0.9672ms 1.0339 KOps/s 1.0247 KOps/s $\color{#35bf28}+0.89\%$
test_func_call_runtime[True-compile-overhead] 0.5687ms 0.5067ms 1.9736 KOps/s 1.9515 KOps/s $\color{#35bf28}+1.13\%$
test_func_call_cm_runtime[False-eager] 0.8796ms 0.7956ms 1.2569 KOps/s 1.2750 KOps/s $\color{#d91a1a}-1.42\%$
test_func_call_cm_runtime[False-compile] 1.0924ms 0.9453ms 1.0579 KOps/s 1.0580 KOps/s $\color{#d91a1a}-0.01\%$
test_func_call_cm_runtime[False-compile-overhead] 0.6283ms 0.4728ms 2.1148 KOps/s 2.1065 KOps/s $\color{#35bf28}+0.40\%$
test_func_call_cm_runtime[True-eager] 1.3016ms 1.1667ms 857.0903 Ops/s 839.4947 Ops/s $\color{#35bf28}+2.10\%$
test_func_call_cm_runtime[True-compile] 1.1484ms 1.0094ms 990.6633 Ops/s 996.7503 Ops/s $\color{#d91a1a}-0.61\%$
test_func_call_cm_runtime[True-compile-overhead] 0.6873ms 0.5315ms 1.8815 KOps/s 1.8602 KOps/s $\color{#35bf28}+1.15\%$
test_vmap_func_call_cm_runtime[eager] 2.8457ms 2.3673ms 422.4147 Ops/s 415.2454 Ops/s $\color{#35bf28}+1.73\%$
test_vmap_func_call_cm_runtime[compile] 3.1312ms 2.6048ms 383.9109 Ops/s 383.4665 Ops/s $\color{#35bf28}+0.12\%$
test_vmap_func_call_cm_runtime[compile-overhead] 3.0315ms 2.5603ms 390.5799 Ops/s 383.4868 Ops/s $\color{#35bf28}+1.85\%$
test_distributed 1.6019ms 0.1571ms 6.3661 KOps/s 6.6817 KOps/s $\color{#d91a1a}-4.72\%$
test_tdmodule 0.3720ms 0.1869ms 5.3515 KOps/s 5.1840 KOps/s $\color{#35bf28}+3.23\%$
test_tdmodule_dispatch 0.4079ms 0.3605ms 2.7741 KOps/s 2.7551 KOps/s $\color{#35bf28}+0.69\%$
test_tdseq 44.6630μs 25.7389μs 38.8516 KOps/s 38.3656 KOps/s $\color{#35bf28}+1.27\%$
test_tdseq_dispatch 0.2514ms 0.2111ms 4.7365 KOps/s 4.7353 KOps/s $\color{#35bf28}+0.03\%$
test_instantiation_functorch 2.1011ms 1.9585ms 510.6051 Ops/s 506.2499 Ops/s $\color{#35bf28}+0.86\%$
test_exec_functorch 0.2167ms 0.1751ms 5.7115 KOps/s 5.6369 KOps/s $\color{#35bf28}+1.32\%$
test_exec_functional_call 0.2158ms 0.1569ms 6.3726 KOps/s 6.3553 KOps/s $\color{#35bf28}+0.27\%$
test_exec_td_decorator 0.4835ms 0.2433ms 4.1098 KOps/s 4.0601 KOps/s $\color{#35bf28}+1.22\%$
test_vmap_mlp_speed_decorator[True-True] 1.1759ms 1.0143ms 985.8754 Ops/s 967.4665 Ops/s $\color{#35bf28}+1.90\%$
test_vmap_mlp_speed_decorator[True-False] 1.2025ms 1.0100ms 990.1131 Ops/s 974.1977 Ops/s $\color{#35bf28}+1.63\%$
test_vmap_mlp_speed_decorator[False-True] 0.9364ms 0.7208ms 1.3873 KOps/s 1.3324 KOps/s $\color{#35bf28}+4.12\%$
test_vmap_mlp_speed_decorator[False-False] 0.9124ms 0.7203ms 1.3883 KOps/s 1.3612 KOps/s $\color{#35bf28}+1.99\%$
test_vmap_transformer_speed_decorator[True-True] 21.7042ms 20.7863ms 48.1086 Ops/s 47.4200 Ops/s $\color{#35bf28}+1.45\%$
test_vmap_transformer_speed_decorator[True-False] 21.2133ms 20.7226ms 48.2564 Ops/s 47.4414 Ops/s $\color{#35bf28}+1.72\%$
test_vmap_transformer_speed_decorator[False-True] 21.0579ms 20.3601ms 49.1157 Ops/s 48.3616 Ops/s $\color{#35bf28}+1.56\%$
test_vmap_transformer_speed_decorator[False-False] 21.0224ms 20.3860ms 49.0532 Ops/s 48.2600 Ops/s $\color{#35bf28}+1.64\%$
test_to_module_speed[True] 1.6834ms 1.5770ms 634.1172 Ops/s 629.7749 Ops/s $\color{#35bf28}+0.69\%$
test_to_module_speed[False] 1.9512ms 1.5551ms 643.0476 Ops/s 635.0288 Ops/s $\color{#35bf28}+1.26\%$
test_to_module_plain_tensor_speed[None] 2.1689ms 1.6753ms 596.9216 Ops/s 589.3783 Ops/s $\color{#35bf28}+1.28\%$
test_to_module_plain_tensor_speed[False] 1.9845ms 1.5301ms 653.5674 Ops/s 650.2576 Ops/s $\color{#35bf28}+0.51\%$
test_to_module_plain_tensor_speed[True] 3.3560ms 3.2627ms 306.4931 Ops/s 306.7955 Ops/s $\color{#d91a1a}-0.10\%$
test_tc_init 76.1750μs 41.8190μs 23.9126 KOps/s 24.3410 KOps/s $\color{#d91a1a}-1.76\%$
test_tc_init_tensor_only 80.6850μs 8.7043μs 114.8859 KOps/s 116.4978 KOps/s $\color{#d91a1a}-1.38\%$
test_tc_init_nested 0.1237ms 82.1726μs 12.1695 KOps/s 12.3623 KOps/s $\color{#d91a1a}-1.56\%$
test_tc_init_many_fields 54.9940μs 14.4058μs 69.4167 KOps/s 71.4443 KOps/s $\color{#d91a1a}-2.84\%$
test_tc_first_layer_tensor 29.6810μs 1.6935μs 590.4989 KOps/s 586.9932 KOps/s $\color{#35bf28}+0.60\%$
test_tc_first_layer_tensor_only 1.9832μs 0.3790μs 2.6386 MOps/s 2.5472 MOps/s $\color{#35bf28}+3.59\%$
test_tc_first_layer_tensor_set 30.9920μs 3.6919μs 270.8628 KOps/s 267.6720 KOps/s $\color{#35bf28}+1.19\%$
test_tc_first_layer_tensor_only_set 25.2110μs 3.1250μs 319.9980 KOps/s 322.5943 KOps/s $\color{#d91a1a}-0.80\%$
test_tc_first_layer_nontensor 34.9020μs 5.8541μs 170.8218 KOps/s 162.6207 KOps/s $\textbf{\color{#35bf28}+5.04\%}$
test_tc_second_layer_tensor 41.5830μs 4.1536μs 240.7561 KOps/s 236.7273 KOps/s $\color{#35bf28}+1.70\%$
test_tc_second_layer_nontensor 36.6420μs 8.2118μs 121.7765 KOps/s 114.9566 KOps/s $\textbf{\color{#35bf28}+5.93\%}$
test_unbind 0.2789s 13.5686ms 73.6998 Ops/s 71.4464 Ops/s $\color{#35bf28}+3.15\%$
test_full_like 4.5482ms 4.3594ms 229.3882 Ops/s 228.0881 Ops/s $\color{#35bf28}+0.57\%$
test_zeros_like 4.7957ms 4.3371ms 230.5698 Ops/s 228.3874 Ops/s $\color{#35bf28}+0.96\%$
test_ones_like 4.5174ms 4.3464ms 230.0755 Ops/s 227.9054 Ops/s $\color{#35bf28}+0.95\%$
test_clone 6.6588ms 6.3986ms 156.2848 Ops/s 155.5287 Ops/s $\color{#35bf28}+0.49\%$
test_squeeze 77.7750μs 13.7690μs 72.6267 KOps/s 72.3235 KOps/s $\color{#35bf28}+0.42\%$
test_unsqueeze 0.1808ms 0.1129ms 8.8586 KOps/s 8.9626 KOps/s $\color{#d91a1a}-1.16\%$
test_split 0.2395ms 0.1816ms 5.5064 KOps/s 5.6997 KOps/s $\color{#d91a1a}-3.39\%$
test_permute 0.2602ms 0.2094ms 4.7750 KOps/s 4.9694 KOps/s $\color{#d91a1a}-3.91\%$
test_stack 18.3405ms 18.0028ms 55.5468 Ops/s 19.5301 Ops/s $\textbf{\color{#35bf28}+184.42\%}$
test_cat 18.2297ms 17.8909ms 55.8943 Ops/s 19.5227 Ops/s $\textbf{\color{#35bf28}+186.30\%}$
test_sequential_tensordict 0.9884ms 0.5630ms 1.7763 KOps/s 1.7581 KOps/s $\color{#35bf28}+1.03\%$
test_sequential_graph_module 0.1663ms 0.1221ms 8.1892 KOps/s 8.2761 KOps/s $\color{#d91a1a}-1.05\%$
test_nested_tensordict 1.1944ms 0.8115ms 1.2322 KOps/s 1.2583 KOps/s $\color{#d91a1a}-2.07\%$
test_nested_graph_module 0.5231ms 0.1311ms 7.6305 KOps/s 7.7521 KOps/s $\color{#d91a1a}-1.57\%$

@github-actions

github-actions Bot commented Jun 4, 2026

Copy link
Copy Markdown
Contributor

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

Total Benchmarks: 344. Improved: $\large\color{#35bf28}30$. 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.2173ms 0.1646ms 6.0737 KOps/s 6.0891 KOps/s $\color{#d91a1a}-0.25\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.3403ms 0.2847ms 3.5125 KOps/s 3.5149 KOps/s $\color{#d91a1a}-0.07\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.8603ms 0.7485ms 1.3360 KOps/s 1.3257 KOps/s $\color{#35bf28}+0.77\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.4615ms 0.3423ms 2.9218 KOps/s 2.9788 KOps/s $\color{#d91a1a}-1.91\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.3486ms 1.1556ms 865.3181 Ops/s 914.3775 Ops/s $\textbf{\color{#d91a1a}-5.37\%}$
test_collector_preallocated_write[stack_out-cpu-4096-32] 8.1235ms 7.3532ms 135.9954 Ops/s 136.4748 Ops/s $\color{#d91a1a}-0.35\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.3128ms 0.2475ms 4.0411 KOps/s 4.1373 KOps/s $\color{#d91a1a}-2.32\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.4347ms 0.3750ms 2.6665 KOps/s 2.8195 KOps/s $\textbf{\color{#d91a1a}-5.43\%}$
test_collector_preallocated_write[stack_out-cuda-32-32] 1.1271ms 0.8021ms 1.2467 KOps/s 1.2737 KOps/s $\color{#d91a1a}-2.12\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.3140ms 0.2565ms 3.8993 KOps/s 4.1342 KOps/s $\textbf{\color{#d91a1a}-5.68\%}$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.5534ms 0.3772ms 2.6510 KOps/s 2.7737 KOps/s $\color{#d91a1a}-4.42\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 0.9919ms 0.8621ms 1.1600 KOps/s 1.1783 KOps/s $\color{#d91a1a}-1.55\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2701ms 0.2150ms 4.6513 KOps/s 4.7953 KOps/s $\color{#d91a1a}-3.00\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 0.9550ms 0.8014ms 1.2478 KOps/s 1.2400 KOps/s $\color{#35bf28}+0.63\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.2701ms 3.1381ms 318.6629 Ops/s 311.3215 Ops/s $\color{#35bf28}+2.36\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.5064ms 0.3549ms 2.8179 KOps/s 2.6271 KOps/s $\textbf{\color{#35bf28}+7.26\%}$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 2.0593ms 1.6592ms 602.7103 Ops/s 618.8134 Ops/s $\color{#d91a1a}-2.60\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 11.1173ms 10.5887ms 94.4407 Ops/s 98.6764 Ops/s $\color{#d91a1a}-4.29\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.5200ms 0.3896ms 2.5668 KOps/s 2.6001 KOps/s $\color{#d91a1a}-1.28\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.6122ms 1.4293ms 699.6674 Ops/s 700.2586 Ops/s $\color{#d91a1a}-0.08\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 5.8134ms 5.6771ms 176.1473 Ops/s 178.1842 Ops/s $\color{#d91a1a}-1.14\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.4507ms 0.3856ms 2.5932 KOps/s 2.5951 KOps/s $\color{#d91a1a}-0.07\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.5630ms 1.4460ms 691.5802 Ops/s 688.5701 Ops/s $\color{#35bf28}+0.44\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 5.7578ms 5.6809ms 176.0289 Ops/s 176.8251 Ops/s $\color{#d91a1a}-0.45\%$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.6262ms 0.1684ms 5.9368 KOps/s 5.9272 KOps/s $\color{#35bf28}+0.16\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.7669ms 0.6549ms 1.5268 KOps/s 1.5080 KOps/s $\color{#35bf28}+1.25\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 2.6954ms 2.5937ms 385.5474 Ops/s 384.8996 Ops/s $\color{#35bf28}+0.17\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.4120ms 0.3048ms 3.2803 KOps/s 3.1353 KOps/s $\color{#35bf28}+4.63\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.9943ms 1.5303ms 653.4705 Ops/s 705.6741 Ops/s $\textbf{\color{#d91a1a}-7.40\%}$
test_collector_preallocated_write[copy_at-cpu-4096-32] 10.0187ms 9.6080ms 104.0801 Ops/s 105.6788 Ops/s $\color{#d91a1a}-1.51\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.4613ms 0.3241ms 3.0850 KOps/s 3.1295 KOps/s $\color{#d91a1a}-1.42\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.3041ms 1.1909ms 839.6684 Ops/s 841.3932 Ops/s $\color{#d91a1a}-0.20\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.8872ms 4.6569ms 214.7338 Ops/s 215.4950 Ops/s $\color{#d91a1a}-0.35\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.3978ms 0.3216ms 3.1095 KOps/s 3.1300 KOps/s $\color{#d91a1a}-0.66\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.2869ms 1.1958ms 836.2714 Ops/s 837.2449 Ops/s $\color{#d91a1a}-0.12\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 4.9666ms 4.7021ms 212.6707 Ops/s 211.7983 Ops/s $\color{#35bf28}+0.41\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.1703ms 0.1167ms 8.5656 KOps/s 8.5959 KOps/s $\color{#d91a1a}-0.35\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.5692ms 0.4573ms 2.1868 KOps/s 2.2150 KOps/s $\color{#d91a1a}-1.27\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 1.8837ms 1.8209ms 549.1816 Ops/s 551.3775 Ops/s $\color{#d91a1a}-0.40\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.6763ms 0.2546ms 3.9273 KOps/s 3.8399 KOps/s $\color{#35bf28}+2.28\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.5522ms 1.3726ms 728.5326 Ops/s 784.3684 Ops/s $\textbf{\color{#d91a1a}-7.12\%}$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 9.5768ms 8.9384ms 111.8774 Ops/s 117.5297 Ops/s $\color{#d91a1a}-4.81\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.3873ms 0.2935ms 3.4070 KOps/s 3.6683 KOps/s $\textbf{\color{#d91a1a}-7.12\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.1672ms 1.0907ms 916.8622 Ops/s 995.8951 Ops/s $\textbf{\color{#d91a1a}-7.94\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 4.3818ms 4.2416ms 235.7614 Ops/s 254.8461 Ops/s $\textbf{\color{#d91a1a}-7.49\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.3505ms 0.2925ms 3.4187 KOps/s 3.6960 KOps/s $\textbf{\color{#d91a1a}-7.50\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.1479ms 1.0922ms 915.6003 Ops/s 992.5475 Ops/s $\textbf{\color{#d91a1a}-7.75\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 4.3804ms 4.2820ms 233.5343 Ops/s 250.6993 Ops/s $\textbf{\color{#d91a1a}-6.85\%}$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.2824ms 15.7297ms 63.5740 Ops/s 63.9903 Ops/s $\color{#d91a1a}-0.65\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 32.1827ms 31.9343ms 31.3143 Ops/s 31.7717 Ops/s $\color{#d91a1a}-1.44\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 88.4724ms 88.2971ms 11.3254 Ops/s 12.1792 Ops/s $\textbf{\color{#d91a1a}-7.01\%}$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.2241s 0.2214s 4.5161 Ops/s 5.1253 Ops/s $\textbf{\color{#d91a1a}-11.89\%}$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 13.5426ms 13.1078ms 76.2903 Ops/s 75.9915 Ops/s $\color{#35bf28}+0.39\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 26.6612ms 25.8462ms 38.6904 Ops/s 38.8752 Ops/s $\color{#d91a1a}-0.48\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 13.4776ms 13.3044ms 75.1633 Ops/s 75.4669 Ops/s $\color{#d91a1a}-0.40\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 26.4526ms 26.2270ms 38.1287 Ops/s 38.3470 Ops/s $\color{#d91a1a}-0.57\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 50.7068ms 50.4332ms 19.8282 Ops/s 19.8087 Ops/s $\color{#35bf28}+0.10\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1020s 0.1017s 9.8313 Ops/s 9.8730 Ops/s $\color{#d91a1a}-0.42\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1494s 0.1488s 6.7188 Ops/s 7.1485 Ops/s $\textbf{\color{#d91a1a}-6.01\%}$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3586s 0.3482s 2.8723 Ops/s 3.2161 Ops/s $\textbf{\color{#d91a1a}-10.69\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 78.1487ms 77.6831ms 12.8728 Ops/s 13.9519 Ops/s $\textbf{\color{#d91a1a}-7.73\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1560s 0.1555s 6.4300 Ops/s 6.9775 Ops/s $\textbf{\color{#d91a1a}-7.85\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 78.8600ms 78.5601ms 12.7291 Ops/s 13.8464 Ops/s $\textbf{\color{#d91a1a}-8.07\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1572s 0.1565s 6.3899 Ops/s 6.7985 Ops/s $\textbf{\color{#d91a1a}-6.01\%}$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 29.3635ms 29.2577ms 34.1790 Ops/s 33.4482 Ops/s $\color{#35bf28}+2.18\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 60.1041ms 59.9224ms 16.6882 Ops/s 16.6872 Ops/s $+0.01\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1228s 0.1216s 8.2260 Ops/s 8.6515 Ops/s $\color{#d91a1a}-4.92\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2703s 0.2662s 3.7564 Ops/s 3.8370 Ops/s $\color{#d91a1a}-2.10\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 43.0139ms 42.6901ms 23.4246 Ops/s 23.5098 Ops/s $\color{#d91a1a}-0.36\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 85.9616ms 85.7445ms 11.6626 Ops/s 11.7767 Ops/s $\color{#d91a1a}-0.97\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 44.0875ms 43.0789ms 23.2132 Ops/s 23.3897 Ops/s $\color{#d91a1a}-0.75\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 86.4599ms 86.2070ms 11.6000 Ops/s 11.6954 Ops/s $\color{#d91a1a}-0.82\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 20.1972ms 19.8897ms 50.2773 Ops/s 50.3834 Ops/s $\color{#d91a1a}-0.21\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 40.6611ms 40.4589ms 24.7164 Ops/s 24.6874 Ops/s $\color{#35bf28}+0.12\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1073s 0.1070s 9.3476 Ops/s 9.5721 Ops/s $\color{#d91a1a}-2.35\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2473s 0.2441s 4.0961 Ops/s 4.2013 Ops/s $\color{#d91a1a}-2.51\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 39.1090ms 38.7128ms 25.8313 Ops/s 28.1913 Ops/s $\textbf{\color{#d91a1a}-8.37\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 78.6394ms 78.1054ms 12.8032 Ops/s 14.0607 Ops/s $\textbf{\color{#d91a1a}-8.94\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 39.2059ms 38.9447ms 25.6775 Ops/s 28.0743 Ops/s $\textbf{\color{#d91a1a}-8.54\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 77.7041ms 77.2747ms 12.9409 Ops/s 14.0256 Ops/s $\textbf{\color{#d91a1a}-7.73\%}$
test_plain_set_nested 54.6510μs 15.0282μs 66.5418 KOps/s 68.0032 KOps/s $\color{#d91a1a}-2.15\%$
test_plain_set_stack_nested 47.0210μs 14.6058μs 68.4658 KOps/s 69.1399 KOps/s $\color{#d91a1a}-0.98\%$
test_plain_set_nested_inplace 44.8620μs 16.1590μs 61.8850 KOps/s 62.3969 KOps/s $\color{#d91a1a}-0.82\%$
test_plain_set_stack_nested_inplace 68.3520μs 15.9207μs 62.8113 KOps/s 63.0302 KOps/s $\color{#d91a1a}-0.35\%$
test_items 34.2610μs 5.5222μs 181.0886 KOps/s 180.7767 KOps/s $\color{#35bf28}+0.17\%$
test_items_nested 0.5050ms 0.4548ms 2.1988 KOps/s 2.1986 KOps/s $+0.01\%$
test_items_nested_locked 0.5657ms 0.4574ms 2.1864 KOps/s 2.2017 KOps/s $\color{#d91a1a}-0.69\%$
test_items_nested_leaf 0.1375ms 92.5834μs 10.8011 KOps/s 10.9330 KOps/s $\color{#d91a1a}-1.21\%$
test_items_stack_nested 0.5454ms 0.4538ms 2.2036 KOps/s 2.2253 KOps/s $\color{#d91a1a}-0.98\%$
test_items_stack_nested_leaf 0.1333ms 92.2269μs 10.8428 KOps/s 10.9290 KOps/s $\color{#d91a1a}-0.79\%$
test_items_stack_nested_locked 0.5239ms 0.4621ms 2.1642 KOps/s 2.1996 KOps/s $\color{#d91a1a}-1.61\%$
test_keys 25.9300μs 4.1252μs 242.4136 KOps/s 241.2135 KOps/s $\color{#35bf28}+0.50\%$
test_keys_nested 0.1700ms 0.1297ms 7.7089 KOps/s 7.8002 KOps/s $\color{#d91a1a}-1.17\%$
test_keys_nested_locked 2.0508ms 0.1362ms 7.3414 KOps/s 7.3117 KOps/s $\color{#35bf28}+0.41\%$
test_keys_nested_leaf 0.1621ms 0.1186ms 8.4308 KOps/s 8.3570 KOps/s $\color{#35bf28}+0.88\%$
test_keys_stack_nested 0.1875ms 0.1291ms 7.7484 KOps/s 7.8226 KOps/s $\color{#d91a1a}-0.95\%$
test_keys_stack_nested_leaf 0.1638ms 0.1190ms 8.4050 KOps/s 8.5205 KOps/s $\color{#d91a1a}-1.36\%$
test_keys_stack_nested_locked 0.1863ms 0.1368ms 7.3109 KOps/s 7.3341 KOps/s $\color{#d91a1a}-0.32\%$
test_values 6.1342μs 1.0017μs 998.3335 KOps/s 1.0009 MOps/s $\color{#d91a1a}-0.26\%$
test_values_nested 83.6530μs 52.3577μs 19.0994 KOps/s 19.6018 KOps/s $\color{#d91a1a}-2.56\%$
test_values_nested_locked 90.8530μs 55.8590μs 17.9022 KOps/s 18.1056 KOps/s $\color{#d91a1a}-1.12\%$
test_values_nested_leaf 92.9930μs 59.5913μs 16.7810 KOps/s 16.8157 KOps/s $\color{#d91a1a}-0.21\%$
test_values_stack_nested 78.9720μs 52.0629μs 19.2075 KOps/s 19.3594 KOps/s $\color{#d91a1a}-0.78\%$
test_values_stack_nested_leaf 84.2830μs 59.5301μs 16.7982 KOps/s 16.8195 KOps/s $\color{#d91a1a}-0.13\%$
test_values_stack_nested_locked 83.1120μs 55.6796μs 17.9599 KOps/s 18.1563 KOps/s $\color{#d91a1a}-1.08\%$
test_membership 4.8368μs 0.8002μs 1.2497 MOps/s 1.2714 MOps/s $\color{#d91a1a}-1.71\%$
test_membership_nested 33.0800μs 2.8048μs 356.5329 KOps/s 355.8796 KOps/s $\color{#35bf28}+0.18\%$
test_membership_nested_leaf 39.1910μs 2.8121μs 355.6011 KOps/s 369.0687 KOps/s $\color{#d91a1a}-3.65\%$
test_membership_stacked_nested 29.7010μs 2.7278μs 366.5994 KOps/s 371.5203 KOps/s $\color{#d91a1a}-1.32\%$
test_membership_stacked_nested_leaf 26.9810μs 2.7140μs 368.4533 KOps/s 364.9196 KOps/s $\color{#35bf28}+0.97\%$
test_membership_nested_last 41.6510μs 4.2141μs 237.2989 KOps/s 240.4751 KOps/s $\color{#d91a1a}-1.32\%$
test_membership_nested_leaf_last 34.6110μs 4.1942μs 238.4254 KOps/s 238.8474 KOps/s $\color{#d91a1a}-0.18\%$
test_membership_stacked_nested_last 35.4810μs 4.0960μs 244.1381 KOps/s 246.3615 KOps/s $\color{#d91a1a}-0.90\%$
test_membership_stacked_nested_leaf_last 32.3710μs 4.1466μs 241.1635 KOps/s 244.4379 KOps/s $\color{#d91a1a}-1.34\%$
test_nested_getleaf 48.5420μs 20.3821μs 49.0628 KOps/s 48.7694 KOps/s $\color{#35bf28}+0.60\%$
test_nested_get 45.2810μs 19.5425μs 51.1705 KOps/s 51.4043 KOps/s $\color{#d91a1a}-0.45\%$
test_stacked_getleaf 45.7420μs 20.4325μs 48.9416 KOps/s 49.1886 KOps/s $\color{#d91a1a}-0.50\%$
test_stacked_get 45.5610μs 19.3215μs 51.7559 KOps/s 51.3816 KOps/s $\color{#35bf28}+0.73\%$
test_nested_getitemleaf 43.6710μs 21.0121μs 47.5917 KOps/s 47.3127 KOps/s $\color{#35bf28}+0.59\%$
test_nested_getitem 50.0010μs 19.9383μs 50.1546 KOps/s 50.3685 KOps/s $\color{#d91a1a}-0.42\%$
test_stacked_getitemleaf 50.7820μs 21.1727μs 47.2307 KOps/s 47.5316 KOps/s $\color{#d91a1a}-0.63\%$
test_stacked_getitem 51.1010μs 20.0856μs 49.7870 KOps/s 50.2363 KOps/s $\color{#d91a1a}-0.89\%$
test_lock_nested 0.5455ms 0.4713ms 2.1219 KOps/s 2.0758 KOps/s $\color{#35bf28}+2.22\%$
test_lock_stack_nested 0.5136ms 0.4805ms 2.0813 KOps/s 2.0741 KOps/s $\color{#35bf28}+0.34\%$
test_unlock_nested 0.4424ms 0.3779ms 2.6460 KOps/s 2.6597 KOps/s $\color{#d91a1a}-0.51\%$
test_unlock_stack_nested 0.4064ms 0.3785ms 2.6421 KOps/s 2.6316 KOps/s $\color{#35bf28}+0.40\%$
test_flatten_speed 0.1700ms 0.1151ms 8.6850 KOps/s 8.6855 KOps/s $-0.00\%$
test_unflatten_speed 0.6257ms 0.5466ms 1.8294 KOps/s 1.8184 KOps/s $\color{#35bf28}+0.61\%$
test_common_ops 99.9382ms 1.9326ms 517.4444 Ops/s 1.3473 KOps/s $\textbf{\color{#d91a1a}-61.59\%}$
test_creation 0.1471ms 2.9492μs 339.0707 KOps/s 342.1176 KOps/s $\color{#d91a1a}-0.89\%$
test_creation_empty 25.7900μs 6.5306μs 153.1253 KOps/s 154.6366 KOps/s $\color{#d91a1a}-0.98\%$
test_creation_nested_1 32.1410μs 10.7669μs 92.8772 KOps/s 93.0302 KOps/s $\color{#d91a1a}-0.16\%$
test_creation_nested_2 39.2410μs 12.7205μs 78.6135 KOps/s 79.0801 KOps/s $\color{#d91a1a}-0.59\%$
test_creation_many_keys[10] 49.2610μs 19.1875μs 52.1172 KOps/s 52.1290 KOps/s $\color{#d91a1a}-0.02\%$
test_creation_many_keys[50] 0.1358ms 82.7155μs 12.0896 KOps/s 12.1531 KOps/s $\color{#d91a1a}-0.52\%$
test_creation_many_keys[100] 0.2159ms 0.1637ms 6.1091 KOps/s 6.1879 KOps/s $\color{#d91a1a}-1.27\%$
test_creation_nested_many_keys[10] 70.2020μs 41.4293μs 24.1375 KOps/s 23.9877 KOps/s $\color{#35bf28}+0.62\%$
test_creation_nested_many_keys[50] 0.2339ms 0.1687ms 5.9294 KOps/s 5.8995 KOps/s $\color{#35bf28}+0.51\%$
test_clone 31.5010μs 12.8779μs 77.6521 KOps/s 78.0593 KOps/s $\color{#d91a1a}-0.52\%$
test_getitem[int] 1.5598ms 14.4713μs 69.1023 KOps/s 69.6065 KOps/s $\color{#d91a1a}-0.72\%$
test_getitem[slice_int] 0.1382ms 23.7518μs 42.1021 KOps/s 43.3990 KOps/s $\color{#d91a1a}-2.99\%$
test_getitem[range] 0.1890ms 61.6947μs 16.2088 KOps/s 16.2119 KOps/s $\color{#d91a1a}-0.02\%$
test_getitem[tuple] 0.1391ms 23.2177μs 43.0705 KOps/s 43.0808 KOps/s $\color{#d91a1a}-0.02\%$
test_getitem[list] 0.1851ms 57.3862μs 17.4258 KOps/s 17.6714 KOps/s $\color{#d91a1a}-1.39\%$
test_setitem_dim[int] 47.3620μs 25.4600μs 39.2773 KOps/s 40.8344 KOps/s $\color{#d91a1a}-3.81\%$
test_setitem_dim[slice_int] 64.7420μs 42.2630μs 23.6614 KOps/s 24.2398 KOps/s $\color{#d91a1a}-2.39\%$
test_setitem_dim[range] 0.1280ms 95.9714μs 10.4198 KOps/s 10.8733 KOps/s $\color{#d91a1a}-4.17\%$
test_setitem_dim[tuple] 68.7320μs 37.7898μs 26.4621 KOps/s 25.8582 KOps/s $\color{#35bf28}+2.34\%$
test_setitem 49.5410μs 16.8593μs 59.3145 KOps/s 54.6049 KOps/s $\textbf{\color{#35bf28}+8.62\%}$
test_set 49.6910μs 16.0153μs 62.4404 KOps/s 59.4197 KOps/s $\textbf{\color{#35bf28}+5.08\%}$
test_set_shared 0.6905ms 0.2094ms 4.7748 KOps/s 4.6504 KOps/s $\color{#35bf28}+2.68\%$
test_update 0.3622ms 20.8751μs 47.9041 KOps/s 44.2545 KOps/s $\textbf{\color{#35bf28}+8.25\%}$
test_update_nested 68.5720μs 32.6111μs 30.6645 KOps/s 28.8767 KOps/s $\textbf{\color{#35bf28}+6.19\%}$
test_update__nested 0.5112ms 32.0760μs 31.1760 KOps/s 29.8578 KOps/s $\color{#35bf28}+4.41\%$
test_set_nested 53.9320μs 18.1583μs 55.0713 KOps/s 51.8182 KOps/s $\textbf{\color{#35bf28}+6.28\%}$
test_set_nested_new 69.4020μs 22.0813μs 45.2872 KOps/s 41.0974 KOps/s $\textbf{\color{#35bf28}+10.19\%}$
test_select 86.2320μs 37.5551μs 26.6275 KOps/s 24.4043 KOps/s $\textbf{\color{#35bf28}+9.11\%}$
test_select_nested 0.1035ms 68.5971μs 14.5779 KOps/s 14.3120 KOps/s $\color{#35bf28}+1.86\%$
test_exclude_nested 0.1440ms 84.6729μs 11.8102 KOps/s 11.5513 KOps/s $\color{#35bf28}+2.24\%$
test_empty[True] 0.4213ms 0.3854ms 2.5947 KOps/s 2.5699 KOps/s $\color{#35bf28}+0.96\%$
test_empty[False] 8.4127μs 1.2916μs 774.2358 KOps/s 769.6512 KOps/s $\color{#35bf28}+0.60\%$
test_to 0.1118ms 68.8450μs 14.5254 KOps/s 14.0176 KOps/s $\color{#35bf28}+3.62\%$
test_to_nonblocking 0.1127ms 67.5219μs 14.8100 KOps/s 14.6423 KOps/s $\color{#35bf28}+1.15\%$
test_unbind_speed 0.4332ms 0.3259ms 3.0681 KOps/s 3.1602 KOps/s $\color{#d91a1a}-2.92\%$
test_unbind_speed_stack0 0.3897ms 0.3151ms 3.1737 KOps/s 3.1841 KOps/s $\color{#d91a1a}-0.33\%$
test_unbind_speed_stack1 0.1158s 0.8150ms 1.2270 KOps/s 1.2047 KOps/s $\color{#35bf28}+1.86\%$
test_split 0.1162s 1.2205ms 819.3155 Ops/s 718.5855 Ops/s $\textbf{\color{#35bf28}+14.02\%}$
test_chunk 1.1321ms 1.0388ms 962.6653 Ops/s 958.8923 Ops/s $\color{#35bf28}+0.39\%$
test_to_cpu_blocking 20.0668ms 19.8979ms 50.2566 Ops/s 49.7340 Ops/s $\color{#35bf28}+1.05\%$
test_to_cpu_global_sync 11.9378ms 11.7142ms 85.3665 Ops/s 84.9934 Ops/s $\color{#35bf28}+0.44\%$
test_to_cpu_event_sync 13.1156ms 12.7524ms 78.4164 Ops/s 77.9123 Ops/s $\color{#35bf28}+0.65\%$
test_to_cpu_default 13.1367ms 12.7986ms 78.1335 Ops/s 69.7292 Ops/s $\textbf{\color{#35bf28}+12.05\%}$
test_consolidate[False-None] 4.0605ms 3.9953ms 250.2922 Ops/s 249.4491 Ops/s $\color{#35bf28}+0.34\%$
test_consolidate[default-None] 3.2986ms 2.0549ms 486.6449 Ops/s 479.8286 Ops/s $\color{#35bf28}+1.42\%$
test_consolidate[reduce-overhead-None] 3.5411ms 1.9928ms 501.7999 Ops/s 490.9385 Ops/s $\color{#35bf28}+2.21\%$
test_consolidate_njt[False-None] 8.4090ms 8.3090ms 120.3516 Ops/s 121.6453 Ops/s $\color{#d91a1a}-1.06\%$
test_to[False-False-None] 2.2842ms 2.1488ms 465.3792 Ops/s 464.7096 Ops/s $\color{#35bf28}+0.14\%$
test_to[True-False-None] 2.1150ms 1.8165ms 550.5020 Ops/s 533.8617 Ops/s $\color{#35bf28}+3.12\%$
test_to[within-False-None] 6.2891ms 5.9511ms 168.0369 Ops/s 164.1708 Ops/s $\color{#35bf28}+2.35\%$
test_to[True-default-None] 10.5245ms 10.3449ms 96.6659 Ops/s 94.1637 Ops/s $\color{#35bf28}+2.66\%$
test_to_njt[False-False-None] 8.4086ms 8.2894ms 120.6365 Ops/s 117.1680 Ops/s $\color{#35bf28}+2.96\%$
test_to_njt[True-False-None] 6.8193ms 6.5965ms 151.5946 Ops/s 145.9143 Ops/s $\color{#35bf28}+3.89\%$
test_to_njt[within-False-None] 15.1205ms 14.8670ms 67.2629 Ops/s 65.1751 Ops/s $\color{#35bf28}+3.20\%$
test_creation[device0] 0.3469ms 0.1125ms 8.8876 KOps/s 8.6659 KOps/s $\color{#35bf28}+2.56\%$
test_creation_from_tensor 0.3575ms 0.1101ms 9.0860 KOps/s 8.7179 KOps/s $\color{#35bf28}+4.22\%$
test_add_one[memmap_tensor0] 0.1747ms 6.4459μs 155.1380 KOps/s 150.7909 KOps/s $\color{#35bf28}+2.88\%$
test_contiguous[memmap_tensor0] 27.6300μs 0.6454μs 1.5494 MOps/s 2.3542 MOps/s $\textbf{\color{#d91a1a}-34.19\%}$
test_stack[memmap_tensor0] 36.7610μs 4.1844μs 238.9852 KOps/s 235.3583 KOps/s $\color{#35bf28}+1.54\%$
test_memmaptd_index 1.1319ms 0.2655ms 3.7669 KOps/s 3.7567 KOps/s $\color{#35bf28}+0.27\%$
test_memmaptd_index_astensor 0.4733ms 0.3657ms 2.7346 KOps/s 2.7034 KOps/s $\color{#35bf28}+1.15\%$
test_memmaptd_index_op 0.8402ms 0.6153ms 1.6253 KOps/s 1.6050 KOps/s $\color{#35bf28}+1.26\%$
test_serialize_model 0.1347s 0.1331s 7.5117 Ops/s 7.4295 Ops/s $\color{#35bf28}+1.11\%$
test_serialize_model_pickle 1.3474s 1.2094s 0.8269 Ops/s 0.8234 Ops/s $\color{#35bf28}+0.42\%$
test_serialize_weights 0.1350s 0.1330s 7.5173 Ops/s 3.4118 Ops/s $\textbf{\color{#35bf28}+120.34\%}$
test_serialize_weights_returnearly 0.3143s 77.9225ms 12.8333 Ops/s 16.1052 Ops/s $\textbf{\color{#d91a1a}-20.32\%}$
test_serialize_weights_pickle 1.3473s 1.2131s 0.8243 Ops/s 0.8230 Ops/s $\color{#35bf28}+0.17\%$
test_reshape_pytree 0.1994ms 30.9193μs 32.3422 KOps/s 32.0598 KOps/s $\color{#35bf28}+0.88\%$
test_reshape_td 87.0030μs 42.8383μs 23.3436 KOps/s 22.6084 KOps/s $\color{#35bf28}+3.25\%$
test_view_pytree 0.2127ms 30.6892μs 32.5847 KOps/s 32.4759 KOps/s $\color{#35bf28}+0.34\%$
test_view_td 93.6730μs 51.7464μs 19.3250 KOps/s 18.5355 KOps/s $\color{#35bf28}+4.26\%$
test_unbind_pytree 0.2209ms 34.7471μs 28.7794 KOps/s 27.9648 KOps/s $\color{#35bf28}+2.91\%$
test_unbind_td 82.5320μs 48.6663μs 20.5481 KOps/s 19.8966 KOps/s $\color{#35bf28}+3.27\%$
test_split_pytree 0.2662ms 40.1030μs 24.9358 KOps/s 23.9012 KOps/s $\color{#35bf28}+4.33\%$
test_split_td 0.1082ms 63.0592μs 15.8581 KOps/s 15.8639 KOps/s $\color{#d91a1a}-0.04\%$
test_add_pytree 0.2264ms 39.8659μs 25.0841 KOps/s 24.4563 KOps/s $\color{#35bf28}+2.57\%$
test_add_td 97.2430μs 55.1971μs 18.1169 KOps/s 18.0737 KOps/s $\color{#35bf28}+0.24\%$
test_compile_add_one_nested[tensordict-compile] 0.3310ms 0.2318ms 4.3137 KOps/s 4.4715 KOps/s $\color{#d91a1a}-3.53\%$
test_compile_add_one_nested[tensordict-eager] 0.2847ms 0.1960ms 5.1023 KOps/s 4.1648 KOps/s $\textbf{\color{#35bf28}+22.51\%}$
test_compile_add_one_nested[pytree-compile] 0.2724ms 0.1354ms 7.3845 KOps/s 7.3868 KOps/s $\color{#d91a1a}-0.03\%$
test_compile_add_one_nested[pytree-eager] 0.4285ms 0.1714ms 5.8341 KOps/s 5.8011 KOps/s $\color{#35bf28}+0.57\%$
test_compile_copy_nested[tensordict-compile] 72.0620μs 16.2153μs 61.6700 KOps/s 62.4362 KOps/s $\color{#d91a1a}-1.23\%$
test_compile_copy_nested[tensordict-eager] 0.1387ms 50.3314μs 19.8683 KOps/s 19.6588 KOps/s $\color{#35bf28}+1.07\%$
test_compile_copy_nested[pytree-compile] 0.1311ms 15.3286μs 65.2375 KOps/s 63.5898 KOps/s $\color{#35bf28}+2.59\%$
test_compile_copy_nested[pytree-eager] 0.3618ms 63.8417μs 15.6638 KOps/s 15.6125 KOps/s $\color{#35bf28}+0.33\%$
test_compile_add_one_flat[tensordict-compile] 0.2857ms 0.2406ms 4.1556 KOps/s 3.7323 KOps/s $\textbf{\color{#35bf28}+11.34\%}$
test_compile_add_one_flat[tensordict-eager] 0.4065ms 0.2766ms 3.6153 KOps/s 3.6145 KOps/s $\color{#35bf28}+0.02\%$
test_compile_add_one_flat[tensorclass-compile] 0.2228ms 0.1803ms 5.5448 KOps/s 5.2666 KOps/s $\textbf{\color{#35bf28}+5.28\%}$
test_compile_add_one_flat[tensorclass-eager] 0.1832ms 72.5321μs 13.7870 KOps/s 13.6306 KOps/s $\color{#35bf28}+1.15\%$
test_compile_add_one_flat[pytree-compile] 0.4653ms 0.1874ms 5.3352 KOps/s 5.1543 KOps/s $\color{#35bf28}+3.51\%$
test_compile_add_one_flat[pytree-eager] 0.7929ms 0.5045ms 1.9820 KOps/s 1.9308 KOps/s $\color{#35bf28}+2.65\%$
test_compile_add_self_flat[tensordict-eager] 0.4556ms 0.3225ms 3.1009 KOps/s 3.0848 KOps/s $\color{#35bf28}+0.52\%$
test_compile_add_self_flat[tensordict-compile] 0.5966ms 0.2459ms 4.0660 KOps/s 3.8654 KOps/s $\textbf{\color{#35bf28}+5.19\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1489ms 88.3931μs 11.3131 KOps/s 11.2125 KOps/s $\color{#35bf28}+0.90\%$
test_compile_add_self_flat[tensorclass-compile] 0.3553ms 0.1837ms 5.4449 KOps/s 5.1667 KOps/s $\textbf{\color{#35bf28}+5.38\%}$
test_compile_add_self_flat[pytree-eager] 0.6416ms 0.4202ms 2.3799 KOps/s 2.3490 KOps/s $\color{#35bf28}+1.32\%$
test_compile_add_self_flat[pytree-compile] 0.4479ms 0.1890ms 5.2908 KOps/s 5.1116 KOps/s $\color{#35bf28}+3.51\%$
test_compile_copy_flat[tensordict-compile] 64.1020μs 19.2629μs 51.9131 KOps/s 48.2803 KOps/s $\textbf{\color{#35bf28}+7.52\%}$
test_compile_copy_flat[tensordict-eager] 69.1920μs 39.8492μs 25.0946 KOps/s 25.0476 KOps/s $\color{#35bf28}+0.19\%$
test_compile_copy_flat[pytree-compile] 0.2215ms 20.9571μs 47.7165 KOps/s 46.8703 KOps/s $\color{#35bf28}+1.81\%$
test_compile_copy_flat[pytree-eager] 0.3411ms 50.9778μs 19.6164 KOps/s 19.5883 KOps/s $\color{#35bf28}+0.14\%$
test_compile_assign_and_add[tensordict-compile] 2.1250ms 0.2362ms 4.2346 KOps/s 3.9220 KOps/s $\textbf{\color{#35bf28}+7.97\%}$
test_compile_assign_and_add[tensordict-eager] 3.4406ms 3.2026ms 312.2497 Ops/s 303.4525 Ops/s $\color{#35bf28}+2.90\%$
test_compile_assign_and_add[pytree-compile] 2.0791ms 0.1754ms 5.7005 KOps/s 5.5773 KOps/s $\color{#35bf28}+2.21\%$
test_compile_assign_and_add[pytree-eager] 2.7958ms 2.6611ms 375.7820 Ops/s 367.3775 Ops/s $\color{#35bf28}+2.29\%$
test_compile_indexing[tensor-tensordict-compile] 0.2147ms 0.1362ms 7.3447 KOps/s 7.5122 KOps/s $\color{#d91a1a}-2.23\%$
test_compile_indexing[tensor-tensordict-eager] 0.3007ms 75.8184μs 13.1894 KOps/s 13.2359 KOps/s $\color{#d91a1a}-0.35\%$
test_compile_indexing[tensor-tensorclass-compile] 0.2185ms 0.1149ms 8.7032 KOps/s 8.4774 KOps/s $\color{#35bf28}+2.66\%$
test_compile_indexing[tensor-tensorclass-eager] 0.2552ms 43.3147μs 23.0868 KOps/s 22.8027 KOps/s $\color{#35bf28}+1.25\%$
test_compile_indexing[tensor-pytree-compile] 0.1640ms 0.1159ms 8.6280 KOps/s 8.4158 KOps/s $\color{#35bf28}+2.52\%$
test_compile_indexing[tensor-pytree-eager] 0.2645ms 43.4629μs 23.0081 KOps/s 22.6912 KOps/s $\color{#35bf28}+1.40\%$
test_compile_indexing[slice-tensordict-compile] 0.2283ms 81.4729μs 12.2740 KOps/s 11.7783 KOps/s $\color{#35bf28}+4.21\%$
test_compile_indexing[slice-tensordict-eager] 0.2029ms 26.8254μs 37.2782 KOps/s 37.1404 KOps/s $\color{#35bf28}+0.37\%$
test_compile_indexing[slice-tensorclass-compile] 0.1505ms 66.9207μs 14.9431 KOps/s 14.0010 KOps/s $\textbf{\color{#35bf28}+6.73\%}$
test_compile_indexing[slice-tensorclass-eager] 0.2426ms 21.0455μs 47.5160 KOps/s 46.8484 KOps/s $\color{#35bf28}+1.43\%$
test_compile_indexing[slice-pytree-compile] 0.1177ms 69.7870μs 14.3293 KOps/s 13.9592 KOps/s $\color{#35bf28}+2.65\%$
test_compile_indexing[slice-pytree-eager] 0.2476ms 21.1874μs 47.1979 KOps/s 46.3367 KOps/s $\color{#35bf28}+1.86\%$
test_compile_indexing[int-tensordict-compile] 0.1339ms 82.2052μs 12.1647 KOps/s 11.5668 KOps/s $\textbf{\color{#35bf28}+5.17\%}$
test_compile_indexing[int-tensordict-eager] 0.2791ms 26.6527μs 37.5197 KOps/s 37.4222 KOps/s $\color{#35bf28}+0.26\%$
test_compile_indexing[int-tensorclass-compile] 0.1081ms 68.7223μs 14.5513 KOps/s 13.9922 KOps/s $\color{#35bf28}+4.00\%$
test_compile_indexing[int-tensorclass-eager] 0.2437ms 20.9263μs 47.7868 KOps/s 47.0251 KOps/s $\color{#35bf28}+1.62\%$
test_compile_indexing[int-pytree-compile] 0.1076ms 68.4830μs 14.6022 KOps/s 14.1692 KOps/s $\color{#35bf28}+3.06\%$
test_compile_indexing[int-pytree-eager] 0.2931ms 21.0271μs 47.5578 KOps/s 47.4310 KOps/s $\color{#35bf28}+0.27\%$
test_compile_replace[single-eager] 95.2130μs 46.5871μs 21.4652 KOps/s 21.2345 KOps/s $\color{#35bf28}+1.09\%$
test_compile_replace[single-compile] 0.2716ms 0.1268ms 7.8841 KOps/s 7.3567 KOps/s $\textbf{\color{#35bf28}+7.17\%}$
test_compile_replace[multi-eager] 0.6543ms 0.5498ms 1.8187 KOps/s 1.7472 KOps/s $\color{#35bf28}+4.10\%$
test_compile_replace[multi-compile] 0.2958ms 0.1347ms 7.4242 KOps/s 7.0134 KOps/s $\textbf{\color{#35bf28}+5.86\%}$
test_compile_tc_getattr_20[eager] 0.1979ms 0.1606ms 6.2264 KOps/s 6.1468 KOps/s $\color{#35bf28}+1.30\%$
test_compile_tc_getattr_20[compile] 0.3942ms 0.1427ms 7.0070 KOps/s 6.5527 KOps/s $\textbf{\color{#35bf28}+6.93\%}$
test_compile_clone_shallow[20-eager] 62.2110μs 18.5192μs 53.9981 KOps/s 53.9167 KOps/s $\color{#35bf28}+0.15\%$
test_compile_clone_shallow[20-compile] 92.8530μs 16.4914μs 60.6378 KOps/s 54.6728 KOps/s $\textbf{\color{#35bf28}+10.91\%}$
test_compile_clone_shallow[40-eager] 62.6520μs 32.7609μs 30.5242 KOps/s 31.0060 KOps/s $\color{#d91a1a}-1.55\%$
test_compile_clone_shallow[40-compile] 74.8420μs 17.9614μs 55.6751 KOps/s 54.1039 KOps/s $\color{#35bf28}+2.90\%$
test_compile_clone_shallow[80-eager] 89.6920μs 60.6340μs 16.4924 KOps/s 16.5004 KOps/s $\color{#d91a1a}-0.05\%$
test_compile_clone_shallow[80-compile] 91.0130μs 21.3231μs 46.8975 KOps/s 45.2146 KOps/s $\color{#35bf28}+3.72\%$
test_compile_update_inplace[eager] 0.1004ms 58.8136μs 17.0029 KOps/s 16.6588 KOps/s $\color{#35bf28}+2.07\%$
test_compile_update_inplace[compile] 0.2829ms 0.1397ms 7.1558 KOps/s 6.8786 KOps/s $\color{#35bf28}+4.03\%$
test_mod_add[eager] 0.2713ms 0.2183ms 4.5805 KOps/s 4.5314 KOps/s $\color{#35bf28}+1.08\%$
test_mod_add[compile] 0.2123ms 0.1275ms 7.8449 KOps/s 7.3087 KOps/s $\textbf{\color{#35bf28}+7.34\%}$
test_mod_add[compile-overhead] 0.3866ms 0.1736ms 5.7598 KOps/s 5.5804 KOps/s $\color{#35bf28}+3.21\%$
test_mod_wrap[eager] 0.5334ms 0.4555ms 2.1953 KOps/s 2.1547 KOps/s $\color{#35bf28}+1.88\%$
test_mod_wrap[compile] 0.4660ms 0.3830ms 2.6108 KOps/s 2.5410 KOps/s $\color{#35bf28}+2.75\%$
test_mod_wrap[compile-overhead] 9.0447ms 4.7667ms 209.7870 Ops/s 213.6846 Ops/s $\color{#d91a1a}-1.82\%$
test_mod_wrap_and_backward[eager] 1.8579ms 1.6642ms 600.8831 Ops/s 577.8644 Ops/s $\color{#35bf28}+3.98\%$
test_mod_wrap_and_backward[compile] 2.5595ms 1.5624ms 640.0368 Ops/s 666.8516 Ops/s $\color{#d91a1a}-4.02\%$
test_mod_wrap_and_backward[compile-overhead] 1.3035ms 0.9168ms 1.0907 KOps/s 1.0693 KOps/s $\color{#35bf28}+2.00\%$
test_seq_add[eager] 0.5780ms 0.4946ms 2.0217 KOps/s 2.0223 KOps/s $\color{#d91a1a}-0.03\%$
test_seq_add[compile] 0.2476ms 0.1370ms 7.2983 KOps/s 7.0755 KOps/s $\color{#35bf28}+3.15\%$
test_seq_add[compile-overhead] 0.2988ms 0.1778ms 5.6253 KOps/s 5.4600 KOps/s $\color{#35bf28}+3.03\%$
test_seq_wrap[eager] 1.4464ms 1.3776ms 725.8843 Ops/s 720.9275 Ops/s $\color{#35bf28}+0.69\%$
test_seq_wrap[compile] 0.5115ms 0.3960ms 2.5252 KOps/s 2.4362 KOps/s $\color{#35bf28}+3.65\%$
test_seq_wrap[compile-overhead] 0.3872ms 0.2970ms 3.3673 KOps/s 3.2888 KOps/s $\color{#35bf28}+2.39\%$
test_func_call_runtime[False-eager] 0.9033ms 0.8248ms 1.2125 KOps/s 1.2316 KOps/s $\color{#d91a1a}-1.55\%$
test_func_call_runtime[False-compile] 1.0528ms 0.9430ms 1.0605 KOps/s 1.0570 KOps/s $\color{#35bf28}+0.33\%$
test_func_call_runtime[False-compile-overhead] 0.6314ms 0.4922ms 2.0316 KOps/s 2.0218 KOps/s $\color{#35bf28}+0.49\%$
test_func_call_runtime[True-eager] 1.1314ms 1.0304ms 970.4681 Ops/s 972.1703 Ops/s $\color{#d91a1a}-0.18\%$
test_func_call_runtime[True-compile] 1.0568ms 0.9534ms 1.0489 KOps/s 1.0393 KOps/s $\color{#35bf28}+0.93\%$
test_func_call_runtime[True-compile-overhead] 0.5505ms 0.5059ms 1.9766 KOps/s 1.9519 KOps/s $\color{#35bf28}+1.26\%$
test_func_call_cm_runtime[False-eager] 0.9931ms 0.7767ms 1.2875 KOps/s 1.2904 KOps/s $\color{#d91a1a}-0.23\%$
test_func_call_cm_runtime[False-compile] 1.0776ms 0.9188ms 1.0884 KOps/s 1.0716 KOps/s $\color{#35bf28}+1.56\%$
test_func_call_cm_runtime[False-compile-overhead] 0.6171ms 0.4724ms 2.1168 KOps/s 2.0939 KOps/s $\color{#35bf28}+1.09\%$
test_func_call_cm_runtime[True-eager] 1.2782ms 1.1724ms 852.9268 Ops/s 841.6784 Ops/s $\color{#35bf28}+1.34\%$
test_func_call_cm_runtime[True-compile] 1.0435ms 0.9833ms 1.0170 KOps/s 1.0125 KOps/s $\color{#35bf28}+0.45\%$
test_func_call_cm_runtime[True-compile-overhead] 0.6773ms 0.5279ms 1.8943 KOps/s 1.8656 KOps/s $\color{#35bf28}+1.54\%$
test_vmap_func_call_cm_runtime[eager] 2.8731ms 2.3661ms 422.6303 Ops/s 419.5252 Ops/s $\color{#35bf28}+0.74\%$
test_vmap_func_call_cm_runtime[compile] 3.0255ms 2.5456ms 392.8310 Ops/s 389.9190 Ops/s $\color{#35bf28}+0.75\%$
test_vmap_func_call_cm_runtime[compile-overhead] 3.0192ms 2.5405ms 393.6166 Ops/s 390.7505 Ops/s $\color{#35bf28}+0.73\%$
test_distributed 2.6350ms 0.1689ms 5.9218 KOps/s 6.6744 KOps/s $\textbf{\color{#d91a1a}-11.28\%}$
test_tdmodule 0.2871ms 0.1868ms 5.3535 KOps/s 5.2579 KOps/s $\color{#35bf28}+1.82\%$
test_tdmodule_dispatch 0.3883ms 0.3627ms 2.7571 KOps/s 2.7348 KOps/s $\color{#35bf28}+0.81\%$
test_tdseq 43.8410μs 25.7198μs 38.8805 KOps/s 38.6624 KOps/s $\color{#35bf28}+0.56\%$
test_tdseq_dispatch 0.2380ms 0.2099ms 4.7637 KOps/s 4.7625 KOps/s $\color{#35bf28}+0.02\%$
test_instantiation_functorch 2.0564ms 1.9720ms 507.1086 Ops/s 503.3109 Ops/s $\color{#35bf28}+0.75\%$
test_exec_functorch 0.2268ms 0.1751ms 5.7109 KOps/s 5.6701 KOps/s $\color{#35bf28}+0.72\%$
test_exec_functional_call 0.2442ms 0.1552ms 6.4417 KOps/s 6.4374 KOps/s $\color{#35bf28}+0.07\%$
test_exec_td_decorator 0.4776ms 0.2411ms 4.1470 KOps/s 4.1074 KOps/s $\color{#35bf28}+0.96\%$
test_vmap_mlp_speed_decorator[True-True] 1.1673ms 1.0163ms 983.9526 Ops/s 983.5999 Ops/s $\color{#35bf28}+0.04\%$
test_vmap_mlp_speed_decorator[True-False] 1.2121ms 1.0084ms 991.6639 Ops/s 982.6267 Ops/s $\color{#35bf28}+0.92\%$
test_vmap_mlp_speed_decorator[False-True] 0.9538ms 0.7205ms 1.3879 KOps/s 1.3632 KOps/s $\color{#35bf28}+1.81\%$
test_vmap_mlp_speed_decorator[False-False] 0.9146ms 0.7216ms 1.3859 KOps/s 1.3819 KOps/s $\color{#35bf28}+0.29\%$
test_vmap_transformer_speed_decorator[True-True] 21.9025ms 20.8276ms 48.0133 Ops/s 47.8968 Ops/s $\color{#35bf28}+0.24\%$
test_vmap_transformer_speed_decorator[True-False] 20.8946ms 20.7422ms 48.2108 Ops/s 47.8481 Ops/s $\color{#35bf28}+0.76\%$
test_vmap_transformer_speed_decorator[False-True] 20.9816ms 20.3512ms 49.1371 Ops/s 48.8855 Ops/s $\color{#35bf28}+0.51\%$
test_vmap_transformer_speed_decorator[False-False] 21.1970ms 20.3813ms 49.0647 Ops/s 48.8366 Ops/s $\color{#35bf28}+0.47\%$
test_to_module_speed[True] 2.1371ms 1.5696ms 637.1083 Ops/s 630.9796 Ops/s $\color{#35bf28}+0.97\%$
test_to_module_speed[False] 2.1146ms 1.5386ms 649.9543 Ops/s 639.7726 Ops/s $\color{#35bf28}+1.59\%$
test_to_module_plain_tensor_speed[None] 2.1539ms 1.6589ms 602.8232 Ops/s 595.1573 Ops/s $\color{#35bf28}+1.29\%$
test_to_module_plain_tensor_speed[False] 1.9552ms 1.5046ms 664.6200 Ops/s 656.0320 Ops/s $\color{#35bf28}+1.31\%$
test_to_module_plain_tensor_speed[True] 3.3316ms 3.2295ms 309.6490 Ops/s 304.2319 Ops/s $\color{#35bf28}+1.78\%$
test_tc_init 95.2230μs 41.8520μs 23.8937 KOps/s 24.0224 KOps/s $\color{#d91a1a}-0.54\%$
test_tc_init_tensor_only 31.8810μs 8.5298μs 117.2358 KOps/s 117.9112 KOps/s $\color{#d91a1a}-0.57\%$
test_tc_init_nested 0.1319ms 83.8342μs 11.9283 KOps/s 12.3236 KOps/s $\color{#d91a1a}-3.21\%$
test_tc_init_many_fields 38.3110μs 14.1393μs 70.7246 KOps/s 70.9629 KOps/s $\color{#d91a1a}-0.34\%$
test_tc_first_layer_tensor 26.5810μs 1.7419μs 574.1020 KOps/s 584.8148 KOps/s $\color{#d91a1a}-1.83\%$
test_tc_first_layer_tensor_only 3.1851μs 0.3867μs 2.5862 MOps/s 2.6156 MOps/s $\color{#d91a1a}-1.12\%$
test_tc_first_layer_tensor_set 1.8544ms 3.6657μs 272.8016 KOps/s 272.0461 KOps/s $\color{#35bf28}+0.28\%$
test_tc_first_layer_tensor_only_set 19.3610μs 3.1033μs 322.2360 KOps/s 324.5809 KOps/s $\color{#d91a1a}-0.72\%$
test_tc_first_layer_nontensor 24.0610μs 5.7582μs 173.6652 KOps/s 165.7327 KOps/s $\color{#35bf28}+4.79\%$
test_tc_second_layer_tensor 30.5710μs 4.1342μs 241.8866 KOps/s 240.7939 KOps/s $\color{#35bf28}+0.45\%$
test_tc_second_layer_nontensor 46.3720μs 8.1810μs 122.2347 KOps/s 117.8033 KOps/s $\color{#35bf28}+3.76\%$
test_unbind 0.2684s 13.4556ms 74.3185 Ops/s 69.3976 Ops/s $\textbf{\color{#35bf28}+7.09\%}$
test_full_like 4.5701ms 4.3654ms 229.0717 Ops/s 232.8613 Ops/s $\color{#d91a1a}-1.63\%$
test_zeros_like 4.5314ms 4.3391ms 230.4649 Ops/s 228.5197 Ops/s $\color{#35bf28}+0.85\%$
test_ones_like 4.4621ms 4.3379ms 230.5252 Ops/s 229.2833 Ops/s $\color{#35bf28}+0.54\%$
test_clone 6.6650ms 6.3855ms 156.6055 Ops/s 155.6255 Ops/s $\color{#35bf28}+0.63\%$
test_squeeze 63.5420μs 13.8000μs 72.4637 KOps/s 68.1887 KOps/s $\textbf{\color{#35bf28}+6.27\%}$
test_unsqueeze 0.1624ms 0.1067ms 9.3723 KOps/s 8.8306 KOps/s $\textbf{\color{#35bf28}+6.13\%}$
test_split 0.2340ms 0.1740ms 5.7477 KOps/s 5.6862 KOps/s $\color{#35bf28}+1.08\%$
test_permute 0.2798ms 0.2061ms 4.8526 KOps/s 4.9116 KOps/s $\color{#d91a1a}-1.20\%$
test_stack 18.2243ms 18.0583ms 55.3761 Ops/s 28.6853 Ops/s $\textbf{\color{#35bf28}+93.05\%}$
test_cat 28.4927ms 27.0843ms 36.9217 Ops/s 28.7617 Ops/s $\textbf{\color{#35bf28}+28.37\%}$
test_sequential_tensordict 0.6940ms 0.5555ms 1.8002 KOps/s 1.7968 KOps/s $\color{#35bf28}+0.19\%$
test_sequential_graph_module 0.4881ms 0.1202ms 8.3176 KOps/s 8.4967 KOps/s $\color{#d91a1a}-2.11\%$
test_nested_tensordict 1.0211ms 0.8029ms 1.2455 KOps/s 1.2638 KOps/s $\color{#d91a1a}-1.45\%$
test_nested_graph_module 0.5363ms 0.1306ms 7.6561 KOps/s 7.7957 KOps/s $\color{#d91a1a}-1.79\%$

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

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

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant