Skip to content

[Linter] Fix linter - #1712

Merged
vmoens merged 1 commit into
mainfrom
fix-linter
May 21, 2026
Merged

[Linter] Fix linter#1712
vmoens merged 1 commit into
mainfrom
fix-linter

Conversation

@vmoens

@vmoens vmoens commented May 21, 2026

Copy link
Copy Markdown
Collaborator

No description provided.

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

github-actions Bot commented May 21, 2026

Copy link
Copy Markdown
Contributor

PR Title Label Error

Unknown or invalid prefix [Linter].

Current title: [Linter] Fix linter

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.

@vmoens
vmoens merged commit 8d1b400 into main May 21, 2026
28 of 45 checks passed
@github-actions

github-actions Bot commented May 21, 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}34$. Worsened: $\large\color{#d91a1a}6$.

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_collector_preallocated_write[stack_out-cpu-32-2] 0.2028ms 0.1657ms 6.0364 KOps/s 5.7312 KOps/s $\textbf{\color{#35bf28}+5.32\%}$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.3497ms 0.2945ms 3.3957 KOps/s 3.3330 KOps/s $\color{#35bf28}+1.88\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.8741ms 0.7761ms 1.2886 KOps/s 1.2702 KOps/s $\color{#35bf28}+1.45\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.3798ms 0.3380ms 2.9584 KOps/s 2.8916 KOps/s $\color{#35bf28}+2.31\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.2930ms 1.0915ms 916.1439 Ops/s 903.0696 Ops/s $\color{#35bf28}+1.45\%$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.9106ms 7.4164ms 134.8361 Ops/s 133.3441 Ops/s $\color{#35bf28}+1.12\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.2923ms 0.2449ms 4.0840 KOps/s 3.9698 KOps/s $\color{#35bf28}+2.88\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.4347ms 0.3673ms 2.7227 KOps/s 2.6656 KOps/s $\color{#35bf28}+2.14\%$
test_collector_preallocated_write[stack_out-cuda-32-32] 0.9312ms 0.8183ms 1.2220 KOps/s 1.2079 KOps/s $\color{#35bf28}+1.16\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.3124ms 0.2528ms 3.9557 KOps/s 3.7824 KOps/s $\color{#35bf28}+4.58\%$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.4216ms 0.3679ms 2.7184 KOps/s 2.6270 KOps/s $\color{#35bf28}+3.48\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 0.9588ms 0.8826ms 1.1330 KOps/s 1.1262 KOps/s $\color{#35bf28}+0.60\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2602ms 0.2131ms 4.6936 KOps/s 4.5004 KOps/s $\color{#35bf28}+4.29\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 1.1929ms 0.8257ms 1.2112 KOps/s 1.1837 KOps/s $\color{#35bf28}+2.32\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.4961ms 3.2840ms 304.5102 Ops/s 295.5339 Ops/s $\color{#35bf28}+3.04\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.4590ms 0.3720ms 2.6883 KOps/s 2.6230 KOps/s $\color{#35bf28}+2.49\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 2.5116ms 1.7197ms 581.4827 Ops/s 578.8899 Ops/s $\color{#35bf28}+0.45\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 10.9056ms 10.4391ms 95.7937 Ops/s 95.5608 Ops/s $\color{#35bf28}+0.24\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.4537ms 0.4048ms 2.4703 KOps/s 2.5224 KOps/s $\color{#d91a1a}-2.06\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.7537ms 1.4942ms 669.2500 Ops/s 648.1996 Ops/s $\color{#35bf28}+3.25\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 6.0975ms 5.9636ms 167.6833 Ops/s 163.6847 Ops/s $\color{#35bf28}+2.44\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.4606ms 0.3909ms 2.5582 KOps/s 2.3881 KOps/s $\textbf{\color{#35bf28}+7.12\%}$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.5806ms 1.4707ms 679.9317 Ops/s 645.8044 Ops/s $\textbf{\color{#35bf28}+5.28\%}$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 6.0707ms 5.7661ms 173.4282 Ops/s 162.8539 Ops/s $\textbf{\color{#35bf28}+6.49\%}$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.2433ms 0.1734ms 5.7657 KOps/s 5.6149 KOps/s $\color{#35bf28}+2.69\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.7766ms 0.6770ms 1.4770 KOps/s 1.4482 KOps/s $\color{#35bf28}+1.99\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 3.2726ms 2.7002ms 370.3408 Ops/s 360.8276 Ops/s $\color{#35bf28}+2.64\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.5260ms 0.3487ms 2.8677 KOps/s 3.0607 KOps/s $\textbf{\color{#d91a1a}-6.30\%}$
test_collector_preallocated_write[copy_at-cpu-4096-8] 2.3117ms 1.5474ms 646.2476 Ops/s 667.9026 Ops/s $\color{#d91a1a}-3.24\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 9.8713ms 9.6581ms 103.5402 Ops/s 102.7398 Ops/s $\color{#35bf28}+0.78\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.3942ms 0.3203ms 3.1218 KOps/s 3.0548 KOps/s $\color{#35bf28}+2.19\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.3669ms 1.1960ms 836.1275 Ops/s 812.9463 Ops/s $\color{#35bf28}+2.85\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.7825ms 4.6868ms 213.3655 Ops/s 208.2491 Ops/s $\color{#35bf28}+2.46\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.4200ms 0.3223ms 3.1030 KOps/s 3.0310 KOps/s $\color{#35bf28}+2.38\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.3131ms 1.1998ms 833.4456 Ops/s 813.4395 Ops/s $\color{#35bf28}+2.46\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 4.9066ms 4.6960ms 212.9462 Ops/s 206.5705 Ops/s $\color{#35bf28}+3.09\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.1715ms 0.1208ms 8.2805 KOps/s 8.0108 KOps/s $\color{#35bf28}+3.37\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.5752ms 0.4786ms 2.0894 KOps/s 2.0549 KOps/s $\color{#35bf28}+1.68\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 1.9915ms 1.9040ms 525.1973 Ops/s 516.9477 Ops/s $\color{#35bf28}+1.60\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.3962ms 0.2605ms 3.8384 KOps/s 3.8834 KOps/s $\color{#d91a1a}-1.16\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.6330ms 1.4060ms 711.2180 Ops/s 804.9211 Ops/s $\textbf{\color{#d91a1a}-11.64\%}$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 8.9227ms 8.7141ms 114.7559 Ops/s 114.2151 Ops/s $\color{#35bf28}+0.47\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.3779ms 0.2718ms 3.6785 KOps/s 3.5559 KOps/s $\color{#35bf28}+3.45\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.1596ms 1.0080ms 992.0458 Ops/s 955.8648 Ops/s $\color{#35bf28}+3.79\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 4.1947ms 3.9482ms 253.2821 Ops/s 243.5446 Ops/s $\color{#35bf28}+4.00\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.3794ms 0.2724ms 3.6705 KOps/s 3.5491 KOps/s $\color{#35bf28}+3.42\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.1458ms 1.0094ms 990.6996 Ops/s 956.3877 Ops/s $\color{#35bf28}+3.59\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 4.1457ms 3.9863ms 250.8587 Ops/s 242.0213 Ops/s $\color{#35bf28}+3.65\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.6664ms 16.3094ms 61.3144 Ops/s 60.6897 Ops/s $\color{#35bf28}+1.03\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 33.0389ms 32.8152ms 30.4737 Ops/s 30.3919 Ops/s $\color{#35bf28}+0.27\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 87.5786ms 86.7705ms 11.5247 Ops/s 11.5806 Ops/s $\color{#d91a1a}-0.48\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.2043s 0.2036s 4.9114 Ops/s 4.9825 Ops/s $\color{#d91a1a}-1.43\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 13.9770ms 13.8799ms 72.0464 Ops/s 72.4708 Ops/s $\color{#d91a1a}-0.59\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 27.6430ms 27.3205ms 36.6026 Ops/s 37.0639 Ops/s $\color{#d91a1a}-1.24\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 14.0874ms 13.9415ms 71.7285 Ops/s 71.6519 Ops/s $\color{#35bf28}+0.11\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 27.6917ms 27.5067ms 36.3548 Ops/s 36.6588 Ops/s $\color{#d91a1a}-0.83\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 51.9498ms 51.7765ms 19.3138 Ops/s 19.0387 Ops/s $\color{#35bf28}+1.44\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1046s 0.1042s 9.5982 Ops/s 9.4555 Ops/s $\color{#35bf28}+1.51\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1474s 0.1465s 6.8266 Ops/s 6.8354 Ops/s $\color{#d91a1a}-0.13\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3233s 0.3225s 3.1004 Ops/s 3.1016 Ops/s $\color{#d91a1a}-0.04\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 78.3351ms 76.3777ms 13.0928 Ops/s 13.4158 Ops/s $\color{#d91a1a}-2.41\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1541s 0.1511s 6.6202 Ops/s 6.4121 Ops/s $\color{#35bf28}+3.25\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 77.7316ms 75.4867ms 13.2474 Ops/s 12.7287 Ops/s $\color{#35bf28}+4.08\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1567s 0.1544s 6.4782 Ops/s 6.6783 Ops/s $\color{#d91a1a}-3.00\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 30.5592ms 30.3805ms 32.9159 Ops/s 32.4522 Ops/s $\color{#35bf28}+1.43\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 61.8087ms 61.5108ms 16.2573 Ops/s 16.0170 Ops/s $\color{#35bf28}+1.50\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1223s 0.1211s 8.2569 Ops/s 8.3874 Ops/s $\color{#d91a1a}-1.56\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2778s 0.2760s 3.6226 Ops/s 3.7207 Ops/s $\color{#d91a1a}-2.64\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 43.2631ms 42.7440ms 23.3951 Ops/s 22.9728 Ops/s $\color{#35bf28}+1.84\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 86.2791ms 85.7836ms 11.6572 Ops/s 11.5006 Ops/s $\color{#35bf28}+1.36\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 43.4148ms 43.0433ms 23.2324 Ops/s 22.8296 Ops/s $\color{#35bf28}+1.76\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 86.5665ms 86.2207ms 11.5981 Ops/s 11.3690 Ops/s $\color{#35bf28}+2.02\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 20.8492ms 20.4795ms 48.8293 Ops/s 47.3464 Ops/s $\color{#35bf28}+3.13\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 42.3799ms 42.0533ms 23.7794 Ops/s 23.1564 Ops/s $\color{#35bf28}+2.69\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1111s 0.1095s 9.1320 Ops/s 9.2162 Ops/s $\color{#d91a1a}-0.91\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2530s 0.2524s 3.9615 Ops/s 4.0431 Ops/s $\color{#d91a1a}-2.02\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 36.7807ms 36.0738ms 27.7209 Ops/s 26.7132 Ops/s $\color{#35bf28}+3.77\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 73.2819ms 72.5553ms 13.7826 Ops/s 13.4000 Ops/s $\color{#35bf28}+2.86\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 36.6880ms 36.2011ms 27.6234 Ops/s 26.7097 Ops/s $\color{#35bf28}+3.42\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 73.6780ms 71.8224ms 13.9232 Ops/s 13.3154 Ops/s $\color{#35bf28}+4.56\%$
test_plain_set_nested 0.5418ms 15.6491μs 63.9016 KOps/s 63.7708 KOps/s $\color{#35bf28}+0.21\%$
test_plain_set_stack_nested 41.3000μs 15.3302μs 65.2308 KOps/s 65.7422 KOps/s $\color{#d91a1a}-0.78\%$
test_plain_set_nested_inplace 40.3600μs 16.9982μs 58.8299 KOps/s 59.2186 KOps/s $\color{#d91a1a}-0.66\%$
test_plain_set_stack_nested_inplace 84.1420μs 16.6965μs 59.8929 KOps/s 59.8192 KOps/s $\color{#35bf28}+0.12\%$
test_items 31.8710μs 6.0344μs 165.7166 KOps/s 163.9691 KOps/s $\color{#35bf28}+1.07\%$
test_items_nested 0.5478ms 0.4678ms 2.1377 KOps/s 2.1263 KOps/s $\color{#35bf28}+0.53\%$
test_items_nested_locked 0.5413ms 0.4676ms 2.1386 KOps/s 2.1117 KOps/s $\color{#35bf28}+1.27\%$
test_items_nested_leaf 0.1365ms 97.3922μs 10.2678 KOps/s 10.2087 KOps/s $\color{#35bf28}+0.58\%$
test_items_stack_nested 0.5434ms 0.4661ms 2.1456 KOps/s 2.1278 KOps/s $\color{#35bf28}+0.83\%$
test_items_stack_nested_leaf 0.1341ms 98.0100μs 10.2030 KOps/s 10.2624 KOps/s $\color{#d91a1a}-0.58\%$
test_items_stack_nested_locked 0.5227ms 0.4679ms 2.1373 KOps/s 2.1103 KOps/s $\color{#35bf28}+1.28\%$
test_keys 30.7710μs 4.2234μs 236.7776 KOps/s 225.1145 KOps/s $\textbf{\color{#35bf28}+5.18\%}$
test_keys_nested 0.2025ms 0.1303ms 7.6769 KOps/s 7.5758 KOps/s $\color{#35bf28}+1.33\%$
test_keys_nested_locked 0.7370ms 0.1399ms 7.1475 KOps/s 7.1267 KOps/s $\color{#35bf28}+0.29\%$
test_keys_nested_leaf 0.1569ms 0.1215ms 8.2329 KOps/s 8.1518 KOps/s $\color{#35bf28}+1.00\%$
test_keys_stack_nested 0.1740ms 0.1307ms 7.6531 KOps/s 7.6400 KOps/s $\color{#35bf28}+0.17\%$
test_keys_stack_nested_leaf 0.1618ms 0.1213ms 8.2413 KOps/s 8.2423 KOps/s $\color{#d91a1a}-0.01\%$
test_keys_stack_nested_locked 0.2140ms 0.1393ms 7.1798 KOps/s 7.1440 KOps/s $\color{#35bf28}+0.50\%$
test_values 6.9180μs 1.0237μs 976.8824 KOps/s 988.5056 KOps/s $\color{#d91a1a}-1.18\%$
test_values_nested 85.9910μs 52.9123μs 18.8992 KOps/s 18.6553 KOps/s $\color{#35bf28}+1.31\%$
test_values_nested_locked 83.7220μs 55.9363μs 17.8775 KOps/s 17.5506 KOps/s $\color{#35bf28}+1.86\%$
test_values_nested_leaf 0.1351ms 60.6013μs 16.5013 KOps/s 16.4045 KOps/s $\color{#35bf28}+0.59\%$
test_values_stack_nested 80.7420μs 52.6676μs 18.9870 KOps/s 18.6737 KOps/s $\color{#35bf28}+1.68\%$
test_values_stack_nested_leaf 0.1058ms 60.0357μs 16.6568 KOps/s 16.3197 KOps/s $\color{#35bf28}+2.07\%$
test_values_stack_nested_locked 88.4820μs 56.1436μs 17.8115 KOps/s 17.5959 KOps/s $\color{#35bf28}+1.23\%$
test_membership 5.2583μs 0.8420μs 1.1877 MOps/s 1.1887 MOps/s $\color{#d91a1a}-0.09\%$
test_membership_nested 75.2110μs 3.0042μs 332.8725 KOps/s 333.6424 KOps/s $\color{#d91a1a}-0.23\%$
test_membership_nested_leaf 25.5000μs 3.0342μs 329.5717 KOps/s 331.0379 KOps/s $\color{#d91a1a}-0.44\%$
test_membership_stacked_nested 30.3910μs 2.8799μs 347.2352 KOps/s 342.6170 KOps/s $\color{#35bf28}+1.35\%$
test_membership_stacked_nested_leaf 31.8900μs 2.8849μs 346.6312 KOps/s 342.6271 KOps/s $\color{#35bf28}+1.17\%$
test_membership_nested_last 31.1200μs 4.5000μs 222.2210 KOps/s 225.2886 KOps/s $\color{#d91a1a}-1.36\%$
test_membership_nested_leaf_last 27.6310μs 4.4985μs 222.2979 KOps/s 223.2962 KOps/s $\color{#d91a1a}-0.45\%$
test_membership_stacked_nested_last 30.0800μs 4.3747μs 228.5879 KOps/s 228.6091 KOps/s $-0.01\%$
test_membership_stacked_nested_leaf_last 32.7310μs 4.3448μs 230.1619 KOps/s 228.0597 KOps/s $\color{#35bf28}+0.92\%$
test_nested_getleaf 46.2310μs 21.9839μs 45.4879 KOps/s 45.5179 KOps/s $\color{#d91a1a}-0.07\%$
test_nested_get 44.5610μs 20.7447μs 48.2051 KOps/s 48.0159 KOps/s $\color{#35bf28}+0.39\%$
test_stacked_getleaf 55.5210μs 21.8656μs 45.7340 KOps/s 45.4974 KOps/s $\color{#35bf28}+0.52\%$
test_stacked_get 47.1210μs 20.7430μs 48.2090 KOps/s 47.8823 KOps/s $\color{#35bf28}+0.68\%$
test_nested_getitemleaf 61.7610μs 22.2137μs 45.0173 KOps/s 44.4722 KOps/s $\color{#35bf28}+1.23\%$
test_nested_getitem 86.0320μs 21.0651μs 47.4718 KOps/s 47.4618 KOps/s $\color{#35bf28}+0.02\%$
test_stacked_getitemleaf 54.9710μs 22.2677μs 44.9080 KOps/s 45.1534 KOps/s $\color{#d91a1a}-0.54\%$
test_stacked_getitem 60.3010μs 21.0040μs 47.6101 KOps/s 47.6075 KOps/s $+0.01\%$
test_lock_nested 7.9635ms 0.4854ms 2.0600 KOps/s 2.0143 KOps/s $\color{#35bf28}+2.27\%$
test_lock_stack_nested 0.5763ms 0.4826ms 2.0720 KOps/s 1.9998 KOps/s $\color{#35bf28}+3.61\%$
test_unlock_nested 0.4646ms 0.3879ms 2.5778 KOps/s 2.4908 KOps/s $\color{#35bf28}+3.49\%$
test_unlock_stack_nested 0.4292ms 0.3890ms 2.5707 KOps/s 2.4673 KOps/s $\color{#35bf28}+4.19\%$
test_flatten_speed 0.1642ms 0.1216ms 8.2261 KOps/s 8.2003 KOps/s $\color{#35bf28}+0.31\%$
test_unflatten_speed 0.7001ms 0.5753ms 1.7383 KOps/s 1.7532 KOps/s $\color{#d91a1a}-0.85\%$
test_common_ops 99.8123ms 1.4910ms 670.6899 Ops/s 265.3692 Ops/s $\textbf{\color{#35bf28}+152.74\%}$
test_creation 0.1028ms 3.1141μs 321.1176 KOps/s 322.9906 KOps/s $\color{#d91a1a}-0.58\%$
test_creation_empty 32.0700μs 6.9857μs 143.1506 KOps/s 143.5515 KOps/s $\color{#d91a1a}-0.28\%$
test_creation_nested_1 72.5610μs 11.1511μs 89.6775 KOps/s 86.8248 KOps/s $\color{#35bf28}+3.29\%$
test_creation_nested_2 87.1220μs 13.4158μs 74.5391 KOps/s 72.8894 KOps/s $\color{#35bf28}+2.26\%$
test_creation_many_keys[10] 64.0910μs 20.5633μs 48.6302 KOps/s 47.4903 KOps/s $\color{#35bf28}+2.40\%$
test_creation_many_keys[50] 0.1301ms 89.8248μs 11.1328 KOps/s 11.0533 KOps/s $\color{#35bf28}+0.72\%$
test_creation_many_keys[100] 0.2326ms 0.1776ms 5.6310 KOps/s 5.6588 KOps/s $\color{#d91a1a}-0.49\%$
test_creation_nested_many_keys[10] 83.8120μs 44.8774μs 22.2829 KOps/s 21.8913 KOps/s $\color{#35bf28}+1.79\%$
test_creation_nested_many_keys[50] 0.2616ms 0.1829ms 5.4672 KOps/s 5.4021 KOps/s $\color{#35bf28}+1.21\%$
test_clone 47.6910μs 13.3991μs 74.6318 KOps/s 74.0376 KOps/s $\color{#35bf28}+0.80\%$
test_getitem[int] 1.5442ms 15.1697μs 65.9211 KOps/s 63.2836 KOps/s $\color{#35bf28}+4.17\%$
test_getitem[slice_int] 0.1336ms 24.4968μs 40.8217 KOps/s 39.7158 KOps/s $\color{#35bf28}+2.78\%$
test_getitem[range] 0.1754ms 67.1121μs 14.9004 KOps/s 15.3164 KOps/s $\color{#d91a1a}-2.72\%$
test_getitem[tuple] 0.1416ms 24.2165μs 41.2942 KOps/s 40.0350 KOps/s $\color{#35bf28}+3.15\%$
test_getitem[list] 0.1820ms 60.1911μs 16.6138 KOps/s 16.5419 KOps/s $\color{#35bf28}+0.43\%$
test_setitem_dim[int] 45.2810μs 24.8158μs 40.2969 KOps/s 37.5808 KOps/s $\textbf{\color{#35bf28}+7.23\%}$
test_setitem_dim[slice_int] 84.4120μs 41.7285μs 23.9644 KOps/s 21.9458 KOps/s $\textbf{\color{#35bf28}+9.20\%}$
test_setitem_dim[range] 0.1248ms 94.1330μs 10.6233 KOps/s 9.8502 KOps/s $\textbf{\color{#35bf28}+7.85\%}$
test_setitem_dim[tuple] 59.2110μs 38.6451μs 25.8765 KOps/s 25.1861 KOps/s $\color{#35bf28}+2.74\%$
test_setitem 53.7910μs 17.6846μs 56.5464 KOps/s 54.7236 KOps/s $\color{#35bf28}+3.33\%$
test_set 40.0810μs 16.9397μs 59.0331 KOps/s 56.8553 KOps/s $\color{#35bf28}+3.83\%$
test_set_shared 0.4915ms 0.2081ms 4.8049 KOps/s 4.8451 KOps/s $\color{#d91a1a}-0.83\%$
test_update 0.1973ms 21.8097μs 45.8511 KOps/s 45.6974 KOps/s $\color{#35bf28}+0.34\%$
test_update_nested 71.3910μs 33.5029μs 29.8482 KOps/s 29.0280 KOps/s $\color{#35bf28}+2.83\%$
test_update__nested 0.5034ms 33.3462μs 29.9884 KOps/s 30.2595 KOps/s $\color{#d91a1a}-0.90\%$
test_set_nested 55.4910μs 18.9797μs 52.6878 KOps/s 51.6684 KOps/s $\color{#35bf28}+1.97\%$
test_set_nested_new 61.9510μs 23.5718μs 42.4236 KOps/s 41.3827 KOps/s $\color{#35bf28}+2.52\%$
test_select 95.6820μs 40.7461μs 24.5423 KOps/s 24.6774 KOps/s $\color{#d91a1a}-0.55\%$
test_select_nested 0.2042ms 74.3763μs 13.4451 KOps/s 13.4296 KOps/s $\color{#35bf28}+0.12\%$
test_exclude_nested 0.1364ms 91.9197μs 10.8791 KOps/s 10.8222 KOps/s $\color{#35bf28}+0.53\%$
test_empty[True] 0.4448ms 0.4019ms 2.4879 KOps/s 2.4871 KOps/s $\color{#35bf28}+0.03\%$
test_empty[False] 12.0903μs 1.3716μs 729.0696 KOps/s 712.1955 KOps/s $\color{#35bf28}+2.37\%$
test_to 0.1131ms 70.0162μs 14.2824 KOps/s 14.3394 KOps/s $\color{#d91a1a}-0.40\%$
test_to_nonblocking 0.1204ms 69.4500μs 14.3988 KOps/s 14.5082 KOps/s $\color{#d91a1a}-0.75\%$
test_unbind_speed 1.9259ms 0.3396ms 2.9446 KOps/s 2.8767 KOps/s $\color{#35bf28}+2.36\%$
test_unbind_speed_stack0 0.4121ms 0.3345ms 2.9891 KOps/s 2.9141 KOps/s $\color{#35bf28}+2.58\%$
test_unbind_speed_stack1 0.1140s 0.8551ms 1.1694 KOps/s 879.2669 Ops/s $\textbf{\color{#35bf28}+33.00\%}$
test_split 1.2413ms 1.1412ms 876.2463 Ops/s 873.8892 Ops/s $\color{#35bf28}+0.27\%$
test_chunk 0.1139s 1.2211ms 818.9593 Ops/s 910.5515 Ops/s $\textbf{\color{#d91a1a}-10.06\%}$
test_to_cpu_blocking 19.5527ms 19.2182ms 52.0339 Ops/s 51.3818 Ops/s $\color{#35bf28}+1.27\%$
test_to_cpu_global_sync 11.8767ms 11.7621ms 85.0185 Ops/s 84.0632 Ops/s $\color{#35bf28}+1.14\%$
test_to_cpu_event_sync 13.0264ms 12.5646ms 79.5885 Ops/s 69.9191 Ops/s $\textbf{\color{#35bf28}+13.83\%}$
test_to_cpu_default 12.7840ms 12.5331ms 79.7887 Ops/s 78.3840 Ops/s $\color{#35bf28}+1.79\%$
test_consolidate[False-None] 0.1434s 4.7131ms 212.1733 Ops/s 237.0543 Ops/s $\textbf{\color{#d91a1a}-10.50\%}$
test_consolidate[default-None] 2.1143ms 2.0286ms 492.9555 Ops/s 464.4573 Ops/s $\textbf{\color{#35bf28}+6.14\%}$
test_consolidate[reduce-overhead-None] 2.2488ms 1.9647ms 508.9716 Ops/s 480.1920 Ops/s $\textbf{\color{#35bf28}+5.99\%}$
test_consolidate_njt[False-None] 9.0361ms 8.5070ms 117.5504 Ops/s 115.9438 Ops/s $\color{#35bf28}+1.39\%$
test_to[False-False-None] 2.2353ms 2.1717ms 460.4687 Ops/s 458.0850 Ops/s $\color{#35bf28}+0.52\%$
test_to[True-False-None] 2.1638ms 1.8740ms 533.6121 Ops/s 523.0143 Ops/s $\color{#35bf28}+2.03\%$
test_to[within-False-None] 6.4995ms 6.2177ms 160.8315 Ops/s 161.0717 Ops/s $\color{#d91a1a}-0.15\%$
test_to[True-default-None] 11.5478ms 10.5878ms 94.4484 Ops/s 90.8817 Ops/s $\color{#35bf28}+3.92\%$
test_to_njt[False-False-None] 9.0128ms 8.6923ms 115.0450 Ops/s 115.9851 Ops/s $\color{#d91a1a}-0.81\%$
test_to_njt[True-False-None] 7.5350ms 7.2859ms 137.2513 Ops/s 142.5455 Ops/s $\color{#d91a1a}-3.71\%$
test_to_njt[within-False-None] 16.3227ms 15.5680ms 64.2345 Ops/s 63.0661 Ops/s $\color{#35bf28}+1.85\%$
test_creation[device0] 0.3947ms 0.1219ms 8.2043 KOps/s 8.4221 KOps/s $\color{#d91a1a}-2.59\%$
test_creation_from_tensor 0.3982ms 0.1189ms 8.4136 KOps/s 8.6772 KOps/s $\color{#d91a1a}-3.04\%$
test_add_one[memmap_tensor0] 0.1933ms 6.7051μs 149.1408 KOps/s 145.8352 KOps/s $\color{#35bf28}+2.27\%$
test_contiguous[memmap_tensor0] 28.9600μs 0.6666μs 1.5002 MOps/s 2.0389 MOps/s $\textbf{\color{#d91a1a}-26.42\%}$
test_stack[memmap_tensor0] 69.7410μs 4.1659μs 240.0452 KOps/s 228.7900 KOps/s $\color{#35bf28}+4.92\%$
test_memmaptd_index 1.2050ms 0.2763ms 3.6194 KOps/s 3.4778 KOps/s $\color{#35bf28}+4.07\%$
test_memmaptd_index_astensor 0.5278ms 0.3803ms 2.6296 KOps/s 2.5546 KOps/s $\color{#35bf28}+2.94\%$
test_memmaptd_index_op 0.8665ms 0.6412ms 1.5596 KOps/s 1.5095 KOps/s $\color{#35bf28}+3.32\%$
test_serialize_model 0.1371s 0.1348s 7.4208 Ops/s 7.4641 Ops/s $\color{#d91a1a}-0.58\%$
test_serialize_model_pickle 1.3859s 1.2147s 0.8232 Ops/s 0.8188 Ops/s $\color{#35bf28}+0.54\%$
test_serialize_weights 0.1382s 0.1353s 7.3883 Ops/s 7.5106 Ops/s $\color{#d91a1a}-1.63\%$
test_serialize_weights_returnearly 73.1943ms 64.4052ms 15.5267 Ops/s 11.5558 Ops/s $\textbf{\color{#35bf28}+34.36\%}$
test_serialize_weights_pickle 1.3653s 1.2130s 0.8244 Ops/s 0.8231 Ops/s $\color{#35bf28}+0.15\%$
test_reshape_pytree 0.2001ms 32.8376μs 30.4529 KOps/s 29.7669 KOps/s $\color{#35bf28}+2.30\%$
test_reshape_td 78.5420μs 46.8056μs 21.3650 KOps/s 22.1182 KOps/s $\color{#d91a1a}-3.41\%$
test_view_pytree 0.2427ms 32.5084μs 30.7613 KOps/s 30.3231 KOps/s $\color{#35bf28}+1.45\%$
test_view_td 92.4920μs 54.9307μs 18.2048 KOps/s 18.5532 KOps/s $\color{#d91a1a}-1.88\%$
test_unbind_pytree 0.2598ms 36.9216μs 27.0844 KOps/s 26.2021 KOps/s $\color{#35bf28}+3.37\%$
test_unbind_td 0.1955ms 50.0396μs 19.9842 KOps/s 19.2912 KOps/s $\color{#35bf28}+3.59\%$
test_split_pytree 0.2420ms 42.2650μs 23.6602 KOps/s 22.9894 KOps/s $\color{#35bf28}+2.92\%$
test_split_td 0.1779ms 64.5777μs 15.4852 KOps/s 15.3102 KOps/s $\color{#35bf28}+1.14\%$
test_add_pytree 0.2267ms 41.8970μs 23.8681 KOps/s 23.1210 KOps/s $\color{#35bf28}+3.23\%$
test_add_td 0.1044ms 54.6849μs 18.2866 KOps/s 16.7264 KOps/s $\textbf{\color{#35bf28}+9.33\%}$
test_compile_add_one_nested[tensordict-compile] 0.3711ms 0.2291ms 4.3643 KOps/s 4.2645 KOps/s $\color{#35bf28}+2.34\%$
test_compile_add_one_nested[tensordict-eager] 0.6550ms 0.2023ms 4.9431 KOps/s 4.9844 KOps/s $\color{#d91a1a}-0.83\%$
test_compile_add_one_nested[pytree-compile] 0.2682ms 0.1338ms 7.4720 KOps/s 7.0484 KOps/s $\textbf{\color{#35bf28}+6.01\%}$
test_compile_add_one_nested[pytree-eager] 0.4038ms 0.1759ms 5.6845 KOps/s 5.5039 KOps/s $\color{#35bf28}+3.28\%$
test_compile_copy_nested[tensordict-compile] 0.1095ms 15.6065μs 64.0759 KOps/s 64.8822 KOps/s $\color{#d91a1a}-1.24\%$
test_compile_copy_nested[tensordict-eager] 0.3371ms 53.0422μs 18.8529 KOps/s 18.2573 KOps/s $\color{#35bf28}+3.26\%$
test_compile_copy_nested[pytree-compile] 0.1433ms 15.8991μs 62.8965 KOps/s 62.6653 KOps/s $\color{#35bf28}+0.37\%$
test_compile_copy_nested[pytree-eager] 0.3792ms 67.8896μs 14.7298 KOps/s 14.4605 KOps/s $\color{#35bf28}+1.86\%$
test_compile_add_one_flat[tensordict-compile] 0.2922ms 0.2461ms 4.0626 KOps/s 2.7160 KOps/s $\textbf{\color{#35bf28}+49.58\%}$
test_compile_add_one_flat[tensordict-eager] 0.3471ms 0.2841ms 3.5194 KOps/s 3.5075 KOps/s $\color{#35bf28}+0.34\%$
test_compile_add_one_flat[tensorclass-compile] 0.2530ms 0.1823ms 5.4848 KOps/s 4.8097 KOps/s $\textbf{\color{#35bf28}+14.04\%}$
test_compile_add_one_flat[tensorclass-eager] 0.2350ms 74.2595μs 13.4663 KOps/s 13.0411 KOps/s $\color{#35bf28}+3.26\%$
test_compile_add_one_flat[pytree-compile] 0.5435ms 0.1904ms 5.2511 KOps/s 4.8378 KOps/s $\textbf{\color{#35bf28}+8.54\%}$
test_compile_add_one_flat[pytree-eager] 0.7808ms 0.5117ms 1.9542 KOps/s 1.8570 KOps/s $\textbf{\color{#35bf28}+5.24\%}$
test_compile_add_self_flat[tensordict-eager] 0.4894ms 0.3307ms 3.0237 KOps/s 2.9643 KOps/s $\color{#35bf28}+2.01\%$
test_compile_add_self_flat[tensordict-compile] 0.2961ms 0.2488ms 4.0188 KOps/s 3.6912 KOps/s $\textbf{\color{#35bf28}+8.88\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1609ms 91.7008μs 10.9050 KOps/s 10.5239 KOps/s $\color{#35bf28}+3.62\%$
test_compile_add_self_flat[tensorclass-compile] 0.3276ms 0.1854ms 5.3946 KOps/s 4.8371 KOps/s $\textbf{\color{#35bf28}+11.52\%}$
test_compile_add_self_flat[pytree-eager] 0.6593ms 0.4208ms 2.3765 KOps/s 2.2794 KOps/s $\color{#35bf28}+4.26\%$
test_compile_add_self_flat[pytree-compile] 0.5297ms 0.1942ms 5.1492 KOps/s 4.8729 KOps/s $\textbf{\color{#35bf28}+5.67\%}$
test_compile_copy_flat[tensordict-compile] 55.1110μs 19.6920μs 50.7820 KOps/s 48.4716 KOps/s $\color{#35bf28}+4.77\%$
test_compile_copy_flat[tensordict-eager] 0.1116ms 41.0179μs 24.3796 KOps/s 23.8063 KOps/s $\color{#35bf28}+2.41\%$
test_compile_copy_flat[pytree-compile] 0.2352ms 21.5474μs 46.4093 KOps/s 44.9835 KOps/s $\color{#35bf28}+3.17\%$
test_compile_copy_flat[pytree-eager] 0.3283ms 52.1221μs 19.1857 KOps/s 19.0139 KOps/s $\color{#35bf28}+0.90\%$
test_compile_assign_and_add[tensordict-compile] 2.1658ms 0.2363ms 4.2310 KOps/s 4.0246 KOps/s $\textbf{\color{#35bf28}+5.13\%}$
test_compile_assign_and_add[tensordict-eager] 3.4286ms 3.2467ms 308.0054 Ops/s 303.3714 Ops/s $\color{#35bf28}+1.53\%$
test_compile_assign_and_add[pytree-compile] 2.0873ms 0.1783ms 5.6083 KOps/s 5.3654 KOps/s $\color{#35bf28}+4.53\%$
test_compile_assign_and_add[pytree-eager] 3.0675ms 2.7016ms 370.1491 Ops/s 353.8780 Ops/s $\color{#35bf28}+4.60\%$
test_compile_indexing[tensor-tensordict-compile] 0.2729ms 0.1345ms 7.4325 KOps/s 6.9803 KOps/s $\textbf{\color{#35bf28}+6.48\%}$
test_compile_indexing[tensor-tensordict-eager] 0.3143ms 77.8841μs 12.8396 KOps/s 12.6200 KOps/s $\color{#35bf28}+1.74\%$
test_compile_indexing[tensor-tensorclass-compile] 0.2198ms 0.1175ms 8.5101 KOps/s 8.3467 KOps/s $\color{#35bf28}+1.96\%$
test_compile_indexing[tensor-tensorclass-eager] 0.2518ms 44.6446μs 22.3991 KOps/s 21.2237 KOps/s $\textbf{\color{#35bf28}+5.54\%}$
test_compile_indexing[tensor-pytree-compile] 0.1526ms 0.1188ms 8.4146 KOps/s 8.1534 KOps/s $\color{#35bf28}+3.20\%$
test_compile_indexing[tensor-pytree-eager] 0.2297ms 47.3128μs 21.1359 KOps/s 20.9643 KOps/s $\color{#35bf28}+0.82\%$
test_compile_indexing[slice-tensordict-compile] 0.1870ms 82.0179μs 12.1925 KOps/s 11.9312 KOps/s $\color{#35bf28}+2.19\%$
test_compile_indexing[slice-tensordict-eager] 0.2038ms 29.2761μs 34.1576 KOps/s 35.1256 KOps/s $\color{#d91a1a}-2.76\%$
test_compile_indexing[slice-tensorclass-compile] 0.1571ms 69.3534μs 14.4189 KOps/s 13.9925 KOps/s $\color{#35bf28}+3.05\%$
test_compile_indexing[slice-tensorclass-eager] 0.2346ms 22.8132μs 43.8343 KOps/s 43.8124 KOps/s $\color{#35bf28}+0.05\%$
test_compile_indexing[slice-pytree-compile] 0.1172ms 70.7303μs 14.1382 KOps/s 13.8350 KOps/s $\color{#35bf28}+2.19\%$
test_compile_indexing[slice-pytree-eager] 0.2682ms 22.6327μs 44.1838 KOps/s 42.9738 KOps/s $\color{#35bf28}+2.82\%$
test_compile_indexing[int-tensordict-compile] 0.1804ms 83.5074μs 11.9750 KOps/s 11.8937 KOps/s $\color{#35bf28}+0.68\%$
test_compile_indexing[int-tensordict-eager] 0.2651ms 28.0248μs 35.6826 KOps/s 35.0502 KOps/s $\color{#35bf28}+1.80\%$
test_compile_indexing[int-tensorclass-compile] 0.1149ms 70.1948μs 14.2461 KOps/s 13.9551 KOps/s $\color{#35bf28}+2.08\%$
test_compile_indexing[int-tensorclass-eager] 0.2379ms 22.4933μs 44.4577 KOps/s 44.2614 KOps/s $\color{#35bf28}+0.44\%$
test_compile_indexing[int-pytree-compile] 0.1136ms 70.5033μs 14.1837 KOps/s 13.7161 KOps/s $\color{#35bf28}+3.41\%$
test_compile_indexing[int-pytree-eager] 0.2319ms 22.4728μs 44.4983 KOps/s 44.0779 KOps/s $\color{#35bf28}+0.95\%$
test_compile_replace[single-eager] 0.1066ms 49.7515μs 20.0999 KOps/s 20.3379 KOps/s $\color{#d91a1a}-1.17\%$
test_compile_replace[single-compile] 0.2362ms 0.1345ms 7.4368 KOps/s 7.3174 KOps/s $\color{#35bf28}+1.63\%$
test_compile_replace[multi-eager] 0.6644ms 0.5662ms 1.7661 KOps/s 1.6836 KOps/s $\color{#35bf28}+4.90\%$
test_compile_replace[multi-compile] 0.3412ms 0.1370ms 7.2991 KOps/s 7.2093 KOps/s $\color{#35bf28}+1.25\%$
test_compile_tc_getattr_20[eager] 0.2157ms 0.1691ms 5.9134 KOps/s 5.8492 KOps/s $\color{#35bf28}+1.10\%$
test_compile_tc_getattr_20[compile] 0.3426ms 0.1474ms 6.7852 KOps/s 6.4992 KOps/s $\color{#35bf28}+4.40\%$
test_compile_clone_shallow[20-eager] 51.8210μs 19.2130μs 52.0480 KOps/s 51.1214 KOps/s $\color{#35bf28}+1.81\%$
test_compile_clone_shallow[20-compile] 85.7920μs 17.5312μs 57.0410 KOps/s 47.0683 KOps/s $\textbf{\color{#35bf28}+21.19\%}$
test_compile_clone_shallow[40-eager] 62.9910μs 33.4244μs 29.9182 KOps/s 28.7472 KOps/s $\color{#35bf28}+4.07\%$
test_compile_clone_shallow[40-compile] 75.0710μs 18.6913μs 53.5007 KOps/s 52.6609 KOps/s $\color{#35bf28}+1.59\%$
test_compile_clone_shallow[80-eager] 94.3320μs 62.0842μs 16.1072 KOps/s 15.5758 KOps/s $\color{#35bf28}+3.41\%$
test_compile_clone_shallow[80-compile] 84.6110μs 21.7027μs 46.0772 KOps/s 44.7494 KOps/s $\color{#35bf28}+2.97\%$
test_compile_update_inplace[eager] 0.2123ms 60.9794μs 16.3990 KOps/s 16.0825 KOps/s $\color{#35bf28}+1.97\%$
test_compile_update_inplace[compile] 0.2690ms 0.1453ms 6.8840 KOps/s 6.7496 KOps/s $\color{#35bf28}+1.99\%$
test_mod_add[eager] 0.3065ms 0.2264ms 4.4175 KOps/s 4.3833 KOps/s $\color{#35bf28}+0.78\%$
test_mod_add[compile] 0.2841ms 0.1347ms 7.4246 KOps/s 7.4645 KOps/s $\color{#d91a1a}-0.54\%$
test_mod_add[compile-overhead] 0.2631ms 0.1796ms 5.5685 KOps/s 5.6084 KOps/s $\color{#d91a1a}-0.71\%$
test_mod_wrap[eager] 0.5700ms 0.4672ms 2.1403 KOps/s 2.1114 KOps/s $\color{#35bf28}+1.37\%$
test_mod_wrap[compile] 0.4915ms 0.3942ms 2.5368 KOps/s 2.3583 KOps/s $\textbf{\color{#35bf28}+7.57\%}$
test_mod_wrap[compile-overhead] 8.5320ms 4.7307ms 211.3865 Ops/s 213.8562 Ops/s $\color{#d91a1a}-1.15\%$
test_mod_wrap_and_backward[eager] 1.8118ms 1.6733ms 597.6385 Ops/s 563.5361 Ops/s $\textbf{\color{#35bf28}+6.05\%}$
test_mod_wrap_and_backward[compile] 1.8975ms 1.5260ms 655.3015 Ops/s 593.3393 Ops/s $\textbf{\color{#35bf28}+10.44\%}$
test_mod_wrap_and_backward[compile-overhead] 1.3074ms 0.9343ms 1.0703 KOps/s 926.6824 Ops/s $\textbf{\color{#35bf28}+15.50\%}$
test_seq_add[eager] 0.5665ms 0.5088ms 1.9655 KOps/s 1.9607 KOps/s $\color{#35bf28}+0.24\%$
test_seq_add[compile] 0.2575ms 0.1419ms 7.0473 KOps/s 6.8334 KOps/s $\color{#35bf28}+3.13\%$
test_seq_add[compile-overhead] 0.3233ms 0.1795ms 5.5726 KOps/s 5.2937 KOps/s $\textbf{\color{#35bf28}+5.27\%}$
test_seq_wrap[eager] 1.4852ms 1.4037ms 712.4166 Ops/s 705.8594 Ops/s $\color{#35bf28}+0.93\%$
test_seq_wrap[compile] 0.5355ms 0.4078ms 2.4519 KOps/s 2.3627 KOps/s $\color{#35bf28}+3.78\%$
test_seq_wrap[compile-overhead] 0.4034ms 0.2989ms 3.3454 KOps/s 3.2309 KOps/s $\color{#35bf28}+3.55\%$
test_func_call_runtime[False-eager] 0.8500ms 0.7920ms 1.2626 KOps/s 1.2349 KOps/s $\color{#35bf28}+2.25\%$
test_func_call_runtime[False-compile] 1.0720ms 0.9740ms 1.0266 KOps/s 987.1824 Ops/s $\color{#35bf28}+4.00\%$
test_func_call_runtime[False-compile-overhead] 0.5751ms 0.5077ms 1.9696 KOps/s 1.9328 KOps/s $\color{#35bf28}+1.91\%$
test_func_call_runtime[True-eager] 1.1519ms 1.0332ms 967.8904 Ops/s 957.5693 Ops/s $\color{#35bf28}+1.08\%$
test_func_call_runtime[True-compile] 1.0478ms 0.9929ms 1.0072 KOps/s 994.7495 Ops/s $\color{#35bf28}+1.25\%$
test_func_call_runtime[True-compile-overhead] 0.5743ms 0.5219ms 1.9161 KOps/s 1.8646 KOps/s $\color{#35bf28}+2.76\%$
test_func_call_cm_runtime[False-eager] 0.8948ms 0.7906ms 1.2648 KOps/s 1.2453 KOps/s $\color{#35bf28}+1.57\%$
test_func_call_cm_runtime[False-compile] 1.2913ms 0.9640ms 1.0374 KOps/s 1.0186 KOps/s $\color{#35bf28}+1.84\%$
test_func_call_cm_runtime[False-compile-overhead] 0.7424ms 0.4940ms 2.0242 KOps/s 2.0043 KOps/s $\color{#35bf28}+1.00\%$
test_func_call_cm_runtime[True-eager] 1.2977ms 1.1821ms 845.9733 Ops/s 832.5363 Ops/s $\color{#35bf28}+1.61\%$
test_func_call_cm_runtime[True-compile] 1.0961ms 1.0026ms 997.4236 Ops/s 952.1209 Ops/s $\color{#35bf28}+4.76\%$
test_func_call_cm_runtime[True-compile-overhead] 0.6361ms 0.5392ms 1.8546 KOps/s 1.8067 KOps/s $\color{#35bf28}+2.65\%$
test_vmap_func_call_cm_runtime[eager] 2.8438ms 2.3426ms 426.8781 Ops/s 419.2753 Ops/s $\color{#35bf28}+1.81\%$
test_vmap_func_call_cm_runtime[compile] 1.2158ms 1.0326ms 968.4589 Ops/s 959.5202 Ops/s $\color{#35bf28}+0.93\%$
test_vmap_func_call_cm_runtime[compile-overhead] 0.6473ms 0.5296ms 1.8882 KOps/s 1.8352 KOps/s $\color{#35bf28}+2.89\%$
test_distributed 0.6493ms 0.1470ms 6.8007 KOps/s 6.6247 KOps/s $\color{#35bf28}+2.66\%$
test_tdmodule 0.6094ms 0.1935ms 5.1687 KOps/s 5.1041 KOps/s $\color{#35bf28}+1.26\%$
test_tdmodule_dispatch 0.4134ms 0.3715ms 2.6914 KOps/s 2.6927 KOps/s $\color{#d91a1a}-0.05\%$
test_tdseq 48.4700μs 26.7056μs 37.4453 KOps/s 37.3049 KOps/s $\color{#35bf28}+0.38\%$
test_tdseq_dispatch 0.2674ms 0.2182ms 4.5820 KOps/s 4.6123 KOps/s $\color{#d91a1a}-0.66\%$
test_instantiation_functorch 2.1657ms 2.0797ms 480.8431 Ops/s 477.0462 Ops/s $\color{#35bf28}+0.80\%$
test_exec_functorch 0.2252ms 0.1796ms 5.5671 KOps/s 5.4390 KOps/s $\color{#35bf28}+2.36\%$
test_exec_functional_call 0.2122ms 0.1603ms 6.2385 KOps/s 6.1359 KOps/s $\color{#35bf28}+1.67\%$
test_exec_td_decorator 0.4588ms 0.2374ms 4.2130 KOps/s 4.1677 KOps/s $\color{#35bf28}+1.09\%$
test_vmap_mlp_speed_decorator[True-True] 1.1999ms 1.0081ms 991.9412 Ops/s 988.0694 Ops/s $\color{#35bf28}+0.39\%$
test_vmap_mlp_speed_decorator[True-False] 1.2062ms 1.0040ms 995.9810 Ops/s 986.5871 Ops/s $\color{#35bf28}+0.95\%$
test_vmap_mlp_speed_decorator[False-True] 0.9182ms 0.7081ms 1.4123 KOps/s 1.3959 KOps/s $\color{#35bf28}+1.17\%$
test_vmap_mlp_speed_decorator[False-False] 0.9347ms 0.7089ms 1.4106 KOps/s 1.3969 KOps/s $\color{#35bf28}+0.98\%$
test_vmap_transformer_speed_decorator[True-True] 21.6231ms 20.6271ms 48.4799 Ops/s 47.9781 Ops/s $\color{#35bf28}+1.05\%$
test_vmap_transformer_speed_decorator[True-False] 20.7637ms 20.5481ms 48.6663 Ops/s 48.0304 Ops/s $\color{#35bf28}+1.32\%$
test_vmap_transformer_speed_decorator[False-True] 20.3132ms 20.1861ms 49.5389 Ops/s 48.8937 Ops/s $\color{#35bf28}+1.32\%$
test_vmap_transformer_speed_decorator[False-False] 20.3363ms 20.1757ms 49.5645 Ops/s 49.0765 Ops/s $\color{#35bf28}+0.99\%$
test_to_module_speed[True] 1.5875ms 1.4793ms 675.9986 Ops/s 674.9134 Ops/s $\color{#35bf28}+0.16\%$
test_to_module_speed[False] 1.5616ms 1.4679ms 681.2590 Ops/s 676.3325 Ops/s $\color{#35bf28}+0.73\%$
test_tc_init 63.6810μs 42.6835μs 23.4282 KOps/s 23.2298 KOps/s $\color{#35bf28}+0.85\%$
test_tc_init_tensor_only 81.4410μs 9.0477μs 110.5256 KOps/s 110.4144 KOps/s $\color{#35bf28}+0.10\%$
test_tc_init_nested 0.1204ms 85.9159μs 11.6393 KOps/s 11.5180 KOps/s $\color{#35bf28}+1.05\%$
test_tc_init_many_fields 33.6310μs 14.9999μs 66.6669 KOps/s 67.6012 KOps/s $\color{#d91a1a}-1.38\%$
test_tc_first_layer_tensor 21.9400μs 1.7976μs 556.2849 KOps/s 545.2922 KOps/s $\color{#35bf28}+2.02\%$
test_tc_first_layer_tensor_only 1.5650μs 0.3964μs 2.5230 MOps/s 2.5091 MOps/s $\color{#35bf28}+0.55\%$
test_tc_first_layer_tensor_set 27.0510μs 3.9323μs 254.3046 KOps/s 254.7049 KOps/s $\color{#d91a1a}-0.16\%$
test_tc_first_layer_tensor_only_set 23.5510μs 3.2325μs 309.3533 KOps/s 305.3519 KOps/s $\color{#35bf28}+1.31\%$
test_tc_first_layer_nontensor 32.6510μs 6.1671μs 162.1506 KOps/s 154.7477 KOps/s $\color{#35bf28}+4.78\%$
test_tc_second_layer_tensor 34.4210μs 4.4157μs 226.4634 KOps/s 226.6138 KOps/s $\color{#d91a1a}-0.07\%$
test_tc_second_layer_nontensor 87.8720μs 8.6889μs 115.0894 KOps/s 110.7008 KOps/s $\color{#35bf28}+3.96\%$
test_unbind 0.2792s 14.0343ms 71.2539 Ops/s 69.0745 Ops/s $\color{#35bf28}+3.16\%$
test_full_like 4.5300ms 4.4064ms 226.9449 Ops/s 226.6137 Ops/s $\color{#35bf28}+0.15\%$
test_zeros_like 4.5087ms 4.3383ms 230.5075 Ops/s 227.3606 Ops/s $\color{#35bf28}+1.38\%$
test_ones_like 4.4781ms 4.3181ms 231.5850 Ops/s 227.0997 Ops/s $\color{#35bf28}+1.98\%$
test_clone 6.8429ms 6.5909ms 151.7244 Ops/s 150.5408 Ops/s $\color{#35bf28}+0.79\%$
test_squeeze 73.1020μs 14.1272μs 70.7852 KOps/s 71.9484 KOps/s $\color{#d91a1a}-1.62\%$
test_unsqueeze 0.1577ms 0.1112ms 8.9956 KOps/s 8.9657 KOps/s $\color{#35bf28}+0.33\%$
test_split 0.3777ms 0.1801ms 5.5519 KOps/s 5.4669 KOps/s $\color{#35bf28}+1.55\%$
test_permute 0.2401ms 0.2080ms 4.8075 KOps/s 4.8171 KOps/s $\color{#d91a1a}-0.20\%$
test_stack 52.1271ms 51.8561ms 19.2841 Ops/s 19.2833 Ops/s $+0.00\%$
test_cat 52.2105ms 51.8387ms 19.2906 Ops/s 22.9831 Ops/s $\textbf{\color{#d91a1a}-16.07\%}$
test_sequential_tensordict 0.6065ms 0.5639ms 1.7734 KOps/s 1.7417 KOps/s $\color{#35bf28}+1.82\%$
test_sequential_graph_module 0.1672ms 0.1144ms 8.7420 KOps/s 7.9477 KOps/s $\textbf{\color{#35bf28}+9.99\%}$
test_nested_tensordict 1.7275ms 0.8200ms 1.2195 KOps/s 1.2178 KOps/s $\color{#35bf28}+0.14\%$
test_nested_graph_module 0.1617ms 0.1253ms 7.9832 KOps/s 7.3725 KOps/s $\textbf{\color{#35bf28}+8.28\%}$

@github-actions

github-actions Bot commented May 21, 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}19$. Worsened: $\large\color{#d91a1a}9$.

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_collector_preallocated_write[stack_out-cpu-32-2] 0.2353ms 0.1702ms 5.8755 KOps/s 5.9210 KOps/s $\color{#d91a1a}-0.77\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.3317ms 0.2947ms 3.3930 KOps/s 3.3399 KOps/s $\color{#35bf28}+1.59\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 1.0309ms 0.7798ms 1.2824 KOps/s 1.2769 KOps/s $\color{#35bf28}+0.43\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.3770ms 0.3371ms 2.9668 KOps/s 2.9285 KOps/s $\color{#35bf28}+1.31\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.3331ms 1.1238ms 889.8073 Ops/s 872.9821 Ops/s $\color{#35bf28}+1.93\%$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.7264ms 7.3640ms 135.7964 Ops/s 134.1136 Ops/s $\color{#35bf28}+1.25\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.3258ms 0.2547ms 3.9267 KOps/s 4.0226 KOps/s $\color{#d91a1a}-2.38\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.4838ms 0.3737ms 2.6762 KOps/s 2.6831 KOps/s $\color{#d91a1a}-0.26\%$
test_collector_preallocated_write[stack_out-cuda-32-32] 1.0398ms 0.8226ms 1.2156 KOps/s 1.2157 KOps/s $-0.01\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.3567ms 0.2536ms 3.9429 KOps/s 3.9518 KOps/s $\color{#d91a1a}-0.23\%$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.4774ms 0.3750ms 2.6669 KOps/s 2.6759 KOps/s $\color{#d91a1a}-0.34\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 1.0685ms 0.8848ms 1.1302 KOps/s 1.1340 KOps/s $\color{#d91a1a}-0.34\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2818ms 0.2159ms 4.6319 KOps/s 4.5820 KOps/s $\color{#35bf28}+1.09\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 0.9286ms 0.8437ms 1.1852 KOps/s 1.1755 KOps/s $\color{#35bf28}+0.83\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.8346ms 3.3671ms 296.9940 Ops/s 299.0302 Ops/s $\color{#d91a1a}-0.68\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.4504ms 0.3734ms 2.6779 KOps/s 2.6795 KOps/s $\color{#d91a1a}-0.06\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 1.9969ms 1.7570ms 569.1463 Ops/s 597.3454 Ops/s $\color{#d91a1a}-4.72\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 16.1424ms 10.4987ms 95.2502 Ops/s 96.4805 Ops/s $\color{#d91a1a}-1.28\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.5251ms 0.4093ms 2.4432 KOps/s 2.5545 KOps/s $\color{#d91a1a}-4.36\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.6637ms 1.5254ms 655.5617 Ops/s 686.8296 Ops/s $\color{#d91a1a}-4.55\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 6.1388ms 6.0213ms 166.0769 Ops/s 175.1179 Ops/s $\textbf{\color{#d91a1a}-5.16\%}$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.4945ms 0.4112ms 2.4318 KOps/s 2.5569 KOps/s $\color{#d91a1a}-4.89\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.6732ms 1.5383ms 650.0548 Ops/s 684.3543 Ops/s $\textbf{\color{#d91a1a}-5.01\%}$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 6.3757ms 6.0444ms 165.4426 Ops/s 173.3081 Ops/s $\color{#d91a1a}-4.54\%$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.2326ms 0.1729ms 5.7831 KOps/s 5.7350 KOps/s $\color{#35bf28}+0.84\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.8360ms 0.6752ms 1.4810 KOps/s 1.4677 KOps/s $\color{#35bf28}+0.90\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 3.0022ms 2.6802ms 373.1132 Ops/s 371.6958 Ops/s $\color{#35bf28}+0.38\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 1.8983ms 0.3204ms 3.1214 KOps/s 3.0607 KOps/s $\color{#35bf28}+1.98\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.8064ms 1.5262ms 655.2325 Ops/s 671.2792 Ops/s $\color{#d91a1a}-2.39\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 9.9198ms 9.6674ms 103.4404 Ops/s 105.1109 Ops/s $\color{#d91a1a}-1.59\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.4051ms 0.3230ms 3.0961 KOps/s 3.0479 KOps/s $\color{#35bf28}+1.58\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.9097ms 1.2081ms 827.7362 Ops/s 831.5427 Ops/s $\color{#d91a1a}-0.46\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 5.1271ms 4.7166ms 212.0150 Ops/s 212.6260 Ops/s $\color{#d91a1a}-0.29\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.7416ms 0.3238ms 3.0883 KOps/s 3.0753 KOps/s $\color{#35bf28}+0.42\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.6296ms 1.2060ms 829.1723 Ops/s 828.8067 Ops/s $\color{#35bf28}+0.04\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 5.1528ms 4.7183ms 211.9423 Ops/s 212.1466 Ops/s $\color{#d91a1a}-0.10\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.5407ms 0.1190ms 8.4065 KOps/s 8.2412 KOps/s $\color{#35bf28}+2.01\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.8925ms 0.4674ms 2.1393 KOps/s 2.1078 KOps/s $\color{#35bf28}+1.49\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 1.9014ms 1.8601ms 537.5968 Ops/s 529.5007 Ops/s $\color{#35bf28}+1.53\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.7004ms 0.2514ms 3.9772 KOps/s 3.8407 KOps/s $\color{#35bf28}+3.55\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.8595ms 1.2936ms 773.0091 Ops/s 795.7805 Ops/s $\color{#d91a1a}-2.86\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 8.9132ms 8.6865ms 115.1216 Ops/s 116.8909 Ops/s $\color{#d91a1a}-1.51\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.3442ms 0.2756ms 3.6279 KOps/s 3.6615 KOps/s $\color{#d91a1a}-0.92\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.1666ms 1.0256ms 975.0855 Ops/s 983.6406 Ops/s $\color{#d91a1a}-0.87\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 4.6943ms 4.0529ms 246.7395 Ops/s 251.2389 Ops/s $\color{#d91a1a}-1.79\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.3523ms 0.2761ms 3.6216 KOps/s 3.6684 KOps/s $\color{#d91a1a}-1.28\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.1495ms 1.0325ms 968.5354 Ops/s 985.8391 Ops/s $\color{#d91a1a}-1.76\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 4.6397ms 4.0335ms 247.9230 Ops/s 249.8938 Ops/s $\color{#d91a1a}-0.79\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.8362ms 16.4103ms 60.9374 Ops/s 61.5439 Ops/s $\color{#d91a1a}-0.99\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 33.3546ms 33.1088ms 30.2035 Ops/s 30.2962 Ops/s $\color{#d91a1a}-0.31\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 87.8766ms 86.1232ms 11.6113 Ops/s 11.6701 Ops/s $\color{#d91a1a}-0.50\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.2038s 0.2028s 4.9303 Ops/s 5.0403 Ops/s $\color{#d91a1a}-2.18\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 14.0102ms 13.8093ms 72.4149 Ops/s 73.0853 Ops/s $\color{#d91a1a}-0.92\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 27.3225ms 27.1869ms 36.7825 Ops/s 37.3763 Ops/s $\color{#d91a1a}-1.59\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 14.1821ms 13.9382ms 71.7452 Ops/s 71.8212 Ops/s $\color{#d91a1a}-0.11\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 27.5760ms 27.4572ms 36.4203 Ops/s 36.5598 Ops/s $\color{#d91a1a}-0.38\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 51.5839ms 51.3687ms 19.4671 Ops/s 19.3126 Ops/s $\color{#35bf28}+0.80\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1047s 0.1046s 9.5644 Ops/s 9.5183 Ops/s $\color{#35bf28}+0.49\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1486s 0.1475s 6.7807 Ops/s 6.8269 Ops/s $\color{#d91a1a}-0.68\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3256s 0.3237s 3.0888 Ops/s 3.0719 Ops/s $\color{#35bf28}+0.55\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 77.5834ms 75.0419ms 13.3259 Ops/s 12.7624 Ops/s $\color{#35bf28}+4.42\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1528s 0.1503s 6.6538 Ops/s 6.3825 Ops/s $\color{#35bf28}+4.25\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 74.9561ms 74.4584ms 13.4303 Ops/s 12.6272 Ops/s $\textbf{\color{#35bf28}+6.36\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1552s 0.1510s 6.6207 Ops/s 6.3125 Ops/s $\color{#35bf28}+4.88\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 30.5807ms 30.2589ms 33.0481 Ops/s 32.9107 Ops/s $\color{#35bf28}+0.42\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 61.7662ms 61.3132ms 16.3097 Ops/s 16.3613 Ops/s $\color{#d91a1a}-0.32\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1232s 0.1216s 8.2236 Ops/s 8.2380 Ops/s $\color{#d91a1a}-0.18\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2758s 0.2748s 3.6392 Ops/s 3.7479 Ops/s $\color{#d91a1a}-2.90\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 43.4383ms 43.0368ms 23.2359 Ops/s 23.4744 Ops/s $\color{#d91a1a}-1.02\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 86.5478ms 86.0871ms 11.6161 Ops/s 11.7013 Ops/s $\color{#d91a1a}-0.73\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 43.5713ms 43.1910ms 23.1530 Ops/s 23.2804 Ops/s $\color{#d91a1a}-0.55\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 86.9813ms 86.5887ms 11.5489 Ops/s 11.5988 Ops/s $\color{#d91a1a}-0.43\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 20.6867ms 20.2639ms 49.3489 Ops/s 48.9310 Ops/s $\color{#35bf28}+0.85\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 42.0367ms 41.5843ms 24.0475 Ops/s 24.0467 Ops/s $+0.00\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1120s 0.1112s 8.9917 Ops/s 9.1045 Ops/s $\color{#d91a1a}-1.24\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2486s 0.2467s 4.0528 Ops/s 4.1086 Ops/s $\color{#d91a1a}-1.36\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 36.9902ms 36.5202ms 27.3821 Ops/s 27.5356 Ops/s $\color{#d91a1a}-0.56\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 74.0863ms 73.3908ms 13.6257 Ops/s 13.7973 Ops/s $\color{#d91a1a}-1.24\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 36.9677ms 36.6833ms 27.2604 Ops/s 27.6460 Ops/s $\color{#d91a1a}-1.39\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 73.6069ms 72.8732ms 13.7225 Ops/s 13.7631 Ops/s $\color{#d91a1a}-0.30\%$
test_plain_set_nested 41.8810μs 15.5257μs 64.4095 KOps/s 65.1478 KOps/s $\color{#d91a1a}-1.13\%$
test_plain_set_stack_nested 32.8810μs 15.1292μs 66.0975 KOps/s 65.9903 KOps/s $\color{#35bf28}+0.16\%$
test_plain_set_nested_inplace 44.0810μs 16.9548μs 58.9803 KOps/s 59.9366 KOps/s $\color{#d91a1a}-1.60\%$
test_plain_set_stack_nested_inplace 45.5510μs 16.5662μs 60.3640 KOps/s 60.7819 KOps/s $\color{#d91a1a}-0.69\%$
test_items 30.2200μs 6.0151μs 166.2473 KOps/s 165.2886 KOps/s $\color{#35bf28}+0.58\%$
test_items_nested 0.5682ms 0.4640ms 2.1552 KOps/s 2.1523 KOps/s $\color{#35bf28}+0.13\%$
test_items_nested_locked 0.6542ms 0.4677ms 2.1382 KOps/s 2.1578 KOps/s $\color{#d91a1a}-0.91\%$
test_items_nested_leaf 0.1679ms 96.8693μs 10.3232 KOps/s 10.3297 KOps/s $\color{#d91a1a}-0.06\%$
test_items_stack_nested 0.6418ms 0.4637ms 2.1568 KOps/s 2.1982 KOps/s $\color{#d91a1a}-1.89\%$
test_items_stack_nested_leaf 0.1450ms 96.6425μs 10.3474 KOps/s 10.4135 KOps/s $\color{#d91a1a}-0.63\%$
test_items_stack_nested_locked 0.6430ms 0.4703ms 2.1262 KOps/s 2.1533 KOps/s $\color{#d91a1a}-1.26\%$
test_keys 25.9210μs 4.2246μs 236.7102 KOps/s 237.6925 KOps/s $\color{#d91a1a}-0.41\%$
test_keys_nested 0.1950ms 0.1293ms 7.7332 KOps/s 7.6910 KOps/s $\color{#35bf28}+0.55\%$
test_keys_nested_locked 0.7640ms 0.1394ms 7.1713 KOps/s 7.2463 KOps/s $\color{#d91a1a}-1.04\%$
test_keys_nested_leaf 0.1854ms 0.1199ms 8.3422 KOps/s 8.2853 KOps/s $\color{#35bf28}+0.69\%$
test_keys_stack_nested 0.2046ms 0.1297ms 7.7093 KOps/s 7.6852 KOps/s $\color{#35bf28}+0.31\%$
test_keys_stack_nested_leaf 0.1967ms 0.1198ms 8.3462 KOps/s 8.3820 KOps/s $\color{#d91a1a}-0.43\%$
test_keys_stack_nested_locked 0.2002ms 0.1383ms 7.2317 KOps/s 7.2810 KOps/s $\color{#d91a1a}-0.68\%$
test_values 9.6302μs 1.0477μs 954.4456 KOps/s 985.6962 KOps/s $\color{#d91a1a}-3.17\%$
test_values_nested 0.4597ms 52.2559μs 19.1366 KOps/s 19.0373 KOps/s $\color{#35bf28}+0.52\%$
test_values_nested_locked 0.4725ms 55.6911μs 17.9562 KOps/s 18.0410 KOps/s $\color{#d91a1a}-0.47\%$
test_values_nested_leaf 87.8620μs 59.9762μs 16.6733 KOps/s 16.7458 KOps/s $\color{#d91a1a}-0.43\%$
test_values_stack_nested 0.4808ms 51.9994μs 19.2310 KOps/s 19.0409 KOps/s $\color{#35bf28}+1.00\%$
test_values_stack_nested_leaf 0.5074ms 59.2135μs 16.8880 KOps/s 16.7015 KOps/s $\color{#35bf28}+1.12\%$
test_values_stack_nested_locked 0.4725ms 55.7372μs 17.9413 KOps/s 18.0581 KOps/s $\color{#d91a1a}-0.65\%$
test_membership 70.0313μs 0.8434μs 1.1857 MOps/s 1.1997 MOps/s $\color{#d91a1a}-1.17\%$
test_membership_nested 0.4253ms 2.9947μs 333.9211 KOps/s 340.3419 KOps/s $\color{#d91a1a}-1.89\%$
test_membership_nested_leaf 23.4400μs 2.9677μs 336.9605 KOps/s 338.8074 KOps/s $\color{#d91a1a}-0.55\%$
test_membership_stacked_nested 0.4327ms 2.8815μs 347.0361 KOps/s 349.7513 KOps/s $\color{#d91a1a}-0.78\%$
test_membership_stacked_nested_leaf 39.3210μs 2.8444μs 351.5689 KOps/s 345.8421 KOps/s $\color{#35bf28}+1.66\%$
test_membership_nested_last 0.4295ms 4.4456μs 224.9429 KOps/s 226.4306 KOps/s $\color{#d91a1a}-0.66\%$
test_membership_nested_leaf_last 27.4900μs 4.4596μs 224.2354 KOps/s 227.8663 KOps/s $\color{#d91a1a}-1.59\%$
test_membership_stacked_nested_last 29.0510μs 4.3126μs 231.8760 KOps/s 230.6311 KOps/s $\color{#35bf28}+0.54\%$
test_membership_stacked_nested_leaf_last 33.7610μs 4.3116μs 231.9314 KOps/s 232.3467 KOps/s $\color{#d91a1a}-0.18\%$
test_nested_getleaf 0.4467ms 21.8543μs 45.7575 KOps/s 46.1670 KOps/s $\color{#d91a1a}-0.89\%$
test_nested_get 0.4348ms 20.8262μs 48.0165 KOps/s 49.1362 KOps/s $\color{#d91a1a}-2.28\%$
test_stacked_getleaf 0.4453ms 21.8976μs 45.6672 KOps/s 46.4032 KOps/s $\color{#d91a1a}-1.59\%$
test_stacked_get 0.4578ms 20.5564μs 48.6468 KOps/s 48.7523 KOps/s $\color{#d91a1a}-0.22\%$
test_nested_getitemleaf 58.2010μs 22.2822μs 44.8790 KOps/s 44.8430 KOps/s $\color{#35bf28}+0.08\%$
test_nested_getitem 0.4438ms 21.0273μs 47.5573 KOps/s 47.6961 KOps/s $\color{#d91a1a}-0.29\%$
test_stacked_getitemleaf 0.4416ms 22.0828μs 45.2842 KOps/s 46.4366 KOps/s $\color{#d91a1a}-2.48\%$
test_stacked_getitem 0.4404ms 21.1832μs 47.2072 KOps/s 48.5807 KOps/s $\color{#d91a1a}-2.83\%$
test_lock_nested 8.0081ms 0.4858ms 2.0585 KOps/s 2.0818 KOps/s $\color{#d91a1a}-1.12\%$
test_lock_stack_nested 0.5780ms 0.4822ms 2.0740 KOps/s 2.0646 KOps/s $\color{#35bf28}+0.46\%$
test_unlock_nested 0.4584ms 0.3878ms 2.5787 KOps/s 2.5690 KOps/s $\color{#35bf28}+0.38\%$
test_unlock_stack_nested 0.4336ms 0.3891ms 2.5701 KOps/s 2.5449 KOps/s $\color{#35bf28}+0.99\%$
test_flatten_speed 0.1663ms 0.1214ms 8.2345 KOps/s 8.1741 KOps/s $\color{#35bf28}+0.74\%$
test_unflatten_speed 0.9928ms 0.5711ms 1.7512 KOps/s 1.7669 KOps/s $\color{#d91a1a}-0.89\%$
test_common_ops 0.1002s 1.5930ms 627.7362 Ops/s 339.4292 Ops/s $\textbf{\color{#35bf28}+84.94\%}$
test_creation 0.1212ms 3.1264μs 319.8517 KOps/s 318.9934 KOps/s $\color{#35bf28}+0.27\%$
test_creation_empty 84.6920μs 7.0338μs 142.1709 KOps/s 143.3403 KOps/s $\color{#d91a1a}-0.82\%$
test_creation_nested_1 34.3600μs 11.5072μs 86.9023 KOps/s 86.7188 KOps/s $\color{#35bf28}+0.21\%$
test_creation_nested_2 44.2910μs 13.4705μs 74.2360 KOps/s 73.9327 KOps/s $\color{#35bf28}+0.41\%$
test_creation_many_keys[10] 84.2220μs 21.3199μs 46.9046 KOps/s 47.5346 KOps/s $\color{#d91a1a}-1.33\%$
test_creation_many_keys[50] 0.1688ms 90.3603μs 11.0668 KOps/s 10.9665 KOps/s $\color{#35bf28}+0.91\%$
test_creation_many_keys[100] 0.2297ms 0.1787ms 5.5947 KOps/s 5.6004 KOps/s $\color{#d91a1a}-0.10\%$
test_creation_nested_many_keys[10] 0.1092ms 45.1548μs 22.1460 KOps/s 22.3406 KOps/s $\color{#d91a1a}-0.87\%$
test_creation_nested_many_keys[50] 0.2526ms 0.1836ms 5.4473 KOps/s 5.4135 KOps/s $\color{#35bf28}+0.63\%$
test_clone 46.7510μs 13.5673μs 73.7068 KOps/s 73.7171 KOps/s $\color{#d91a1a}-0.01\%$
test_getitem[int] 1.6187ms 15.0655μs 66.3768 KOps/s 66.3316 KOps/s $\color{#35bf28}+0.07\%$
test_getitem[slice_int] 0.1515ms 24.3793μs 41.0184 KOps/s 41.4686 KOps/s $\color{#d91a1a}-1.09\%$
test_getitem[range] 0.1761ms 65.1493μs 15.3494 KOps/s 15.2948 KOps/s $\color{#35bf28}+0.36\%$
test_getitem[tuple] 0.1361ms 23.9296μs 41.7892 KOps/s 41.3261 KOps/s $\color{#35bf28}+1.12\%$
test_getitem[list] 0.1871ms 59.6154μs 16.7742 KOps/s 16.7218 KOps/s $\color{#35bf28}+0.31\%$
test_setitem_dim[int] 44.6810μs 25.7123μs 38.8920 KOps/s 38.3765 KOps/s $\color{#35bf28}+1.34\%$
test_setitem_dim[slice_int] 62.1020μs 41.8055μs 23.9203 KOps/s 23.4863 KOps/s $\color{#35bf28}+1.85\%$
test_setitem_dim[range] 0.1214ms 96.9767μs 10.3118 KOps/s 10.3282 KOps/s $\color{#d91a1a}-0.16\%$
test_setitem_dim[tuple] 59.8010μs 39.9467μs 25.0333 KOps/s 25.3610 KOps/s $\color{#d91a1a}-1.29\%$
test_setitem 54.2710μs 17.9520μs 55.7042 KOps/s 55.5860 KOps/s $\color{#35bf28}+0.21\%$
test_set 53.0510μs 16.9338μs 59.0535 KOps/s 57.9417 KOps/s $\color{#35bf28}+1.92\%$
test_set_shared 0.5161ms 0.2035ms 4.9140 KOps/s 4.9109 KOps/s $\color{#35bf28}+0.06\%$
test_update 0.1351ms 22.1391μs 45.1690 KOps/s 44.9789 KOps/s $\color{#35bf28}+0.42\%$
test_update_nested 76.6120μs 33.5476μs 29.8084 KOps/s 29.2602 KOps/s $\color{#35bf28}+1.87\%$
test_update__nested 0.4813ms 33.5378μs 29.8171 KOps/s 29.7165 KOps/s $\color{#35bf28}+0.34\%$
test_set_nested 45.5510μs 19.1191μs 52.3038 KOps/s 51.9552 KOps/s $\color{#35bf28}+0.67\%$
test_set_nested_new 80.4320μs 24.1723μs 41.3696 KOps/s 42.8143 KOps/s $\color{#d91a1a}-3.37\%$
test_select 76.8410μs 40.7106μs 24.5636 KOps/s 24.9811 KOps/s $\color{#d91a1a}-1.67\%$
test_select_nested 0.1172ms 74.5272μs 13.4179 KOps/s 13.3822 KOps/s $\color{#35bf28}+0.27\%$
test_exclude_nested 0.1341ms 91.5470μs 10.9234 KOps/s 10.9285 KOps/s $\color{#d91a1a}-0.05\%$
test_empty[True] 0.4385ms 0.4016ms 2.4900 KOps/s 2.5117 KOps/s $\color{#d91a1a}-0.86\%$
test_empty[False] 11.4975μs 1.3811μs 724.0362 KOps/s 718.3894 KOps/s $\color{#35bf28}+0.79\%$
test_to 0.1058ms 68.4554μs 14.6080 KOps/s 14.4618 KOps/s $\color{#35bf28}+1.01\%$
test_to_nonblocking 0.1136ms 68.7986μs 14.5352 KOps/s 14.6365 KOps/s $\color{#d91a1a}-0.69\%$
test_unbind_speed 1.8886ms 0.3397ms 2.9440 KOps/s 2.9733 KOps/s $\color{#d91a1a}-0.98\%$
test_unbind_speed_stack0 0.3982ms 0.3367ms 2.9700 KOps/s 3.0089 KOps/s $\color{#d91a1a}-1.29\%$
test_unbind_speed_stack1 0.1219s 0.8590ms 1.1642 KOps/s 1.1669 KOps/s $\color{#d91a1a}-0.23\%$
test_split 1.2558ms 1.1386ms 878.2744 Ops/s 721.4850 Ops/s $\textbf{\color{#35bf28}+21.73\%}$
test_chunk 0.1148s 1.2266ms 815.2902 Ops/s 920.4819 Ops/s $\textbf{\color{#d91a1a}-11.43\%}$
test_to_cpu_blocking 20.0043ms 19.8801ms 50.3016 Ops/s 49.6268 Ops/s $\color{#35bf28}+1.36\%$
test_to_cpu_global_sync 11.8971ms 11.6893ms 85.5486 Ops/s 84.9950 Ops/s $\color{#35bf28}+0.65\%$
test_to_cpu_event_sync 13.0398ms 12.7152ms 78.6462 Ops/s 78.4282 Ops/s $\color{#35bf28}+0.28\%$
test_to_cpu_default 13.0239ms 12.6868ms 78.8219 Ops/s 69.7675 Ops/s $\textbf{\color{#35bf28}+12.98\%}$
test_consolidate[False-None] 0.1440s 4.7476ms 210.6339 Ops/s 240.1920 Ops/s $\textbf{\color{#d91a1a}-12.31\%}$
test_consolidate[default-None] 2.4750ms 2.0419ms 489.7352 Ops/s 479.0950 Ops/s $\color{#35bf28}+2.22\%$
test_consolidate[reduce-overhead-None] 2.1138ms 1.9544ms 511.6548 Ops/s 496.7667 Ops/s $\color{#35bf28}+3.00\%$
test_consolidate_njt[False-None] 8.7696ms 8.5431ms 117.0539 Ops/s 114.1743 Ops/s $\color{#35bf28}+2.52\%$
test_to[False-False-None] 2.3016ms 2.1627ms 462.3866 Ops/s 459.3013 Ops/s $\color{#35bf28}+0.67\%$
test_to[True-False-None] 2.1091ms 1.8949ms 527.7292 Ops/s 534.7328 Ops/s $\color{#d91a1a}-1.31\%$
test_to[within-False-None] 6.4704ms 6.2073ms 161.1007 Ops/s 161.5852 Ops/s $\color{#d91a1a}-0.30\%$
test_to[True-default-None] 11.1491ms 10.8167ms 92.4495 Ops/s 91.9778 Ops/s $\color{#35bf28}+0.51\%$
test_to_njt[False-False-None] 8.9634ms 8.6614ms 115.4548 Ops/s 113.9422 Ops/s $\color{#35bf28}+1.33\%$
test_to_njt[True-False-None] 7.4004ms 6.9612ms 143.6540 Ops/s 142.0034 Ops/s $\color{#35bf28}+1.16\%$
test_to_njt[within-False-None] 16.0578ms 15.5791ms 64.1886 Ops/s 62.7623 Ops/s $\color{#35bf28}+2.27\%$
test_creation[device0] 0.3897ms 0.1199ms 8.3414 KOps/s 8.4980 KOps/s $\color{#d91a1a}-1.84\%$
test_creation_from_tensor 0.5400ms 0.1162ms 8.6035 KOps/s 8.6992 KOps/s $\color{#d91a1a}-1.10\%$
test_add_one[memmap_tensor0] 0.3072ms 6.8475μs 146.0394 KOps/s 141.7874 KOps/s $\color{#35bf28}+3.00\%$
test_contiguous[memmap_tensor0] 14.3700μs 0.6516μs 1.5347 MOps/s 2.2207 MOps/s $\textbf{\color{#d91a1a}-30.89\%}$
test_stack[memmap_tensor0] 40.6010μs 4.2866μs 233.2826 KOps/s 231.8112 KOps/s $\color{#35bf28}+0.63\%$
test_memmaptd_index 1.1715ms 0.2732ms 3.6600 KOps/s 3.6343 KOps/s $\color{#35bf28}+0.71\%$
test_memmaptd_index_astensor 0.8160ms 0.3781ms 2.6446 KOps/s 2.6416 KOps/s $\color{#35bf28}+0.11\%$
test_memmaptd_index_op 1.0582ms 0.6458ms 1.5484 KOps/s 1.5179 KOps/s $\color{#35bf28}+2.01\%$
test_serialize_model 0.1385s 0.1343s 7.4442 Ops/s 7.4784 Ops/s $\color{#d91a1a}-0.46\%$
test_serialize_model_pickle 1.8285s 1.3007s 0.7688 Ops/s 0.6822 Ops/s $\textbf{\color{#35bf28}+12.69\%}$
test_serialize_weights 0.3222s 0.1605s 6.2302 Ops/s 7.4734 Ops/s $\textbf{\color{#d91a1a}-16.64\%}$
test_serialize_weights_returnearly 80.1078ms 64.2700ms 15.5594 Ops/s 6.4982 Ops/s $\textbf{\color{#35bf28}+139.44\%}$
test_serialize_weights_pickle 1.3723s 1.2132s 0.8242 Ops/s 0.8244 Ops/s $\color{#d91a1a}-0.01\%$
test_reshape_pytree 0.2069ms 32.8300μs 30.4600 KOps/s 30.6121 KOps/s $\color{#d91a1a}-0.50\%$
test_reshape_td 0.1283ms 44.4305μs 22.5071 KOps/s 22.0558 KOps/s $\color{#35bf28}+2.05\%$
test_view_pytree 0.2173ms 32.2935μs 30.9660 KOps/s 31.0559 KOps/s $\color{#d91a1a}-0.29\%$
test_view_td 95.6720μs 53.2124μs 18.7926 KOps/s 17.7855 KOps/s $\textbf{\color{#35bf28}+5.66\%}$
test_unbind_pytree 0.2454ms 36.6451μs 27.2888 KOps/s 26.8922 KOps/s $\color{#35bf28}+1.47\%$
test_unbind_td 0.1871ms 50.3480μs 19.8618 KOps/s 19.7119 KOps/s $\color{#35bf28}+0.76\%$
test_split_pytree 0.2441ms 42.9416μs 23.2874 KOps/s 23.0889 KOps/s $\color{#35bf28}+0.86\%$
test_split_td 0.1146ms 65.4879μs 15.2700 KOps/s 15.1103 KOps/s $\color{#35bf28}+1.06\%$
test_add_pytree 0.2333ms 43.0400μs 23.2342 KOps/s 23.2386 KOps/s $\color{#d91a1a}-0.02\%$
test_add_td 91.9920μs 57.6116μs 17.3576 KOps/s 17.7065 KOps/s $\color{#d91a1a}-1.97\%$
test_compile_add_one_nested[tensordict-compile] 0.3400ms 0.2326ms 4.2985 KOps/s 4.1645 KOps/s $\color{#35bf28}+3.22\%$
test_compile_add_one_nested[tensordict-eager] 0.4112ms 0.2085ms 4.7966 KOps/s 5.0158 KOps/s $\color{#d91a1a}-4.37\%$
test_compile_add_one_nested[pytree-compile] 0.2918ms 0.1338ms 7.4730 KOps/s 7.0829 KOps/s $\textbf{\color{#35bf28}+5.51\%}$
test_compile_add_one_nested[pytree-eager] 0.4185ms 0.1786ms 5.5981 KOps/s 5.5512 KOps/s $\color{#35bf28}+0.84\%$
test_compile_copy_nested[tensordict-compile] 0.1846ms 15.2097μs 65.7477 KOps/s 66.6309 KOps/s $\color{#d91a1a}-1.33\%$
test_compile_copy_nested[tensordict-eager] 0.2927ms 53.5823μs 18.6629 KOps/s 18.1523 KOps/s $\color{#35bf28}+2.81\%$
test_compile_copy_nested[pytree-compile] 0.1114ms 15.6702μs 63.8154 KOps/s 61.7636 KOps/s $\color{#35bf28}+3.32\%$
test_compile_copy_nested[pytree-eager] 0.3888ms 67.3701μs 14.8434 KOps/s 14.6858 KOps/s $\color{#35bf28}+1.07\%$
test_compile_add_one_flat[tensordict-compile] 0.2962ms 0.2465ms 4.0571 KOps/s 3.8755 KOps/s $\color{#35bf28}+4.68\%$
test_compile_add_one_flat[tensordict-eager] 0.3431ms 0.2812ms 3.5565 KOps/s 3.5634 KOps/s $\color{#d91a1a}-0.20\%$
test_compile_add_one_flat[tensorclass-compile] 0.3774ms 0.1839ms 5.4382 KOps/s 5.2490 KOps/s $\color{#35bf28}+3.60\%$
test_compile_add_one_flat[tensorclass-eager] 0.1247ms 75.8599μs 13.1822 KOps/s 13.1830 KOps/s $-0.01\%$
test_compile_add_one_flat[pytree-compile] 0.4633ms 0.1923ms 5.1991 KOps/s 5.0804 KOps/s $\color{#35bf28}+2.33\%$
test_compile_add_one_flat[pytree-eager] 0.8341ms 0.5209ms 1.9196 KOps/s 1.9086 KOps/s $\color{#35bf28}+0.58\%$
test_compile_add_self_flat[tensordict-eager] 0.4798ms 0.3281ms 3.0477 KOps/s 3.0223 KOps/s $\color{#35bf28}+0.84\%$
test_compile_add_self_flat[tensordict-compile] 0.2992ms 0.2484ms 4.0252 KOps/s 3.7251 KOps/s $\textbf{\color{#35bf28}+8.06\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1447ms 92.0834μs 10.8597 KOps/s 11.0562 KOps/s $\color{#d91a1a}-1.78\%$
test_compile_add_self_flat[tensorclass-compile] 0.4098ms 0.1890ms 5.2921 KOps/s 4.8965 KOps/s $\textbf{\color{#35bf28}+8.08\%}$
test_compile_add_self_flat[pytree-eager] 0.6403ms 0.4304ms 2.3233 KOps/s 2.3060 KOps/s $\color{#35bf28}+0.75\%$
test_compile_add_self_flat[pytree-compile] 0.4360ms 0.1927ms 5.1889 KOps/s 4.8431 KOps/s $\textbf{\color{#35bf28}+7.14\%}$
test_compile_copy_flat[tensordict-compile] 72.1420μs 19.4695μs 51.3624 KOps/s 47.9669 KOps/s $\textbf{\color{#35bf28}+7.08\%}$
test_compile_copy_flat[tensordict-eager] 73.1720μs 41.2763μs 24.2270 KOps/s 24.0845 KOps/s $\color{#35bf28}+0.59\%$
test_compile_copy_flat[pytree-compile] 0.1559ms 20.9797μs 47.6651 KOps/s 45.1640 KOps/s $\textbf{\color{#35bf28}+5.54\%}$
test_compile_copy_flat[pytree-eager] 0.3294ms 52.4462μs 19.0672 KOps/s 19.1861 KOps/s $\color{#d91a1a}-0.62\%$
test_compile_assign_and_add[tensordict-compile] 2.2215ms 0.2373ms 4.2135 KOps/s 3.7957 KOps/s $\textbf{\color{#35bf28}+11.01\%}$
test_compile_assign_and_add[tensordict-eager] 3.4181ms 3.3123ms 301.9014 Ops/s 303.7161 Ops/s $\color{#d91a1a}-0.60\%$
test_compile_assign_and_add[pytree-compile] 2.0853ms 0.1787ms 5.5953 KOps/s 5.4429 KOps/s $\color{#35bf28}+2.80\%$
test_compile_assign_and_add[pytree-eager] 2.9152ms 2.7918ms 358.1911 Ops/s 358.8006 Ops/s $\color{#d91a1a}-0.17\%$
test_compile_indexing[tensor-tensordict-compile] 0.2539ms 0.1319ms 7.5798 KOps/s 7.2412 KOps/s $\color{#35bf28}+4.68\%$
test_compile_indexing[tensor-tensordict-eager] 0.3102ms 75.0450μs 13.3253 KOps/s 13.5884 KOps/s $\color{#d91a1a}-1.94\%$
test_compile_indexing[tensor-tensorclass-compile] 0.2560ms 0.1186ms 8.4293 KOps/s 8.4148 KOps/s $\color{#35bf28}+0.17\%$
test_compile_indexing[tensor-tensorclass-eager] 0.2556ms 45.5211μs 21.9678 KOps/s 22.1032 KOps/s $\color{#d91a1a}-0.61\%$
test_compile_indexing[tensor-pytree-compile] 0.2035ms 0.1228ms 8.1445 KOps/s 8.1796 KOps/s $\color{#d91a1a}-0.43\%$
test_compile_indexing[tensor-pytree-eager] 0.2637ms 45.9864μs 21.7456 KOps/s 22.1114 KOps/s $\color{#d91a1a}-1.65\%$
test_compile_indexing[slice-tensordict-compile] 0.2189ms 83.4873μs 11.9779 KOps/s 11.5020 KOps/s $\color{#35bf28}+4.14\%$
test_compile_indexing[slice-tensordict-eager] 0.2042ms 27.8652μs 35.8871 KOps/s 36.6441 KOps/s $\color{#d91a1a}-2.07\%$
test_compile_indexing[slice-tensorclass-compile] 0.2025ms 70.4003μs 14.2045 KOps/s 13.4355 KOps/s $\textbf{\color{#35bf28}+5.72\%}$
test_compile_indexing[slice-tensorclass-eager] 0.2355ms 22.0754μs 45.2994 KOps/s 45.2265 KOps/s $\color{#35bf28}+0.16\%$
test_compile_indexing[slice-pytree-compile] 0.1125ms 71.0538μs 14.0738 KOps/s 13.4388 KOps/s $\color{#35bf28}+4.73\%$
test_compile_indexing[slice-pytree-eager] 0.2691ms 21.9556μs 45.5466 KOps/s 45.8667 KOps/s $\color{#d91a1a}-0.70\%$
test_compile_indexing[int-tensordict-compile] 0.1491ms 84.9050μs 11.7779 KOps/s 11.1729 KOps/s $\textbf{\color{#35bf28}+5.41\%}$
test_compile_indexing[int-tensordict-eager] 0.2379ms 27.5902μs 36.2448 KOps/s 36.4529 KOps/s $\color{#d91a1a}-0.57\%$
test_compile_indexing[int-tensorclass-compile] 0.1167ms 71.0517μs 14.0743 KOps/s 14.0556 KOps/s $\color{#35bf28}+0.13\%$
test_compile_indexing[int-tensorclass-eager] 0.2288ms 22.0532μs 45.3449 KOps/s 45.6507 KOps/s $\color{#d91a1a}-0.67\%$
test_compile_indexing[int-pytree-compile] 0.1150ms 70.7427μs 14.1357 KOps/s 13.8993 KOps/s $\color{#35bf28}+1.70\%$
test_compile_indexing[int-pytree-eager] 0.2701ms 22.1318μs 45.1839 KOps/s 45.2950 KOps/s $\color{#d91a1a}-0.25\%$
test_compile_replace[single-eager] 0.2121ms 47.9890μs 20.8381 KOps/s 20.7849 KOps/s $\color{#35bf28}+0.26\%$
test_compile_replace[single-compile] 0.2266ms 0.1300ms 7.6947 KOps/s 7.5798 KOps/s $\color{#35bf28}+1.52\%$
test_compile_replace[multi-eager] 0.6091ms 0.5625ms 1.7778 KOps/s 1.7759 KOps/s $\color{#35bf28}+0.11\%$
test_compile_replace[multi-compile] 0.3057ms 0.1369ms 7.3029 KOps/s 7.2011 KOps/s $\color{#35bf28}+1.41\%$
test_compile_tc_getattr_20[eager] 0.2444ms 0.1638ms 6.1034 KOps/s 6.1207 KOps/s $\color{#d91a1a}-0.28\%$
test_compile_tc_getattr_20[compile] 0.3791ms 0.1474ms 6.7848 KOps/s 6.6084 KOps/s $\color{#35bf28}+2.67\%$
test_compile_clone_shallow[20-eager] 44.9610μs 19.0914μs 52.3795 KOps/s 52.3164 KOps/s $\color{#35bf28}+0.12\%$
test_compile_clone_shallow[20-compile] 72.2720μs 17.0070μs 58.7993 KOps/s 58.3522 KOps/s $\color{#35bf28}+0.77\%$
test_compile_clone_shallow[40-eager] 63.0920μs 33.6746μs 29.6960 KOps/s 29.1265 KOps/s $\color{#35bf28}+1.96\%$
test_compile_clone_shallow[40-compile] 69.0720μs 17.9153μs 55.8181 KOps/s 54.5046 KOps/s $\color{#35bf28}+2.41\%$
test_compile_clone_shallow[80-eager] 91.6420μs 62.6298μs 15.9668 KOps/s 15.7039 KOps/s $\color{#35bf28}+1.67\%$
test_compile_clone_shallow[80-compile] 0.1195ms 21.9788μs 45.4984 KOps/s 46.1080 KOps/s $\color{#d91a1a}-1.32\%$
test_compile_update_inplace[eager] 0.2124ms 60.9889μs 16.3964 KOps/s 16.1487 KOps/s $\color{#35bf28}+1.53\%$
test_compile_update_inplace[compile] 0.3710ms 0.1454ms 6.8764 KOps/s 6.7197 KOps/s $\color{#35bf28}+2.33\%$
test_mod_add[eager] 0.3526ms 0.2264ms 4.4162 KOps/s 4.4163 KOps/s $-0.00\%$
test_mod_add[compile] 0.3213ms 0.1307ms 7.6507 KOps/s 7.3972 KOps/s $\color{#35bf28}+3.43\%$
test_mod_add[compile-overhead] 0.3112ms 0.1752ms 5.7083 KOps/s 5.5530 KOps/s $\color{#35bf28}+2.80\%$
test_mod_wrap[eager] 0.6025ms 0.4714ms 2.1213 KOps/s 2.1258 KOps/s $\color{#d91a1a}-0.21\%$
test_mod_wrap[compile] 0.4546ms 0.3937ms 2.5403 KOps/s 2.4802 KOps/s $\color{#35bf28}+2.43\%$
test_mod_wrap[compile-overhead] 8.6247ms 4.6703ms 214.1211 Ops/s 215.5298 Ops/s $\color{#d91a1a}-0.65\%$
test_mod_wrap_and_backward[eager] 2.2615ms 1.7995ms 555.7204 Ops/s 587.6498 Ops/s $\textbf{\color{#d91a1a}-5.43\%}$
test_mod_wrap_and_backward[compile] 2.4714ms 1.5977ms 625.9100 Ops/s 642.9546 Ops/s $\color{#d91a1a}-2.65\%$
test_mod_wrap_and_backward[compile-overhead] 1.3661ms 0.9432ms 1.0602 KOps/s 1.0353 KOps/s $\color{#35bf28}+2.41\%$
test_seq_add[eager] 0.5683ms 0.5065ms 1.9743 KOps/s 1.9634 KOps/s $\color{#35bf28}+0.55\%$
test_seq_add[compile] 0.2755ms 0.1456ms 6.8671 KOps/s 6.9276 KOps/s $\color{#d91a1a}-0.87\%$
test_seq_add[compile-overhead] 0.3406ms 0.1834ms 5.4525 KOps/s 5.3205 KOps/s $\color{#35bf28}+2.48\%$
test_seq_wrap[eager] 1.6057ms 1.4246ms 701.9700 Ops/s 702.8493 Ops/s $\color{#d91a1a}-0.13\%$
test_seq_wrap[compile] 0.5338ms 0.4258ms 2.3486 KOps/s 2.3709 KOps/s $\color{#d91a1a}-0.94\%$
test_seq_wrap[compile-overhead] 0.3770ms 0.3072ms 3.2553 KOps/s 3.2276 KOps/s $\color{#35bf28}+0.86\%$
test_func_call_runtime[False-eager] 0.9266ms 0.8274ms 1.2086 KOps/s 1.2188 KOps/s $\color{#d91a1a}-0.83\%$
test_func_call_runtime[False-compile] 1.1018ms 0.9867ms 1.0134 KOps/s 1.0174 KOps/s $\color{#d91a1a}-0.39\%$
test_func_call_runtime[False-compile-overhead] 0.6393ms 0.5139ms 1.9459 KOps/s 1.9185 KOps/s $\color{#35bf28}+1.43\%$
test_func_call_runtime[True-eager] 1.1528ms 1.0574ms 945.7221 Ops/s 961.0041 Ops/s $\color{#d91a1a}-1.59\%$
test_func_call_runtime[True-compile] 1.0926ms 0.9940ms 1.0060 KOps/s 997.5370 Ops/s $\color{#35bf28}+0.85\%$
test_func_call_runtime[True-compile-overhead] 0.7029ms 0.5278ms 1.8945 KOps/s 1.8653 KOps/s $\color{#35bf28}+1.57\%$
test_func_call_cm_runtime[False-eager] 0.8716ms 0.7995ms 1.2508 KOps/s 1.1889 KOps/s $\textbf{\color{#35bf28}+5.21\%}$
test_func_call_cm_runtime[False-compile] 1.0773ms 0.9656ms 1.0357 KOps/s 1.0298 KOps/s $\color{#35bf28}+0.57\%$
test_func_call_cm_runtime[False-compile-overhead] 0.5915ms 0.4941ms 2.0239 KOps/s 1.9954 KOps/s $\color{#35bf28}+1.42\%$
test_func_call_cm_runtime[True-eager] 1.3216ms 1.1892ms 840.8864 Ops/s 839.3855 Ops/s $\color{#35bf28}+0.18\%$
test_func_call_cm_runtime[True-compile] 1.0876ms 1.0160ms 984.2384 Ops/s 980.5285 Ops/s $\color{#35bf28}+0.38\%$
test_func_call_cm_runtime[True-compile-overhead] 0.6140ms 0.5445ms 1.8365 KOps/s 1.8161 KOps/s $\color{#35bf28}+1.12\%$
test_vmap_func_call_cm_runtime[eager] 2.8537ms 2.3964ms 417.2945 Ops/s 418.7817 Ops/s $\color{#d91a1a}-0.36\%$
test_vmap_func_call_cm_runtime[compile] 1.2429ms 1.0395ms 961.9578 Ops/s 972.4932 Ops/s $\color{#d91a1a}-1.08\%$
test_vmap_func_call_cm_runtime[compile-overhead] 0.5912ms 0.5408ms 1.8490 KOps/s 1.8269 KOps/s $\color{#35bf28}+1.21\%$
test_distributed 0.4675ms 0.1473ms 6.7906 KOps/s 6.5317 KOps/s $\color{#35bf28}+3.96\%$
test_tdmodule 0.2429ms 0.1925ms 5.1952 KOps/s 5.0706 KOps/s $\color{#35bf28}+2.46\%$
test_tdmodule_dispatch 0.5906ms 0.3710ms 2.6955 KOps/s 2.6746 KOps/s $\color{#35bf28}+0.78\%$
test_tdseq 44.1910μs 26.1015μs 38.3120 KOps/s 37.5868 KOps/s $\color{#35bf28}+1.93\%$
test_tdseq_dispatch 0.2548ms 0.2155ms 4.6411 KOps/s 4.6054 KOps/s $\color{#35bf28}+0.78\%$
test_instantiation_functorch 2.1564ms 2.0791ms 480.9807 Ops/s 479.8739 Ops/s $\color{#35bf28}+0.23\%$
test_exec_functorch 0.2300ms 0.1829ms 5.4678 KOps/s 5.4237 KOps/s $\color{#35bf28}+0.81\%$
test_exec_functional_call 0.2254ms 0.1639ms 6.1016 KOps/s 6.0770 KOps/s $\color{#35bf28}+0.40\%$
test_exec_td_decorator 0.4557ms 0.2402ms 4.1625 KOps/s 4.1822 KOps/s $\color{#d91a1a}-0.47\%$
test_vmap_mlp_speed_decorator[True-True] 1.2050ms 1.0135ms 986.7190 Ops/s 987.2571 Ops/s $\color{#d91a1a}-0.05\%$
test_vmap_mlp_speed_decorator[True-False] 1.2078ms 1.0111ms 988.9870 Ops/s 987.3552 Ops/s $\color{#35bf28}+0.17\%$
test_vmap_mlp_speed_decorator[False-True] 0.9209ms 0.7223ms 1.3846 KOps/s 1.4028 KOps/s $\color{#d91a1a}-1.30\%$
test_vmap_mlp_speed_decorator[False-False] 0.9244ms 0.7209ms 1.3871 KOps/s 1.4062 KOps/s $\color{#d91a1a}-1.36\%$
test_vmap_transformer_speed_decorator[True-True] 21.6842ms 21.0874ms 47.4218 Ops/s 47.9753 Ops/s $\color{#d91a1a}-1.15\%$
test_vmap_transformer_speed_decorator[True-False] 21.4670ms 20.9437ms 47.7470 Ops/s 48.1212 Ops/s $\color{#d91a1a}-0.78\%$
test_vmap_transformer_speed_decorator[False-True] 21.2782ms 20.6935ms 48.3243 Ops/s 48.8061 Ops/s $\color{#d91a1a}-0.99\%$
test_vmap_transformer_speed_decorator[False-False] 21.2337ms 20.6719ms 48.3748 Ops/s 48.9250 Ops/s $\color{#d91a1a}-1.12\%$
test_to_module_speed[True] 2.2525ms 1.4785ms 676.3723 Ops/s 671.4678 Ops/s $\color{#35bf28}+0.73\%$
test_to_module_speed[False] 1.5390ms 1.4313ms 698.6665 Ops/s 692.7697 Ops/s $\color{#35bf28}+0.85\%$
test_tc_init 67.4010μs 42.6866μs 23.4266 KOps/s 23.1296 KOps/s $\color{#35bf28}+1.28\%$
test_tc_init_tensor_only 29.6410μs 9.0071μs 111.0232 KOps/s 110.3810 KOps/s $\color{#35bf28}+0.58\%$
test_tc_init_nested 0.1218ms 84.3200μs 11.8596 KOps/s 11.5721 KOps/s $\color{#35bf28}+2.48\%$
test_tc_init_many_fields 37.7710μs 14.8142μs 67.5028 KOps/s 67.0037 KOps/s $\color{#35bf28}+0.74\%$
test_tc_first_layer_tensor 28.0500μs 1.8175μs 550.2137 KOps/s 553.1388 KOps/s $\color{#d91a1a}-0.53\%$
test_tc_first_layer_tensor_only 1.8181μs 0.3967μs 2.5207 MOps/s 2.4828 MOps/s $\color{#35bf28}+1.53\%$
test_tc_first_layer_tensor_set 20.9300μs 3.9401μs 253.7991 KOps/s 252.2069 KOps/s $\color{#35bf28}+0.63\%$
test_tc_first_layer_tensor_only_set 18.4710μs 3.2819μs 304.6983 KOps/s 301.1154 KOps/s $\color{#35bf28}+1.19\%$
test_tc_first_layer_nontensor 32.4710μs 6.2479μs 160.0534 KOps/s 161.8605 KOps/s $\color{#d91a1a}-1.12\%$
test_tc_second_layer_tensor 27.5200μs 4.3826μs 228.1742 KOps/s 230.1691 KOps/s $\color{#d91a1a}-0.87\%$
test_tc_second_layer_nontensor 38.7500μs 8.7840μs 113.8433 KOps/s 114.7795 KOps/s $\color{#d91a1a}-0.82\%$
test_unbind 0.2809s 16.5459ms 60.4379 Ops/s 69.9081 Ops/s $\textbf{\color{#d91a1a}-13.55\%}$
test_full_like 4.5631ms 4.3962ms 227.4715 Ops/s 229.6209 Ops/s $\color{#d91a1a}-0.94\%$
test_zeros_like 4.5326ms 4.3800ms 228.3131 Ops/s 231.9214 Ops/s $\color{#d91a1a}-1.56\%$
test_ones_like 4.5554ms 4.3899ms 227.7981 Ops/s 231.3961 Ops/s $\color{#d91a1a}-1.55\%$
test_clone 6.8149ms 6.5727ms 152.1452 Ops/s 151.3756 Ops/s $\color{#35bf28}+0.51\%$
test_squeeze 61.2410μs 14.2833μs 70.0120 KOps/s 71.7559 KOps/s $\color{#d91a1a}-2.43\%$
test_unsqueeze 0.1607ms 0.1119ms 8.9392 KOps/s 8.8724 KOps/s $\color{#35bf28}+0.75\%$
test_split 0.3870ms 0.1818ms 5.5012 KOps/s 5.5049 KOps/s $\color{#d91a1a}-0.07\%$
test_permute 0.3271ms 0.2090ms 4.7836 KOps/s 4.8422 KOps/s $\color{#d91a1a}-1.21\%$
test_stack 27.4592ms 26.7532ms 37.3787 Ops/s 19.3979 Ops/s $\textbf{\color{#35bf28}+92.69\%}$
test_cat 27.4271ms 26.8542ms 37.2381 Ops/s 19.4222 Ops/s $\textbf{\color{#35bf28}+91.73\%}$
test_sequential_tensordict 0.6253ms 0.5722ms 1.7475 KOps/s 1.7457 KOps/s $\color{#35bf28}+0.10\%$
test_sequential_graph_module 0.1736ms 0.1229ms 8.1361 KOps/s 8.2697 KOps/s $\color{#d91a1a}-1.62\%$
test_nested_tensordict 0.9139ms 0.8314ms 1.2028 KOps/s 1.2219 KOps/s $\color{#d91a1a}-1.56\%$
test_nested_graph_module 0.1990ms 0.1377ms 7.2618 KOps/s 7.6815 KOps/s $\textbf{\color{#d91a1a}-5.46\%}$

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

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