Skip to content

[CI] Add TensorDict PR label prefixes - #1727

Merged
vmoens merged 1 commit into
mainfrom
codex/update-pr-label-prefixes
Jun 18, 2026
Merged

[CI] Add TensorDict PR label prefixes#1727
vmoens merged 1 commit into
mainfrom
codex/update-pr-label-prefixes

Conversation

@vmoens

@vmoens vmoens commented Jun 18, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Add missing title-prefix mappings for regular TensorDict labels: [BE], [Versioning], and [Release].
  • Treat [DTensor] as a distributed-work alias and map it to the existing Distributed label.
  • Update the help-comment prefix table to match the mappings.

Diagnosis

  • Checked 265 failing PR Label workflow runs.
  • The high-volume missing prefix was [DTensor] (99 failing runs), which should map to the existing Distributed label.
  • [Versioning] failed on the 0.13.0 version bump and is listed in the repo's canonical PR tags, so it now maps to versioning.
  • [Release] failed on a release/version-bump PR and maps to the existing release label.
  • Existing supported prefixes such as [BugFix], [Distributed], [Benchmark], [Performance], [Compile], [Fix], and [Perf] were already covered; recent failures there were either transient GitHub errors or titles that did not start with a prefix at run time.
  • I did not add [Linter] or no-prefix titles; those should use an existing canonical prefix such as [CI] / [Quality] or add a prefix.
  • I also did not map [Minor] because this repo currently has no small change label; mapping it to suitable for minor would conflate two different meanings.

Validation

  • bash -n /tmp/td-pr-label-step.sh
  • Stubbed gh and checked mappings for [BE], [DTensor], [Distributed], [Release], [Versioning], [BugFix], [Fix], and [Perf]
  • git diff --check -- .github/workflows/pr-label.yml

@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 18, 2026
@github-actions github-actions Bot added the CI label Jun 18, 2026
@github-actions

github-actions Bot commented Jun 18, 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}12$. Worsened: $\large\color{#d91a1a}20$.

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_collector_preallocated_write[stack_out-cpu-32-2] 0.2219ms 0.1672ms 5.9796 KOps/s 5.8953 KOps/s $\color{#35bf28}+1.43\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.3749ms 0.2990ms 3.3444 KOps/s 3.3838 KOps/s $\color{#d91a1a}-1.16\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.9331ms 0.7828ms 1.2775 KOps/s 1.2676 KOps/s $\color{#35bf28}+0.78\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.4650ms 0.3418ms 2.9253 KOps/s 2.9026 KOps/s $\color{#35bf28}+0.78\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.3646ms 1.2426ms 804.7363 Ops/s 905.2369 Ops/s $\textbf{\color{#d91a1a}-11.10\%}$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.9140ms 7.3768ms 135.5609 Ops/s 135.2391 Ops/s $\color{#35bf28}+0.24\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.3033ms 0.2516ms 3.9741 KOps/s 3.9864 KOps/s $\color{#d91a1a}-0.31\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.4798ms 0.3786ms 2.6411 KOps/s 2.6809 KOps/s $\color{#d91a1a}-1.49\%$
test_collector_preallocated_write[stack_out-cuda-32-32] 0.9236ms 0.8302ms 1.2045 KOps/s 1.2187 KOps/s $\color{#d91a1a}-1.16\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.3474ms 0.2547ms 3.9261 KOps/s 3.9939 KOps/s $\color{#d91a1a}-1.70\%$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.4689ms 0.3768ms 2.6539 KOps/s 2.6745 KOps/s $\color{#d91a1a}-0.77\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 0.9862ms 0.8913ms 1.1219 KOps/s 1.1249 KOps/s $\color{#d91a1a}-0.26\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2635ms 0.2146ms 4.6592 KOps/s 4.5452 KOps/s $\color{#35bf28}+2.51\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 0.9532ms 0.8227ms 1.2155 KOps/s 1.2014 KOps/s $\color{#35bf28}+1.17\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.3799ms 3.2825ms 304.6413 Ops/s 305.2551 Ops/s $\color{#d91a1a}-0.20\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.7180ms 0.3642ms 2.7458 KOps/s 2.6416 KOps/s $\color{#35bf28}+3.94\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 1.9107ms 1.6996ms 588.3625 Ops/s 604.6232 Ops/s $\color{#d91a1a}-2.69\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 10.4391ms 10.2863ms 97.2169 Ops/s 97.7495 Ops/s $\color{#d91a1a}-0.54\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.4182ms 0.3885ms 2.5738 KOps/s 2.5756 KOps/s $\color{#d91a1a}-0.07\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.6104ms 1.4439ms 692.5705 Ops/s 688.2842 Ops/s $\color{#35bf28}+0.62\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 5.8598ms 5.6746ms 176.2234 Ops/s 167.2193 Ops/s $\textbf{\color{#35bf28}+5.38\%}$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.5268ms 0.3868ms 2.5852 KOps/s 2.5369 KOps/s $\color{#35bf28}+1.91\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.6340ms 1.4518ms 688.7811 Ops/s 654.2763 Ops/s $\textbf{\color{#35bf28}+5.27\%}$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 5.9129ms 5.7209ms 174.7978 Ops/s 165.9378 Ops/s $\textbf{\color{#35bf28}+5.34\%}$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.6177ms 0.1723ms 5.8028 KOps/s 5.8509 KOps/s $\color{#d91a1a}-0.82\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.8031ms 0.6692ms 1.4944 KOps/s 1.4994 KOps/s $\color{#d91a1a}-0.33\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 2.7641ms 2.6397ms 378.8303 Ops/s 378.2463 Ops/s $\color{#35bf28}+0.15\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.4349ms 0.3209ms 3.1166 KOps/s 3.2055 KOps/s $\color{#d91a1a}-2.77\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.9917ms 1.4867ms 672.6366 Ops/s 688.2541 Ops/s $\color{#d91a1a}-2.27\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 9.6830ms 9.5504ms 104.7073 Ops/s 104.8739 Ops/s $\color{#d91a1a}-0.16\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.3902ms 0.3260ms 3.0678 KOps/s 3.0660 KOps/s $\color{#35bf28}+0.06\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.3484ms 1.2062ms 829.0661 Ops/s 828.7398 Ops/s $\color{#35bf28}+0.04\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.8196ms 4.7165ms 212.0222 Ops/s 212.0018 Ops/s $+0.01\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.3709ms 0.3251ms 3.0762 KOps/s 3.0607 KOps/s $\color{#35bf28}+0.50\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.2952ms 1.2088ms 827.2338 Ops/s 824.5258 Ops/s $\color{#35bf28}+0.33\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 4.8293ms 4.7553ms 210.2897 Ops/s 210.7096 Ops/s $\color{#d91a1a}-0.20\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.1626ms 0.1197ms 8.3541 KOps/s 8.3829 KOps/s $\color{#d91a1a}-0.34\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.5425ms 0.4720ms 2.1187 KOps/s 2.1392 KOps/s $\color{#d91a1a}-0.96\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 2.1148ms 1.8727ms 533.9748 Ops/s 541.3324 Ops/s $\color{#d91a1a}-1.36\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.3297ms 0.2509ms 3.9857 KOps/s 4.0342 KOps/s $\color{#d91a1a}-1.20\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.6443ms 1.2670ms 789.2773 Ops/s 821.9811 Ops/s $\color{#d91a1a}-3.98\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 9.2531ms 8.6946ms 115.0141 Ops/s 117.2434 Ops/s $\color{#d91a1a}-1.90\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.3511ms 0.2922ms 3.4229 KOps/s 3.6150 KOps/s $\textbf{\color{#d91a1a}-5.31\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.1505ms 1.0838ms 922.7072 Ops/s 979.9481 Ops/s $\textbf{\color{#d91a1a}-5.84\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 4.3759ms 4.2646ms 234.4881 Ops/s 248.9631 Ops/s $\textbf{\color{#d91a1a}-5.81\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.3752ms 0.2929ms 3.4140 KOps/s 3.6290 KOps/s $\textbf{\color{#d91a1a}-5.92\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.2239ms 1.0723ms 932.6033 Ops/s 969.8325 Ops/s $\color{#d91a1a}-3.84\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 4.0505ms 3.9743ms 251.6170 Ops/s 248.3396 Ops/s $\color{#35bf28}+1.32\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.4698ms 16.2969ms 61.3614 Ops/s 60.2511 Ops/s $\color{#35bf28}+1.84\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 33.1996ms 32.9084ms 30.3874 Ops/s 30.1999 Ops/s $\color{#35bf28}+0.62\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 86.1902ms 84.8557ms 11.7847 Ops/s 11.9063 Ops/s $\color{#d91a1a}-1.02\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.1997s 0.1986s 5.0341 Ops/s 5.0254 Ops/s $\color{#35bf28}+0.17\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 14.0134ms 13.7948ms 72.4910 Ops/s 72.8408 Ops/s $\color{#d91a1a}-0.48\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 27.1867ms 26.9516ms 37.1035 Ops/s 36.8693 Ops/s $\color{#35bf28}+0.64\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 14.0257ms 13.9411ms 71.7303 Ops/s 71.7415 Ops/s $\color{#d91a1a}-0.02\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 27.4187ms 27.3138ms 36.6115 Ops/s 36.4361 Ops/s $\color{#35bf28}+0.48\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 51.9894ms 51.8403ms 19.2900 Ops/s 18.5231 Ops/s $\color{#35bf28}+4.14\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1052s 0.1050s 9.5242 Ops/s 9.3381 Ops/s $\color{#35bf28}+1.99\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1440s 0.1432s 6.9829 Ops/s 6.9489 Ops/s $\color{#35bf28}+0.49\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3226s 0.3220s 3.1060 Ops/s 3.1465 Ops/s $\color{#d91a1a}-1.29\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 78.7054ms 75.1032ms 13.3150 Ops/s 12.9756 Ops/s $\color{#35bf28}+2.62\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1579s 0.1566s 6.3865 Ops/s 6.7048 Ops/s $\color{#d91a1a}-4.75\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 79.2949ms 75.5239ms 13.2408 Ops/s 13.3774 Ops/s $\color{#d91a1a}-1.02\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1486s 0.1479s 6.7604 Ops/s 6.6332 Ops/s $\color{#35bf28}+1.92\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 30.8131ms 30.2535ms 33.0541 Ops/s 33.2199 Ops/s $\color{#d91a1a}-0.50\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 61.3407ms 61.1378ms 16.3565 Ops/s 16.4184 Ops/s $\color{#d91a1a}-0.38\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1186s 0.1177s 8.4956 Ops/s 8.3459 Ops/s $\color{#35bf28}+1.79\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2644s 0.2633s 3.7978 Ops/s 3.8645 Ops/s $\color{#d91a1a}-1.73\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 43.6464ms 43.3680ms 23.0585 Ops/s 23.0450 Ops/s $\color{#35bf28}+0.06\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 87.1014ms 86.7722ms 11.5244 Ops/s 11.5426 Ops/s $\color{#d91a1a}-0.16\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 43.7959ms 43.5724ms 22.9503 Ops/s 23.0214 Ops/s $\color{#d91a1a}-0.31\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 87.5747ms 87.2331ms 11.4635 Ops/s 11.4673 Ops/s $\color{#d91a1a}-0.03\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 20.6351ms 20.4191ms 48.9738 Ops/s 49.6341 Ops/s $\color{#d91a1a}-1.33\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 41.9267ms 41.6599ms 24.0039 Ops/s 24.1439 Ops/s $\color{#d91a1a}-0.58\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1091s 0.1080s 9.2623 Ops/s 9.3199 Ops/s $\color{#d91a1a}-0.62\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2432s 0.2412s 4.1456 Ops/s 4.0829 Ops/s $\color{#35bf28}+1.54\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 36.4032ms 36.1303ms 27.6776 Ops/s 27.3526 Ops/s $\color{#35bf28}+1.19\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 74.3913ms 72.2662ms 13.8377 Ops/s 13.6142 Ops/s $\color{#35bf28}+1.64\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 38.7390ms 38.0614ms 26.2733 Ops/s 27.2999 Ops/s $\color{#d91a1a}-3.76\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 77.3611ms 75.5407ms 13.2379 Ops/s 13.5949 Ops/s $\color{#d91a1a}-2.63\%$
test_plain_set_nested 50.4010μs 15.1496μs 66.0082 KOps/s 66.6262 KOps/s $\color{#d91a1a}-0.93\%$
test_plain_set_stack_nested 43.5510μs 14.8115μs 67.5151 KOps/s 68.0367 KOps/s $\color{#d91a1a}-0.77\%$
test_plain_set_nested_inplace 58.8910μs 16.1566μs 61.8941 KOps/s 60.7945 KOps/s $\color{#35bf28}+1.81\%$
test_plain_set_stack_nested_inplace 44.6010μs 15.9725μs 62.6074 KOps/s 61.8243 KOps/s $\color{#35bf28}+1.27\%$
test_items 30.2110μs 6.0323μs 165.7747 KOps/s 165.2963 KOps/s $\color{#35bf28}+0.29\%$
test_items_nested 0.4903ms 0.4621ms 2.1639 KOps/s 2.1646 KOps/s $\color{#d91a1a}-0.03\%$
test_items_nested_locked 0.5562ms 0.4679ms 2.1373 KOps/s 2.1485 KOps/s $\color{#d91a1a}-0.52\%$
test_items_nested_leaf 0.1346ms 0.1010ms 9.9046 KOps/s 10.2010 KOps/s $\color{#d91a1a}-2.91\%$
test_items_stack_nested 0.4928ms 0.4654ms 2.1488 KOps/s 2.1647 KOps/s $\color{#d91a1a}-0.73\%$
test_items_stack_nested_leaf 0.1365ms 99.2709μs 10.0734 KOps/s 10.2524 KOps/s $\color{#d91a1a}-1.75\%$
test_items_stack_nested_locked 0.5631ms 0.4669ms 2.1420 KOps/s 2.1555 KOps/s $\color{#d91a1a}-0.63\%$
test_keys 25.3410μs 4.2638μs 234.5349 KOps/s 235.4087 KOps/s $\color{#d91a1a}-0.37\%$
test_keys_nested 0.1537ms 0.1280ms 7.8154 KOps/s 7.7184 KOps/s $\color{#35bf28}+1.26\%$
test_keys_nested_locked 0.7491ms 0.1364ms 7.3288 KOps/s 7.2457 KOps/s $\color{#35bf28}+1.15\%$
test_keys_nested_leaf 0.1557ms 0.1193ms 8.3855 KOps/s 8.3046 KOps/s $\color{#35bf28}+0.97\%$
test_keys_stack_nested 0.1552ms 0.1275ms 7.8409 KOps/s 7.7405 KOps/s $\color{#35bf28}+1.30\%$
test_keys_stack_nested_leaf 0.1794ms 0.1189ms 8.4117 KOps/s 8.3372 KOps/s $\color{#35bf28}+0.89\%$
test_keys_stack_nested_locked 0.1649ms 0.1362ms 7.3434 KOps/s 7.3110 KOps/s $\color{#35bf28}+0.44\%$
test_values 6.6762μs 1.0136μs 986.5621 KOps/s 964.5076 KOps/s $\color{#35bf28}+2.29\%$
test_values_nested 71.7110μs 51.9951μs 19.2326 KOps/s 19.0919 KOps/s $\color{#35bf28}+0.74\%$
test_values_nested_locked 86.1720μs 55.2520μs 18.0989 KOps/s 17.9941 KOps/s $\color{#35bf28}+0.58\%$
test_values_nested_leaf 89.1720μs 59.6251μs 16.7715 KOps/s 16.7345 KOps/s $\color{#35bf28}+0.22\%$
test_values_stack_nested 81.3620μs 51.6582μs 19.3580 KOps/s 19.2634 KOps/s $\color{#35bf28}+0.49\%$
test_values_stack_nested_leaf 0.1236ms 58.9516μs 16.9631 KOps/s 16.8602 KOps/s $\color{#35bf28}+0.61\%$
test_values_stack_nested_locked 89.1320μs 55.1044μs 18.1474 KOps/s 18.1325 KOps/s $\color{#35bf28}+0.08\%$
test_membership 5.1333μs 0.8501μs 1.1764 MOps/s 1.0170 MOps/s $\textbf{\color{#35bf28}+15.67\%}$
test_membership_nested 30.7110μs 2.9672μs 337.0204 KOps/s 343.7267 KOps/s $\color{#d91a1a}-1.95\%$
test_membership_nested_leaf 32.5710μs 2.9575μs 338.1240 KOps/s 358.9459 KOps/s $\textbf{\color{#d91a1a}-5.80\%}$
test_membership_stacked_nested 26.5410μs 2.8594μs 349.7195 KOps/s 356.4626 KOps/s $\color{#d91a1a}-1.89\%$
test_membership_stacked_nested_leaf 23.3600μs 2.8464μs 351.3215 KOps/s 357.1767 KOps/s $\color{#d91a1a}-1.64\%$
test_membership_nested_last 37.3010μs 4.4517μs 224.6321 KOps/s 229.2331 KOps/s $\color{#d91a1a}-2.01\%$
test_membership_nested_leaf_last 40.1110μs 4.4336μs 225.5512 KOps/s 231.2827 KOps/s $\color{#d91a1a}-2.48\%$
test_membership_stacked_nested_last 28.7300μs 4.2760μs 233.8609 KOps/s 233.7864 KOps/s $\color{#35bf28}+0.03\%$
test_membership_stacked_nested_leaf_last 25.3900μs 4.2549μs 235.0252 KOps/s 235.1376 KOps/s $\color{#d91a1a}-0.05\%$
test_nested_getleaf 55.8720μs 21.6437μs 46.2029 KOps/s 46.9654 KOps/s $\color{#d91a1a}-1.62\%$
test_nested_get 50.8210μs 20.4187μs 48.9747 KOps/s 49.9384 KOps/s $\color{#d91a1a}-1.93\%$
test_stacked_getleaf 47.2610μs 20.9227μs 47.7949 KOps/s 47.7132 KOps/s $\color{#35bf28}+0.17\%$
test_stacked_get 53.7820μs 20.0637μs 49.8412 KOps/s 49.3559 KOps/s $\color{#35bf28}+0.98\%$
test_nested_getitemleaf 46.5110μs 22.1184μs 45.2111 KOps/s 45.6168 KOps/s $\color{#d91a1a}-0.89\%$
test_nested_getitem 55.5210μs 20.7192μs 48.2643 KOps/s 48.2576 KOps/s $\color{#35bf28}+0.01\%$
test_stacked_getitemleaf 50.1410μs 21.8869μs 45.6894 KOps/s 46.1658 KOps/s $\color{#d91a1a}-1.03\%$
test_stacked_getitem 49.2510μs 20.7464μs 48.2012 KOps/s 48.3087 KOps/s $\color{#d91a1a}-0.22\%$
test_lock_nested 5.2336ms 0.5124ms 1.9516 KOps/s 1.9606 KOps/s $\color{#d91a1a}-0.45\%$
test_lock_stack_nested 0.5774ms 0.5093ms 1.9636 KOps/s 1.9703 KOps/s $\color{#d91a1a}-0.34\%$
test_unlock_nested 0.4825ms 0.4006ms 2.4964 KOps/s 2.5135 KOps/s $\color{#d91a1a}-0.68\%$
test_unlock_stack_nested 0.4368ms 0.4036ms 2.4778 KOps/s 2.4849 KOps/s $\color{#d91a1a}-0.28\%$
test_flatten_speed 0.1682ms 0.1228ms 8.1460 KOps/s 8.1069 KOps/s $\color{#35bf28}+0.48\%$
test_unflatten_speed 0.6457ms 0.5613ms 1.7816 KOps/s 1.7702 KOps/s $\color{#35bf28}+0.64\%$
test_common_ops 0.1002s 1.5476ms 646.1721 Ops/s 291.8522 Ops/s $\textbf{\color{#35bf28}+121.40\%}$
test_creation 0.1113ms 3.1819μs 314.2801 KOps/s 326.7097 KOps/s $\color{#d91a1a}-3.80\%$
test_creation_empty 38.7210μs 6.9521μs 143.8414 KOps/s 143.2969 KOps/s $\color{#35bf28}+0.38\%$
test_creation_nested_1 33.3510μs 11.4164μs 87.5930 KOps/s 87.1702 KOps/s $\color{#35bf28}+0.48\%$
test_creation_nested_2 41.2910μs 13.3373μs 74.9776 KOps/s 74.8880 KOps/s $\color{#35bf28}+0.12\%$
test_creation_many_keys[10] 61.2710μs 20.9655μs 47.6973 KOps/s 47.5410 KOps/s $\color{#35bf28}+0.33\%$
test_creation_many_keys[50] 0.1286ms 90.3667μs 11.0660 KOps/s 10.9314 KOps/s $\color{#35bf28}+1.23\%$
test_creation_many_keys[100] 0.2306ms 0.1780ms 5.6165 KOps/s 5.6549 KOps/s $\color{#d91a1a}-0.68\%$
test_creation_nested_many_keys[10] 77.6120μs 44.9042μs 22.2696 KOps/s 22.1845 KOps/s $\color{#35bf28}+0.38\%$
test_creation_nested_many_keys[50] 0.2413ms 0.1840ms 5.4361 KOps/s 5.3801 KOps/s $\color{#35bf28}+1.04\%$
test_clone 0.2993ms 13.8070μs 72.4270 KOps/s 73.7503 KOps/s $\color{#d91a1a}-1.79\%$
test_getitem[int] 1.5002ms 15.4463μs 64.7403 KOps/s 65.5869 KOps/s $\color{#d91a1a}-1.29\%$
test_getitem[slice_int] 0.1428ms 24.8879μs 40.1802 KOps/s 40.8881 KOps/s $\color{#d91a1a}-1.73\%$
test_getitem[range] 0.1845ms 66.4545μs 15.0479 KOps/s 15.7399 KOps/s $\color{#d91a1a}-4.40\%$
test_getitem[tuple] 0.1437ms 24.4711μs 40.8645 KOps/s 41.6335 KOps/s $\color{#d91a1a}-1.85\%$
test_getitem[list] 0.1793ms 61.0386μs 16.3831 KOps/s 17.2908 KOps/s $\textbf{\color{#d91a1a}-5.25\%}$
test_setitem_dim[int] 60.7110μs 25.4627μs 39.2732 KOps/s 39.8540 KOps/s $\color{#d91a1a}-1.46\%$
test_setitem_dim[slice_int] 68.3110μs 44.8076μs 22.3177 KOps/s 23.5461 KOps/s $\textbf{\color{#d91a1a}-5.22\%}$
test_setitem_dim[range] 0.1409ms 98.8807μs 10.1132 KOps/s 10.4490 KOps/s $\color{#d91a1a}-3.21\%$
test_setitem_dim[tuple] 67.1120μs 40.3638μs 24.7747 KOps/s 25.9605 KOps/s $\color{#d91a1a}-4.57\%$
test_setitem 50.2710μs 18.0411μs 55.4289 KOps/s 55.6244 KOps/s $\color{#d91a1a}-0.35\%$
test_set 39.2810μs 17.3149μs 57.7539 KOps/s 56.0014 KOps/s $\color{#35bf28}+3.13\%$
test_set_shared 0.6733ms 0.2143ms 4.6656 KOps/s 4.5967 KOps/s $\color{#35bf28}+1.50\%$
test_update 0.1998ms 22.4708μs 44.5022 KOps/s 43.9135 KOps/s $\color{#35bf28}+1.34\%$
test_update_nested 71.5610μs 34.5820μs 28.9168 KOps/s 28.5938 KOps/s $\color{#35bf28}+1.13\%$
test_update__nested 0.5245ms 34.2839μs 29.1682 KOps/s 28.3156 KOps/s $\color{#35bf28}+3.01\%$
test_set_nested 58.0410μs 19.3215μs 51.7557 KOps/s 53.8032 KOps/s $\color{#d91a1a}-3.81\%$
test_set_nested_new 67.1920μs 24.3371μs 41.0896 KOps/s 43.0246 KOps/s $\color{#d91a1a}-4.50\%$
test_select 71.4510μs 40.8240μs 24.4954 KOps/s 24.5021 KOps/s $\color{#d91a1a}-0.03\%$
test_select_nested 0.1078ms 73.2843μs 13.6455 KOps/s 13.4249 KOps/s $\color{#35bf28}+1.64\%$
test_exclude_nested 0.1256ms 89.6546μs 11.1539 KOps/s 11.0315 KOps/s $\color{#35bf28}+1.11\%$
test_empty[True] 0.4747ms 0.3890ms 2.5710 KOps/s 2.5198 KOps/s $\color{#35bf28}+2.03\%$
test_empty[False] 7.9653μs 1.3572μs 736.8130 KOps/s 720.3753 KOps/s $\color{#35bf28}+2.28\%$
test_to 0.1090ms 71.0089μs 14.0827 KOps/s 14.2637 KOps/s $\color{#d91a1a}-1.27\%$
test_to_nonblocking 0.1299ms 71.2589μs 14.0333 KOps/s 14.5610 KOps/s $\color{#d91a1a}-3.62\%$
test_unbind_speed 1.9634ms 0.3462ms 2.8886 KOps/s 2.9499 KOps/s $\color{#d91a1a}-2.08\%$
test_unbind_speed_stack0 0.4111ms 0.3400ms 2.9413 KOps/s 3.0185 KOps/s $\color{#d91a1a}-2.56\%$
test_unbind_speed_stack1 0.1145s 0.8519ms 1.1739 KOps/s 1.1671 KOps/s $\color{#35bf28}+0.58\%$
test_split 1.2630ms 1.1403ms 876.9882 Ops/s 873.4443 Ops/s $\color{#35bf28}+0.41\%$
test_chunk 0.1142s 1.2313ms 812.1362 Ops/s 912.9578 Ops/s $\textbf{\color{#d91a1a}-11.04\%}$
test_to_cpu_blocking 20.3945ms 20.2134ms 49.4722 Ops/s 49.1974 Ops/s $\color{#35bf28}+0.56\%$
test_to_cpu_global_sync 12.1817ms 11.9718ms 83.5297 Ops/s 83.4531 Ops/s $\color{#35bf28}+0.09\%$
test_to_cpu_event_sync 14.3528ms 12.9636ms 77.1388 Ops/s 68.7151 Ops/s $\textbf{\color{#35bf28}+12.26\%}$
test_to_cpu_default 13.3093ms 12.9817ms 77.0314 Ops/s 76.8158 Ops/s $\color{#35bf28}+0.28\%$
test_consolidate[False-None] 0.1440s 4.7499ms 210.5295 Ops/s 238.7978 Ops/s $\textbf{\color{#d91a1a}-11.84\%}$
test_consolidate[default-None] 3.5014ms 2.1783ms 459.0839 Ops/s 456.1753 Ops/s $\color{#35bf28}+0.64\%$
test_consolidate[reduce-overhead-None] 0.2019s 2.5306ms 395.1687 Ops/s 253.4429 Ops/s $\textbf{\color{#35bf28}+55.92\%}$
test_consolidate_njt[False-None] 9.4447ms 8.3104ms 120.3310 Ops/s 117.2164 Ops/s $\color{#35bf28}+2.66\%$
test_to[False-False-None] 2.2770ms 2.1730ms 460.1930 Ops/s 457.9141 Ops/s $\color{#35bf28}+0.50\%$
test_to[True-False-None] 2.1357ms 1.8993ms 526.4959 Ops/s 523.4112 Ops/s $\color{#35bf28}+0.59\%$
test_to[within-False-None] 6.4597ms 6.1556ms 162.4543 Ops/s 161.5984 Ops/s $\color{#35bf28}+0.53\%$
test_to[True-default-None] 10.8314ms 10.5931ms 94.4008 Ops/s 92.5252 Ops/s $\color{#35bf28}+2.03\%$
test_to_njt[False-False-None] 8.7493ms 8.6164ms 116.0579 Ops/s 114.7961 Ops/s $\color{#35bf28}+1.10\%$
test_to_njt[True-False-None] 7.0387ms 6.8192ms 146.6444 Ops/s 145.7784 Ops/s $\color{#35bf28}+0.59\%$
test_to_njt[within-False-None] 0.2030s 18.3073ms 54.6231 Ops/s 64.7832 Ops/s $\textbf{\color{#d91a1a}-15.68\%}$
test_creation[device0] 0.3998ms 0.1171ms 8.5408 KOps/s 8.5329 KOps/s $\color{#35bf28}+0.09\%$
test_creation_from_tensor 0.4143ms 0.1141ms 8.7659 KOps/s 8.7290 KOps/s $\color{#35bf28}+0.42\%$
test_add_one[memmap_tensor0] 0.3120ms 6.8733μs 145.4908 KOps/s 140.1517 KOps/s $\color{#35bf28}+3.81\%$
test_contiguous[memmap_tensor0] 27.2710μs 0.6502μs 1.5380 MOps/s 2.2139 MOps/s $\textbf{\color{#d91a1a}-30.53\%}$
test_stack[memmap_tensor0] 29.5910μs 4.3950μs 227.5335 KOps/s 232.0620 KOps/s $\color{#d91a1a}-1.95\%$
test_memmaptd_index 1.3184ms 0.2777ms 3.6006 KOps/s 3.6842 KOps/s $\color{#d91a1a}-2.27\%$
test_memmaptd_index_astensor 0.5364ms 0.3743ms 2.6714 KOps/s 2.6768 KOps/s $\color{#d91a1a}-0.20\%$
test_memmaptd_index_op 0.8031ms 0.6377ms 1.5681 KOps/s 1.5736 KOps/s $\color{#d91a1a}-0.35\%$
test_serialize_model 0.1387s 0.1352s 7.3958 Ops/s 7.4284 Ops/s $\color{#d91a1a}-0.44\%$
test_serialize_model_pickle 1.3716s 1.2146s 0.8233 Ops/s 0.8238 Ops/s $\color{#d91a1a}-0.06\%$
test_serialize_weights 0.1367s 0.1340s 7.4602 Ops/s 7.4714 Ops/s $\color{#d91a1a}-0.15\%$
test_serialize_weights_returnearly 0.4677s 89.9542ms 11.1168 Ops/s 15.5288 Ops/s $\textbf{\color{#d91a1a}-28.41\%}$
test_serialize_weights_pickle 1.3652s 1.2116s 0.8254 Ops/s 0.8233 Ops/s $\color{#35bf28}+0.25\%$
test_reshape_pytree 0.2101ms 32.6218μs 30.6544 KOps/s 30.5810 KOps/s $\color{#35bf28}+0.24\%$
test_reshape_td 79.1910μs 45.4824μs 21.9865 KOps/s 22.1334 KOps/s $\color{#d91a1a}-0.66\%$
test_view_pytree 0.2344ms 32.0465μs 31.2046 KOps/s 31.0559 KOps/s $\color{#35bf28}+0.48\%$
test_view_td 99.8120μs 53.5682μs 18.6678 KOps/s 18.7333 KOps/s $\color{#d91a1a}-0.35\%$
test_unbind_pytree 0.2434ms 36.7935μs 27.1787 KOps/s 26.8399 KOps/s $\color{#35bf28}+1.26\%$
test_unbind_td 0.1989ms 51.4388μs 19.4406 KOps/s 19.8458 KOps/s $\color{#d91a1a}-2.04\%$
test_split_pytree 0.2631ms 42.2028μs 23.6951 KOps/s 23.6975 KOps/s $\color{#d91a1a}-0.01\%$
test_split_td 0.1752ms 65.8830μs 15.1784 KOps/s 15.2475 KOps/s $\color{#d91a1a}-0.45\%$
test_add_pytree 0.2367ms 41.3153μs 24.2041 KOps/s 23.7557 KOps/s $\color{#35bf28}+1.89\%$
test_add_td 0.1124ms 54.2650μs 18.4281 KOps/s 18.5396 KOps/s $\color{#d91a1a}-0.60\%$
test_compile_add_one_nested[tensordict-compile] 0.3198ms 0.2295ms 4.3565 KOps/s 4.2574 KOps/s $\color{#35bf28}+2.33\%$
test_compile_add_one_nested[tensordict-eager] 0.2917ms 0.1998ms 5.0061 KOps/s 5.0344 KOps/s $\color{#d91a1a}-0.56\%$
test_compile_add_one_nested[pytree-compile] 0.3642ms 0.1372ms 7.2885 KOps/s 7.1778 KOps/s $\color{#35bf28}+1.54\%$
test_compile_add_one_nested[pytree-eager] 0.5060ms 0.1797ms 5.5663 KOps/s 5.5394 KOps/s $\color{#35bf28}+0.49\%$
test_compile_copy_nested[tensordict-compile] 80.9320μs 14.9377μs 66.9449 KOps/s 66.3319 KOps/s $\color{#35bf28}+0.92\%$
test_compile_copy_nested[tensordict-eager] 0.3589ms 52.8786μs 18.9112 KOps/s 18.4460 KOps/s $\color{#35bf28}+2.52\%$
test_compile_copy_nested[pytree-compile] 0.1400ms 15.3781μs 65.0274 KOps/s 65.0148 KOps/s $\color{#35bf28}+0.02\%$
test_compile_copy_nested[pytree-eager] 0.4098ms 67.2632μs 14.8670 KOps/s 14.5673 KOps/s $\color{#35bf28}+2.06\%$
test_compile_add_one_flat[tensordict-compile] 0.3788ms 0.2552ms 3.9190 KOps/s 3.7421 KOps/s $\color{#35bf28}+4.72\%$
test_compile_add_one_flat[tensordict-eager] 0.3366ms 0.2818ms 3.5490 KOps/s 3.5335 KOps/s $\color{#35bf28}+0.44\%$
test_compile_add_one_flat[tensorclass-compile] 0.3824ms 0.1889ms 5.2942 KOps/s 5.1154 KOps/s $\color{#35bf28}+3.50\%$
test_compile_add_one_flat[tensorclass-eager] 0.1182ms 73.9418μs 13.5241 KOps/s 13.7000 KOps/s $\color{#d91a1a}-1.28\%$
test_compile_add_one_flat[pytree-compile] 0.2972ms 0.1934ms 5.1705 KOps/s 5.0560 KOps/s $\color{#35bf28}+2.26\%$
test_compile_add_one_flat[pytree-eager] 0.8094ms 0.5322ms 1.8790 KOps/s 1.9173 KOps/s $\color{#d91a1a}-2.00\%$
test_compile_add_self_flat[tensordict-eager] 0.4962ms 0.3312ms 3.0195 KOps/s 3.0266 KOps/s $\color{#d91a1a}-0.23\%$
test_compile_add_self_flat[tensordict-compile] 0.3803ms 0.2543ms 3.9318 KOps/s 3.6209 KOps/s $\textbf{\color{#35bf28}+8.59\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1539ms 90.6481μs 11.0317 KOps/s 10.7459 KOps/s $\color{#35bf28}+2.66\%$
test_compile_add_self_flat[tensorclass-compile] 0.3118ms 0.1935ms 5.1670 KOps/s 4.9313 KOps/s $\color{#35bf28}+4.78\%$
test_compile_add_self_flat[pytree-eager] 0.6900ms 0.4397ms 2.2742 KOps/s 2.2997 KOps/s $\color{#d91a1a}-1.11\%$
test_compile_add_self_flat[pytree-compile] 0.2691ms 0.1931ms 5.1780 KOps/s 5.0260 KOps/s $\color{#35bf28}+3.02\%$
test_compile_copy_flat[tensordict-compile] 57.5610μs 19.6166μs 50.9772 KOps/s 51.5445 KOps/s $\color{#d91a1a}-1.10\%$
test_compile_copy_flat[tensordict-eager] 97.0820μs 40.6915μs 24.5751 KOps/s 24.1760 KOps/s $\color{#35bf28}+1.65\%$
test_compile_copy_flat[pytree-compile] 0.2222ms 21.4104μs 46.7062 KOps/s 47.1363 KOps/s $\color{#d91a1a}-0.91\%$
test_compile_copy_flat[pytree-eager] 0.1946s 62.1313μs 16.0949 KOps/s 19.2861 KOps/s $\textbf{\color{#d91a1a}-16.55\%}$
test_compile_assign_and_add[tensordict-compile] 2.1826ms 0.2382ms 4.1983 KOps/s 3.8825 KOps/s $\textbf{\color{#35bf28}+8.14\%}$
test_compile_assign_and_add[tensordict-eager] 3.4878ms 3.2973ms 303.2798 Ops/s 304.6364 Ops/s $\color{#d91a1a}-0.45\%$
test_compile_assign_and_add[pytree-compile] 2.1055ms 0.1829ms 5.4673 KOps/s 5.3322 KOps/s $\color{#35bf28}+2.53\%$
test_compile_assign_and_add[pytree-eager] 3.0226ms 2.8213ms 354.4453 Ops/s 358.1774 Ops/s $\color{#d91a1a}-1.04\%$
test_compile_indexing[tensor-tensordict-compile] 0.2498ms 0.1341ms 7.4552 KOps/s 7.1573 KOps/s $\color{#35bf28}+4.16\%$
test_compile_indexing[tensor-tensordict-eager] 0.3766ms 73.8761μs 13.5362 KOps/s 13.7719 KOps/s $\color{#d91a1a}-1.71\%$
test_compile_indexing[tensor-tensorclass-compile] 0.2612ms 0.1191ms 8.3992 KOps/s 8.3178 KOps/s $\color{#35bf28}+0.98\%$
test_compile_indexing[tensor-tensorclass-eager] 0.3243ms 44.0408μs 22.7062 KOps/s 22.8532 KOps/s $\color{#d91a1a}-0.64\%$
test_compile_indexing[tensor-pytree-compile] 0.1685ms 0.1197ms 8.3550 KOps/s 8.2837 KOps/s $\color{#35bf28}+0.86\%$
test_compile_indexing[tensor-pytree-eager] 0.2848ms 44.1385μs 22.6560 KOps/s 22.8771 KOps/s $\color{#d91a1a}-0.97\%$
test_compile_indexing[slice-tensordict-compile] 0.2409ms 83.0641μs 12.0389 KOps/s 11.5537 KOps/s $\color{#35bf28}+4.20\%$
test_compile_indexing[slice-tensordict-eager] 0.2206ms 30.5280μs 32.7568 KOps/s 37.3000 KOps/s $\textbf{\color{#d91a1a}-12.18\%}$
test_compile_indexing[slice-tensorclass-compile] 0.1907ms 69.6109μs 14.3656 KOps/s 14.1306 KOps/s $\color{#35bf28}+1.66\%$
test_compile_indexing[slice-tensorclass-eager] 0.2516ms 22.0698μs 45.3108 KOps/s 44.8692 KOps/s $\color{#35bf28}+0.98\%$
test_compile_indexing[slice-pytree-compile] 0.1152ms 70.2016μs 14.2447 KOps/s 14.0492 KOps/s $\color{#35bf28}+1.39\%$
test_compile_indexing[slice-pytree-eager] 0.2700ms 21.9480μs 45.5622 KOps/s 45.0827 KOps/s $\color{#35bf28}+1.06\%$
test_compile_indexing[int-tensordict-compile] 0.1380ms 83.6678μs 11.9520 KOps/s 11.4053 KOps/s $\color{#35bf28}+4.79\%$
test_compile_indexing[int-tensordict-eager] 0.2146ms 26.8520μs 37.2412 KOps/s 37.7654 KOps/s $\color{#d91a1a}-1.39\%$
test_compile_indexing[int-tensorclass-compile] 0.1077ms 69.9354μs 14.2989 KOps/s 14.1251 KOps/s $\color{#35bf28}+1.23\%$
test_compile_indexing[int-tensorclass-eager] 0.2425ms 21.9639μs 45.5293 KOps/s 45.1383 KOps/s $\color{#35bf28}+0.87\%$
test_compile_indexing[int-pytree-compile] 0.1289ms 69.7192μs 14.3433 KOps/s 13.9790 KOps/s $\color{#35bf28}+2.61\%$
test_compile_indexing[int-pytree-eager] 0.2666ms 22.0564μs 45.3383 KOps/s 45.3734 KOps/s $\color{#d91a1a}-0.08\%$
test_compile_replace[single-eager] 91.9720μs 46.4575μs 21.5251 KOps/s 21.5109 KOps/s $\color{#35bf28}+0.07\%$
test_compile_replace[single-compile] 0.2554ms 0.1321ms 7.5675 KOps/s 7.5429 KOps/s $\color{#35bf28}+0.33\%$
test_compile_replace[multi-eager] 0.6502ms 0.5593ms 1.7880 KOps/s 1.8039 KOps/s $\color{#d91a1a}-0.88\%$
test_compile_replace[multi-compile] 0.3618ms 0.1381ms 7.2392 KOps/s 7.0511 KOps/s $\color{#35bf28}+2.67\%$
test_compile_tc_getattr_20[eager] 0.2130ms 0.1751ms 5.7095 KOps/s 5.8933 KOps/s $\color{#d91a1a}-3.12\%$
test_compile_tc_getattr_20[compile] 0.3289ms 0.1478ms 6.7653 KOps/s 6.6614 KOps/s $\color{#35bf28}+1.56\%$
test_compile_clone_shallow[20-eager] 56.2820μs 19.3532μs 51.6710 KOps/s 52.0547 KOps/s $\color{#d91a1a}-0.74\%$
test_compile_clone_shallow[20-compile] 82.0910μs 16.6449μs 60.0783 KOps/s 57.8808 KOps/s $\color{#35bf28}+3.80\%$
test_compile_clone_shallow[40-eager] 65.3310μs 33.5056μs 29.8457 KOps/s 29.7970 KOps/s $\color{#35bf28}+0.16\%$
test_compile_clone_shallow[40-compile] 77.1020μs 18.0766μs 55.3200 KOps/s 53.6467 KOps/s $\color{#35bf28}+3.12\%$
test_compile_clone_shallow[80-eager] 95.8820μs 61.6443μs 16.2221 KOps/s 15.8449 KOps/s $\color{#35bf28}+2.38\%$
test_compile_clone_shallow[80-compile] 83.7920μs 21.7068μs 46.0685 KOps/s 45.3615 KOps/s $\color{#35bf28}+1.56\%$
test_compile_update_inplace[eager] 0.1049ms 61.0081μs 16.3913 KOps/s 16.2366 KOps/s $\color{#35bf28}+0.95\%$
test_compile_update_inplace[compile] 0.2794ms 0.1456ms 6.8703 KOps/s 6.7599 KOps/s $\color{#35bf28}+1.63\%$
test_mod_add[eager] 0.2907ms 0.2257ms 4.4302 KOps/s 4.3975 KOps/s $\color{#35bf28}+0.74\%$
test_mod_add[compile] 0.2764ms 0.1310ms 7.6362 KOps/s 7.4320 KOps/s $\color{#35bf28}+2.75\%$
test_mod_add[compile-overhead] 0.3565ms 0.1771ms 5.6464 KOps/s 5.5478 KOps/s $\color{#35bf28}+1.78\%$
test_mod_wrap[eager] 0.5869ms 0.4688ms 2.1329 KOps/s 2.1307 KOps/s $\color{#35bf28}+0.10\%$
test_mod_wrap[compile] 0.5092ms 0.4018ms 2.4886 KOps/s 2.4780 KOps/s $\color{#35bf28}+0.43\%$
test_mod_wrap[compile-overhead] 8.2357ms 4.6162ms 216.6261 Ops/s 218.2148 Ops/s $\color{#d91a1a}-0.73\%$
test_mod_wrap_and_backward[eager] 3.8548ms 1.7515ms 570.9515 Ops/s 586.1861 Ops/s $\color{#d91a1a}-2.60\%$
test_mod_wrap_and_backward[compile] 2.8493ms 1.6974ms 589.1206 Ops/s 645.6032 Ops/s $\textbf{\color{#d91a1a}-8.75\%}$
test_mod_wrap_and_backward[compile-overhead] 1.5599ms 1.0553ms 947.6404 Ops/s 1.0448 KOps/s $\textbf{\color{#d91a1a}-9.30\%}$
test_seq_add[eager] 0.5777ms 0.5069ms 1.9728 KOps/s 1.9683 KOps/s $\color{#35bf28}+0.23\%$
test_seq_add[compile] 0.2632ms 0.1404ms 7.1245 KOps/s 6.7784 KOps/s $\textbf{\color{#35bf28}+5.11\%}$
test_seq_add[compile-overhead] 0.3088ms 0.1871ms 5.3434 KOps/s 5.2867 KOps/s $\color{#35bf28}+1.07\%$
test_seq_wrap[eager] 1.5940ms 1.4132ms 707.6287 Ops/s 711.6851 Ops/s $\color{#d91a1a}-0.57\%$
test_seq_wrap[compile] 0.5328ms 0.4195ms 2.3838 KOps/s 2.3674 KOps/s $\color{#35bf28}+0.69\%$
test_seq_wrap[compile-overhead] 0.4273ms 0.3039ms 3.2906 KOps/s 3.2413 KOps/s $\color{#35bf28}+1.52\%$
test_func_call_runtime[False-eager] 0.9216ms 0.8515ms 1.1744 KOps/s 1.2579 KOps/s $\textbf{\color{#d91a1a}-6.64\%}$
test_func_call_runtime[False-compile] 1.0943ms 0.9843ms 1.0160 KOps/s 1.0043 KOps/s $\color{#35bf28}+1.16\%$
test_func_call_runtime[False-compile-overhead] 0.5544ms 0.5054ms 1.9787 KOps/s 1.9481 KOps/s $\color{#35bf28}+1.57\%$
test_func_call_runtime[True-eager] 1.1289ms 1.0705ms 934.1602 Ops/s 935.9326 Ops/s $\color{#d91a1a}-0.19\%$
test_func_call_runtime[True-compile] 1.1392ms 0.9986ms 1.0014 KOps/s 996.7498 Ops/s $\color{#35bf28}+0.47\%$
test_func_call_runtime[True-compile-overhead] 0.5671ms 0.5217ms 1.9170 KOps/s 1.8984 KOps/s $\color{#35bf28}+0.98\%$
test_func_call_cm_runtime[False-eager] 0.9191ms 0.8066ms 1.2398 KOps/s 1.2568 KOps/s $\color{#d91a1a}-1.35\%$
test_func_call_cm_runtime[False-compile] 1.1485ms 0.9669ms 1.0342 KOps/s 1.0218 KOps/s $\color{#35bf28}+1.21\%$
test_func_call_cm_runtime[False-compile-overhead] 0.6989ms 0.4864ms 2.0558 KOps/s 2.0202 KOps/s $\color{#35bf28}+1.77\%$
test_func_call_cm_runtime[True-eager] 1.3953ms 1.2188ms 820.5026 Ops/s 820.6464 Ops/s $\color{#d91a1a}-0.02\%$
test_func_call_cm_runtime[True-compile] 1.1116ms 1.0255ms 975.1204 Ops/s 967.4203 Ops/s $\color{#35bf28}+0.80\%$
test_func_call_cm_runtime[True-compile-overhead] 0.6151ms 0.5457ms 1.8325 KOps/s 1.7959 KOps/s $\color{#35bf28}+2.04\%$
test_vmap_func_call_cm_runtime[eager] 2.9065ms 2.4164ms 413.8377 Ops/s 413.2458 Ops/s $\color{#35bf28}+0.14\%$
test_vmap_func_call_cm_runtime[compile] 3.0785ms 2.5884ms 386.3464 Ops/s 384.3738 Ops/s $\color{#35bf28}+0.51\%$
test_vmap_func_call_cm_runtime[compile-overhead] 3.0786ms 2.5996ms 384.6718 Ops/s 385.0924 Ops/s $\color{#d91a1a}-0.11\%$
test_distributed 2.8918ms 0.1620ms 6.1741 KOps/s 6.5775 KOps/s $\textbf{\color{#d91a1a}-6.13\%}$
test_tdmodule 0.6916ms 0.1938ms 5.1602 KOps/s 5.0388 KOps/s $\color{#35bf28}+2.41\%$
test_tdmodule_dispatch 0.4285ms 0.3746ms 2.6695 KOps/s 2.6408 KOps/s $\color{#35bf28}+1.09\%$
test_tdseq 46.5710μs 26.5519μs 37.6621 KOps/s 36.4841 KOps/s $\color{#35bf28}+3.23\%$
test_tdseq_dispatch 0.2630ms 0.2182ms 4.5838 KOps/s 4.5688 KOps/s $\color{#35bf28}+0.33\%$
test_instantiation_functorch 2.1952ms 2.0641ms 484.4756 Ops/s 486.9358 Ops/s $\color{#d91a1a}-0.51\%$
test_exec_functorch 0.2344ms 0.1828ms 5.4695 KOps/s 5.5144 KOps/s $\color{#d91a1a}-0.82\%$
test_exec_functional_call 0.2162ms 0.1603ms 6.2364 KOps/s 6.2756 KOps/s $\color{#d91a1a}-0.62\%$
test_exec_td_decorator 0.5177ms 0.2479ms 4.0346 KOps/s 4.0104 KOps/s $\color{#35bf28}+0.60\%$
test_vmap_mlp_speed_decorator[True-True] 1.2552ms 1.0431ms 958.6889 Ops/s 958.5653 Ops/s $\color{#35bf28}+0.01\%$
test_vmap_mlp_speed_decorator[True-False] 1.2631ms 1.0318ms 969.1881 Ops/s 958.6505 Ops/s $\color{#35bf28}+1.10\%$
test_vmap_mlp_speed_decorator[False-True] 0.9378ms 0.7392ms 1.3528 KOps/s 1.3540 KOps/s $\color{#d91a1a}-0.09\%$
test_vmap_mlp_speed_decorator[False-False] 0.9204ms 0.7377ms 1.3556 KOps/s 1.3594 KOps/s $\color{#d91a1a}-0.28\%$
test_vmap_transformer_speed_decorator[True-True] 21.2685ms 21.0902ms 47.4154 Ops/s 47.3088 Ops/s $\color{#35bf28}+0.23\%$
test_vmap_transformer_speed_decorator[True-False] 21.9044ms 21.2126ms 47.1419 Ops/s 47.2440 Ops/s $\color{#d91a1a}-0.22\%$
test_vmap_transformer_speed_decorator[False-True] 21.6541ms 20.7686ms 48.1497 Ops/s 48.1540 Ops/s $-0.01\%$
test_vmap_transformer_speed_decorator[False-False] 21.5101ms 20.7573ms 48.1759 Ops/s 48.1553 Ops/s $\color{#35bf28}+0.04\%$
test_to_module_speed[True] 2.2670ms 1.6403ms 609.6499 Ops/s 605.6901 Ops/s $\color{#35bf28}+0.65\%$
test_to_module_speed[False] 2.3557ms 1.6137ms 619.6902 Ops/s 621.4832 Ops/s $\color{#d91a1a}-0.29\%$
test_to_module_plain_tensor_speed[None] 2.4136ms 1.7174ms 582.2760 Ops/s 580.0463 Ops/s $\color{#35bf28}+0.38\%$
test_to_module_plain_tensor_speed[False] 2.2521ms 1.5770ms 634.1041 Ops/s 635.7084 Ops/s $\color{#d91a1a}-0.25\%$
test_to_module_plain_tensor_speed[True] 3.9883ms 3.3487ms 298.6200 Ops/s 296.0909 Ops/s $\color{#35bf28}+0.85\%$
test_tc_init 78.8320μs 42.8518μs 23.3363 KOps/s 23.1978 KOps/s $\color{#35bf28}+0.60\%$
test_tc_init_tensor_only 44.5010μs 9.3408μs 107.0569 KOps/s 111.0841 KOps/s $\color{#d91a1a}-3.63\%$
test_tc_init_nested 0.1222ms 84.7901μs 11.7938 KOps/s 11.8080 KOps/s $\color{#d91a1a}-0.12\%$
test_tc_init_many_fields 45.1810μs 14.5791μs 68.5913 KOps/s 66.2025 KOps/s $\color{#35bf28}+3.61\%$
test_tc_first_layer_tensor 35.0010μs 1.7952μs 557.0540 KOps/s 557.2644 KOps/s $\color{#d91a1a}-0.04\%$
test_tc_first_layer_tensor_only 1.6500μs 0.3994μs 2.5036 MOps/s 2.5366 MOps/s $\color{#d91a1a}-1.30\%$
test_tc_first_layer_tensor_set 36.3700μs 3.9415μs 253.7097 KOps/s 253.9350 KOps/s $\color{#d91a1a}-0.09\%$
test_tc_first_layer_tensor_only_set 22.7200μs 3.2516μs 307.5439 KOps/s 306.2275 KOps/s $\color{#35bf28}+0.43\%$
test_tc_first_layer_nontensor 29.5400μs 6.1777μs 161.8738 KOps/s 163.0323 KOps/s $\color{#d91a1a}-0.71\%$
test_tc_second_layer_tensor 26.5600μs 4.3920μs 227.6878 KOps/s 227.0323 KOps/s $\color{#35bf28}+0.29\%$
test_tc_second_layer_nontensor 35.8710μs 8.6561μs 115.5252 KOps/s 116.4225 KOps/s $\color{#d91a1a}-0.77\%$
test_unbind 0.2777s 13.6953ms 73.0175 Ops/s 67.7747 Ops/s $\textbf{\color{#35bf28}+7.74\%}$
test_full_like 4.4960ms 4.3776ms 228.4376 Ops/s 227.7737 Ops/s $\color{#35bf28}+0.29\%$
test_zeros_like 4.7965ms 4.3671ms 228.9832 Ops/s 228.4952 Ops/s $\color{#35bf28}+0.21\%$
test_ones_like 4.6222ms 4.3710ms 228.7819 Ops/s 228.1559 Ops/s $\color{#35bf28}+0.27\%$
test_clone 6.8888ms 6.5034ms 153.7648 Ops/s 154.5405 Ops/s $\color{#d91a1a}-0.50\%$
test_squeeze 74.0810μs 14.1212μs 70.8153 KOps/s 69.7651 KOps/s $\color{#35bf28}+1.51\%$
test_unsqueeze 0.1690ms 0.1116ms 8.9643 KOps/s 9.0364 KOps/s $\color{#d91a1a}-0.80\%$
test_split 0.3796ms 0.1806ms 5.5372 KOps/s 5.5336 KOps/s $\color{#35bf28}+0.06\%$
test_permute 0.2626ms 0.2044ms 4.8925 KOps/s 4.5767 KOps/s $\textbf{\color{#35bf28}+6.90\%}$
test_stack 18.3821ms 18.1433ms 55.1167 Ops/s 54.9151 Ops/s $\color{#35bf28}+0.37\%$
test_cat 26.7808ms 26.4076ms 37.8679 Ops/s 54.9261 Ops/s $\textbf{\color{#d91a1a}-31.06\%}$
test_sequential_tensordict 0.7186ms 0.5720ms 1.7483 KOps/s 1.7417 KOps/s $\color{#35bf28}+0.38\%$
test_sequential_graph_module 0.1609ms 0.1165ms 8.5851 KOps/s 8.2759 KOps/s $\color{#35bf28}+3.74\%$
test_nested_tensordict 0.9915ms 0.8182ms 1.2223 KOps/s 1.2129 KOps/s $\color{#35bf28}+0.77\%$
test_nested_graph_module 0.1934ms 0.1267ms 7.8937 KOps/s 7.7616 KOps/s $\color{#35bf28}+1.70\%$

@github-actions

github-actions Bot commented Jun 18, 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}16$. Worsened: $\large\color{#d91a1a}17$.

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_collector_preallocated_write[stack_out-cpu-32-2] 0.2277ms 0.1699ms 5.8871 KOps/s 5.9948 KOps/s $\color{#d91a1a}-1.80\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.4257ms 0.2995ms 3.3384 KOps/s 3.3242 KOps/s $\color{#35bf28}+0.43\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.9337ms 0.7846ms 1.2745 KOps/s 1.2573 KOps/s $\color{#35bf28}+1.36\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.4001ms 0.3430ms 2.9155 KOps/s 2.8404 KOps/s $\color{#35bf28}+2.64\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.3382ms 1.1800ms 847.4253 Ops/s 898.5357 Ops/s $\textbf{\color{#d91a1a}-5.69\%}$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.9106ms 7.5092ms 133.1709 Ops/s 132.7159 Ops/s $\color{#35bf28}+0.34\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.2980ms 0.2526ms 3.9581 KOps/s 4.0134 KOps/s $\color{#d91a1a}-1.38\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.4762ms 0.3781ms 2.6449 KOps/s 2.5833 KOps/s $\color{#35bf28}+2.39\%$
test_collector_preallocated_write[stack_out-cuda-32-32] 0.9410ms 0.8338ms 1.1994 KOps/s 1.1966 KOps/s $\color{#35bf28}+0.23\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.3404ms 0.2621ms 3.8151 KOps/s 3.8290 KOps/s $\color{#d91a1a}-0.36\%$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.4690ms 0.3784ms 2.6429 KOps/s 2.5710 KOps/s $\color{#35bf28}+2.80\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 0.9702ms 0.8915ms 1.1217 KOps/s 1.1166 KOps/s $\color{#35bf28}+0.46\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2723ms 0.2167ms 4.6144 KOps/s 4.5894 KOps/s $\color{#35bf28}+0.54\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 0.8952ms 0.8411ms 1.1889 KOps/s 1.1699 KOps/s $\color{#35bf28}+1.63\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.4559ms 3.3588ms 297.7252 Ops/s 295.5500 Ops/s $\color{#35bf28}+0.74\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.4564ms 0.3689ms 2.7106 KOps/s 2.6546 KOps/s $\color{#35bf28}+2.11\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 2.1429ms 1.7377ms 575.4587 Ops/s 574.0516 Ops/s $\color{#35bf28}+0.25\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 11.0156ms 10.4782ms 95.4360 Ops/s 95.7056 Ops/s $\color{#d91a1a}-0.28\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.4800ms 0.3979ms 2.5133 KOps/s 2.5687 KOps/s $\color{#d91a1a}-2.16\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.5836ms 1.4862ms 672.8377 Ops/s 642.3122 Ops/s $\color{#35bf28}+4.75\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 5.9446ms 5.8233ms 171.7238 Ops/s 164.0000 Ops/s $\color{#35bf28}+4.71\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.5225ms 0.4029ms 2.4821 KOps/s 2.4030 KOps/s $\color{#35bf28}+3.29\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.6332ms 1.5112ms 661.7089 Ops/s 641.6665 Ops/s $\color{#35bf28}+3.12\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 6.1822ms 5.9046ms 169.3583 Ops/s 163.7768 Ops/s $\color{#35bf28}+3.41\%$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.5707ms 0.1770ms 5.6491 KOps/s 5.8126 KOps/s $\color{#d91a1a}-2.81\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.7647ms 0.6894ms 1.4505 KOps/s 1.4809 KOps/s $\color{#d91a1a}-2.06\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 2.8366ms 2.7362ms 365.4754 Ops/s 373.1307 Ops/s $\color{#d91a1a}-2.05\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.3999ms 0.3223ms 3.1025 KOps/s 3.1189 KOps/s $\color{#d91a1a}-0.53\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.7131ms 1.4581ms 685.8220 Ops/s 685.5735 Ops/s $\color{#35bf28}+0.04\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 9.9887ms 9.7393ms 102.6768 Ops/s 103.7295 Ops/s $\color{#d91a1a}-1.01\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.4573ms 0.3285ms 3.0442 KOps/s 3.1038 KOps/s $\color{#d91a1a}-1.92\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.3144ms 1.2213ms 818.8164 Ops/s 837.8333 Ops/s $\color{#d91a1a}-2.27\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.9244ms 4.7905ms 208.7464 Ops/s 213.9089 Ops/s $\color{#d91a1a}-2.41\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.3882ms 0.3299ms 3.0308 KOps/s 3.1006 KOps/s $\color{#d91a1a}-2.25\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.3152ms 1.2266ms 815.2355 Ops/s 833.6806 Ops/s $\color{#d91a1a}-2.21\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 4.9175ms 4.8099ms 207.9032 Ops/s 213.2442 Ops/s $\color{#d91a1a}-2.50\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.1621ms 0.1256ms 7.9638 KOps/s 8.3228 KOps/s $\color{#d91a1a}-4.31\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.5599ms 0.4890ms 2.0450 KOps/s 2.1207 KOps/s $\color{#d91a1a}-3.57\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 2.0204ms 1.9536ms 511.8660 Ops/s 532.1663 Ops/s $\color{#d91a1a}-3.81\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.3528ms 0.2565ms 3.8993 KOps/s 3.9385 KOps/s $\color{#d91a1a}-1.00\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.4952ms 1.2992ms 769.7074 Ops/s 827.1608 Ops/s $\textbf{\color{#d91a1a}-6.95\%}$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 9.2147ms 8.7964ms 113.6827 Ops/s 114.3878 Ops/s $\color{#d91a1a}-0.62\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.4250ms 0.2870ms 3.4844 KOps/s 3.5048 KOps/s $\color{#d91a1a}-0.58\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.2009ms 1.0767ms 928.7998 Ops/s 961.7163 Ops/s $\color{#d91a1a}-3.42\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 4.3822ms 4.1903ms 238.6449 Ops/s 247.3404 Ops/s $\color{#d91a1a}-3.52\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.3860ms 0.2881ms 3.4708 KOps/s 3.5107 KOps/s $\color{#d91a1a}-1.14\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.2161ms 1.0770ms 928.5296 Ops/s 953.8306 Ops/s $\color{#d91a1a}-2.65\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 4.4130ms 4.2234ms 236.7742 Ops/s 245.9905 Ops/s $\color{#d91a1a}-3.75\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.7188ms 16.5575ms 60.3955 Ops/s 60.5482 Ops/s $\color{#d91a1a}-0.25\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 34.3534ms 33.3916ms 29.9476 Ops/s 30.0409 Ops/s $\color{#d91a1a}-0.31\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 88.0439ms 86.9260ms 11.5040 Ops/s 11.5450 Ops/s $\color{#d91a1a}-0.35\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.2067s 0.2054s 4.8681 Ops/s 4.9970 Ops/s $\color{#d91a1a}-2.58\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 14.0085ms 13.8124ms 72.3985 Ops/s 72.0485 Ops/s $\color{#35bf28}+0.49\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 28.0878ms 27.0795ms 36.9283 Ops/s 37.2379 Ops/s $\color{#d91a1a}-0.83\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 14.2126ms 13.9879ms 71.4903 Ops/s 71.7496 Ops/s $\color{#d91a1a}-0.36\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 27.5453ms 27.3864ms 36.5145 Ops/s 36.6538 Ops/s $\color{#d91a1a}-0.38\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 52.3975ms 52.1669ms 19.1693 Ops/s 19.5759 Ops/s $\color{#d91a1a}-2.08\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1068s 0.1062s 9.4122 Ops/s 9.6618 Ops/s $\color{#d91a1a}-2.58\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1475s 0.1464s 6.8285 Ops/s 6.9589 Ops/s $\color{#d91a1a}-1.87\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3263s 0.3243s 3.0831 Ops/s 3.0716 Ops/s $\color{#35bf28}+0.38\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 75.9729ms 75.3845ms 13.2653 Ops/s 12.9924 Ops/s $\color{#35bf28}+2.10\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1522s 0.1515s 6.5998 Ops/s 6.2921 Ops/s $\color{#35bf28}+4.89\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 76.3362ms 75.8248ms 13.1883 Ops/s 12.7401 Ops/s $\color{#35bf28}+3.52\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1531s 0.1522s 6.5684 Ops/s 6.2410 Ops/s $\textbf{\color{#35bf28}+5.25\%}$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 31.0700ms 30.9812ms 32.2776 Ops/s 33.2168 Ops/s $\color{#d91a1a}-2.83\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 62.4842ms 62.2978ms 16.0519 Ops/s 16.3625 Ops/s $\color{#d91a1a}-1.90\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1231s 0.1205s 8.2954 Ops/s 8.4442 Ops/s $\color{#d91a1a}-1.76\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2734s 0.2722s 3.6741 Ops/s 3.7441 Ops/s $\color{#d91a1a}-1.87\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 44.0469ms 43.7450ms 22.8598 Ops/s 23.2887 Ops/s $\color{#d91a1a}-1.84\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 88.0216ms 87.6418ms 11.4101 Ops/s 11.6762 Ops/s $\color{#d91a1a}-2.28\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 44.4145ms 44.1508ms 22.6497 Ops/s 23.1318 Ops/s $\color{#d91a1a}-2.08\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 90.3651ms 88.5642ms 11.2912 Ops/s 11.6245 Ops/s $\color{#d91a1a}-2.87\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 21.2000ms 21.0568ms 47.4905 Ops/s 48.8715 Ops/s $\color{#d91a1a}-2.83\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 43.1461ms 42.9390ms 23.2888 Ops/s 23.5174 Ops/s $\color{#d91a1a}-0.97\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1100s 0.1091s 9.1699 Ops/s 9.4610 Ops/s $\color{#d91a1a}-3.08\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2496s 0.2483s 4.0278 Ops/s 4.0680 Ops/s $\color{#d91a1a}-0.99\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 38.8608ms 38.3091ms 26.1035 Ops/s 26.2713 Ops/s $\color{#d91a1a}-0.64\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 77.7911ms 77.0073ms 12.9858 Ops/s 12.9544 Ops/s $\color{#35bf28}+0.24\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 39.2549ms 38.6747ms 25.8567 Ops/s 25.8706 Ops/s $\color{#d91a1a}-0.05\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 77.5515ms 77.1095ms 12.9686 Ops/s 13.7676 Ops/s $\textbf{\color{#d91a1a}-5.80\%}$
test_plain_set_nested 30.6310μs 15.1629μs 65.9504 KOps/s 66.3192 KOps/s $\color{#d91a1a}-0.56\%$
test_plain_set_stack_nested 56.9810μs 14.7685μs 67.7116 KOps/s 66.3898 KOps/s $\color{#35bf28}+1.99\%$
test_plain_set_nested_inplace 49.2710μs 16.6107μs 60.2023 KOps/s 61.0418 KOps/s $\color{#d91a1a}-1.38\%$
test_plain_set_stack_nested_inplace 44.8300μs 16.2446μs 61.5591 KOps/s 61.5202 KOps/s $\color{#35bf28}+0.06\%$
test_items 26.7500μs 6.0367μs 165.6533 KOps/s 164.7256 KOps/s $\color{#35bf28}+0.56\%$
test_items_nested 0.4992ms 0.4667ms 2.1425 KOps/s 2.1886 KOps/s $\color{#d91a1a}-2.11\%$
test_items_nested_locked 0.5786ms 0.4703ms 2.1261 KOps/s 2.1475 KOps/s $\color{#d91a1a}-0.99\%$
test_items_nested_leaf 0.1344ms 98.9862μs 10.1024 KOps/s 10.0547 KOps/s $\color{#35bf28}+0.48\%$
test_items_stack_nested 0.5515ms 0.4609ms 2.1695 KOps/s 2.1838 KOps/s $\color{#d91a1a}-0.65\%$
test_items_stack_nested_leaf 0.1357ms 97.4476μs 10.2619 KOps/s 10.1896 KOps/s $\color{#35bf28}+0.71\%$
test_items_stack_nested_locked 0.5440ms 0.4630ms 2.1597 KOps/s 2.1407 KOps/s $\color{#35bf28}+0.89\%$
test_keys 52.9810μs 4.1806μs 239.2013 KOps/s 237.0018 KOps/s $\color{#35bf28}+0.93\%$
test_keys_nested 0.2177ms 0.1279ms 7.8180 KOps/s 7.7164 KOps/s $\color{#35bf28}+1.32\%$
test_keys_nested_locked 0.6612ms 0.1379ms 7.2542 KOps/s 7.2697 KOps/s $\color{#d91a1a}-0.21\%$
test_keys_nested_leaf 0.1638ms 0.1212ms 8.2519 KOps/s 8.3792 KOps/s $\color{#d91a1a}-1.52\%$
test_keys_stack_nested 0.1578ms 0.1296ms 7.7176 KOps/s 7.7386 KOps/s $\color{#d91a1a}-0.27\%$
test_keys_stack_nested_leaf 0.1614ms 0.1199ms 8.3410 KOps/s 8.3635 KOps/s $\color{#d91a1a}-0.27\%$
test_keys_stack_nested_locked 0.1770ms 0.1384ms 7.2260 KOps/s 7.3098 KOps/s $\color{#d91a1a}-1.15\%$
test_values 6.2320μs 1.0178μs 982.5107 KOps/s 983.1348 KOps/s $\color{#d91a1a}-0.06\%$
test_values_nested 77.6920μs 52.9143μs 18.8985 KOps/s 19.2028 KOps/s $\color{#d91a1a}-1.58\%$
test_values_nested_locked 88.7410μs 56.2495μs 17.7779 KOps/s 18.0391 KOps/s $\color{#d91a1a}-1.45\%$
test_values_nested_leaf 0.1257ms 60.6107μs 16.4987 KOps/s 16.8038 KOps/s $\color{#d91a1a}-1.82\%$
test_values_stack_nested 0.1180ms 52.1344μs 19.1812 KOps/s 19.2069 KOps/s $\color{#d91a1a}-0.13\%$
test_values_stack_nested_leaf 0.1270ms 60.0772μs 16.6452 KOps/s 16.7975 KOps/s $\color{#d91a1a}-0.91\%$
test_values_stack_nested_locked 0.1074ms 55.6807μs 17.9595 KOps/s 18.0741 KOps/s $\color{#d91a1a}-0.63\%$
test_membership 29.7800μs 0.9866μs 1.0135 MOps/s 1.1830 MOps/s $\textbf{\color{#d91a1a}-14.32\%}$
test_membership_nested 35.6510μs 2.9739μs 336.2560 KOps/s 337.1853 KOps/s $\color{#d91a1a}-0.28\%$
test_membership_nested_leaf 19.0105μs 2.8675μs 348.7370 KOps/s 340.1163 KOps/s $\color{#35bf28}+2.53\%$
test_membership_stacked_nested 69.8810μs 2.8869μs 346.3920 KOps/s 352.3989 KOps/s $\color{#d91a1a}-1.70\%$
test_membership_stacked_nested_leaf 34.4710μs 2.8808μs 347.1214 KOps/s 351.2872 KOps/s $\color{#d91a1a}-1.19\%$
test_membership_nested_last 27.3110μs 4.5123μs 221.6171 KOps/s 225.8817 KOps/s $\color{#d91a1a}-1.89\%$
test_membership_nested_leaf_last 34.7510μs 4.4615μs 224.1390 KOps/s 224.4620 KOps/s $\color{#d91a1a}-0.14\%$
test_membership_stacked_nested_last 35.2300μs 4.3206μs 231.4516 KOps/s 232.4703 KOps/s $\color{#d91a1a}-0.44\%$
test_membership_stacked_nested_leaf_last 29.8710μs 4.3270μs 231.1045 KOps/s 234.1321 KOps/s $\color{#d91a1a}-1.29\%$
test_nested_getleaf 43.8910μs 21.7739μs 45.9265 KOps/s 46.2595 KOps/s $\color{#d91a1a}-0.72\%$
test_nested_get 89.9720μs 20.4902μs 48.8038 KOps/s 49.3642 KOps/s $\color{#d91a1a}-1.14\%$
test_stacked_getleaf 52.8710μs 21.3349μs 46.8716 KOps/s 46.5830 KOps/s $\color{#35bf28}+0.62\%$
test_stacked_get 44.8910μs 20.1638μs 49.5938 KOps/s 49.3224 KOps/s $\color{#35bf28}+0.55\%$
test_nested_getitemleaf 52.7810μs 22.1917μs 45.0619 KOps/s 44.9541 KOps/s $\color{#35bf28}+0.24\%$
test_nested_getitem 60.3510μs 21.2774μs 46.9981 KOps/s 48.5680 KOps/s $\color{#d91a1a}-3.23\%$
test_stacked_getitemleaf 64.5610μs 21.7466μs 45.9841 KOps/s 45.6699 KOps/s $\color{#35bf28}+0.69\%$
test_stacked_getitem 51.3110μs 20.6532μs 48.4186 KOps/s 48.4760 KOps/s $\color{#d91a1a}-0.12\%$
test_lock_nested 5.2971ms 0.5200ms 1.9230 KOps/s 1.9633 KOps/s $\color{#d91a1a}-2.05\%$
test_lock_stack_nested 0.6291ms 0.5178ms 1.9312 KOps/s 1.9698 KOps/s $\color{#d91a1a}-1.96\%$
test_unlock_nested 0.4882ms 0.4120ms 2.4269 KOps/s 2.5099 KOps/s $\color{#d91a1a}-3.31\%$
test_unlock_stack_nested 0.5653ms 0.4136ms 2.4178 KOps/s 2.4898 KOps/s $\color{#d91a1a}-2.89\%$
test_flatten_speed 0.2012ms 0.1217ms 8.2137 KOps/s 8.0720 KOps/s $\color{#35bf28}+1.76\%$
test_unflatten_speed 0.6134ms 0.5668ms 1.7643 KOps/s 1.7671 KOps/s $\color{#d91a1a}-0.16\%$
test_common_ops 0.1024s 1.9357ms 516.5982 Ops/s 276.0618 Ops/s $\textbf{\color{#35bf28}+87.13\%}$
test_creation 73.2410μs 3.1668μs 315.7712 KOps/s 323.9654 KOps/s $\color{#d91a1a}-2.53\%$
test_creation_empty 31.8710μs 6.9440μs 144.0086 KOps/s 142.0097 KOps/s $\color{#35bf28}+1.41\%$
test_creation_nested_1 38.4910μs 11.4472μs 87.3572 KOps/s 86.1503 KOps/s $\color{#35bf28}+1.40\%$
test_creation_nested_2 43.4310μs 13.4419μs 74.3944 KOps/s 73.7021 KOps/s $\color{#35bf28}+0.94\%$
test_creation_many_keys[10] 98.5520μs 20.7822μs 48.1182 KOps/s 47.3887 KOps/s $\color{#35bf28}+1.54\%$
test_creation_many_keys[50] 0.1249ms 90.1053μs 11.0981 KOps/s 11.0516 KOps/s $\color{#35bf28}+0.42\%$
test_creation_many_keys[100] 0.2017ms 0.1765ms 5.6645 KOps/s 5.6096 KOps/s $\color{#35bf28}+0.98\%$
test_creation_nested_many_keys[10] 0.1416ms 43.7367μs 22.8641 KOps/s 22.2044 KOps/s $\color{#35bf28}+2.97\%$
test_creation_nested_many_keys[50] 0.2220ms 0.1807ms 5.5351 KOps/s 5.4084 KOps/s $\color{#35bf28}+2.34\%$
test_clone 34.5900μs 13.7851μs 72.5421 KOps/s 73.7532 KOps/s $\color{#d91a1a}-1.64\%$
test_getitem[int] 1.5689ms 15.6265μs 63.9938 KOps/s 65.5298 KOps/s $\color{#d91a1a}-2.34\%$
test_getitem[slice_int] 0.1424ms 24.8891μs 40.1782 KOps/s 40.7967 KOps/s $\color{#d91a1a}-1.52\%$
test_getitem[range] 0.1776ms 64.1343μs 15.5923 KOps/s 15.3943 KOps/s $\color{#35bf28}+1.29\%$
test_getitem[tuple] 0.1442ms 25.0854μs 39.8638 KOps/s 40.9909 KOps/s $\color{#d91a1a}-2.75\%$
test_getitem[list] 0.1817ms 59.4471μs 16.8217 KOps/s 16.4940 KOps/s $\color{#35bf28}+1.99\%$
test_setitem_dim[int] 45.3510μs 26.6768μs 37.4858 KOps/s 37.8449 KOps/s $\color{#d91a1a}-0.95\%$
test_setitem_dim[slice_int] 75.1810μs 43.8249μs 22.8181 KOps/s 22.9611 KOps/s $\color{#d91a1a}-0.62\%$
test_setitem_dim[range] 0.1605ms 96.0780μs 10.4082 KOps/s 10.1809 KOps/s $\color{#35bf28}+2.23\%$
test_setitem_dim[tuple] 62.6910μs 40.2430μs 24.8491 KOps/s 24.7244 KOps/s $\color{#35bf28}+0.50\%$
test_setitem 42.1610μs 18.2315μs 54.8500 KOps/s 55.1445 KOps/s $\color{#d91a1a}-0.53\%$
test_set 57.4810μs 17.4047μs 57.4558 KOps/s 57.9701 KOps/s $\color{#d91a1a}-0.89\%$
test_set_shared 0.5025ms 0.2089ms 4.7879 KOps/s 4.6984 KOps/s $\color{#35bf28}+1.90\%$
test_update 0.1063ms 22.3258μs 44.7913 KOps/s 44.7465 KOps/s $\color{#35bf28}+0.10\%$
test_update_nested 68.8510μs 34.8997μs 28.6536 KOps/s 28.9073 KOps/s $\color{#d91a1a}-0.88\%$
test_update__nested 0.4590ms 34.2887μs 29.1641 KOps/s 24.1969 KOps/s $\textbf{\color{#35bf28}+20.53\%}$
test_set_nested 71.4910μs 19.2780μs 51.8725 KOps/s 52.4479 KOps/s $\color{#d91a1a}-1.10\%$
test_set_nested_new 61.3010μs 24.1334μs 41.4364 KOps/s 42.4625 KOps/s $\color{#d91a1a}-2.42\%$
test_select 87.1220μs 41.6310μs 24.0206 KOps/s 24.5591 KOps/s $\color{#d91a1a}-2.19\%$
test_select_nested 0.1106ms 74.4094μs 13.4392 KOps/s 13.5802 KOps/s $\color{#d91a1a}-1.04\%$
test_exclude_nested 0.1235ms 91.6773μs 10.9078 KOps/s 11.0198 KOps/s $\color{#d91a1a}-1.02\%$
test_empty[True] 0.4327ms 0.3973ms 2.5171 KOps/s 2.5539 KOps/s $\color{#d91a1a}-1.44\%$
test_empty[False] 7.9052μs 1.4126μs 707.9137 KOps/s 728.8663 KOps/s $\color{#d91a1a}-2.87\%$
test_to 0.1033ms 71.1469μs 14.0554 KOps/s 13.9633 KOps/s $\color{#35bf28}+0.66\%$
test_to_nonblocking 0.1254ms 70.7096μs 14.1423 KOps/s 14.4247 KOps/s $\color{#d91a1a}-1.96\%$
test_unbind_speed 0.4201ms 0.3484ms 2.8702 KOps/s 2.9634 KOps/s $\color{#d91a1a}-3.15\%$
test_unbind_speed_stack0 0.3887ms 0.3450ms 2.8982 KOps/s 3.0189 KOps/s $\color{#d91a1a}-4.00\%$
test_unbind_speed_stack1 0.1133s 0.8712ms 1.1479 KOps/s 1.3127 KOps/s $\textbf{\color{#d91a1a}-12.56\%}$
test_split 1.2123ms 1.1573ms 864.0485 Ops/s 877.9310 Ops/s $\color{#d91a1a}-1.58\%$
test_chunk 0.1131s 1.2386ms 807.3798 Ops/s 917.5818 Ops/s $\textbf{\color{#d91a1a}-12.01\%}$
test_to_cpu_blocking 20.4863ms 20.3792ms 49.0697 Ops/s 45.2748 Ops/s $\textbf{\color{#35bf28}+8.38\%}$
test_to_cpu_global_sync 12.0774ms 11.9746ms 83.5100 Ops/s 85.4742 Ops/s $\color{#d91a1a}-2.30\%$
test_to_cpu_event_sync 13.2524ms 12.9846ms 77.0143 Ops/s 78.9898 Ops/s $\color{#d91a1a}-2.50\%$
test_to_cpu_default 14.4588ms 13.0112ms 76.8566 Ops/s 78.7182 Ops/s $\color{#d91a1a}-2.36\%$
test_consolidate[False-None] 4.3593ms 4.2209ms 236.9181 Ops/s 240.1672 Ops/s $\color{#d91a1a}-1.35\%$
test_consolidate[default-None] 3.6026ms 2.2330ms 447.8184 Ops/s 461.6643 Ops/s $\color{#d91a1a}-3.00\%$
test_consolidate[reduce-overhead-None] 3.5422ms 2.1435ms 466.5175 Ops/s 474.1260 Ops/s $\color{#d91a1a}-1.60\%$
test_consolidate_njt[False-None] 8.7659ms 8.5679ms 116.7141 Ops/s 60.5243 Ops/s $\textbf{\color{#35bf28}+92.84\%}$
test_to[False-False-None] 2.2525ms 2.1860ms 457.4626 Ops/s 465.6336 Ops/s $\color{#d91a1a}-1.75\%$
test_to[True-False-None] 0.2040s 2.3650ms 422.8308 Ops/s 528.5799 Ops/s $\textbf{\color{#d91a1a}-20.01\%}$
test_to[within-False-None] 6.3751ms 6.2978ms 158.7861 Ops/s 161.7769 Ops/s $\color{#d91a1a}-1.85\%$
test_to[True-default-None] 11.2873ms 10.8421ms 92.2332 Ops/s 92.5259 Ops/s $\color{#d91a1a}-0.32\%$
test_to_njt[False-False-None] 8.9507ms 8.7069ms 114.8511 Ops/s 114.8127 Ops/s $\color{#35bf28}+0.03\%$
test_to_njt[True-False-None] 7.1977ms 6.9786ms 143.2950 Ops/s 144.8128 Ops/s $\color{#d91a1a}-1.05\%$
test_to_njt[within-False-None] 15.8223ms 15.5726ms 64.2153 Ops/s 64.3859 Ops/s $\color{#d91a1a}-0.26\%$
test_creation[device0] 0.4506ms 0.1199ms 8.3396 KOps/s 8.6008 KOps/s $\color{#d91a1a}-3.04\%$
test_creation_from_tensor 0.4508ms 0.1172ms 8.5321 KOps/s 8.7420 KOps/s $\color{#d91a1a}-2.40\%$
test_add_one[memmap_tensor0] 0.5786ms 7.0077μs 142.6992 KOps/s 146.7001 KOps/s $\color{#d91a1a}-2.73\%$
test_contiguous[memmap_tensor0] 17.2110μs 0.6429μs 1.5555 MOps/s 2.1725 MOps/s $\textbf{\color{#d91a1a}-28.40\%}$
test_stack[memmap_tensor0] 91.6110μs 4.4736μs 223.5358 KOps/s 230.0935 KOps/s $\color{#d91a1a}-2.85\%$
test_memmaptd_index 1.3326ms 0.2906ms 3.4417 KOps/s 3.6578 KOps/s $\textbf{\color{#d91a1a}-5.91\%}$
test_memmaptd_index_astensor 0.5535ms 0.3939ms 2.5390 KOps/s 2.6561 KOps/s $\color{#d91a1a}-4.41\%$
test_memmaptd_index_op 0.8295ms 0.6794ms 1.4719 KOps/s 1.5513 KOps/s $\textbf{\color{#d91a1a}-5.12\%}$
test_serialize_model 0.3263s 0.1623s 6.1631 Ops/s 5.8428 Ops/s $\textbf{\color{#35bf28}+5.48\%}$
test_serialize_model_pickle 1.3735s 1.2144s 0.8234 Ops/s 0.8237 Ops/s $\color{#d91a1a}-0.04\%$
test_serialize_weights 0.1357s 0.1341s 7.4587 Ops/s 7.4824 Ops/s $\color{#d91a1a}-0.32\%$
test_serialize_weights_returnearly 0.4615s 89.7522ms 11.1418 Ops/s 15.2467 Ops/s $\textbf{\color{#d91a1a}-26.92\%}$
test_serialize_weights_pickle 1.3656s 1.2136s 0.8240 Ops/s 0.8227 Ops/s $\color{#35bf28}+0.16\%$
test_reshape_pytree 0.2069ms 33.1097μs 30.2027 KOps/s 30.5867 KOps/s $\color{#d91a1a}-1.26\%$
test_reshape_td 87.3410μs 45.6011μs 21.9293 KOps/s 22.4827 KOps/s $\color{#d91a1a}-2.46\%$
test_view_pytree 0.2242ms 32.5575μs 30.7149 KOps/s 30.8953 KOps/s $\color{#d91a1a}-0.58\%$
test_view_td 0.1151ms 55.5569μs 17.9996 KOps/s 18.7026 KOps/s $\color{#d91a1a}-3.76\%$
test_unbind_pytree 0.2522ms 37.0670μs 26.9782 KOps/s 26.9412 KOps/s $\color{#35bf28}+0.14\%$
test_unbind_td 0.1171ms 51.6612μs 19.3569 KOps/s 19.7146 KOps/s $\color{#d91a1a}-1.81\%$
test_split_pytree 0.2483ms 42.8289μs 23.3487 KOps/s 23.6556 KOps/s $\color{#d91a1a}-1.30\%$
test_split_td 0.2295ms 66.6185μs 15.0108 KOps/s 15.0186 KOps/s $\color{#d91a1a}-0.05\%$
test_add_pytree 0.1972ms 43.0855μs 23.2096 KOps/s 23.1766 KOps/s $\color{#35bf28}+0.14\%$
test_add_td 0.1185ms 54.4819μs 18.3547 KOps/s 17.7017 KOps/s $\color{#35bf28}+3.69\%$
test_compile_add_one_nested[tensordict-compile] 0.3788ms 0.2355ms 4.2462 KOps/s 4.1947 KOps/s $\color{#35bf28}+1.23\%$
test_compile_add_one_nested[tensordict-eager] 0.4254ms 0.2001ms 4.9972 KOps/s 5.0583 KOps/s $\color{#d91a1a}-1.21\%$
test_compile_add_one_nested[pytree-compile] 0.3106ms 0.1383ms 7.2329 KOps/s 7.0720 KOps/s $\color{#35bf28}+2.28\%$
test_compile_add_one_nested[pytree-eager] 0.4995ms 0.1809ms 5.5284 KOps/s 5.5229 KOps/s $\color{#35bf28}+0.10\%$
test_compile_copy_nested[tensordict-compile] 0.1352ms 15.3675μs 65.0726 KOps/s 65.4306 KOps/s $\color{#d91a1a}-0.55\%$
test_compile_copy_nested[tensordict-eager] 0.3032ms 52.7013μs 18.9749 KOps/s 18.3017 KOps/s $\color{#35bf28}+3.68\%$
test_compile_copy_nested[pytree-compile] 0.1100ms 15.7290μs 63.5768 KOps/s 63.8371 KOps/s $\color{#d91a1a}-0.41\%$
test_compile_copy_nested[pytree-eager] 0.3904ms 67.1814μs 14.8851 KOps/s 14.4598 KOps/s $\color{#35bf28}+2.94\%$
test_compile_add_one_flat[tensordict-compile] 0.3586ms 0.2537ms 3.9411 KOps/s 3.7247 KOps/s $\textbf{\color{#35bf28}+5.81\%}$
test_compile_add_one_flat[tensordict-eager] 0.3722ms 0.2837ms 3.5251 KOps/s 3.4885 KOps/s $\color{#35bf28}+1.05\%$
test_compile_add_one_flat[tensorclass-compile] 0.4223ms 0.1885ms 5.3047 KOps/s 4.9119 KOps/s $\textbf{\color{#35bf28}+8.00\%}$
test_compile_add_one_flat[tensorclass-eager] 0.1210ms 74.7968μs 13.3696 KOps/s 13.0166 KOps/s $\color{#35bf28}+2.71\%$
test_compile_add_one_flat[pytree-compile] 0.3063ms 0.1927ms 5.1894 KOps/s 4.9962 KOps/s $\color{#35bf28}+3.87\%$
test_compile_add_one_flat[pytree-eager] 0.8043ms 0.5401ms 1.8514 KOps/s 1.9133 KOps/s $\color{#d91a1a}-3.24\%$
test_compile_add_self_flat[tensordict-eager] 0.4228ms 0.3329ms 3.0041 KOps/s 2.9850 KOps/s $\color{#35bf28}+0.64\%$
test_compile_add_self_flat[tensordict-compile] 0.3217ms 0.2523ms 3.9639 KOps/s 3.5515 KOps/s $\textbf{\color{#35bf28}+11.61\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1833ms 91.4116μs 10.9395 KOps/s 10.6253 KOps/s $\color{#35bf28}+2.96\%$
test_compile_add_self_flat[tensorclass-compile] 0.2775ms 0.1910ms 5.2355 KOps/s 5.0331 KOps/s $\color{#35bf28}+4.02\%$
test_compile_add_self_flat[pytree-eager] 0.6965ms 0.4402ms 2.2718 KOps/s 2.3235 KOps/s $\color{#d91a1a}-2.23\%$
test_compile_add_self_flat[pytree-compile] 0.4295ms 0.1928ms 5.1870 KOps/s 4.9967 KOps/s $\color{#35bf28}+3.81\%$
test_compile_copy_flat[tensordict-compile] 59.4420μs 19.7662μs 50.5915 KOps/s 50.4219 KOps/s $\color{#35bf28}+0.34\%$
test_compile_copy_flat[tensordict-eager] 71.2610μs 41.3310μs 24.1949 KOps/s 24.0828 KOps/s $\color{#35bf28}+0.47\%$
test_compile_copy_flat[pytree-compile] 0.1307ms 21.3533μs 46.8312 KOps/s 45.1568 KOps/s $\color{#35bf28}+3.71\%$
test_compile_copy_flat[pytree-eager] 0.3649ms 51.8554μs 19.2844 KOps/s 19.2292 KOps/s $\color{#35bf28}+0.29\%$
test_compile_assign_and_add[tensordict-compile] 2.2440ms 0.2431ms 4.1141 KOps/s 3.9095 KOps/s $\textbf{\color{#35bf28}+5.23\%}$
test_compile_assign_and_add[tensordict-eager] 3.8009ms 3.3571ms 297.8717 Ops/s 299.3277 Ops/s $\color{#d91a1a}-0.49\%$
test_compile_assign_and_add[pytree-compile] 2.1260ms 0.1822ms 5.4891 KOps/s 5.3168 KOps/s $\color{#35bf28}+3.24\%$
test_compile_assign_and_add[pytree-eager] 3.3214ms 2.8680ms 348.6789 Ops/s 357.7546 Ops/s $\color{#d91a1a}-2.54\%$
test_compile_indexing[tensor-tensordict-compile] 0.2004ms 0.1345ms 7.4330 KOps/s 6.8642 KOps/s $\textbf{\color{#35bf28}+8.29\%}$
test_compile_indexing[tensor-tensordict-eager] 0.5344ms 77.0431μs 12.9797 KOps/s 12.5883 KOps/s $\color{#35bf28}+3.11\%$
test_compile_indexing[tensor-tensorclass-compile] 0.5514ms 0.1202ms 8.3163 KOps/s 7.8295 KOps/s $\textbf{\color{#35bf28}+6.22\%}$
test_compile_indexing[tensor-tensorclass-eager] 0.4783ms 46.0638μs 21.7090 KOps/s 21.0651 KOps/s $\color{#35bf28}+3.06\%$
test_compile_indexing[tensor-pytree-compile] 0.5543ms 0.1206ms 8.2929 KOps/s 7.7423 KOps/s $\textbf{\color{#35bf28}+7.11\%}$
test_compile_indexing[tensor-pytree-eager] 0.2487ms 45.8146μs 21.8271 KOps/s 21.1476 KOps/s $\color{#35bf28}+3.21\%$
test_compile_indexing[slice-tensordict-compile] 0.1808ms 83.8301μs 11.9289 KOps/s 11.0461 KOps/s $\textbf{\color{#35bf28}+7.99\%}$
test_compile_indexing[slice-tensordict-eager] 0.5057ms 29.0779μs 34.3904 KOps/s 33.9475 KOps/s $\color{#35bf28}+1.30\%$
test_compile_indexing[slice-tensorclass-compile] 0.4991ms 73.1732μs 13.6662 KOps/s 13.3071 KOps/s $\color{#35bf28}+2.70\%$
test_compile_indexing[slice-tensorclass-eager] 0.4453ms 22.8680μs 43.7293 KOps/s 45.4852 KOps/s $\color{#d91a1a}-3.86\%$
test_compile_indexing[slice-pytree-compile] 0.5005ms 71.7430μs 13.9386 KOps/s 13.9764 KOps/s $\color{#d91a1a}-0.27\%$
test_compile_indexing[slice-pytree-eager] 0.4595ms 22.7837μs 43.8911 KOps/s 44.9514 KOps/s $\color{#d91a1a}-2.36\%$
test_compile_indexing[int-tensordict-compile] 0.1207ms 85.1107μs 11.7494 KOps/s 11.4681 KOps/s $\color{#35bf28}+2.45\%$
test_compile_indexing[int-tensordict-eager] 0.4540ms 27.7186μs 36.0769 KOps/s 37.0211 KOps/s $\color{#d91a1a}-2.55\%$
test_compile_indexing[int-tensorclass-compile] 0.1581ms 71.8765μs 13.9128 KOps/s 13.8902 KOps/s $\color{#35bf28}+0.16\%$
test_compile_indexing[int-tensorclass-eager] 0.4470ms 22.7687μs 43.9199 KOps/s 44.7193 KOps/s $\color{#d91a1a}-1.79\%$
test_compile_indexing[int-pytree-compile] 0.5044ms 71.7174μs 13.9436 KOps/s 13.8849 KOps/s $\color{#35bf28}+0.42\%$
test_compile_indexing[int-pytree-eager] 0.4468ms 22.7080μs 44.0373 KOps/s 45.0875 KOps/s $\color{#d91a1a}-2.33\%$
test_compile_replace[single-eager] 0.4771ms 47.7761μs 20.9309 KOps/s 21.1706 KOps/s $\color{#d91a1a}-1.13\%$
test_compile_replace[single-compile] 0.5623ms 0.1307ms 7.6531 KOps/s 7.4859 KOps/s $\color{#35bf28}+2.23\%$
test_compile_replace[multi-eager] 1.0136ms 0.5720ms 1.7481 KOps/s 1.7764 KOps/s $\color{#d91a1a}-1.59\%$
test_compile_replace[multi-compile] 0.5858ms 0.1390ms 7.1945 KOps/s 7.0716 KOps/s $\color{#35bf28}+1.74\%$
test_compile_tc_getattr_20[eager] 0.6088ms 0.1786ms 5.5994 KOps/s 6.0657 KOps/s $\textbf{\color{#d91a1a}-7.69\%}$
test_compile_tc_getattr_20[compile] 0.3161ms 0.1488ms 6.7208 KOps/s 6.5207 KOps/s $\color{#35bf28}+3.07\%$
test_compile_clone_shallow[20-eager] 0.4572ms 19.0952μs 52.3692 KOps/s 52.7319 KOps/s $\color{#d91a1a}-0.69\%$
test_compile_clone_shallow[20-compile] 0.4522ms 17.5068μs 57.1208 KOps/s 59.6851 KOps/s $\color{#d91a1a}-4.30\%$
test_compile_clone_shallow[40-eager] 0.4541ms 33.5180μs 29.8347 KOps/s 29.3804 KOps/s $\color{#35bf28}+1.55\%$
test_compile_clone_shallow[40-compile] 0.4589ms 19.3137μs 51.7766 KOps/s 53.1719 KOps/s $\color{#d91a1a}-2.62\%$
test_compile_clone_shallow[80-eager] 0.4883ms 62.3744μs 16.0322 KOps/s 15.8816 KOps/s $\color{#35bf28}+0.95\%$
test_compile_clone_shallow[80-compile] 89.3820μs 22.5680μs 44.3106 KOps/s 45.9568 KOps/s $\color{#d91a1a}-3.58\%$
test_compile_update_inplace[eager] 0.4984ms 62.2788μs 16.0568 KOps/s 16.2238 KOps/s $\color{#d91a1a}-1.03\%$
test_compile_update_inplace[compile] 0.5832ms 0.1468ms 6.8104 KOps/s 6.6559 KOps/s $\color{#35bf28}+2.32\%$
test_mod_add[eager] 0.6688ms 0.2266ms 4.4126 KOps/s 4.3816 KOps/s $\color{#35bf28}+0.71\%$
test_mod_add[compile] 0.5622ms 0.1325ms 7.5457 KOps/s 7.3473 KOps/s $\color{#35bf28}+2.70\%$
test_mod_add[compile-overhead] 0.3456ms 0.1790ms 5.5869 KOps/s 5.4936 KOps/s $\color{#35bf28}+1.70\%$
test_mod_wrap[eager] 0.9093ms 0.4719ms 2.1189 KOps/s 2.0714 KOps/s $\color{#35bf28}+2.30\%$
test_mod_wrap[compile] 0.5312ms 0.4226ms 2.3664 KOps/s 2.4910 KOps/s $\textbf{\color{#d91a1a}-5.00\%}$
test_mod_wrap[compile-overhead] 8.2564ms 4.5570ms 219.4418 Ops/s 217.1575 Ops/s $\color{#35bf28}+1.05\%$
test_mod_wrap_and_backward[eager] 1.9950ms 1.7513ms 570.9924 Ops/s 587.6582 Ops/s $\color{#d91a1a}-2.84\%$
test_mod_wrap_and_backward[compile] 2.4466ms 1.6267ms 614.7375 Ops/s 645.0919 Ops/s $\color{#d91a1a}-4.71\%$
test_mod_wrap_and_backward[compile-overhead] 1.3424ms 0.9434ms 1.0600 KOps/s 1.0444 KOps/s $\color{#35bf28}+1.50\%$
test_seq_add[eager] 0.5891ms 0.5151ms 1.9414 KOps/s 1.9413 KOps/s $+0.00\%$
test_seq_add[compile] 0.2431ms 0.1451ms 6.8927 KOps/s 6.6670 KOps/s $\color{#35bf28}+3.38\%$
test_seq_add[compile-overhead] 0.2414ms 0.1903ms 5.2537 KOps/s 5.2751 KOps/s $\color{#d91a1a}-0.41\%$
test_seq_wrap[eager] 1.5189ms 1.4324ms 698.1237 Ops/s 706.9025 Ops/s $\color{#d91a1a}-1.24\%$
test_seq_wrap[compile] 0.5681ms 0.4152ms 2.4083 KOps/s 2.3939 KOps/s $\color{#35bf28}+0.60\%$
test_seq_wrap[compile-overhead] 0.4271ms 0.3049ms 3.2794 KOps/s 3.2385 KOps/s $\color{#35bf28}+1.26\%$
test_func_call_runtime[False-eager] 0.8820ms 0.8169ms 1.2242 KOps/s 1.2313 KOps/s $\color{#d91a1a}-0.58\%$
test_func_call_runtime[False-compile] 1.1564ms 1.0010ms 999.0123 Ops/s 1.0098 KOps/s $\color{#d91a1a}-1.06\%$
test_func_call_runtime[False-compile-overhead] 0.5908ms 0.5101ms 1.9602 KOps/s 1.9550 KOps/s $\color{#35bf28}+0.27\%$
test_func_call_runtime[True-eager] 1.2091ms 1.0980ms 910.7071 Ops/s 931.2276 Ops/s $\color{#d91a1a}-2.20\%$
test_func_call_runtime[True-compile] 1.0950ms 1.0166ms 983.7023 Ops/s 994.8146 Ops/s $\color{#d91a1a}-1.12\%$
test_func_call_runtime[True-compile-overhead] 0.5965ms 0.5239ms 1.9088 KOps/s 1.8811 KOps/s $\color{#35bf28}+1.47\%$
test_func_call_cm_runtime[False-eager] 0.9521ms 0.8204ms 1.2190 KOps/s 1.2462 KOps/s $\color{#d91a1a}-2.18\%$
test_func_call_cm_runtime[False-compile] 1.0716ms 0.9827ms 1.0176 KOps/s 1.0234 KOps/s $\color{#d91a1a}-0.56\%$
test_func_call_cm_runtime[False-compile-overhead] 0.7227ms 0.4914ms 2.0351 KOps/s 2.0120 KOps/s $\color{#35bf28}+1.15\%$
test_func_call_cm_runtime[True-eager] 1.3431ms 1.2327ms 811.2489 Ops/s 816.0072 Ops/s $\color{#d91a1a}-0.58\%$
test_func_call_cm_runtime[True-compile] 1.1026ms 1.0380ms 963.4213 Ops/s 971.8865 Ops/s $\color{#d91a1a}-0.87\%$
test_func_call_cm_runtime[True-compile-overhead] 0.6794ms 0.5521ms 1.8112 KOps/s 1.7930 KOps/s $\color{#35bf28}+1.01\%$
test_vmap_func_call_cm_runtime[eager] 2.9432ms 2.4370ms 410.3331 Ops/s 411.7820 Ops/s $\color{#d91a1a}-0.35\%$
test_vmap_func_call_cm_runtime[compile] 3.0829ms 2.5973ms 385.0131 Ops/s 382.6829 Ops/s $\color{#35bf28}+0.61\%$
test_vmap_func_call_cm_runtime[compile-overhead] 3.2862ms 2.6164ms 382.2058 Ops/s 382.8303 Ops/s $\color{#d91a1a}-0.16\%$
test_distributed 0.6753ms 0.1482ms 6.7488 KOps/s 6.3841 KOps/s $\textbf{\color{#35bf28}+5.71\%}$
test_tdmodule 0.7227ms 0.1944ms 5.1452 KOps/s 5.0510 KOps/s $\color{#35bf28}+1.86\%$
test_tdmodule_dispatch 0.4120ms 0.3733ms 2.6785 KOps/s 2.6597 KOps/s $\color{#35bf28}+0.71\%$
test_tdseq 45.2210μs 26.9029μs 37.1707 KOps/s 36.3414 KOps/s $\color{#35bf28}+2.28\%$
test_tdseq_dispatch 0.2702ms 0.2185ms 4.5777 KOps/s 4.5537 KOps/s $\color{#35bf28}+0.53\%$
test_instantiation_functorch 2.2744ms 2.0872ms 479.1200 Ops/s 485.6909 Ops/s $\color{#d91a1a}-1.35\%$
test_exec_functorch 0.2307ms 0.1835ms 5.4510 KOps/s 5.4122 KOps/s $\color{#35bf28}+0.72\%$
test_exec_functional_call 0.2335ms 0.1692ms 5.9102 KOps/s 6.1863 KOps/s $\color{#d91a1a}-4.46\%$
test_exec_td_decorator 0.5088ms 0.2550ms 3.9218 KOps/s 3.9215 KOps/s $+0.01\%$
test_vmap_mlp_speed_decorator[True-True] 1.2773ms 1.0411ms 960.5142 Ops/s 954.4513 Ops/s $\color{#35bf28}+0.64\%$
test_vmap_mlp_speed_decorator[True-False] 1.2440ms 1.0381ms 963.2883 Ops/s 956.2405 Ops/s $\color{#35bf28}+0.74\%$
test_vmap_mlp_speed_decorator[False-True] 0.9806ms 0.7394ms 1.3525 KOps/s 1.3413 KOps/s $\color{#35bf28}+0.83\%$
test_vmap_mlp_speed_decorator[False-False] 0.9334ms 0.7392ms 1.3528 KOps/s 1.3251 KOps/s $\color{#35bf28}+2.09\%$
test_vmap_transformer_speed_decorator[True-True] 21.3909ms 21.1662ms 47.2451 Ops/s 47.1616 Ops/s $\color{#35bf28}+0.18\%$
test_vmap_transformer_speed_decorator[True-False] 21.8999ms 21.1885ms 47.1954 Ops/s 47.1618 Ops/s $\color{#35bf28}+0.07\%$
test_vmap_transformer_speed_decorator[False-True] 21.5667ms 20.8042ms 48.0672 Ops/s 48.0879 Ops/s $\color{#d91a1a}-0.04\%$
test_vmap_transformer_speed_decorator[False-False] 21.5383ms 20.8045ms 48.0665 Ops/s 48.1068 Ops/s $\color{#d91a1a}-0.08\%$
test_to_module_speed[True] 2.2794ms 1.6606ms 602.2016 Ops/s 600.4963 Ops/s $\color{#35bf28}+0.28\%$
test_to_module_speed[False] 2.1990ms 1.6085ms 621.7085 Ops/s 611.2565 Ops/s $\color{#35bf28}+1.71\%$
test_to_module_plain_tensor_speed[None] 2.3608ms 1.7385ms 575.2071 Ops/s 577.2225 Ops/s $\color{#d91a1a}-0.35\%$
test_to_module_plain_tensor_speed[False] 2.1479ms 1.5632ms 639.7018 Ops/s 634.7268 Ops/s $\color{#35bf28}+0.78\%$
test_to_module_plain_tensor_speed[True] 3.5310ms 3.3953ms 294.5239 Ops/s 295.5393 Ops/s $\color{#d91a1a}-0.34\%$
test_tc_init 67.3710μs 44.6754μs 22.3837 KOps/s 23.0078 KOps/s $\color{#d91a1a}-2.71\%$
test_tc_init_tensor_only 27.6710μs 9.0566μs 110.4171 KOps/s 110.4745 KOps/s $\color{#d91a1a}-0.05\%$
test_tc_init_nested 0.1235ms 87.4571μs 11.4342 KOps/s 11.5580 KOps/s $\color{#d91a1a}-1.07\%$
test_tc_init_many_fields 50.2010μs 15.7756μs 63.3891 KOps/s 66.9093 KOps/s $\textbf{\color{#d91a1a}-5.26\%}$
test_tc_first_layer_tensor 26.4100μs 1.8326μs 545.6679 KOps/s 553.5666 KOps/s $\color{#d91a1a}-1.43\%$
test_tc_first_layer_tensor_only 1.8185μs 0.3982μs 2.5113 MOps/s 2.5498 MOps/s $\color{#d91a1a}-1.51\%$
test_tc_first_layer_tensor_set 24.0110μs 3.9901μs 250.6207 KOps/s 253.5904 KOps/s $\color{#d91a1a}-1.17\%$
test_tc_first_layer_tensor_only_set 28.0810μs 3.2830μs 304.5960 KOps/s 308.4987 KOps/s $\color{#d91a1a}-1.27\%$
test_tc_first_layer_nontensor 34.8210μs 6.1866μs 161.6398 KOps/s 161.9724 KOps/s $\color{#d91a1a}-0.21\%$
test_tc_second_layer_tensor 25.0500μs 4.3390μs 230.4681 KOps/s 225.6393 KOps/s $\color{#35bf28}+2.14\%$
test_tc_second_layer_nontensor 38.9110μs 8.6665μs 115.3872 KOps/s 115.6879 KOps/s $\color{#d91a1a}-0.26\%$
test_unbind 0.2742s 13.4688ms 74.2455 Ops/s 67.8851 Ops/s $\textbf{\color{#35bf28}+9.37\%}$
test_full_like 4.5961ms 4.3963ms 227.4633 Ops/s 228.3415 Ops/s $\color{#d91a1a}-0.38\%$
test_zeros_like 4.8069ms 4.3735ms 228.6498 Ops/s 228.5324 Ops/s $\color{#35bf28}+0.05\%$
test_ones_like 4.6257ms 4.3904ms 227.7686 Ops/s 228.1314 Ops/s $\color{#d91a1a}-0.16\%$
test_clone 7.1971ms 6.6620ms 150.1044 Ops/s 151.0915 Ops/s $\color{#d91a1a}-0.65\%$
test_squeeze 68.7620μs 14.3103μs 69.8796 KOps/s 68.5464 KOps/s $\color{#35bf28}+1.94\%$
test_unsqueeze 0.2008ms 0.1184ms 8.4478 KOps/s 9.0387 KOps/s $\textbf{\color{#d91a1a}-6.54\%}$
test_split 0.2640ms 0.1884ms 5.3080 KOps/s 5.4954 KOps/s $\color{#d91a1a}-3.41\%$
test_permute 0.2783ms 0.2165ms 4.6184 KOps/s 4.8794 KOps/s $\textbf{\color{#d91a1a}-5.35\%}$
test_stack 19.4110ms 18.6419ms 53.6426 Ops/s 53.8031 Ops/s $\color{#d91a1a}-0.30\%$
test_cat 19.3693ms 18.6734ms 53.5520 Ops/s 53.6207 Ops/s $\color{#d91a1a}-0.13\%$
test_sequential_tensordict 0.6662ms 0.5840ms 1.7124 KOps/s 1.7416 KOps/s $\color{#d91a1a}-1.67\%$
test_sequential_graph_module 0.5236ms 0.1233ms 8.1087 KOps/s 8.4732 KOps/s $\color{#d91a1a}-4.30\%$
test_nested_tensordict 0.9526ms 0.8356ms 1.1967 KOps/s 1.2143 KOps/s $\color{#d91a1a}-1.45\%$
test_nested_graph_module 0.5518ms 0.1357ms 7.3712 KOps/s 7.7773 KOps/s $\textbf{\color{#d91a1a}-5.22\%}$

@vmoens
vmoens merged commit a7c9437 into main Jun 18, 2026
72 of 73 checks passed
@vmoens
vmoens deleted the codex/update-pr-label-prefixes branch June 18, 2026 08:11
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.

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@vmoens