Skip to content

[Versioning] Bump version to 0.13.0 - #1716

Merged
vmoens merged 1 commit into
mainfrom
release-0.13.0-version-bump
May 24, 2026
Merged

[Versioning] Bump version to 0.13.0#1716
vmoens merged 1 commit into
mainfrom
release-0.13.0-version-bump

Conversation

@vmoens

@vmoens vmoens commented May 24, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Bump TensorDict base version to 0.13.0 in all release workflow version files.

Validation

  • Checked all four release version files resolve to 0.13.0.
  • Ran git diff --check.

Release context

This prepares the main branch for the TensorDict 0.13.0 release; release branch/tag/workflow steps should follow after this PR is reviewed and merged.

@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 May 24, 2026
@github-actions

github-actions Bot commented May 24, 2026

Copy link
Copy Markdown
Contributor

PR Title Label Error

Unknown or invalid prefix [Versioning].

Current title: [Versioning] Bump version to 0.13.0

Supported Prefixes

Your PR title must start with exactly one of these prefixes (case-insensitive):

Prefix Label Applied Example
[BugFix] or [Fix] bug [BugFix] Fix memory leak in TensorDict
[Feature] Feature [Feature] Add new storage backend
[Doc] or [Docs] documentation [Doc] Update installation guide
[Refactor] Refactor [Refactor] Clean up module imports
[CI] CI [CI] Fix workflow permissions
[Test] or [Tests] Test [Test] Add unit tests for nn module
[Compile] Compile [Compile] Fix torch.compile issue
[Performance] or [Perf] Performance [Perf] Optimize tensor operations
[Deprecation] Deprecation [Deprecation] Mark old function
[Setup] setup [Setup] Update build configuration
[Distributed] or [Dist] Distributed [Distributed] Add scatter collective
[Benchmark] or [Bench] Benchmarks [Benchmark] Add compile benchmark
[Typing] or [Type] Typing [Typing] Add type stubs
[BC-breaking] or [BC] BC-breaking [BC-breaking] Remove deprecated API
[Formatting] or [Format] Formatting [Format] Fix code style
[Quality] Quality [Quality] Improve error messages

Note: Matching is case-insensitive. Common variations (singular/plural) are supported.

@github-actions github-actions Bot added the CI label May 24, 2026
@vmoens
vmoens merged commit 00283c3 into main May 24, 2026
57 of 67 checks passed
@github-actions

github-actions Bot commented May 24, 2026

Copy link
Copy Markdown
Contributor

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

Total Benchmarks: 341. Improved: $\large\color{#35bf28}16$. Worsened: $\large\color{#d91a1a}8$.

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_collector_preallocated_write[stack_out-cpu-32-2] 0.2134ms 0.1683ms 5.9407 KOps/s 5.8184 KOps/s $\color{#35bf28}+2.10\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.3787ms 0.2957ms 3.3821 KOps/s 3.3478 KOps/s $\color{#35bf28}+1.02\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.8715ms 0.7800ms 1.2821 KOps/s 1.2811 KOps/s $\color{#35bf28}+0.08\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.5836ms 0.3373ms 2.9643 KOps/s 2.9156 KOps/s $\color{#35bf28}+1.67\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.2632ms 1.1339ms 881.9365 Ops/s 930.0345 Ops/s $\textbf{\color{#d91a1a}-5.17\%}$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.6522ms 7.2733ms 137.4884 Ops/s 137.4025 Ops/s $\color{#35bf28}+0.06\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.2792ms 0.2433ms 4.1108 KOps/s 4.0411 KOps/s $\color{#35bf28}+1.72\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.4747ms 0.3663ms 2.7301 KOps/s 2.6742 KOps/s $\color{#35bf28}+2.09\%$
test_collector_preallocated_write[stack_out-cuda-32-32] 0.9156ms 0.8167ms 1.2244 KOps/s 1.2192 KOps/s $\color{#35bf28}+0.42\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.3430ms 0.2552ms 3.9180 KOps/s 3.8895 KOps/s $\color{#35bf28}+0.73\%$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.4366ms 0.3764ms 2.6566 KOps/s 2.7042 KOps/s $\color{#d91a1a}-1.76\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 0.9587ms 0.8810ms 1.1350 KOps/s 1.1361 KOps/s $\color{#d91a1a}-0.09\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2842ms 0.2149ms 4.6538 KOps/s 4.6155 KOps/s $\color{#35bf28}+0.83\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 0.9620ms 0.8328ms 1.2008 KOps/s 1.1845 KOps/s $\color{#35bf28}+1.37\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.4787ms 3.3226ms 300.9665 Ops/s 299.7209 Ops/s $\color{#35bf28}+0.42\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.7745ms 0.3707ms 2.6975 KOps/s 2.6313 KOps/s $\color{#35bf28}+2.51\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 2.1892ms 1.7505ms 571.2619 Ops/s 589.1448 Ops/s $\color{#d91a1a}-3.04\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 10.3796ms 10.1759ms 98.2713 Ops/s 97.7843 Ops/s $\color{#35bf28}+0.50\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.4291ms 0.3878ms 2.5788 KOps/s 2.5384 KOps/s $\color{#35bf28}+1.59\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.6021ms 1.4695ms 680.5008 Ops/s 676.5407 Ops/s $\color{#35bf28}+0.59\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 5.9961ms 5.7689ms 173.3441 Ops/s 172.7970 Ops/s $\color{#35bf28}+0.32\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.5278ms 0.3936ms 2.5407 KOps/s 2.5120 KOps/s $\color{#35bf28}+1.14\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.6811ms 1.4797ms 675.8309 Ops/s 675.5346 Ops/s $\color{#35bf28}+0.04\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 6.0960ms 5.8737ms 170.2504 Ops/s 172.6467 Ops/s $\color{#d91a1a}-1.39\%$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.5799ms 0.1728ms 5.7855 KOps/s 5.7328 KOps/s $\color{#35bf28}+0.92\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.7836ms 0.6732ms 1.4854 KOps/s 1.4745 KOps/s $\color{#35bf28}+0.74\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 2.7668ms 2.6652ms 375.2053 Ops/s 367.5748 Ops/s $\color{#35bf28}+2.08\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.7350ms 0.3259ms 3.0680 KOps/s 3.1815 KOps/s $\color{#d91a1a}-3.57\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.6766ms 1.4777ms 676.7253 Ops/s 681.8885 Ops/s $\color{#d91a1a}-0.76\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 9.8295ms 9.3904ms 106.4918 Ops/s 105.3939 Ops/s $\color{#35bf28}+1.04\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.3927ms 0.3199ms 3.1263 KOps/s 3.0696 KOps/s $\color{#35bf28}+1.85\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.2990ms 1.1937ms 837.7320 Ops/s 824.3485 Ops/s $\color{#35bf28}+1.62\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.7641ms 4.6667ms 214.2845 Ops/s 210.7891 Ops/s $\color{#35bf28}+1.66\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.3818ms 0.3220ms 3.1058 KOps/s 3.0630 KOps/s $\color{#35bf28}+1.40\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.2916ms 1.2113ms 825.5862 Ops/s 821.1421 Ops/s $\color{#35bf28}+0.54\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 4.8330ms 4.7021ms 212.6698 Ops/s 209.4819 Ops/s $\color{#35bf28}+1.52\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.1633ms 0.1209ms 8.2719 KOps/s 8.2662 KOps/s $\color{#35bf28}+0.07\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.5511ms 0.4729ms 2.1144 KOps/s 2.1086 KOps/s $\color{#35bf28}+0.28\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 1.9839ms 1.8863ms 530.1272 Ops/s 530.5151 Ops/s $\color{#d91a1a}-0.07\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.3407ms 0.2505ms 3.9927 KOps/s 3.9202 KOps/s $\color{#35bf28}+1.85\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.6636ms 1.2507ms 799.5816 Ops/s 814.6840 Ops/s $\color{#d91a1a}-1.85\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 8.6147ms 8.4537ms 118.2913 Ops/s 116.1214 Ops/s $\color{#35bf28}+1.87\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.3465ms 0.2714ms 3.6847 KOps/s 3.6242 KOps/s $\color{#35bf28}+1.67\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.1272ms 1.0119ms 988.2484 Ops/s 974.4792 Ops/s $\color{#35bf28}+1.41\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 4.1451ms 3.9820ms 251.1299 Ops/s 248.3815 Ops/s $\color{#35bf28}+1.11\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.3656ms 0.2717ms 3.6804 KOps/s 3.6172 KOps/s $\color{#35bf28}+1.75\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.1292ms 1.0154ms 984.8623 Ops/s 969.5511 Ops/s $\color{#35bf28}+1.58\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 4.1804ms 4.0005ms 249.9657 Ops/s 247.2106 Ops/s $\color{#35bf28}+1.11\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.5260ms 16.3227ms 61.2644 Ops/s 61.3299 Ops/s $\color{#d91a1a}-0.11\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 33.2412ms 32.9993ms 30.3037 Ops/s 30.4894 Ops/s $\color{#d91a1a}-0.61\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 84.1733ms 83.4013ms 11.9902 Ops/s 11.9856 Ops/s $\color{#35bf28}+0.04\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.2012s 0.2006s 4.9846 Ops/s 5.0628 Ops/s $\color{#d91a1a}-1.55\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 14.1329ms 13.8326ms 72.2932 Ops/s 72.6757 Ops/s $\color{#d91a1a}-0.53\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 28.4247ms 27.3099ms 36.6168 Ops/s 37.2031 Ops/s $\color{#d91a1a}-1.58\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 13.9921ms 13.8930ms 71.9788 Ops/s 71.5582 Ops/s $\color{#35bf28}+0.59\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 27.7254ms 27.5792ms 36.2592 Ops/s 36.5905 Ops/s $\color{#d91a1a}-0.91\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 52.7015ms 52.0526ms 19.2113 Ops/s 18.4813 Ops/s $\color{#35bf28}+3.95\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1058s 0.1054s 9.4880 Ops/s 9.4003 Ops/s $\color{#35bf28}+0.93\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1445s 0.1422s 7.0332 Ops/s 6.6641 Ops/s $\textbf{\color{#35bf28}+5.54\%}$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3259s 0.3219s 3.1067 Ops/s 3.1424 Ops/s $\color{#d91a1a}-1.13\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 78.8252ms 77.7338ms 12.8644 Ops/s 13.3839 Ops/s $\color{#d91a1a}-3.88\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1572s 0.1566s 6.3868 Ops/s 6.7251 Ops/s $\textbf{\color{#d91a1a}-5.03\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 79.1189ms 78.6532ms 12.7140 Ops/s 13.3735 Ops/s $\color{#d91a1a}-4.93\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1580s 0.1568s 6.3783 Ops/s 6.6497 Ops/s $\color{#d91a1a}-4.08\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 30.4407ms 30.2519ms 33.0557 Ops/s 32.6675 Ops/s $\color{#35bf28}+1.19\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 61.9292ms 61.5421ms 16.2490 Ops/s 16.2416 Ops/s $\color{#35bf28}+0.05\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1199s 0.1188s 8.4173 Ops/s 8.4408 Ops/s $\color{#d91a1a}-0.28\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2683s 0.2673s 3.7414 Ops/s 3.8277 Ops/s $\color{#d91a1a}-2.26\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 42.9838ms 42.6674ms 23.4371 Ops/s 23.1668 Ops/s $\color{#35bf28}+1.17\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 85.9322ms 85.6567ms 11.6745 Ops/s 11.6027 Ops/s $\color{#35bf28}+0.62\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 43.2038ms 43.0594ms 23.2237 Ops/s 23.0501 Ops/s $\color{#35bf28}+0.75\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 86.6008ms 86.2521ms 11.5939 Ops/s 11.5212 Ops/s $\color{#35bf28}+0.63\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 20.6963ms 20.5171ms 48.7399 Ops/s 47.6847 Ops/s $\color{#35bf28}+2.21\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 42.5634ms 41.9367ms 23.8455 Ops/s 23.7171 Ops/s $\color{#35bf28}+0.54\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1076s 0.1068s 9.3660 Ops/s 9.2067 Ops/s $\color{#35bf28}+1.73\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2395s 0.2376s 4.2091 Ops/s 4.1655 Ops/s $\color{#35bf28}+1.05\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 36.4524ms 36.1133ms 27.6906 Ops/s 27.3702 Ops/s $\color{#35bf28}+1.17\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 72.4809ms 71.4795ms 13.9900 Ops/s 13.7364 Ops/s $\color{#35bf28}+1.85\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 36.4450ms 36.0302ms 27.7545 Ops/s 27.4094 Ops/s $\color{#35bf28}+1.26\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 72.7735ms 71.8562ms 13.9167 Ops/s 13.5392 Ops/s $\color{#35bf28}+2.79\%$
test_plain_set_nested 43.2510μs 15.4547μs 64.7053 KOps/s 65.3823 KOps/s $\color{#d91a1a}-1.04\%$
test_plain_set_stack_nested 39.0210μs 15.2068μs 65.7602 KOps/s 65.5154 KOps/s $\color{#35bf28}+0.37\%$
test_plain_set_nested_inplace 44.4410μs 16.7234μs 59.7966 KOps/s 59.0390 KOps/s $\color{#35bf28}+1.28\%$
test_plain_set_stack_nested_inplace 49.5310μs 16.4781μs 60.6868 KOps/s 59.9899 KOps/s $\color{#35bf28}+1.16\%$
test_items 31.0910μs 5.9904μs 166.9340 KOps/s 167.3999 KOps/s $\color{#d91a1a}-0.28\%$
test_items_nested 0.5352ms 0.4699ms 2.1283 KOps/s 2.1734 KOps/s $\color{#d91a1a}-2.08\%$
test_items_nested_locked 0.5608ms 0.4692ms 2.1312 KOps/s 2.1645 KOps/s $\color{#d91a1a}-1.54\%$
test_items_nested_leaf 0.1480ms 96.1147μs 10.4042 KOps/s 10.3663 KOps/s $\color{#35bf28}+0.37\%$
test_items_stack_nested 0.5500ms 0.4723ms 2.1173 KOps/s 2.1674 KOps/s $\color{#d91a1a}-2.31\%$
test_items_stack_nested_leaf 0.1349ms 96.8443μs 10.3259 KOps/s 10.3787 KOps/s $\color{#d91a1a}-0.51\%$
test_items_stack_nested_locked 0.5613ms 0.4721ms 2.1181 KOps/s 2.1503 KOps/s $\color{#d91a1a}-1.50\%$
test_keys 33.0710μs 4.2166μs 237.1593 KOps/s 237.6799 KOps/s $\color{#d91a1a}-0.22\%$
test_keys_nested 0.1789ms 0.1325ms 7.5473 KOps/s 7.6869 KOps/s $\color{#d91a1a}-1.82\%$
test_keys_nested_locked 2.0298ms 0.1417ms 7.0593 KOps/s 7.2415 KOps/s $\color{#d91a1a}-2.52\%$
test_keys_nested_leaf 0.1688ms 0.1234ms 8.1011 KOps/s 8.3568 KOps/s $\color{#d91a1a}-3.06\%$
test_keys_stack_nested 0.2228ms 0.1323ms 7.5604 KOps/s 7.6576 KOps/s $\color{#d91a1a}-1.27\%$
test_keys_stack_nested_leaf 0.1547ms 0.1218ms 8.2093 KOps/s 8.3375 KOps/s $\color{#d91a1a}-1.54\%$
test_keys_stack_nested_locked 0.2025ms 0.1398ms 7.1548 KOps/s 7.2961 KOps/s $\color{#d91a1a}-1.94\%$
test_values 6.6622μs 1.0052μs 994.8069 KOps/s 982.4111 KOps/s $\color{#35bf28}+1.26\%$
test_values_nested 83.9820μs 53.1292μs 18.8220 KOps/s 18.9483 KOps/s $\color{#d91a1a}-0.67\%$
test_values_nested_locked 0.1102ms 56.4883μs 17.7028 KOps/s 18.0257 KOps/s $\color{#d91a1a}-1.79\%$
test_values_nested_leaf 92.4020μs 60.8628μs 16.4304 KOps/s 16.4636 KOps/s $\color{#d91a1a}-0.20\%$
test_values_stack_nested 0.1012ms 52.5846μs 19.0170 KOps/s 19.0743 KOps/s $\color{#d91a1a}-0.30\%$
test_values_stack_nested_leaf 91.5420μs 60.2646μs 16.5935 KOps/s 16.6944 KOps/s $\color{#d91a1a}-0.60\%$
test_values_stack_nested_locked 90.3420μs 55.8601μs 17.9019 KOps/s 18.0451 KOps/s $\color{#d91a1a}-0.79\%$
test_membership 6.6450μs 0.8171μs 1.2238 MOps/s 1.2045 MOps/s $\color{#35bf28}+1.60\%$
test_membership_nested 53.2810μs 2.9236μs 342.0387 KOps/s 338.4681 KOps/s $\color{#35bf28}+1.05\%$
test_membership_nested_leaf 57.2220μs 2.9377μs 340.4042 KOps/s 337.6273 KOps/s $\color{#35bf28}+0.82\%$
test_membership_stacked_nested 28.8210μs 2.8396μs 352.1603 KOps/s 347.6481 KOps/s $\color{#35bf28}+1.30\%$
test_membership_stacked_nested_leaf 25.8600μs 2.8331μs 352.9697 KOps/s 346.9724 KOps/s $\color{#35bf28}+1.73\%$
test_membership_nested_last 32.0010μs 4.3877μs 227.9115 KOps/s 227.8218 KOps/s $\color{#35bf28}+0.04\%$
test_membership_nested_leaf_last 33.1010μs 4.4150μs 226.5006 KOps/s 226.4246 KOps/s $\color{#35bf28}+0.03\%$
test_membership_stacked_nested_last 33.8210μs 4.2889μs 233.1623 KOps/s 229.4508 KOps/s $\color{#35bf28}+1.62\%$
test_membership_stacked_nested_leaf_last 37.7700μs 4.2530μs 235.1280 KOps/s 231.9699 KOps/s $\color{#35bf28}+1.36\%$
test_nested_getleaf 52.9510μs 21.5418μs 46.4214 KOps/s 45.9083 KOps/s $\color{#35bf28}+1.12\%$
test_nested_get 65.0110μs 20.5067μs 48.7646 KOps/s 49.5409 KOps/s $\color{#d91a1a}-1.57\%$
test_stacked_getleaf 62.8920μs 21.4676μs 46.5819 KOps/s 47.4487 KOps/s $\color{#d91a1a}-1.83\%$
test_stacked_get 56.2310μs 20.3254μs 49.1995 KOps/s 49.5979 KOps/s $\color{#d91a1a}-0.80\%$
test_nested_getitemleaf 63.6910μs 21.8736μs 45.7173 KOps/s 45.8460 KOps/s $\color{#d91a1a}-0.28\%$
test_nested_getitem 79.8620μs 20.6990μs 48.3115 KOps/s 48.5108 KOps/s $\color{#d91a1a}-0.41\%$
test_stacked_getitemleaf 56.6310μs 21.6344μs 46.2227 KOps/s 46.5716 KOps/s $\color{#d91a1a}-0.75\%$
test_stacked_getitem 55.6710μs 20.7774μs 48.1293 KOps/s 48.3260 KOps/s $\color{#d91a1a}-0.41\%$
test_lock_nested 0.6412ms 0.4953ms 2.0191 KOps/s 1.9970 KOps/s $\color{#35bf28}+1.11\%$
test_lock_stack_nested 0.5783ms 0.5079ms 1.9689 KOps/s 1.9813 KOps/s $\color{#d91a1a}-0.63\%$
test_unlock_nested 0.4764ms 0.3947ms 2.5333 KOps/s 2.5369 KOps/s $\color{#d91a1a}-0.14\%$
test_unlock_stack_nested 0.6013ms 0.4018ms 2.4889 KOps/s 2.5117 KOps/s $\color{#d91a1a}-0.91\%$
test_flatten_speed 0.1675ms 0.1223ms 8.1784 KOps/s 8.1282 KOps/s $\color{#35bf28}+0.62\%$
test_unflatten_speed 0.6598ms 0.5726ms 1.7463 KOps/s 1.7662 KOps/s $\color{#d91a1a}-1.13\%$
test_common_ops 99.9523ms 1.5665ms 638.3678 Ops/s 271.6387 Ops/s $\textbf{\color{#35bf28}+135.01\%}$
test_creation 81.5910μs 3.1897μs 313.5101 KOps/s 317.3358 KOps/s $\color{#d91a1a}-1.21\%$
test_creation_empty 44.8210μs 6.9989μs 142.8793 KOps/s 141.9701 KOps/s $\color{#35bf28}+0.64\%$
test_creation_nested_1 38.0810μs 11.4249μs 87.5279 KOps/s 86.5110 KOps/s $\color{#35bf28}+1.18\%$
test_creation_nested_2 47.7210μs 13.5596μs 73.7484 KOps/s 73.1289 KOps/s $\color{#35bf28}+0.85\%$
test_creation_many_keys[10] 50.4810μs 21.0375μs 47.5342 KOps/s 47.1590 KOps/s $\color{#35bf28}+0.80\%$
test_creation_many_keys[50] 0.2198ms 86.7922μs 11.5218 KOps/s 10.9378 KOps/s $\textbf{\color{#35bf28}+5.34\%}$
test_creation_many_keys[100] 0.2496ms 0.1770ms 5.6506 KOps/s 5.5991 KOps/s $\color{#35bf28}+0.92\%$
test_creation_nested_many_keys[10] 90.3320μs 45.0671μs 22.1891 KOps/s 21.8384 KOps/s $\color{#35bf28}+1.61\%$
test_creation_nested_many_keys[50] 0.2634ms 0.1826ms 5.4766 KOps/s 5.3953 KOps/s $\color{#35bf28}+1.51\%$
test_clone 44.4710μs 13.3548μs 74.8793 KOps/s 74.3500 KOps/s $\color{#35bf28}+0.71\%$
test_getitem[int] 1.4979ms 15.0800μs 66.3131 KOps/s 65.5267 KOps/s $\color{#35bf28}+1.20\%$
test_getitem[slice_int] 0.1402ms 24.4346μs 40.9256 KOps/s 40.9591 KOps/s $\color{#d91a1a}-0.08\%$
test_getitem[range] 0.1769ms 64.3368μs 15.5432 KOps/s 15.7769 KOps/s $\color{#d91a1a}-1.48\%$
test_getitem[tuple] 0.1604ms 24.2958μs 41.1593 KOps/s 41.7849 KOps/s $\color{#d91a1a}-1.50\%$
test_getitem[list] 0.1874ms 59.1933μs 16.8938 KOps/s 16.6643 KOps/s $\color{#35bf28}+1.38\%$
test_setitem_dim[int] 35.0310μs 25.9779μs 38.4943 KOps/s 39.3327 KOps/s $\color{#d91a1a}-2.13\%$
test_setitem_dim[slice_int] 67.6610μs 42.8169μs 23.3553 KOps/s 23.5322 KOps/s $\color{#d91a1a}-0.75\%$
test_setitem_dim[range] 0.1239ms 95.5535μs 10.4653 KOps/s 10.3684 KOps/s $\color{#35bf28}+0.93\%$
test_setitem_dim[tuple] 63.0710μs 39.3298μs 25.4260 KOps/s 25.3692 KOps/s $\color{#35bf28}+0.22\%$
test_setitem 75.7620μs 17.7625μs 56.2983 KOps/s 55.5303 KOps/s $\color{#35bf28}+1.38\%$
test_set 51.0420μs 17.0356μs 58.7005 KOps/s 58.7101 KOps/s $\color{#d91a1a}-0.02\%$
test_set_shared 0.5401ms 0.2099ms 4.7641 KOps/s 4.8168 KOps/s $\color{#d91a1a}-1.09\%$
test_update 0.3453ms 21.9362μs 45.5867 KOps/s 45.5717 KOps/s $\color{#35bf28}+0.03\%$
test_update_nested 77.2520μs 33.6997μs 29.6739 KOps/s 29.7060 KOps/s $\color{#d91a1a}-0.11\%$
test_update__nested 0.4670ms 33.4332μs 29.9104 KOps/s 30.0004 KOps/s $\color{#d91a1a}-0.30\%$
test_set_nested 45.2910μs 18.8191μs 53.1376 KOps/s 52.4760 KOps/s $\color{#35bf28}+1.26\%$
test_set_nested_new 73.3310μs 24.0075μs 41.6537 KOps/s 41.7822 KOps/s $\color{#d91a1a}-0.31\%$
test_select 85.0720μs 41.2015μs 24.2710 KOps/s 24.9002 KOps/s $\color{#d91a1a}-2.53\%$
test_select_nested 0.1127ms 75.3213μs 13.2765 KOps/s 13.2701 KOps/s $\color{#35bf28}+0.05\%$
test_exclude_nested 0.1262ms 93.2777μs 10.7207 KOps/s 10.9157 KOps/s $\color{#d91a1a}-1.79\%$
test_empty[True] 0.4471ms 0.4068ms 2.4579 KOps/s 2.4989 KOps/s $\color{#d91a1a}-1.64\%$
test_empty[False] 10.7767μs 1.4084μs 710.0492 KOps/s 723.7296 KOps/s $\color{#d91a1a}-1.89\%$
test_to 0.1335ms 70.5703μs 14.1703 KOps/s 14.4671 KOps/s $\color{#d91a1a}-2.05\%$
test_to_nonblocking 0.1098ms 69.0384μs 14.4847 KOps/s 14.7161 KOps/s $\color{#d91a1a}-1.57\%$
test_unbind_speed 0.3889ms 0.3388ms 2.9514 KOps/s 2.9783 KOps/s $\color{#d91a1a}-0.90\%$
test_unbind_speed_stack0 0.3836ms 0.3290ms 3.0391 KOps/s 3.0276 KOps/s $\color{#35bf28}+0.38\%$
test_unbind_speed_stack1 0.1144s 0.8460ms 1.1820 KOps/s 1.1728 KOps/s $\color{#35bf28}+0.79\%$
test_split 0.1142s 1.2844ms 778.5653 Ops/s 711.5816 Ops/s $\textbf{\color{#35bf28}+9.41\%}$
test_chunk 1.1623ms 1.0942ms 913.8819 Ops/s 918.5037 Ops/s $\color{#d91a1a}-0.50\%$
test_to_cpu_blocking 0.1336s 22.3085ms 44.8259 Ops/s 50.1082 Ops/s $\textbf{\color{#d91a1a}-10.54\%}$
test_to_cpu_global_sync 11.8340ms 11.6809ms 85.6096 Ops/s 85.5128 Ops/s $\color{#35bf28}+0.11\%$
test_to_cpu_event_sync 12.9175ms 12.7052ms 78.7081 Ops/s 79.0136 Ops/s $\color{#d91a1a}-0.39\%$
test_to_cpu_default 12.9385ms 12.6699ms 78.9271 Ops/s 70.3930 Ops/s $\textbf{\color{#35bf28}+12.12\%}$
test_consolidate[False-None] 4.3747ms 4.2297ms 236.4249 Ops/s 238.4315 Ops/s $\color{#d91a1a}-0.84\%$
test_consolidate[default-None] 3.5633ms 2.1756ms 459.6360 Ops/s 459.5015 Ops/s $\color{#35bf28}+0.03\%$
test_consolidate[reduce-overhead-None] 3.4761ms 2.0817ms 480.3807 Ops/s 473.2250 Ops/s $\color{#35bf28}+1.51\%$
test_consolidate_njt[False-None] 8.7902ms 8.5072ms 117.5478 Ops/s 115.7012 Ops/s $\color{#35bf28}+1.60\%$
test_to[False-False-None] 2.3124ms 2.2060ms 453.3169 Ops/s 459.9812 Ops/s $\color{#d91a1a}-1.45\%$
test_to[True-False-None] 2.2081ms 1.9245ms 519.6077 Ops/s 526.3442 Ops/s $\color{#d91a1a}-1.28\%$
test_to[within-False-None] 6.5955ms 6.2875ms 159.0462 Ops/s 160.7681 Ops/s $\color{#d91a1a}-1.07\%$
test_to[True-default-None] 10.8041ms 10.5370ms 94.9041 Ops/s 92.8890 Ops/s $\color{#35bf28}+2.17\%$
test_to_njt[False-False-None] 8.6954ms 8.5824ms 116.5175 Ops/s 114.4123 Ops/s $\color{#35bf28}+1.84\%$
test_to_njt[True-False-None] 7.5663ms 6.9086ms 144.7461 Ops/s 142.2618 Ops/s $\color{#35bf28}+1.75\%$
test_to_njt[within-False-None] 15.6031ms 15.4127ms 64.8817 Ops/s 63.2052 Ops/s $\color{#35bf28}+2.65\%$
test_creation[device0] 0.4067ms 0.1160ms 8.6181 KOps/s 8.5582 KOps/s $\color{#35bf28}+0.70\%$
test_creation_from_tensor 0.4139ms 0.1130ms 8.8504 KOps/s 8.7783 KOps/s $\color{#35bf28}+0.82\%$
test_add_one[memmap_tensor0] 0.3339ms 6.6992μs 149.2721 KOps/s 146.6212 KOps/s $\color{#35bf28}+1.81\%$
test_contiguous[memmap_tensor0] 19.0000μs 0.6705μs 1.4914 MOps/s 1.9853 MOps/s $\textbf{\color{#d91a1a}-24.88\%}$
test_stack[memmap_tensor0] 36.9410μs 4.3165μs 231.6688 KOps/s 229.9784 KOps/s $\color{#35bf28}+0.74\%$
test_memmaptd_index 1.1945ms 0.2731ms 3.6613 KOps/s 3.6841 KOps/s $\color{#d91a1a}-0.62\%$
test_memmaptd_index_astensor 0.5306ms 0.3768ms 2.6539 KOps/s 2.6396 KOps/s $\color{#35bf28}+0.54\%$
test_memmaptd_index_op 0.7985ms 0.6386ms 1.5658 KOps/s 1.5399 KOps/s $\color{#35bf28}+1.69\%$
test_serialize_model 0.1361s 0.1342s 7.4490 Ops/s 7.4666 Ops/s $\color{#d91a1a}-0.24\%$
test_serialize_model_pickle 1.3690s 1.2176s 0.8213 Ops/s 0.8266 Ops/s $\color{#d91a1a}-0.65\%$
test_serialize_weights 0.3150s 0.1585s 6.3105 Ops/s 7.4782 Ops/s $\textbf{\color{#d91a1a}-15.62\%}$
test_serialize_weights_returnearly 75.1029ms 63.7528ms 15.6856 Ops/s 6.8618 Ops/s $\textbf{\color{#35bf28}+128.59\%}$
test_serialize_weights_pickle 1.3666s 1.2132s 0.8242 Ops/s 0.8242 Ops/s $-0.00\%$
test_reshape_pytree 0.2075ms 32.7216μs 30.5608 KOps/s 30.7071 KOps/s $\color{#d91a1a}-0.48\%$
test_reshape_td 70.6710μs 45.2237μs 22.1123 KOps/s 22.6595 KOps/s $\color{#d91a1a}-2.41\%$
test_view_pytree 0.2146ms 32.1630μs 31.0916 KOps/s 30.9022 KOps/s $\color{#35bf28}+0.61\%$
test_view_td 0.1036ms 53.1224μs 18.8245 KOps/s 18.8057 KOps/s $\color{#35bf28}+0.10\%$
test_unbind_pytree 0.2338ms 36.0358μs 27.7502 KOps/s 26.8349 KOps/s $\color{#35bf28}+3.41\%$
test_unbind_td 97.3320μs 49.6324μs 20.1481 KOps/s 20.2172 KOps/s $\color{#d91a1a}-0.34\%$
test_split_pytree 0.1924ms 42.3316μs 23.6230 KOps/s 23.4150 KOps/s $\color{#35bf28}+0.89\%$
test_split_td 0.2209ms 63.7462μs 15.6872 KOps/s 15.5687 KOps/s $\color{#35bf28}+0.76\%$
test_add_pytree 0.2341ms 42.0460μs 23.7835 KOps/s 23.7069 KOps/s $\color{#35bf28}+0.32\%$
test_add_td 0.1154ms 55.2285μs 18.1066 KOps/s 18.0750 KOps/s $\color{#35bf28}+0.17\%$
test_compile_add_one_nested[tensordict-compile] 0.6011ms 0.2348ms 4.2589 KOps/s 4.2952 KOps/s $\color{#d91a1a}-0.84\%$
test_compile_add_one_nested[tensordict-eager] 0.4503ms 0.2116ms 4.7254 KOps/s 5.0209 KOps/s $\textbf{\color{#d91a1a}-5.88\%}$
test_compile_add_one_nested[pytree-compile] 0.3383ms 0.1326ms 7.5410 KOps/s 7.4240 KOps/s $\color{#35bf28}+1.58\%$
test_compile_add_one_nested[pytree-eager] 0.3999ms 0.1752ms 5.7073 KOps/s 5.6193 KOps/s $\color{#35bf28}+1.57\%$
test_compile_copy_nested[tensordict-compile] 77.5220μs 15.8479μs 63.0999 KOps/s 65.3363 KOps/s $\color{#d91a1a}-3.42\%$
test_compile_copy_nested[tensordict-eager] 0.3053ms 57.4838μs 17.3962 KOps/s 18.2294 KOps/s $\color{#d91a1a}-4.57\%$
test_compile_copy_nested[pytree-compile] 0.1559ms 15.4019μs 64.9269 KOps/s 63.9118 KOps/s $\color{#35bf28}+1.59\%$
test_compile_copy_nested[pytree-eager] 0.3652ms 68.1187μs 14.6802 KOps/s 14.5375 KOps/s $\color{#35bf28}+0.98\%$
test_compile_add_one_flat[tensordict-compile] 0.6264ms 0.2464ms 4.0587 KOps/s 3.7411 KOps/s $\textbf{\color{#35bf28}+8.49\%}$
test_compile_add_one_flat[tensordict-eager] 0.3795ms 0.2812ms 3.5557 KOps/s 3.5354 KOps/s $\color{#35bf28}+0.57\%$
test_compile_add_one_flat[tensorclass-compile] 0.3470ms 0.1882ms 5.3146 KOps/s 5.0229 KOps/s $\textbf{\color{#35bf28}+5.81\%}$
test_compile_add_one_flat[tensorclass-eager] 0.1596ms 76.1951μs 13.1242 KOps/s 13.4610 KOps/s $\color{#d91a1a}-2.50\%$
test_compile_add_one_flat[pytree-compile] 0.2937ms 0.1898ms 5.2700 KOps/s 5.0605 KOps/s $\color{#35bf28}+4.14\%$
test_compile_add_one_flat[pytree-eager] 0.8188ms 0.5159ms 1.9383 KOps/s 1.9504 KOps/s $\color{#d91a1a}-0.62\%$
test_compile_add_self_flat[tensordict-eager] 0.4977ms 0.3263ms 3.0642 KOps/s 3.0286 KOps/s $\color{#35bf28}+1.18\%$
test_compile_add_self_flat[tensordict-compile] 0.2943ms 0.2487ms 4.0203 KOps/s 3.7522 KOps/s $\textbf{\color{#35bf28}+7.15\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1417ms 90.5492μs 11.0437 KOps/s 11.0859 KOps/s $\color{#d91a1a}-0.38\%$
test_compile_add_self_flat[tensorclass-compile] 0.2832ms 0.1854ms 5.3936 KOps/s 5.0921 KOps/s $\textbf{\color{#35bf28}+5.92\%}$
test_compile_add_self_flat[pytree-eager] 0.6356ms 0.4204ms 2.3790 KOps/s 2.3410 KOps/s $\color{#35bf28}+1.62\%$
test_compile_add_self_flat[pytree-compile] 0.2367ms 0.1908ms 5.2399 KOps/s 5.0614 KOps/s $\color{#35bf28}+3.53\%$
test_compile_copy_flat[tensordict-compile] 0.1008ms 19.8423μs 50.3973 KOps/s 51.2165 KOps/s $\color{#d91a1a}-1.60\%$
test_compile_copy_flat[tensordict-eager] 0.1050ms 41.6842μs 23.9899 KOps/s 23.8386 KOps/s $\color{#35bf28}+0.63\%$
test_compile_copy_flat[pytree-compile] 0.1929ms 20.9741μs 47.6780 KOps/s 46.9040 KOps/s $\color{#35bf28}+1.65\%$
test_compile_copy_flat[pytree-eager] 0.3346ms 52.7280μs 18.9652 KOps/s 19.0120 KOps/s $\color{#d91a1a}-0.25\%$
test_compile_assign_and_add[tensordict-compile] 2.2035ms 0.2366ms 4.2261 KOps/s 3.8973 KOps/s $\textbf{\color{#35bf28}+8.44\%}$
test_compile_assign_and_add[tensordict-eager] 3.4487ms 3.2814ms 304.7488 Ops/s 304.6471 Ops/s $\color{#35bf28}+0.03\%$
test_compile_assign_and_add[pytree-compile] 2.0814ms 0.1769ms 5.6515 KOps/s 5.4549 KOps/s $\color{#35bf28}+3.60\%$
test_compile_assign_and_add[pytree-eager] 2.8878ms 2.7636ms 361.8462 Ops/s 364.4197 Ops/s $\color{#d91a1a}-0.71\%$
test_compile_indexing[tensor-tensordict-compile] 0.2789ms 0.1317ms 7.5926 KOps/s 7.2196 KOps/s $\textbf{\color{#35bf28}+5.17\%}$
test_compile_indexing[tensor-tensordict-eager] 0.3254ms 75.6875μs 13.2122 KOps/s 13.3369 KOps/s $\color{#d91a1a}-0.93\%$
test_compile_indexing[tensor-tensorclass-compile] 0.2262ms 0.1169ms 8.5572 KOps/s 8.4068 KOps/s $\color{#35bf28}+1.79\%$
test_compile_indexing[tensor-tensorclass-eager] 0.2573ms 44.5810μs 22.4311 KOps/s 22.0396 KOps/s $\color{#35bf28}+1.78\%$
test_compile_indexing[tensor-pytree-compile] 0.2000ms 0.1213ms 8.2430 KOps/s 8.3644 KOps/s $\color{#d91a1a}-1.45\%$
test_compile_indexing[tensor-pytree-eager] 0.2705ms 44.1698μs 22.6399 KOps/s 21.8416 KOps/s $\color{#35bf28}+3.65\%$
test_compile_indexing[slice-tensordict-compile] 0.5034ms 82.3451μs 12.1440 KOps/s 12.0040 KOps/s $\color{#35bf28}+1.17\%$
test_compile_indexing[slice-tensordict-eager] 0.2058ms 28.9528μs 34.5390 KOps/s 36.2038 KOps/s $\color{#d91a1a}-4.60\%$
test_compile_indexing[slice-tensorclass-compile] 0.1600ms 69.2294μs 14.4447 KOps/s 14.1681 KOps/s $\color{#35bf28}+1.95\%$
test_compile_indexing[slice-tensorclass-eager] 0.2338ms 22.0618μs 45.3272 KOps/s 44.2923 KOps/s $\color{#35bf28}+2.34\%$
test_compile_indexing[slice-pytree-compile] 0.1239ms 70.1003μs 14.2653 KOps/s 14.0254 KOps/s $\color{#35bf28}+1.71\%$
test_compile_indexing[slice-pytree-eager] 0.2579ms 21.8060μs 45.8589 KOps/s 44.5862 KOps/s $\color{#35bf28}+2.85\%$
test_compile_indexing[int-tensordict-compile] 0.1492ms 83.2246μs 12.0157 KOps/s 11.9528 KOps/s $\color{#35bf28}+0.53\%$
test_compile_indexing[int-tensordict-eager] 0.2141ms 27.9985μs 35.7162 KOps/s 35.9504 KOps/s $\color{#d91a1a}-0.65\%$
test_compile_indexing[int-tensorclass-compile] 0.1140ms 71.7032μs 13.9464 KOps/s 14.0628 KOps/s $\color{#d91a1a}-0.83\%$
test_compile_indexing[int-tensorclass-eager] 0.2451ms 21.9615μs 45.5342 KOps/s 44.6262 KOps/s $\color{#35bf28}+2.03\%$
test_compile_indexing[int-pytree-compile] 0.1306ms 71.4354μs 13.9987 KOps/s 14.0293 KOps/s $\color{#d91a1a}-0.22\%$
test_compile_indexing[int-pytree-eager] 0.2294ms 21.8758μs 45.7126 KOps/s 44.8523 KOps/s $\color{#35bf28}+1.92\%$
test_compile_replace[single-eager] 0.1083ms 50.0695μs 19.9722 KOps/s 21.0234 KOps/s $\color{#d91a1a}-5.00\%$
test_compile_replace[single-compile] 0.2408ms 0.1338ms 7.4711 KOps/s 7.3606 KOps/s $\color{#35bf28}+1.50\%$
test_compile_replace[multi-eager] 0.6742ms 0.5889ms 1.6980 KOps/s 1.7207 KOps/s $\color{#d91a1a}-1.32\%$
test_compile_replace[multi-compile] 0.3294ms 0.1424ms 7.0230 KOps/s 7.2005 KOps/s $\color{#d91a1a}-2.47\%$
test_compile_tc_getattr_20[eager] 0.2214ms 0.1708ms 5.8538 KOps/s 6.1897 KOps/s $\textbf{\color{#d91a1a}-5.43\%}$
test_compile_tc_getattr_20[compile] 0.3414ms 0.1510ms 6.6209 KOps/s 6.5947 KOps/s $\color{#35bf28}+0.40\%$
test_compile_clone_shallow[20-eager] 70.4810μs 19.6293μs 50.9442 KOps/s 50.5384 KOps/s $\color{#35bf28}+0.80\%$
test_compile_clone_shallow[20-compile] 75.0820μs 17.6722μs 56.5861 KOps/s 49.0768 KOps/s $\textbf{\color{#35bf28}+15.30\%}$
test_compile_clone_shallow[40-eager] 89.4920μs 33.8658μs 29.5284 KOps/s 29.0274 KOps/s $\color{#35bf28}+1.73\%$
test_compile_clone_shallow[40-compile] 88.1120μs 19.4533μs 51.4052 KOps/s 54.5337 KOps/s $\textbf{\color{#d91a1a}-5.74\%}$
test_compile_clone_shallow[80-eager] 0.1029ms 61.9414μs 16.1443 KOps/s 15.7951 KOps/s $\color{#35bf28}+2.21\%$
test_compile_clone_shallow[80-compile] 93.5920μs 22.7209μs 44.0124 KOps/s 44.5716 KOps/s $\color{#d91a1a}-1.25\%$
test_compile_update_inplace[eager] 0.2224ms 61.0083μs 16.3912 KOps/s 16.3473 KOps/s $\color{#35bf28}+0.27\%$
test_compile_update_inplace[compile] 0.3671ms 0.1465ms 6.8280 KOps/s 6.7352 KOps/s $\color{#35bf28}+1.38\%$
test_mod_add[eager] 0.3832ms 0.2248ms 4.4478 KOps/s 4.4434 KOps/s $\color{#35bf28}+0.10\%$
test_mod_add[compile] 0.2558ms 0.1334ms 7.4959 KOps/s 7.5232 KOps/s $\color{#d91a1a}-0.36\%$
test_mod_add[compile-overhead] 0.2980ms 0.1761ms 5.6797 KOps/s 5.5820 KOps/s $\color{#35bf28}+1.75\%$
test_mod_wrap[eager] 0.5482ms 0.4719ms 2.1191 KOps/s 2.1244 KOps/s $\color{#d91a1a}-0.25\%$
test_mod_wrap[compile] 0.5026ms 0.4073ms 2.4553 KOps/s 2.5230 KOps/s $\color{#d91a1a}-2.68\%$
test_mod_wrap[compile-overhead] 8.6904ms 4.7588ms 210.1365 Ops/s 212.5233 Ops/s $\color{#d91a1a}-1.12\%$
test_mod_wrap_and_backward[eager] 1.9777ms 1.7685ms 565.4492 Ops/s 568.2887 Ops/s $\color{#d91a1a}-0.50\%$
test_mod_wrap_and_backward[compile] 2.2699ms 1.5600ms 641.0156 Ops/s 651.2348 Ops/s $\color{#d91a1a}-1.57\%$
test_mod_wrap_and_backward[compile-overhead] 1.3321ms 0.9398ms 1.0640 KOps/s 1.0428 KOps/s $\color{#35bf28}+2.04\%$
test_seq_add[eager] 0.5582ms 0.5086ms 1.9661 KOps/s 1.9669 KOps/s $\color{#d91a1a}-0.04\%$
test_seq_add[compile] 0.2193ms 0.1413ms 7.0780 KOps/s 6.8642 KOps/s $\color{#35bf28}+3.11\%$
test_seq_add[compile-overhead] 0.3540ms 0.1818ms 5.5008 KOps/s 5.0504 KOps/s $\textbf{\color{#35bf28}+8.92\%}$
test_seq_wrap[eager] 1.5035ms 1.4043ms 712.0750 Ops/s 705.2318 Ops/s $\color{#35bf28}+0.97\%$
test_seq_wrap[compile] 0.4971ms 0.4044ms 2.4730 KOps/s 2.3876 KOps/s $\color{#35bf28}+3.57\%$
test_seq_wrap[compile-overhead] 0.3778ms 0.3033ms 3.2968 KOps/s 3.2007 KOps/s $\color{#35bf28}+3.00\%$
test_func_call_runtime[False-eager] 0.8812ms 0.7800ms 1.2821 KOps/s 1.2500 KOps/s $\color{#35bf28}+2.57\%$
test_func_call_runtime[False-compile] 1.1615ms 0.9707ms 1.0301 KOps/s 1.0200 KOps/s $\color{#35bf28}+0.99\%$
test_func_call_runtime[False-compile-overhead] 0.7013ms 0.5096ms 1.9621 KOps/s 1.9299 KOps/s $\color{#35bf28}+1.67\%$
test_func_call_runtime[True-eager] 1.1043ms 1.0274ms 973.3589 Ops/s 949.9622 Ops/s $\color{#35bf28}+2.46\%$
test_func_call_runtime[True-compile] 1.0940ms 0.9809ms 1.0195 KOps/s 999.1011 Ops/s $\color{#35bf28}+2.04\%$
test_func_call_runtime[True-compile-overhead] 0.5972ms 0.5268ms 1.8983 KOps/s 1.8690 KOps/s $\color{#35bf28}+1.57\%$
test_func_call_cm_runtime[False-eager] 0.8933ms 0.7842ms 1.2752 KOps/s 1.2485 KOps/s $\color{#35bf28}+2.14\%$
test_func_call_cm_runtime[False-compile] 1.0762ms 0.9506ms 1.0520 KOps/s 1.0368 KOps/s $\color{#35bf28}+1.47\%$
test_func_call_cm_runtime[False-compile-overhead] 0.6332ms 0.4915ms 2.0344 KOps/s 2.0096 KOps/s $\color{#35bf28}+1.24\%$
test_func_call_cm_runtime[True-eager] 1.2894ms 1.1650ms 858.3375 Ops/s 836.6930 Ops/s $\color{#35bf28}+2.59\%$
test_func_call_cm_runtime[True-compile] 1.0725ms 1.0054ms 994.6329 Ops/s 978.1415 Ops/s $\color{#35bf28}+1.69\%$
test_func_call_cm_runtime[True-compile-overhead] 0.7053ms 0.5496ms 1.8193 KOps/s 1.7812 KOps/s $\color{#35bf28}+2.14\%$
test_vmap_func_call_cm_runtime[eager] 2.8328ms 2.3608ms 423.5811 Ops/s 419.1134 Ops/s $\color{#35bf28}+1.07\%$
test_vmap_func_call_cm_runtime[compile] 1.0980ms 1.0130ms 987.2045 Ops/s 974.4313 Ops/s $\color{#35bf28}+1.31\%$
test_vmap_func_call_cm_runtime[compile-overhead] 0.6178ms 0.5364ms 1.8644 KOps/s 1.8450 KOps/s $\color{#35bf28}+1.05\%$
test_distributed 0.7480ms 0.1467ms 6.8161 KOps/s 5.8101 KOps/s $\textbf{\color{#35bf28}+17.31\%}$
test_tdmodule 0.5404ms 0.1924ms 5.1987 KOps/s 5.0949 KOps/s $\color{#35bf28}+2.04\%$
test_tdmodule_dispatch 0.4073ms 0.3662ms 2.7305 KOps/s 2.6542 KOps/s $\color{#35bf28}+2.88\%$
test_tdseq 49.8010μs 25.9018μs 38.6074 KOps/s 36.8971 KOps/s $\color{#35bf28}+4.64\%$
test_tdseq_dispatch 0.2703ms 0.2146ms 4.6602 KOps/s 4.5498 KOps/s $\color{#35bf28}+2.43\%$
test_instantiation_functorch 2.1347ms 2.0665ms 483.9110 Ops/s 480.2355 Ops/s $\color{#35bf28}+0.77\%$
test_exec_functorch 0.2235ms 0.1806ms 5.5372 KOps/s 5.2536 KOps/s $\textbf{\color{#35bf28}+5.40\%}$
test_exec_functional_call 0.2277ms 0.1617ms 6.1845 KOps/s 5.9262 KOps/s $\color{#35bf28}+4.36\%$
test_exec_td_decorator 0.4633ms 0.2386ms 4.1909 KOps/s 4.1180 KOps/s $\color{#35bf28}+1.77\%$
test_vmap_mlp_speed_decorator[True-True] 1.2462ms 1.0136ms 986.5838 Ops/s 976.3210 Ops/s $\color{#35bf28}+1.05\%$
test_vmap_mlp_speed_decorator[True-False] 1.1964ms 1.0109ms 989.1992 Ops/s 974.6406 Ops/s $\color{#35bf28}+1.49\%$
test_vmap_mlp_speed_decorator[False-True] 0.9079ms 0.7167ms 1.3953 KOps/s 1.3413 KOps/s $\color{#35bf28}+4.03\%$
test_vmap_mlp_speed_decorator[False-False] 0.9215ms 0.7179ms 1.3929 KOps/s 1.3843 KOps/s $\color{#35bf28}+0.62\%$
test_vmap_transformer_speed_decorator[True-True] 21.8610ms 20.9072ms 47.8304 Ops/s 47.9927 Ops/s $\color{#d91a1a}-0.34\%$
test_vmap_transformer_speed_decorator[True-False] 21.0555ms 20.8683ms 47.9196 Ops/s 48.0126 Ops/s $\color{#d91a1a}-0.19\%$
test_vmap_transformer_speed_decorator[False-True] 21.1202ms 20.4861ms 48.8135 Ops/s 48.7509 Ops/s $\color{#35bf28}+0.13\%$
test_vmap_transformer_speed_decorator[False-False] 20.7678ms 20.4943ms 48.7940 Ops/s 48.8351 Ops/s $\color{#d91a1a}-0.08\%$
test_to_module_speed[True] 1.5984ms 1.4958ms 668.5447 Ops/s 666.7075 Ops/s $\color{#35bf28}+0.28\%$
test_to_module_speed[False] 1.5684ms 1.4647ms 682.7445 Ops/s 687.6236 Ops/s $\color{#d91a1a}-0.71\%$
test_tc_init 84.7320μs 43.3318μs 23.0778 KOps/s 22.9542 KOps/s $\color{#35bf28}+0.54\%$
test_tc_init_tensor_only 46.3010μs 9.0459μs 110.5477 KOps/s 109.4644 KOps/s $\color{#35bf28}+0.99\%$
test_tc_init_nested 0.1304ms 87.1960μs 11.4684 KOps/s 11.6870 KOps/s $\color{#d91a1a}-1.87\%$
test_tc_init_many_fields 44.4010μs 14.9056μs 67.0891 KOps/s 67.9946 KOps/s $\color{#d91a1a}-1.33\%$
test_tc_first_layer_tensor 18.3400μs 1.8408μs 543.2335 KOps/s 548.1937 KOps/s $\color{#d91a1a}-0.90\%$
test_tc_first_layer_tensor_only 1.5480μs 0.3910μs 2.5575 MOps/s 2.4698 MOps/s $\color{#35bf28}+3.55\%$
test_tc_first_layer_tensor_set 23.3510μs 3.9751μs 251.5642 KOps/s 250.0534 KOps/s $\color{#35bf28}+0.60\%$
test_tc_first_layer_tensor_only_set 27.1800μs 3.2444μs 308.2259 KOps/s 307.8235 KOps/s $\color{#35bf28}+0.13\%$
test_tc_first_layer_nontensor 31.1900μs 6.1607μs 162.3179 KOps/s 162.4377 KOps/s $\color{#d91a1a}-0.07\%$
test_tc_second_layer_tensor 20.1500μs 4.3837μs 228.1193 KOps/s 227.3065 KOps/s $\color{#35bf28}+0.36\%$
test_tc_second_layer_nontensor 42.2910μs 8.6490μs 115.6209 KOps/s 114.0835 KOps/s $\color{#35bf28}+1.35\%$
test_unbind 0.2787s 14.1119ms 70.8622 Ops/s 71.1129 Ops/s $\color{#d91a1a}-0.35\%$
test_full_like 4.4661ms 4.3295ms 230.9738 Ops/s 228.0014 Ops/s $\color{#35bf28}+1.30\%$
test_zeros_like 4.4227ms 4.2829ms 233.4873 Ops/s 230.9640 Ops/s $\color{#35bf28}+1.09\%$
test_ones_like 4.4633ms 4.3201ms 231.4787 Ops/s 230.4883 Ops/s $\color{#35bf28}+0.43\%$
test_clone 6.8007ms 6.4084ms 156.0448 Ops/s 156.1959 Ops/s $\color{#d91a1a}-0.10\%$
test_squeeze 68.0110μs 14.3843μs 69.5200 KOps/s 68.9075 KOps/s $\color{#35bf28}+0.89\%$
test_unsqueeze 0.1813ms 0.1095ms 9.1358 KOps/s 9.0275 KOps/s $\color{#35bf28}+1.20\%$
test_split 0.2630ms 0.1778ms 5.6249 KOps/s 5.5472 KOps/s $\color{#35bf28}+1.40\%$
test_permute 0.2505ms 0.2042ms 4.8963 KOps/s 4.8360 KOps/s $\color{#35bf28}+1.25\%$
test_stack 51.2050ms 50.9691ms 19.6197 Ops/s 19.5190 Ops/s $\color{#35bf28}+0.52\%$
test_cat 51.4052ms 50.9987ms 19.6083 Ops/s 19.5247 Ops/s $\color{#35bf28}+0.43\%$
test_sequential_tensordict 0.6905ms 0.5661ms 1.7664 KOps/s 1.7340 KOps/s $\color{#35bf28}+1.87\%$
test_sequential_graph_module 0.5449ms 0.1176ms 8.5003 KOps/s 8.1634 KOps/s $\color{#35bf28}+4.13\%$
test_nested_tensordict 0.9929ms 0.8123ms 1.2311 KOps/s 1.2277 KOps/s $\color{#35bf28}+0.28\%$
test_nested_graph_module 0.5485ms 0.1281ms 7.8057 KOps/s 7.5849 KOps/s $\color{#35bf28}+2.91\%$

@github-actions

github-actions Bot commented May 24, 2026

Copy link
Copy Markdown
Contributor

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

Total Benchmarks: 341. Improved: $\large\color{#35bf28}38$. 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.2158ms 0.1713ms 5.8367 KOps/s 5.8378 KOps/s $\color{#d91a1a}-0.02\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.7238ms 0.2978ms 3.3575 KOps/s 3.3121 KOps/s $\color{#35bf28}+1.37\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.8221ms 0.7771ms 1.2868 KOps/s 1.2839 KOps/s $\color{#35bf28}+0.23\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.8099ms 0.3423ms 2.9215 KOps/s 2.9103 KOps/s $\color{#35bf28}+0.39\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.2919ms 1.1391ms 877.9212 Ops/s 907.3152 Ops/s $\color{#d91a1a}-3.24\%$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.6506ms 7.2707ms 137.5392 Ops/s 135.5111 Ops/s $\color{#35bf28}+1.50\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.3378ms 0.2540ms 3.9375 KOps/s 3.9895 KOps/s $\color{#d91a1a}-1.30\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.8170ms 0.3727ms 2.6831 KOps/s 2.6341 KOps/s $\color{#35bf28}+1.86\%$
test_collector_preallocated_write[stack_out-cuda-32-32] 1.2582ms 0.8237ms 1.2140 KOps/s 1.2139 KOps/s $+0.01\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.6848ms 0.2517ms 3.9728 KOps/s 4.0005 KOps/s $\color{#d91a1a}-0.69\%$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.8043ms 0.3757ms 2.6614 KOps/s 2.6777 KOps/s $\color{#d91a1a}-0.61\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 1.3208ms 0.8822ms 1.1335 KOps/s 1.1340 KOps/s $\color{#d91a1a}-0.05\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2655ms 0.2183ms 4.5817 KOps/s 4.5747 KOps/s $\color{#35bf28}+0.15\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 1.2764ms 0.8432ms 1.1859 KOps/s 1.1793 KOps/s $\color{#35bf28}+0.57\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.7944ms 3.3697ms 296.7666 Ops/s 295.5402 Ops/s $\color{#35bf28}+0.41\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.8432ms 0.3721ms 2.6876 KOps/s 2.5741 KOps/s $\color{#35bf28}+4.41\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 2.2222ms 1.7449ms 573.0927 Ops/s 575.1668 Ops/s $\color{#d91a1a}-0.36\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 10.3237ms 10.2086ms 97.9562 Ops/s 96.1059 Ops/s $\color{#35bf28}+1.93\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.4548ms 0.4002ms 2.4990 KOps/s 2.5142 KOps/s $\color{#d91a1a}-0.60\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.6848ms 1.5078ms 663.2219 Ops/s 676.0736 Ops/s $\color{#d91a1a}-1.90\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 6.0858ms 5.9107ms 169.1837 Ops/s 167.1932 Ops/s $\color{#35bf28}+1.19\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.5394ms 0.4019ms 2.4884 KOps/s 2.4679 KOps/s $\color{#35bf28}+0.83\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.6737ms 1.5011ms 666.1776 Ops/s 674.4987 Ops/s $\color{#d91a1a}-1.23\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 6.3423ms 5.9162ms 169.0279 Ops/s 171.2885 Ops/s $\color{#d91a1a}-1.32\%$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.5408ms 0.1757ms 5.6914 KOps/s 5.7564 KOps/s $\color{#d91a1a}-1.13\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 1.1149ms 0.6860ms 1.4577 KOps/s 1.4855 KOps/s $\color{#d91a1a}-1.88\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 3.1571ms 2.7379ms 365.2460 Ops/s 374.0184 Ops/s $\color{#d91a1a}-2.35\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.3992ms 0.3208ms 3.1177 KOps/s 3.1485 KOps/s $\color{#d91a1a}-0.98\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.7484ms 1.5528ms 644.0003 Ops/s 676.3754 Ops/s $\color{#d91a1a}-4.79\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 9.9748ms 9.5186ms 105.0569 Ops/s 105.2527 Ops/s $\color{#d91a1a}-0.19\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.7828ms 0.3350ms 2.9855 KOps/s 3.0545 KOps/s $\color{#d91a1a}-2.26\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.6857ms 1.2488ms 800.7511 Ops/s 822.1724 Ops/s $\color{#d91a1a}-2.61\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.9101ms 4.8860ms 204.6661 Ops/s 210.4262 Ops/s $\color{#d91a1a}-2.74\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.7658ms 0.3371ms 2.9662 KOps/s 3.0527 KOps/s $\color{#d91a1a}-2.84\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.6988ms 1.2604ms 793.4063 Ops/s 817.7698 Ops/s $\color{#d91a1a}-2.98\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 5.3222ms 4.9126ms 203.5597 Ops/s 209.2806 Ops/s $\color{#d91a1a}-2.73\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.5451ms 0.1252ms 7.9866 KOps/s 8.2610 KOps/s $\color{#d91a1a}-3.32\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.9145ms 0.4927ms 2.0295 KOps/s 2.1210 KOps/s $\color{#d91a1a}-4.32\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 2.3851ms 1.9603ms 510.1344 Ops/s 530.1652 Ops/s $\color{#d91a1a}-3.78\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.3432ms 0.2665ms 3.7522 KOps/s 3.8729 KOps/s $\color{#d91a1a}-3.12\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.6670ms 1.2174ms 821.4320 Ops/s 798.9762 Ops/s $\color{#35bf28}+2.81\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 9.0375ms 8.6422ms 115.7114 Ops/s 115.1529 Ops/s $\color{#35bf28}+0.49\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.7347ms 0.2919ms 3.4260 KOps/s 3.5964 KOps/s $\color{#d91a1a}-4.74\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.5171ms 1.0838ms 922.6747 Ops/s 968.9677 Ops/s $\color{#d91a1a}-4.78\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 4.6815ms 4.2644ms 234.4986 Ops/s 248.0506 Ops/s $\textbf{\color{#d91a1a}-5.46\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.7252ms 0.2892ms 3.4576 KOps/s 3.6135 KOps/s $\color{#d91a1a}-4.31\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.5174ms 1.0817ms 924.5015 Ops/s 968.0101 Ops/s $\color{#d91a1a}-4.49\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 4.6981ms 4.2579ms 234.8552 Ops/s 246.9999 Ops/s $\color{#d91a1a}-4.92\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.6703ms 16.3062ms 61.3265 Ops/s 60.7637 Ops/s $\color{#35bf28}+0.93\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 33.1577ms 32.6358ms 30.6412 Ops/s 30.5152 Ops/s $\color{#35bf28}+0.41\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 84.7003ms 83.8764ms 11.9223 Ops/s 11.8555 Ops/s $\color{#35bf28}+0.56\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.1979s 0.1974s 5.0667 Ops/s 5.0342 Ops/s $\color{#35bf28}+0.65\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 14.3439ms 13.8553ms 72.1743 Ops/s 72.5407 Ops/s $\color{#d91a1a}-0.51\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 27.4659ms 27.2911ms 36.6419 Ops/s 37.2434 Ops/s $\color{#d91a1a}-1.61\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 14.3221ms 13.9822ms 71.5196 Ops/s 71.6192 Ops/s $\color{#d91a1a}-0.14\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 28.0736ms 27.6120ms 36.2162 Ops/s 36.6528 Ops/s $\color{#d91a1a}-1.19\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 56.3468ms 55.5084ms 18.0153 Ops/s 17.8625 Ops/s $\color{#35bf28}+0.86\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1073s 0.1071s 9.3380 Ops/s 8.9220 Ops/s $\color{#35bf28}+4.66\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1490s 0.1471s 6.7983 Ops/s 6.5858 Ops/s $\color{#35bf28}+3.23\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3341s 0.3249s 3.0778 Ops/s 3.0152 Ops/s $\color{#35bf28}+2.08\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 76.6704ms 75.8874ms 13.1774 Ops/s 12.5276 Ops/s $\textbf{\color{#35bf28}+5.19\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1529s 0.1521s 6.5753 Ops/s 6.2492 Ops/s $\textbf{\color{#35bf28}+5.22\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 82.8277ms 81.1951ms 12.3160 Ops/s 12.5509 Ops/s $\color{#d91a1a}-1.87\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1642s 0.1641s 6.0944 Ops/s 6.2420 Ops/s $\color{#d91a1a}-2.36\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 31.3538ms 30.8975ms 32.3650 Ops/s 32.7337 Ops/s $\color{#d91a1a}-1.13\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 62.6598ms 62.4671ms 16.0084 Ops/s 16.2586 Ops/s $\color{#d91a1a}-1.54\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1194s 0.1190s 8.4025 Ops/s 8.4150 Ops/s $\color{#d91a1a}-0.15\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2673s 0.2667s 3.7492 Ops/s 3.7335 Ops/s $\color{#35bf28}+0.42\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 45.2211ms 44.7570ms 22.3429 Ops/s 23.0286 Ops/s $\color{#d91a1a}-2.98\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 90.1943ms 89.8699ms 11.1272 Ops/s 11.5365 Ops/s $\color{#d91a1a}-3.55\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 45.3489ms 45.1643ms 22.1414 Ops/s 22.8592 Ops/s $\color{#d91a1a}-3.14\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 92.4286ms 90.5350ms 11.0454 Ops/s 11.4631 Ops/s $\color{#d91a1a}-3.64\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 21.5082ms 21.2565ms 47.0444 Ops/s 48.2930 Ops/s $\color{#d91a1a}-2.59\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 43.3555ms 43.0608ms 23.2230 Ops/s 23.8009 Ops/s $\color{#d91a1a}-2.43\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1084s 0.1077s 9.2815 Ops/s 9.4242 Ops/s $\color{#d91a1a}-1.51\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2403s 0.2397s 4.1725 Ops/s 4.1328 Ops/s $\color{#35bf28}+0.96\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 39.3815ms 38.9230ms 25.6917 Ops/s 27.2547 Ops/s $\textbf{\color{#d91a1a}-5.73\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 78.9109ms 77.8889ms 12.8388 Ops/s 13.7030 Ops/s $\textbf{\color{#d91a1a}-6.31\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 39.6387ms 39.0489ms 25.6089 Ops/s 27.3339 Ops/s $\textbf{\color{#d91a1a}-6.31\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 78.6318ms 78.1040ms 12.8034 Ops/s 13.7019 Ops/s $\textbf{\color{#d91a1a}-6.56\%}$
test_plain_set_nested 33.3710μs 15.5853μs 64.1632 KOps/s 65.6746 KOps/s $\color{#d91a1a}-2.30\%$
test_plain_set_stack_nested 0.4515ms 15.3896μs 64.9791 KOps/s 66.1169 KOps/s $\color{#d91a1a}-1.72\%$
test_plain_set_nested_inplace 48.0400μs 16.7156μs 59.8245 KOps/s 59.2701 KOps/s $\color{#35bf28}+0.94\%$
test_plain_set_stack_nested_inplace 0.4447ms 16.6239μs 60.1543 KOps/s 60.5785 KOps/s $\color{#d91a1a}-0.70\%$
test_items 0.4166ms 6.0031μs 166.5802 KOps/s 165.9787 KOps/s $\color{#35bf28}+0.36\%$
test_items_nested 0.8907ms 0.4679ms 2.1370 KOps/s 2.1656 KOps/s $\color{#d91a1a}-1.32\%$
test_items_nested_locked 0.8938ms 0.4716ms 2.1204 KOps/s 2.1520 KOps/s $\color{#d91a1a}-1.47\%$
test_items_nested_leaf 0.5199ms 97.3980μs 10.2671 KOps/s 10.2820 KOps/s $\color{#d91a1a}-0.14\%$
test_items_stack_nested 0.8887ms 0.4652ms 2.1494 KOps/s 2.1623 KOps/s $\color{#d91a1a}-0.60\%$
test_items_stack_nested_leaf 0.5189ms 96.6808μs 10.3433 KOps/s 10.4696 KOps/s $\color{#d91a1a}-1.21\%$
test_items_stack_nested_locked 0.8978ms 0.4694ms 2.1304 KOps/s 2.1397 KOps/s $\color{#d91a1a}-0.44\%$
test_keys 28.8110μs 4.2256μs 236.6525 KOps/s 234.2355 KOps/s $\color{#35bf28}+1.03\%$
test_keys_nested 0.5582ms 0.1298ms 7.7066 KOps/s 7.7702 KOps/s $\color{#d91a1a}-0.82\%$
test_keys_nested_locked 2.0397ms 0.1402ms 7.1343 KOps/s 7.3273 KOps/s $\color{#d91a1a}-2.63\%$
test_keys_nested_leaf 0.5449ms 0.1218ms 8.2072 KOps/s 8.3618 KOps/s $\color{#d91a1a}-1.85\%$
test_keys_stack_nested 0.5576ms 0.1310ms 7.6361 KOps/s 7.7460 KOps/s $\color{#d91a1a}-1.42\%$
test_keys_stack_nested_leaf 0.5497ms 0.1201ms 8.3268 KOps/s 8.4158 KOps/s $\color{#d91a1a}-1.06\%$
test_keys_stack_nested_locked 0.5756ms 0.1392ms 7.1860 KOps/s 7.3778 KOps/s $\color{#d91a1a}-2.60\%$
test_values 85.4634μs 1.0165μs 983.7950 KOps/s 974.1393 KOps/s $\color{#35bf28}+0.99\%$
test_values_nested 91.1220μs 53.0684μs 18.8436 KOps/s 19.1662 KOps/s $\color{#d91a1a}-1.68\%$
test_values_nested_locked 0.4747ms 56.5110μs 17.6957 KOps/s 17.8935 KOps/s $\color{#d91a1a}-1.11\%$
test_values_nested_leaf 0.4774ms 60.9186μs 16.4153 KOps/s 16.7392 KOps/s $\color{#d91a1a}-1.93\%$
test_values_stack_nested 0.4768ms 52.9154μs 18.8981 KOps/s 19.2145 KOps/s $\color{#d91a1a}-1.65\%$
test_values_stack_nested_leaf 0.4826ms 60.2792μs 16.5895 KOps/s 16.7714 KOps/s $\color{#d91a1a}-1.08\%$
test_values_stack_nested_locked 0.4741ms 56.5821μs 17.6734 KOps/s 18.1387 KOps/s $\color{#d91a1a}-2.57\%$
test_membership 69.9177μs 0.8322μs 1.2016 MOps/s 1.2095 MOps/s $\color{#d91a1a}-0.65\%$
test_membership_nested 24.7800μs 2.9802μs 335.5521 KOps/s 338.1686 KOps/s $\color{#d91a1a}-0.77\%$
test_membership_nested_leaf 0.4268ms 3.0134μs 331.8461 KOps/s 337.1665 KOps/s $\color{#d91a1a}-1.58\%$
test_membership_stacked_nested 39.1000μs 2.8929μs 345.6752 KOps/s 345.0681 KOps/s $\color{#35bf28}+0.18\%$
test_membership_stacked_nested_leaf 0.4268ms 2.8611μs 349.5210 KOps/s 347.3998 KOps/s $\color{#35bf28}+0.61\%$
test_membership_nested_last 33.3800μs 4.4839μs 223.0193 KOps/s 226.6692 KOps/s $\color{#d91a1a}-1.61\%$
test_membership_nested_leaf_last 0.4363ms 4.4654μs 223.9452 KOps/s 226.5310 KOps/s $\color{#d91a1a}-1.14\%$
test_membership_stacked_nested_last 39.9710μs 4.3308μs 230.9031 KOps/s 230.7563 KOps/s $\color{#35bf28}+0.06\%$
test_membership_stacked_nested_leaf_last 0.4388ms 4.3366μs 230.5939 KOps/s 234.8478 KOps/s $\color{#d91a1a}-1.81\%$
test_nested_getleaf 56.5410μs 21.7453μs 45.9869 KOps/s 46.3602 KOps/s $\color{#d91a1a}-0.81\%$
test_nested_get 0.4379ms 20.5392μs 48.6873 KOps/s 49.3537 KOps/s $\color{#d91a1a}-1.35\%$
test_stacked_getleaf 0.4401ms 21.6639μs 46.1597 KOps/s 46.3281 KOps/s $\color{#d91a1a}-0.36\%$
test_stacked_get 0.4397ms 20.7629μs 48.1629 KOps/s 48.8460 KOps/s $\color{#d91a1a}-1.40\%$
test_nested_getitemleaf 0.4443ms 22.1238μs 45.2002 KOps/s 45.0955 KOps/s $\color{#35bf28}+0.23\%$
test_nested_getitem 45.4010μs 21.3730μs 46.7879 KOps/s 48.1039 KOps/s $\color{#d91a1a}-2.74\%$
test_stacked_getitemleaf 0.4519ms 22.2289μs 44.9866 KOps/s 45.5008 KOps/s $\color{#d91a1a}-1.13\%$
test_stacked_getitem 0.4430ms 21.1828μs 47.2080 KOps/s 48.4440 KOps/s $\color{#d91a1a}-2.55\%$
test_lock_nested 0.5929ms 0.5061ms 1.9758 KOps/s 1.9823 KOps/s $\color{#d91a1a}-0.32\%$
test_lock_stack_nested 0.6638ms 0.5157ms 1.9390 KOps/s 1.9652 KOps/s $\color{#d91a1a}-1.33\%$
test_unlock_nested 0.8273ms 0.4047ms 2.4708 KOps/s 2.5093 KOps/s $\color{#d91a1a}-1.53\%$
test_unlock_stack_nested 0.5493ms 0.4095ms 2.4420 KOps/s 2.4956 KOps/s $\color{#d91a1a}-2.15\%$
test_flatten_speed 0.5403ms 0.1226ms 8.1551 KOps/s 8.2158 KOps/s $\color{#d91a1a}-0.74\%$
test_unflatten_speed 0.7516ms 0.5588ms 1.7894 KOps/s 1.7524 KOps/s $\color{#35bf28}+2.12\%$
test_common_ops 0.1929s 1.8314ms 546.0302 Ops/s 420.9152 Ops/s $\textbf{\color{#35bf28}+29.72\%}$
test_creation 0.1302ms 3.1779μs 314.6713 KOps/s 319.5325 KOps/s $\color{#d91a1a}-1.52\%$
test_creation_empty 28.8000μs 7.0006μs 142.8457 KOps/s 142.9011 KOps/s $\color{#d91a1a}-0.04\%$
test_creation_nested_1 48.8610μs 11.4808μs 87.1020 KOps/s 86.7565 KOps/s $\color{#35bf28}+0.40\%$
test_creation_nested_2 36.0610μs 13.7839μs 72.5484 KOps/s 74.2809 KOps/s $\color{#d91a1a}-2.33\%$
test_creation_many_keys[10] 45.8310μs 20.9747μs 47.6765 KOps/s 47.4052 KOps/s $\color{#35bf28}+0.57\%$
test_creation_many_keys[50] 0.1212ms 90.8460μs 11.0076 KOps/s 11.0159 KOps/s $\color{#d91a1a}-0.08\%$
test_creation_many_keys[100] 0.2319ms 0.1780ms 5.6194 KOps/s 5.5810 KOps/s $\color{#35bf28}+0.69\%$
test_creation_nested_many_keys[10] 76.6320μs 44.8024μs 22.3202 KOps/s 21.9906 KOps/s $\color{#35bf28}+1.50\%$
test_creation_nested_many_keys[50] 0.2357ms 0.1851ms 5.4035 KOps/s 5.4107 KOps/s $\color{#d91a1a}-0.13\%$
test_clone 41.5610μs 13.6223μs 73.4093 KOps/s 74.5630 KOps/s $\color{#d91a1a}-1.55\%$
test_getitem[int] 1.5473ms 15.4762μs 64.6154 KOps/s 64.3041 KOps/s $\color{#35bf28}+0.48\%$
test_getitem[slice_int] 0.1411ms 25.2722μs 39.5691 KOps/s 38.7149 KOps/s $\color{#35bf28}+2.21\%$
test_getitem[range] 0.1751ms 65.4125μs 15.2876 KOps/s 14.9152 KOps/s $\color{#35bf28}+2.50\%$
test_getitem[tuple] 0.1482ms 24.9146μs 40.1372 KOps/s 41.3323 KOps/s $\color{#d91a1a}-2.89\%$
test_getitem[list] 0.1955ms 61.1017μs 16.3661 KOps/s 16.2788 KOps/s $\color{#35bf28}+0.54\%$
test_setitem_dim[int] 47.5010μs 26.2691μs 38.0676 KOps/s 35.8387 KOps/s $\textbf{\color{#35bf28}+6.22\%}$
test_setitem_dim[slice_int] 64.8710μs 43.7069μs 22.8797 KOps/s 21.6022 KOps/s $\textbf{\color{#35bf28}+5.91\%}$
test_setitem_dim[range] 0.1271ms 0.1015ms 9.8489 KOps/s 9.7572 KOps/s $\color{#35bf28}+0.94\%$
test_setitem_dim[tuple] 62.1710μs 40.7820μs 24.5206 KOps/s 24.1630 KOps/s $\color{#35bf28}+1.48\%$
test_setitem 53.1010μs 17.9489μs 55.7136 KOps/s 51.9726 KOps/s $\textbf{\color{#35bf28}+7.20\%}$
test_set 55.8610μs 17.8328μs 56.0765 KOps/s 54.7919 KOps/s $\color{#35bf28}+2.34\%$
test_set_shared 0.6794ms 0.2275ms 4.3963 KOps/s 4.5664 KOps/s $\color{#d91a1a}-3.73\%$
test_update 0.2584ms 22.3256μs 44.7916 KOps/s 43.1740 KOps/s $\color{#35bf28}+3.75\%$
test_update_nested 70.2410μs 33.8506μs 29.5416 KOps/s 28.2740 KOps/s $\color{#35bf28}+4.48\%$
test_update__nested 0.5030ms 32.9431μs 30.3554 KOps/s 28.6227 KOps/s $\textbf{\color{#35bf28}+6.05\%}$
test_set_nested 46.3210μs 18.8764μs 52.9761 KOps/s 49.3370 KOps/s $\textbf{\color{#35bf28}+7.38\%}$
test_set_nested_new 56.8110μs 23.8198μs 41.9819 KOps/s 39.7365 KOps/s $\textbf{\color{#35bf28}+5.65\%}$
test_select 77.8010μs 40.3634μs 24.7749 KOps/s 23.2025 KOps/s $\textbf{\color{#35bf28}+6.78\%}$
test_select_nested 0.1079ms 74.4474μs 13.4323 KOps/s 13.4107 KOps/s $\color{#35bf28}+0.16\%$
test_exclude_nested 0.1285ms 91.4682μs 10.9328 KOps/s 11.0117 KOps/s $\color{#d91a1a}-0.72\%$
test_empty[True] 0.4570ms 0.4016ms 2.4903 KOps/s 2.5356 KOps/s $\color{#d91a1a}-1.79\%$
test_empty[False] 8.4975μs 1.3857μs 721.6715 KOps/s 725.9281 KOps/s $\color{#d91a1a}-0.59\%$
test_to 0.1007ms 69.2125μs 14.4483 KOps/s 13.7494 KOps/s $\textbf{\color{#35bf28}+5.08\%}$
test_to_nonblocking 0.1083ms 68.7480μs 14.5459 KOps/s 13.7540 KOps/s $\textbf{\color{#35bf28}+5.76\%}$
test_unbind_speed 0.4004ms 0.3412ms 2.9305 KOps/s 2.8310 KOps/s $\color{#35bf28}+3.51\%$
test_unbind_speed_stack0 0.4292ms 0.3375ms 2.9627 KOps/s 2.8847 KOps/s $\color{#35bf28}+2.71\%$
test_unbind_speed_stack1 0.1147s 0.8636ms 1.1579 KOps/s 1.1304 KOps/s $\color{#35bf28}+2.43\%$
test_split 0.1144s 1.2992ms 769.6773 Ops/s 711.5891 Ops/s $\textbf{\color{#35bf28}+8.16\%}$
test_chunk 1.1492ms 1.1048ms 905.1541 Ops/s 902.8791 Ops/s $\color{#35bf28}+0.25\%$
test_to_cpu_blocking 0.1344s 22.6004ms 44.2470 Ops/s 49.5875 Ops/s $\textbf{\color{#d91a1a}-10.77\%}$
test_to_cpu_global_sync 12.0685ms 11.9886ms 83.4123 Ops/s 84.3003 Ops/s $\color{#d91a1a}-1.05\%$
test_to_cpu_event_sync 13.4424ms 13.0468ms 76.6469 Ops/s 77.5130 Ops/s $\color{#d91a1a}-1.12\%$
test_to_cpu_default 13.3746ms 13.0404ms 76.6850 Ops/s 69.1034 Ops/s $\textbf{\color{#35bf28}+10.97\%}$
test_consolidate[False-None] 4.2900ms 4.2358ms 236.0837 Ops/s 239.7545 Ops/s $\color{#d91a1a}-1.53\%$
test_consolidate[default-None] 3.5071ms 2.2032ms 453.8828 Ops/s 457.6095 Ops/s $\color{#d91a1a}-0.81\%$
test_consolidate[reduce-overhead-None] 3.5764ms 2.1351ms 468.3632 Ops/s 472.3649 Ops/s $\color{#d91a1a}-0.85\%$
test_consolidate_njt[False-None] 8.7891ms 8.5956ms 116.3387 Ops/s 115.2976 Ops/s $\color{#35bf28}+0.90\%$
test_to[False-False-None] 2.3358ms 2.2271ms 449.0063 Ops/s 455.0336 Ops/s $\color{#d91a1a}-1.32\%$
test_to[True-False-None] 2.2934ms 1.9875ms 503.1468 Ops/s 523.9081 Ops/s $\color{#d91a1a}-3.96\%$
test_to[within-False-None] 6.5302ms 6.3759ms 156.8396 Ops/s 160.1714 Ops/s $\color{#d91a1a}-2.08\%$
test_to[True-default-None] 11.1271ms 10.7375ms 93.1319 Ops/s 91.4251 Ops/s $\color{#35bf28}+1.87\%$
test_to_njt[False-False-None] 8.8184ms 8.7137ms 114.7619 Ops/s 112.7035 Ops/s $\color{#35bf28}+1.83\%$
test_to_njt[True-False-None] 7.1600ms 7.0042ms 142.7710 Ops/s 141.5974 Ops/s $\color{#35bf28}+0.83\%$
test_to_njt[within-False-None] 15.8793ms 15.7321ms 63.5644 Ops/s 63.0843 Ops/s $\color{#35bf28}+0.76\%$
test_creation[device0] 0.4722ms 0.1234ms 8.1007 KOps/s 7.2065 KOps/s $\textbf{\color{#35bf28}+12.41\%}$
test_creation_from_tensor 0.4795ms 0.1200ms 8.3341 KOps/s 7.4215 KOps/s $\textbf{\color{#35bf28}+12.30\%}$
test_add_one[memmap_tensor0] 0.3908ms 6.9458μs 143.9711 KOps/s 145.2078 KOps/s $\color{#d91a1a}-0.85\%$
test_contiguous[memmap_tensor0] 26.3200μs 0.6540μs 1.5291 MOps/s 2.2103 MOps/s $\textbf{\color{#d91a1a}-30.82\%}$
test_stack[memmap_tensor0] 40.4910μs 4.4903μs 222.7023 KOps/s 232.3177 KOps/s $\color{#d91a1a}-4.14\%$
test_memmaptd_index 1.1802ms 0.2909ms 3.4378 KOps/s 3.6438 KOps/s $\textbf{\color{#d91a1a}-5.65\%}$
test_memmaptd_index_astensor 0.5431ms 0.3968ms 2.5202 KOps/s 2.6412 KOps/s $\color{#d91a1a}-4.58\%$
test_memmaptd_index_op 0.8324ms 0.6624ms 1.5096 KOps/s 1.5384 KOps/s $\color{#d91a1a}-1.87\%$
test_serialize_model 0.1363s 0.1343s 7.4464 Ops/s 7.3291 Ops/s $\color{#35bf28}+1.60\%$
test_serialize_model_pickle 1.3637s 1.2122s 0.8250 Ops/s 0.8262 Ops/s $\color{#d91a1a}-0.15\%$
test_serialize_weights 0.1352s 0.1337s 7.4810 Ops/s 7.5233 Ops/s $\color{#d91a1a}-0.56\%$
test_serialize_weights_returnearly 66.7864ms 61.7004ms 16.2074 Ops/s 6.8437 Ops/s $\textbf{\color{#35bf28}+136.82\%}$
test_serialize_weights_pickle 1.3508s 1.2110s 0.8257 Ops/s 0.8244 Ops/s $\color{#35bf28}+0.17\%$
test_reshape_pytree 0.2148ms 33.4262μs 29.9167 KOps/s 30.7475 KOps/s $\color{#d91a1a}-2.70\%$
test_reshape_td 88.6320μs 48.2378μs 20.7306 KOps/s 22.6125 KOps/s $\textbf{\color{#d91a1a}-8.32\%}$
test_view_pytree 0.2174ms 32.9305μs 30.3670 KOps/s 31.0778 KOps/s $\color{#d91a1a}-2.29\%$
test_view_td 98.2810μs 57.4346μs 17.4111 KOps/s 18.7679 KOps/s $\textbf{\color{#d91a1a}-7.23\%}$
test_unbind_pytree 0.2417ms 38.6334μs 25.8843 KOps/s 27.1054 KOps/s $\color{#d91a1a}-4.50\%$
test_unbind_td 0.1574ms 53.2319μs 18.7857 KOps/s 20.0680 KOps/s $\textbf{\color{#d91a1a}-6.39\%}$
test_split_pytree 0.1989ms 45.5483μs 21.9547 KOps/s 23.8110 KOps/s $\textbf{\color{#d91a1a}-7.80\%}$
test_split_td 0.1297ms 69.8972μs 14.3067 KOps/s 15.3377 KOps/s $\textbf{\color{#d91a1a}-6.72\%}$
test_add_pytree 0.1953ms 43.1113μs 23.1958 KOps/s 23.6390 KOps/s $\color{#d91a1a}-1.88\%$
test_add_td 0.1020ms 59.2083μs 16.8895 KOps/s 17.6443 KOps/s $\color{#d91a1a}-4.28\%$
test_compile_add_one_nested[tensordict-compile] 0.3764ms 0.2310ms 4.3298 KOps/s 4.1673 KOps/s $\color{#35bf28}+3.90\%$
test_compile_add_one_nested[tensordict-eager] 0.3543ms 0.2011ms 4.9717 KOps/s 5.0084 KOps/s $\color{#d91a1a}-0.73\%$
test_compile_add_one_nested[pytree-compile] 0.3269ms 0.1344ms 7.4419 KOps/s 7.0241 KOps/s $\textbf{\color{#35bf28}+5.95\%}$
test_compile_add_one_nested[pytree-eager] 0.4378ms 0.1815ms 5.5101 KOps/s 5.5409 KOps/s $\color{#d91a1a}-0.56\%$
test_compile_copy_nested[tensordict-compile] 80.3010μs 16.5847μs 60.2967 KOps/s 65.2724 KOps/s $\textbf{\color{#d91a1a}-7.62\%}$
test_compile_copy_nested[tensordict-eager] 0.2816ms 53.0648μs 18.8449 KOps/s 18.5799 KOps/s $\color{#35bf28}+1.43\%$
test_compile_copy_nested[pytree-compile] 0.1537ms 15.9214μs 62.8086 KOps/s 62.7234 KOps/s $\color{#35bf28}+0.14\%$
test_compile_copy_nested[pytree-eager] 0.3362ms 67.8461μs 14.7392 KOps/s 14.6997 KOps/s $\color{#35bf28}+0.27\%$
test_compile_add_one_flat[tensordict-compile] 0.3292ms 0.2462ms 4.0613 KOps/s 3.7707 KOps/s $\textbf{\color{#35bf28}+7.71\%}$
test_compile_add_one_flat[tensordict-eager] 0.3941ms 0.2845ms 3.5149 KOps/s 3.5263 KOps/s $\color{#d91a1a}-0.32\%$
test_compile_add_one_flat[tensorclass-compile] 0.3260ms 0.1834ms 5.4528 KOps/s 5.0842 KOps/s $\textbf{\color{#35bf28}+7.25\%}$
test_compile_add_one_flat[tensorclass-eager] 0.1271ms 77.2763μs 12.9406 KOps/s 13.2036 KOps/s $\color{#d91a1a}-1.99\%$
test_compile_add_one_flat[pytree-compile] 0.4670ms 0.1945ms 5.1421 KOps/s 5.0642 KOps/s $\color{#35bf28}+1.54\%$
test_compile_add_one_flat[pytree-eager] 0.8482ms 0.5430ms 1.8417 KOps/s 1.9028 KOps/s $\color{#d91a1a}-3.21\%$
test_compile_add_self_flat[tensordict-eager] 0.4452ms 0.3296ms 3.0336 KOps/s 3.0072 KOps/s $\color{#35bf28}+0.88\%$
test_compile_add_self_flat[tensordict-compile] 0.6047ms 0.2497ms 4.0055 KOps/s 3.7937 KOps/s $\textbf{\color{#35bf28}+5.58\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1466ms 92.6540μs 10.7928 KOps/s 10.9636 KOps/s $\color{#d91a1a}-1.56\%$
test_compile_add_self_flat[tensorclass-compile] 0.3378ms 0.1925ms 5.1945 KOps/s 5.1131 KOps/s $\color{#35bf28}+1.59\%$
test_compile_add_self_flat[pytree-eager] 0.6882ms 0.4601ms 2.1736 KOps/s 2.2852 KOps/s $\color{#d91a1a}-4.89\%$
test_compile_add_self_flat[pytree-compile] 0.4404ms 0.2399ms 4.1682 KOps/s 5.0429 KOps/s $\textbf{\color{#d91a1a}-17.35\%}$
test_compile_copy_flat[tensordict-compile] 60.3410μs 19.7997μs 50.5058 KOps/s 49.7462 KOps/s $\color{#35bf28}+1.53\%$
test_compile_copy_flat[tensordict-eager] 75.1610μs 40.7457μs 24.5425 KOps/s 24.1280 KOps/s $\color{#35bf28}+1.72\%$
test_compile_copy_flat[pytree-compile] 0.2595ms 21.7914μs 45.8897 KOps/s 45.8615 KOps/s $\color{#35bf28}+0.06\%$
test_compile_copy_flat[pytree-eager] 0.3321ms 51.4993μs 19.4177 KOps/s 18.8789 KOps/s $\color{#35bf28}+2.85\%$
test_compile_assign_and_add[tensordict-compile] 2.2604ms 0.2373ms 4.2144 KOps/s 3.8634 KOps/s $\textbf{\color{#35bf28}+9.09\%}$
test_compile_assign_and_add[tensordict-eager] 3.5110ms 3.3748ms 296.3176 Ops/s 302.6311 Ops/s $\color{#d91a1a}-2.09\%$
test_compile_assign_and_add[pytree-compile] 2.1385ms 0.1788ms 5.5922 KOps/s 5.3994 KOps/s $\color{#35bf28}+3.57\%$
test_compile_assign_and_add[pytree-eager] 3.0168ms 2.8606ms 349.5810 Ops/s 352.2708 Ops/s $\color{#d91a1a}-0.76\%$
test_compile_indexing[tensor-tensordict-compile] 0.2180ms 0.1340ms 7.4636 KOps/s 6.8121 KOps/s $\textbf{\color{#35bf28}+9.56\%}$
test_compile_indexing[tensor-tensordict-eager] 0.3168ms 75.3541μs 13.2707 KOps/s 12.5442 KOps/s $\textbf{\color{#35bf28}+5.79\%}$
test_compile_indexing[tensor-tensorclass-compile] 0.2519ms 0.1178ms 8.4879 KOps/s 7.8897 KOps/s $\textbf{\color{#35bf28}+7.58\%}$
test_compile_indexing[tensor-tensorclass-eager] 0.2501ms 46.1358μs 21.6752 KOps/s 20.6527 KOps/s $\color{#35bf28}+4.95\%$
test_compile_indexing[tensor-pytree-compile] 0.1806ms 0.1194ms 8.3737 KOps/s 7.8473 KOps/s $\textbf{\color{#35bf28}+6.71\%}$
test_compile_indexing[tensor-pytree-eager] 0.2537ms 46.2236μs 21.6340 KOps/s 20.4392 KOps/s $\textbf{\color{#35bf28}+5.85\%}$
test_compile_indexing[slice-tensordict-compile] 0.2356ms 83.2913μs 12.0061 KOps/s 11.2592 KOps/s $\textbf{\color{#35bf28}+6.63\%}$
test_compile_indexing[slice-tensordict-eager] 0.2046ms 28.3836μs 35.2316 KOps/s 34.8336 KOps/s $\color{#35bf28}+1.14\%$
test_compile_indexing[slice-tensorclass-compile] 0.1568ms 70.4191μs 14.2007 KOps/s 13.5390 KOps/s $\color{#35bf28}+4.89\%$
test_compile_indexing[slice-tensorclass-eager] 0.2338ms 22.8198μs 43.8217 KOps/s 45.0965 KOps/s $\color{#d91a1a}-2.83\%$
test_compile_indexing[slice-pytree-compile] 0.1237ms 71.0762μs 14.0694 KOps/s 13.6162 KOps/s $\color{#35bf28}+3.33\%$
test_compile_indexing[slice-pytree-eager] 0.2269ms 22.8314μs 43.7993 KOps/s 44.6202 KOps/s $\color{#d91a1a}-1.84\%$
test_compile_indexing[int-tensordict-compile] 0.1354ms 83.5788μs 11.9648 KOps/s 11.2336 KOps/s $\textbf{\color{#35bf28}+6.51\%}$
test_compile_indexing[int-tensordict-eager] 0.2239ms 28.0965μs 35.5917 KOps/s 34.4465 KOps/s $\color{#35bf28}+3.32\%$
test_compile_indexing[int-tensorclass-compile] 0.1134ms 71.1756μs 14.0498 KOps/s 13.2314 KOps/s $\textbf{\color{#35bf28}+6.18\%}$
test_compile_indexing[int-tensorclass-eager] 0.2481ms 22.7093μs 44.0349 KOps/s 45.1853 KOps/s $\color{#d91a1a}-2.55\%$
test_compile_indexing[int-pytree-compile] 0.1201ms 70.8558μs 14.1132 KOps/s 13.1543 KOps/s $\textbf{\color{#35bf28}+7.29\%}$
test_compile_indexing[int-pytree-eager] 0.2547ms 22.6020μs 44.2439 KOps/s 45.4586 KOps/s $\color{#d91a1a}-2.67\%$
test_compile_replace[single-eager] 85.5810μs 48.1309μs 20.7767 KOps/s 19.6841 KOps/s $\textbf{\color{#35bf28}+5.55\%}$
test_compile_replace[single-compile] 0.2360ms 0.1329ms 7.5257 KOps/s 7.5254 KOps/s $+0.00\%$
test_compile_replace[multi-eager] 0.6609ms 0.5736ms 1.7435 KOps/s 1.6701 KOps/s $\color{#35bf28}+4.39\%$
test_compile_replace[multi-compile] 0.3185ms 0.1413ms 7.0790 KOps/s 7.1644 KOps/s $\color{#d91a1a}-1.19\%$
test_compile_tc_getattr_20[eager] 0.2295ms 0.1757ms 5.6904 KOps/s 6.0279 KOps/s $\textbf{\color{#d91a1a}-5.60\%}$
test_compile_tc_getattr_20[compile] 0.4642ms 0.1470ms 6.8007 KOps/s 6.5199 KOps/s $\color{#35bf28}+4.31\%$
test_compile_clone_shallow[20-eager] 64.7810μs 19.4029μs 51.5387 KOps/s 52.3136 KOps/s $\color{#d91a1a}-1.48\%$
test_compile_clone_shallow[20-compile] 82.9010μs 17.2259μs 58.0522 KOps/s 56.3080 KOps/s $\color{#35bf28}+3.10\%$
test_compile_clone_shallow[40-eager] 66.2210μs 33.5283μs 29.8255 KOps/s 29.7431 KOps/s $\color{#35bf28}+0.28\%$
test_compile_clone_shallow[40-compile] 76.8710μs 18.4662μs 54.1530 KOps/s 53.3192 KOps/s $\color{#35bf28}+1.56\%$
test_compile_clone_shallow[80-eager] 99.4820μs 62.3708μs 16.0331 KOps/s 15.8521 KOps/s $\color{#35bf28}+1.14\%$
test_compile_clone_shallow[80-compile] 60.4210μs 22.4503μs 44.5428 KOps/s 44.5445 KOps/s $-0.00\%$
test_compile_update_inplace[eager] 0.1121ms 60.1062μs 16.6372 KOps/s 16.3633 KOps/s $\color{#35bf28}+1.67\%$
test_compile_update_inplace[compile] 0.3888ms 0.1439ms 6.9498 KOps/s 6.4616 KOps/s $\textbf{\color{#35bf28}+7.56\%}$
test_mod_add[eager] 0.2778ms 0.2283ms 4.3807 KOps/s 4.1664 KOps/s $\textbf{\color{#35bf28}+5.14\%}$
test_mod_add[compile] 0.2976ms 0.1368ms 7.3117 KOps/s 7.4399 KOps/s $\color{#d91a1a}-1.72\%$
test_mod_add[compile-overhead] 0.3962ms 0.1863ms 5.3672 KOps/s 5.1327 KOps/s $\color{#35bf28}+4.57\%$
test_mod_wrap[eager] 0.5865ms 0.4748ms 2.1061 KOps/s 2.1105 KOps/s $\color{#d91a1a}-0.21\%$
test_mod_wrap[compile] 0.4715ms 0.4072ms 2.4557 KOps/s 2.4687 KOps/s $\color{#d91a1a}-0.53\%$
test_mod_wrap[compile-overhead] 8.6296ms 4.7057ms 212.5063 Ops/s 214.2524 Ops/s $\color{#d91a1a}-0.81\%$
test_mod_wrap_and_backward[eager] 2.1717ms 1.7196ms 581.5204 Ops/s 582.2735 Ops/s $\color{#d91a1a}-0.13\%$
test_mod_wrap_and_backward[compile] 1.7057ms 1.5448ms 647.3242 Ops/s 646.0749 Ops/s $\color{#35bf28}+0.19\%$
test_mod_wrap_and_backward[compile-overhead] 1.3511ms 0.9502ms 1.0524 KOps/s 1.0290 KOps/s $\color{#35bf28}+2.27\%$
test_seq_add[eager] 0.5674ms 0.5153ms 1.9407 KOps/s 1.9192 KOps/s $\color{#35bf28}+1.12\%$
test_seq_add[compile] 0.2541ms 0.1397ms 7.1578 KOps/s 6.4612 KOps/s $\textbf{\color{#35bf28}+10.78\%}$
test_seq_add[compile-overhead] 0.2747ms 0.1826ms 5.4755 KOps/s 5.2616 KOps/s $\color{#35bf28}+4.07\%$
test_seq_wrap[eager] 1.5005ms 1.4274ms 700.5772 Ops/s 702.8244 Ops/s $\color{#d91a1a}-0.32\%$
test_seq_wrap[compile] 0.5021ms 0.4253ms 2.3514 KOps/s 2.3544 KOps/s $\color{#d91a1a}-0.13\%$
test_seq_wrap[compile-overhead] 0.3814ms 0.3040ms 3.2894 KOps/s 3.1689 KOps/s $\color{#35bf28}+3.80\%$
test_func_call_runtime[False-eager] 0.8614ms 0.8104ms 1.2339 KOps/s 1.2318 KOps/s $\color{#35bf28}+0.17\%$
test_func_call_runtime[False-compile] 1.1497ms 0.9991ms 1.0009 KOps/s 1.0050 KOps/s $\color{#d91a1a}-0.41\%$
test_func_call_runtime[False-compile-overhead] 0.6658ms 0.5166ms 1.9357 KOps/s 1.9382 KOps/s $\color{#d91a1a}-0.13\%$
test_func_call_runtime[True-eager] 1.1670ms 1.0574ms 945.6718 Ops/s 936.2798 Ops/s $\color{#35bf28}+1.00\%$
test_func_call_runtime[True-compile] 1.0602ms 1.0101ms 990.0074 Ops/s 992.1833 Ops/s $\color{#d91a1a}-0.22\%$
test_func_call_runtime[True-compile-overhead] 0.6320ms 0.5269ms 1.8979 KOps/s 1.8698 KOps/s $\color{#35bf28}+1.50\%$
test_func_call_cm_runtime[False-eager] 0.8701ms 0.8099ms 1.2347 KOps/s 1.2229 KOps/s $\color{#35bf28}+0.96\%$
test_func_call_cm_runtime[False-compile] 1.3761ms 0.9891ms 1.0111 KOps/s 1.0224 KOps/s $\color{#d91a1a}-1.11\%$
test_func_call_cm_runtime[False-compile-overhead] 0.7776ms 0.5103ms 1.9598 KOps/s 2.0078 KOps/s $\color{#d91a1a}-2.39\%$
test_func_call_cm_runtime[True-eager] 1.2884ms 1.1987ms 834.2582 Ops/s 824.0958 Ops/s $\color{#35bf28}+1.23\%$
test_func_call_cm_runtime[True-compile] 1.1331ms 1.0343ms 966.8118 Ops/s 967.2712 Ops/s $\color{#d91a1a}-0.05\%$
test_func_call_cm_runtime[True-compile-overhead] 0.7146ms 0.5518ms 1.8122 KOps/s 1.7932 KOps/s $\color{#35bf28}+1.06\%$
test_vmap_func_call_cm_runtime[eager] 2.8704ms 2.3927ms 417.9401 Ops/s 416.9173 Ops/s $\color{#35bf28}+0.25\%$
test_vmap_func_call_cm_runtime[compile] 1.1068ms 1.0433ms 958.5271 Ops/s 965.9941 Ops/s $\color{#d91a1a}-0.77\%$
test_vmap_func_call_cm_runtime[compile-overhead] 0.6028ms 0.5392ms 1.8546 KOps/s 1.8347 KOps/s $\color{#35bf28}+1.09\%$
test_distributed 2.4737ms 0.1595ms 6.2712 KOps/s 6.6850 KOps/s $\textbf{\color{#d91a1a}-6.19\%}$
test_tdmodule 0.6594ms 0.1926ms 5.1929 KOps/s 4.9287 KOps/s $\textbf{\color{#35bf28}+5.36\%}$
test_tdmodule_dispatch 0.4073ms 0.3725ms 2.6847 KOps/s 2.5997 KOps/s $\color{#35bf28}+3.27\%$
test_tdseq 45.0100μs 26.4640μs 37.7872 KOps/s 34.9596 KOps/s $\textbf{\color{#35bf28}+8.09\%}$
test_tdseq_dispatch 0.2681ms 0.2166ms 4.6172 KOps/s 4.4289 KOps/s $\color{#35bf28}+4.25\%$
test_instantiation_functorch 2.3262ms 2.0912ms 478.2054 Ops/s 484.7598 Ops/s $\color{#d91a1a}-1.35\%$
test_exec_functorch 0.2601ms 0.1852ms 5.3981 KOps/s 5.2478 KOps/s $\color{#35bf28}+2.86\%$
test_exec_functional_call 0.2315ms 0.1633ms 6.1233 KOps/s 5.8670 KOps/s $\color{#35bf28}+4.37\%$
test_exec_td_decorator 0.4687ms 0.2416ms 4.1382 KOps/s 4.0484 KOps/s $\color{#35bf28}+2.22\%$
test_vmap_mlp_speed_decorator[True-True] 1.2640ms 1.0303ms 970.5613 Ops/s 959.3410 Ops/s $\color{#35bf28}+1.17\%$
test_vmap_mlp_speed_decorator[True-False] 1.2211ms 1.0234ms 977.1189 Ops/s 974.9508 Ops/s $\color{#35bf28}+0.22\%$
test_vmap_mlp_speed_decorator[False-True] 0.8938ms 0.7265ms 1.3765 KOps/s 1.3777 KOps/s $\color{#d91a1a}-0.08\%$
test_vmap_mlp_speed_decorator[False-False] 0.9197ms 0.7305ms 1.3690 KOps/s 1.3788 KOps/s $\color{#d91a1a}-0.71\%$
test_vmap_transformer_speed_decorator[True-True] 22.0415ms 21.2513ms 47.0560 Ops/s 47.5115 Ops/s $\color{#d91a1a}-0.96\%$
test_vmap_transformer_speed_decorator[True-False] 23.9100ms 21.7259ms 46.0281 Ops/s 47.4484 Ops/s $\color{#d91a1a}-2.99\%$
test_vmap_transformer_speed_decorator[False-True] 21.7971ms 21.0688ms 47.4635 Ops/s 48.3483 Ops/s $\color{#d91a1a}-1.83\%$
test_vmap_transformer_speed_decorator[False-False] 21.8205ms 21.3126ms 46.9205 Ops/s 48.2281 Ops/s $\color{#d91a1a}-2.71\%$
test_to_module_speed[True] 1.5680ms 1.4822ms 674.6811 Ops/s 661.9823 Ops/s $\color{#35bf28}+1.92\%$
test_to_module_speed[False] 1.5639ms 1.4680ms 681.1989 Ops/s 678.0294 Ops/s $\color{#35bf28}+0.47\%$
test_tc_init 67.9810μs 43.6233μs 22.9235 KOps/s 23.1685 KOps/s $\color{#d91a1a}-1.06\%$
test_tc_init_tensor_only 37.0600μs 8.9795μs 111.3651 KOps/s 111.5046 KOps/s $\color{#d91a1a}-0.13\%$
test_tc_init_nested 0.1239ms 86.9239μs 11.5043 KOps/s 11.8896 KOps/s $\color{#d91a1a}-3.24\%$
test_tc_init_many_fields 51.9610μs 14.7919μs 67.6044 KOps/s 67.4381 KOps/s $\color{#35bf28}+0.25\%$
test_tc_first_layer_tensor 30.9200μs 1.8157μs 550.7449 KOps/s 545.2353 KOps/s $\color{#35bf28}+1.01\%$
test_tc_first_layer_tensor_only 2.9133μs 0.4007μs 2.4956 MOps/s 2.5086 MOps/s $\color{#d91a1a}-0.52\%$
test_tc_first_layer_tensor_set 28.2410μs 3.8945μs 256.7746 KOps/s 254.8258 KOps/s $\color{#35bf28}+0.76\%$
test_tc_first_layer_tensor_only_set 19.4000μs 3.3014μs 302.8973 KOps/s 303.8238 KOps/s $\color{#d91a1a}-0.30\%$
test_tc_first_layer_nontensor 37.7310μs 6.1606μs 162.3227 KOps/s 155.0210 KOps/s $\color{#35bf28}+4.71\%$
test_tc_second_layer_tensor 31.5100μs 4.4254μs 225.9669 KOps/s 226.6627 KOps/s $\color{#d91a1a}-0.31\%$
test_tc_second_layer_nontensor 42.3610μs 8.7908μs 113.7559 KOps/s 111.9844 KOps/s $\color{#35bf28}+1.58\%$
test_unbind 0.2834s 14.1222ms 70.8105 Ops/s 70.9109 Ops/s $\color{#d91a1a}-0.14\%$
test_full_like 4.4961ms 4.3824ms 228.1841 Ops/s 227.7583 Ops/s $\color{#35bf28}+0.19\%$
test_zeros_like 4.5247ms 4.3717ms 228.7436 Ops/s 228.7537 Ops/s $-0.00\%$
test_ones_like 4.5099ms 4.3788ms 228.3748 Ops/s 228.3694 Ops/s $+0.00\%$
test_clone 6.8162ms 6.4247ms 155.6490 Ops/s 155.4545 Ops/s $\color{#35bf28}+0.13\%$
test_squeeze 70.3910μs 14.7362μs 67.8603 KOps/s 70.1152 KOps/s $\color{#d91a1a}-3.22\%$
test_unsqueeze 0.1809ms 0.1131ms 8.8449 KOps/s 8.9547 KOps/s $\color{#d91a1a}-1.23\%$
test_split 0.2612ms 0.1829ms 5.4673 KOps/s 5.5328 KOps/s $\color{#d91a1a}-1.18\%$
test_permute 0.2660ms 0.2083ms 4.8001 KOps/s 4.8560 KOps/s $\color{#d91a1a}-1.15\%$
test_stack 35.6055ms 34.9845ms 28.5841 Ops/s 19.6550 Ops/s $\textbf{\color{#35bf28}+45.43\%}$
test_cat 43.6066ms 42.8148ms 23.3564 Ops/s 23.1847 Ops/s $\color{#35bf28}+0.74\%$
test_sequential_tensordict 0.6273ms 0.5759ms 1.7364 KOps/s 1.7253 KOps/s $\color{#35bf28}+0.64\%$
test_sequential_graph_module 0.5357ms 0.1210ms 8.2677 KOps/s 8.2401 KOps/s $\color{#35bf28}+0.34\%$
test_nested_tensordict 0.9800ms 0.8139ms 1.2286 KOps/s 1.2025 KOps/s $\color{#35bf28}+2.17\%$
test_nested_graph_module 0.5240ms 0.1286ms 7.7780 KOps/s 7.6915 KOps/s $\color{#35bf28}+1.12\%$

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