Skip to content

[BugFix] Fix CI breakage from torch.squeeze and dynamo changes - #1738

Merged
vmoens merged 1 commit into
mainfrom
fix/torch-compat-squeeze-compile
Jul 9, 2026
Merged

[BugFix] Fix CI breakage from torch.squeeze and dynamo changes#1738
vmoens merged 1 commit into
mainfrom
fix/torch-compat-squeeze-compile

Conversation

@vmoens

@vmoens vmoens commented Jul 9, 2026

Copy link
Copy Markdown
Collaborator

Description

Main's CI went red on 2026-07-09 (e.g. this run) after recent torch releases; both stable and nightly jobs fail with the same 49 tests. Two independent root causes, both fixed here:

1. torch.squeeze no longer accepts dim=None (40 failures)

test_squeeze_with_none and test_squeeze_with_none_legacy called torch.squeeze(td, dim=None), which current torch rejects at the argument-parser level even for plain tensors. The tests now call torch.squeeze(td), which has identical semantics on every torch version.

Dropping the explicit dim=None exposed a real library inconsistency: _UnsqueezedTensorDict._legacy_squeeze was missing the dim=None default that TensorDictBase._legacy_squeeze and LazyStackedTensorDict._legacy_squeeze already have, so td.squeeze() with no argument raised a TypeError on that class in legacy mode. Fixed by adding the default.

2. dynamo recursion in tensorclass __setattr__ (9 failures)

The tensorclass machinery stores cls.__setattr_parent__ = object.__setattr__ and calls self.__setattr_parent__(key, value) to perform raw attribute writes. Current dynamo re-dispatches a slot wrapper stored as a class attribute through the type's custom __setattr__, so _setattr recursed into itself infinitely under torch.compile (InternalTorchDynamoError: RecursionError: maximum recursion depth exceeded). The call sites now invoke object.__setattr__(self, key, value) directly, which dynamo special-cases correctly and which is equivalent in eager mode. The __setattr_parent__ class attribute is kept for backward compatibility.

Testing

All 49 previously failing tests pass locally on torch 2.14.0.dev20260708: the full squeeze family (95 tests), the 9 compile tests, plus full runs of test/tensordict/test_methods.py (6441 passed), test/compile/test_compile.py, test/tensorclass, test/nn, and test/tensordict/test_generic.py. The single remaining local test_compile.py failure (TestTTDDynamoCompatibility::test_td_subclass_arithmetic_without_registration) pre-exists on main unchanged and does not appear in the Linux CI failure set.

Generated with Claude Code

Recent torch releases (stable and nightly) broke main's CI in two
independent ways:

- torch.squeeze no longer accepts dim=None; call torch.squeeze(td)
  in the tests instead (equivalent semantics on all torch versions).
  This exposed a missing dim default in
  _UnsqueezedTensorDict._legacy_squeeze, which the base class and
  LazyStackedTensorDict already have.
- dynamo now re-dispatches a slot wrapper stored as a class attribute
  (cls.__setattr_parent__ = object.__setattr__) through the custom
  __setattr__, so tensorclass _setattr recursed infinitely under
  torch.compile (InternalTorchDynamoError: RecursionError). Call
  object.__setattr__(self, ...) directly instead; dynamo handles that
  form. The __setattr_parent__ class attribute is kept for backward
  compatibility.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
@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 Jul 9, 2026
@github-actions github-actions Bot added bug Something isn't working Test tensorclass labels Jul 9, 2026
@github-actions

github-actions Bot commented Jul 9, 2026

Copy link
Copy Markdown
Contributor

$\color{#D29922}\textsf{\Large&amp;#x26A0;\kern{0.2cm}\normalsize Warning}$ Result of CPU Benchmark Tests

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

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_collector_preallocated_write[stack_out-cpu-32-2] 0.2258ms 0.1716ms 5.8268 KOps/s 5.9796 KOps/s $\color{#d91a1a}-2.56\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.3628ms 0.3017ms 3.3146 KOps/s 3.3917 KOps/s $\color{#d91a1a}-2.27\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.8823ms 0.7766ms 1.2876 KOps/s 1.2994 KOps/s $\color{#d91a1a}-0.91\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.3855ms 0.3397ms 2.9437 KOps/s 3.0216 KOps/s $\color{#d91a1a}-2.58\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.2878ms 1.1460ms 872.6181 Ops/s 924.2077 Ops/s $\textbf{\color{#d91a1a}-5.58\%}$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.7208ms 7.3852ms 135.4065 Ops/s 137.1701 Ops/s $\color{#d91a1a}-1.29\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.2838ms 0.2520ms 3.9678 KOps/s 4.0374 KOps/s $\color{#d91a1a}-1.72\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.5421ms 0.3935ms 2.5410 KOps/s 2.7161 KOps/s $\textbf{\color{#d91a1a}-6.45\%}$
test_collector_preallocated_write[stack_out-cuda-32-32] 0.9082ms 0.8267ms 1.2096 KOps/s 1.2237 KOps/s $\color{#d91a1a}-1.15\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.3410ms 0.2663ms 3.7552 KOps/s 4.0643 KOps/s $\textbf{\color{#d91a1a}-7.61\%}$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.4669ms 0.3928ms 2.5458 KOps/s 2.7097 KOps/s $\textbf{\color{#d91a1a}-6.05\%}$
test_collector_preallocated_write[stack_out-cuda-4096-32] 0.9939ms 0.8847ms 1.1303 KOps/s 1.1430 KOps/s $\color{#d91a1a}-1.11\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2741ms 0.2188ms 4.5707 KOps/s 4.7465 KOps/s $\color{#d91a1a}-3.70\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 1.0428ms 0.8358ms 1.1964 KOps/s 1.2356 KOps/s $\color{#d91a1a}-3.17\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.5914ms 3.3251ms 300.7425 Ops/s 315.0856 Ops/s $\color{#d91a1a}-4.55\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.4380ms 0.3827ms 2.6127 KOps/s 2.7686 KOps/s $\textbf{\color{#d91a1a}-5.63\%}$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 1.8708ms 1.7350ms 576.3679 Ops/s 629.4853 Ops/s $\textbf{\color{#d91a1a}-8.44\%}$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 11.4965ms 10.2148ms 97.8973 Ops/s 99.8423 Ops/s $\color{#d91a1a}-1.95\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.4665ms 0.4102ms 2.4380 KOps/s 2.4842 KOps/s $\color{#d91a1a}-1.86\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.6454ms 1.5637ms 639.5145 Ops/s 668.3652 Ops/s $\color{#d91a1a}-4.32\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 6.2906ms 6.1217ms 163.3539 Ops/s 167.5197 Ops/s $\color{#d91a1a}-2.49\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.5066ms 0.4079ms 2.4516 KOps/s 2.4520 KOps/s $\color{#d91a1a}-0.02\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.6808ms 1.5823ms 631.9922 Ops/s 684.4466 Ops/s $\textbf{\color{#d91a1a}-7.66\%}$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 6.7169ms 6.1863ms 161.6462 Ops/s 166.3232 Ops/s $\color{#d91a1a}-2.81\%$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.6002ms 0.1705ms 5.8637 KOps/s 5.8831 KOps/s $\color{#d91a1a}-0.33\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.7165ms 0.6661ms 1.5013 KOps/s 1.5162 KOps/s $\color{#d91a1a}-0.98\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 2.7172ms 2.6461ms 377.9157 Ops/s 382.6439 Ops/s $\color{#d91a1a}-1.24\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.4238ms 0.3190ms 3.1344 KOps/s 3.1246 KOps/s $\color{#35bf28}+0.31\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.6870ms 1.4551ms 687.2388 Ops/s 692.4047 Ops/s $\color{#d91a1a}-0.75\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 11.3459ms 9.6425ms 103.7074 Ops/s 105.7533 Ops/s $\color{#d91a1a}-1.93\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.3934ms 0.3211ms 3.1145 KOps/s 3.1763 KOps/s $\color{#d91a1a}-1.95\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.2835ms 1.1952ms 836.6850 Ops/s 855.5349 Ops/s $\color{#d91a1a}-2.20\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.8701ms 4.6863ms 213.3864 Ops/s 218.3614 Ops/s $\color{#d91a1a}-2.28\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.4085ms 0.3240ms 3.0869 KOps/s 3.1663 KOps/s $\color{#d91a1a}-2.51\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.2661ms 1.2048ms 830.0149 Ops/s 849.0579 Ops/s $\color{#d91a1a}-2.24\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 4.9198ms 4.7402ms 210.9610 Ops/s 216.6304 Ops/s $\color{#d91a1a}-2.62\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.1773ms 0.1217ms 8.2198 KOps/s 8.6773 KOps/s $\textbf{\color{#d91a1a}-5.27\%}$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.5655ms 0.4757ms 2.1022 KOps/s 2.2113 KOps/s $\color{#d91a1a}-4.93\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 1.9761ms 1.9030ms 525.4766 Ops/s 557.0945 Ops/s $\textbf{\color{#d91a1a}-5.68\%}$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.3504ms 0.2718ms 3.6787 KOps/s 3.8157 KOps/s $\color{#d91a1a}-3.59\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.4636ms 1.2806ms 780.8745 Ops/s 818.4294 Ops/s $\color{#d91a1a}-4.59\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 8.8188ms 8.5325ms 117.1992 Ops/s 117.8057 Ops/s $\color{#d91a1a}-0.51\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.3828ms 0.2973ms 3.3636 KOps/s 3.5468 KOps/s $\textbf{\color{#d91a1a}-5.17\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.1587ms 1.1030ms 906.6581 Ops/s 953.0303 Ops/s $\color{#d91a1a}-4.87\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 4.4564ms 4.3241ms 231.2615 Ops/s 248.9245 Ops/s $\textbf{\color{#d91a1a}-7.10\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.3702ms 0.2973ms 3.3634 KOps/s 3.6824 KOps/s $\textbf{\color{#d91a1a}-8.66\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.1961ms 1.1042ms 905.6414 Ops/s 1.0127 KOps/s $\textbf{\color{#d91a1a}-10.57\%}$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 4.4939ms 4.3295ms 230.9724 Ops/s 246.5537 Ops/s $\textbf{\color{#d91a1a}-6.32\%}$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.4117ms 16.0463ms 62.3196 Ops/s 62.3935 Ops/s $\color{#d91a1a}-0.12\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 32.8539ms 32.6657ms 30.6131 Ops/s 30.8453 Ops/s $\color{#d91a1a}-0.75\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 89.3057ms 85.6144ms 11.6803 Ops/s 11.5258 Ops/s $\color{#35bf28}+1.34\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.1979s 0.1972s 5.0699 Ops/s 4.9089 Ops/s $\color{#35bf28}+3.28\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 13.7478ms 13.5752ms 73.6636 Ops/s 73.9005 Ops/s $\color{#d91a1a}-0.32\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 26.7273ms 26.4721ms 37.7756 Ops/s 38.1031 Ops/s $\color{#d91a1a}-0.86\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 14.0639ms 13.7688ms 72.6281 Ops/s 73.2099 Ops/s $\color{#d91a1a}-0.79\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 27.3217ms 26.7658ms 37.3612 Ops/s 37.2480 Ops/s $\color{#35bf28}+0.30\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 52.3172ms 52.0010ms 19.2304 Ops/s 19.8952 Ops/s $\color{#d91a1a}-3.34\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1042s 0.1039s 9.6273 Ops/s 9.9135 Ops/s $\color{#d91a1a}-2.89\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1458s 0.1453s 6.8807 Ops/s 6.6922 Ops/s $\color{#35bf28}+2.82\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3272s 0.3251s 3.0758 Ops/s 3.0266 Ops/s $\color{#35bf28}+1.63\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 79.8734ms 79.0076ms 12.6570 Ops/s 13.7383 Ops/s $\textbf{\color{#d91a1a}-7.87\%}$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1597s 0.1589s 6.2943 Ops/s 6.6194 Ops/s $\color{#d91a1a}-4.91\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 80.5627ms 80.1209ms 12.4811 Ops/s 13.0140 Ops/s $\color{#d91a1a}-4.09\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1615s 0.1607s 6.2235 Ops/s 6.4429 Ops/s $\color{#d91a1a}-3.40\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 30.1222ms 29.9977ms 33.3359 Ops/s 33.8150 Ops/s $\color{#d91a1a}-1.42\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 60.5761ms 60.4027ms 16.5556 Ops/s 16.7416 Ops/s $\color{#d91a1a}-1.11\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1200s 0.1181s 8.4648 Ops/s 8.5621 Ops/s $\color{#d91a1a}-1.14\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2678s 0.2665s 3.7521 Ops/s 3.8309 Ops/s $\color{#d91a1a}-2.06\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 43.3093ms 43.0016ms 23.2549 Ops/s 23.8505 Ops/s $\color{#d91a1a}-2.50\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 86.4153ms 85.9614ms 11.6331 Ops/s 11.8116 Ops/s $\color{#d91a1a}-1.51\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 43.6268ms 43.2242ms 23.1352 Ops/s 23.5863 Ops/s $\color{#d91a1a}-1.91\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 87.5054ms 86.7751ms 11.5240 Ops/s 11.7497 Ops/s $\color{#d91a1a}-1.92\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 20.9300ms 20.7938ms 48.0913 Ops/s 51.1267 Ops/s $\textbf{\color{#d91a1a}-5.94\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 42.6177ms 42.4644ms 23.5492 Ops/s 25.0300 Ops/s $\textbf{\color{#d91a1a}-5.92\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1086s 0.1070s 9.3466 Ops/s 9.5119 Ops/s $\color{#d91a1a}-1.74\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2484s 0.2434s 4.1082 Ops/s 4.2389 Ops/s $\color{#d91a1a}-3.08\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 39.7101ms 37.6879ms 26.5337 Ops/s 28.4753 Ops/s $\textbf{\color{#d91a1a}-6.82\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 79.4609ms 78.5566ms 12.7297 Ops/s 13.7913 Ops/s $\textbf{\color{#d91a1a}-7.70\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 39.9985ms 39.4433ms 25.3528 Ops/s 28.2294 Ops/s $\textbf{\color{#d91a1a}-10.19\%}$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 79.3981ms 78.6612ms 12.7127 Ops/s 13.5534 Ops/s $\textbf{\color{#d91a1a}-6.20\%}$
test_plain_set_nested 43.0010μs 15.0417μs 66.4818 KOps/s 66.7704 KOps/s $\color{#d91a1a}-0.43\%$
test_plain_set_stack_nested 55.1620μs 14.7843μs 67.6392 KOps/s 68.4356 KOps/s $\color{#d91a1a}-1.16\%$
test_plain_set_nested_inplace 74.6220μs 16.2394μs 61.5788 KOps/s 61.2799 KOps/s $\color{#35bf28}+0.49\%$
test_plain_set_stack_nested_inplace 0.1151ms 15.6605μs 63.8549 KOps/s 62.4683 KOps/s $\color{#35bf28}+2.22\%$
test_items 24.6210μs 5.5284μs 180.8841 KOps/s 180.2662 KOps/s $\color{#35bf28}+0.34\%$
test_items_nested 0.5242ms 0.4506ms 2.2195 KOps/s 2.2190 KOps/s $\color{#35bf28}+0.02\%$
test_items_nested_locked 0.5191ms 0.4557ms 2.1946 KOps/s 2.1828 KOps/s $\color{#35bf28}+0.54\%$
test_items_nested_leaf 0.1476ms 95.1144μs 10.5137 KOps/s 10.5546 KOps/s $\color{#d91a1a}-0.39\%$
test_items_stack_nested 0.5381ms 0.4518ms 2.2136 KOps/s 2.1981 KOps/s $\color{#35bf28}+0.71\%$
test_items_stack_nested_leaf 0.1451ms 94.6463μs 10.5656 KOps/s 10.4720 KOps/s $\color{#35bf28}+0.89\%$
test_items_stack_nested_locked 0.5254ms 0.4557ms 2.1946 KOps/s 2.2433 KOps/s $\color{#d91a1a}-2.17\%$
test_keys 43.0510μs 4.1488μs 241.0323 KOps/s 241.2258 KOps/s $\color{#d91a1a}-0.08\%$
test_keys_nested 0.1619ms 0.1273ms 7.8536 KOps/s 7.7484 KOps/s $\color{#35bf28}+1.36\%$
test_keys_nested_locked 0.7780ms 0.1354ms 7.3862 KOps/s 7.2851 KOps/s $\color{#35bf28}+1.39\%$
test_keys_nested_leaf 0.1664ms 0.1182ms 8.4595 KOps/s 8.3374 KOps/s $\color{#35bf28}+1.46\%$
test_keys_stack_nested 0.1960ms 0.1270ms 7.8722 KOps/s 7.7501 KOps/s $\color{#35bf28}+1.57\%$
test_keys_stack_nested_leaf 0.1704ms 0.1175ms 8.5122 KOps/s 8.4770 KOps/s $\color{#35bf28}+0.41\%$
test_keys_stack_nested_locked 0.1895ms 0.1347ms 7.4222 KOps/s 7.3665 KOps/s $\color{#35bf28}+0.76\%$
test_values 6.9780μs 0.9996μs 1.0004 MOps/s 998.2585 KOps/s $\color{#35bf28}+0.21\%$
test_values_nested 0.1048ms 51.4884μs 19.4219 KOps/s 19.1841 KOps/s $\color{#35bf28}+1.24\%$
test_values_nested_locked 94.8820μs 54.5878μs 18.3191 KOps/s 18.1662 KOps/s $\color{#35bf28}+0.84\%$
test_values_nested_leaf 0.1023ms 59.3110μs 16.8603 KOps/s 16.4980 KOps/s $\color{#35bf28}+2.20\%$
test_values_stack_nested 0.1032ms 51.3825μs 19.4619 KOps/s 19.0554 KOps/s $\color{#35bf28}+2.13\%$
test_values_stack_nested_leaf 99.0620μs 59.0733μs 16.9281 KOps/s 16.5950 KOps/s $\color{#35bf28}+2.01\%$
test_values_stack_nested_locked 0.1183ms 54.6712μs 18.2912 KOps/s 17.9964 KOps/s $\color{#35bf28}+1.64\%$
test_membership 13.4920μs 0.8065μs 1.2399 MOps/s 1.2298 MOps/s $\color{#35bf28}+0.82\%$
test_membership_nested 52.9020μs 2.8712μs 348.2815 KOps/s 348.4179 KOps/s $\color{#d91a1a}-0.04\%$
test_membership_nested_leaf 70.9020μs 2.8961μs 345.2876 KOps/s 361.9308 KOps/s $\color{#d91a1a}-4.60\%$
test_membership_stacked_nested 50.7510μs 2.7994μs 357.2184 KOps/s 362.9079 KOps/s $\color{#d91a1a}-1.57\%$
test_membership_stacked_nested_leaf 89.5220μs 2.6701μs 374.5140 KOps/s 361.9187 KOps/s $\color{#35bf28}+3.48\%$
test_membership_nested_last 34.6200μs 4.3645μs 229.1205 KOps/s 229.5882 KOps/s $\color{#d91a1a}-0.20\%$
test_membership_nested_leaf_last 31.9100μs 4.3847μs 228.0642 KOps/s 230.1617 KOps/s $\color{#d91a1a}-0.91\%$
test_membership_stacked_nested_last 47.9610μs 4.1919μs 238.5553 KOps/s 240.0597 KOps/s $\color{#d91a1a}-0.63\%$
test_membership_stacked_nested_leaf_last 0.1038ms 4.0348μs 247.8412 KOps/s 241.8268 KOps/s $\color{#35bf28}+2.49\%$
test_nested_getleaf 94.1820μs 20.2070μs 49.4877 KOps/s 50.2183 KOps/s $\color{#d91a1a}-1.45\%$
test_nested_get 59.4220μs 18.9374μs 52.8055 KOps/s 50.6050 KOps/s $\color{#35bf28}+4.35\%$
test_stacked_getleaf 76.1020μs 19.9805μs 50.0488 KOps/s 48.6164 KOps/s $\color{#35bf28}+2.95\%$
test_stacked_get 58.2410μs 18.9277μs 52.8326 KOps/s 50.3444 KOps/s $\color{#35bf28}+4.94\%$
test_nested_getitemleaf 81.7820μs 21.1315μs 47.3228 KOps/s 46.4075 KOps/s $\color{#35bf28}+1.97\%$
test_nested_getitem 94.8320μs 20.3405μs 49.1630 KOps/s 49.1703 KOps/s $\color{#d91a1a}-0.01\%$
test_stacked_getitemleaf 80.2520μs 20.8097μs 48.0546 KOps/s 47.4142 KOps/s $\color{#35bf28}+1.35\%$
test_stacked_getitem 71.3720μs 19.6850μs 50.8000 KOps/s 49.8378 KOps/s $\color{#35bf28}+1.93\%$
test_lock_nested 7.9286ms 0.5030ms 1.9880 KOps/s 2.0282 KOps/s $\color{#d91a1a}-1.98\%$
test_lock_stack_nested 0.7048ms 0.5024ms 1.9906 KOps/s 2.0305 KOps/s $\color{#d91a1a}-1.96\%$
test_unlock_nested 0.4741ms 0.3932ms 2.5434 KOps/s 2.5793 KOps/s $\color{#d91a1a}-1.39\%$
test_unlock_stack_nested 0.5803ms 0.3991ms 2.5056 KOps/s 2.5593 KOps/s $\color{#d91a1a}-2.10\%$
test_flatten_speed 0.2080ms 0.1181ms 8.4654 KOps/s 8.2925 KOps/s $\color{#35bf28}+2.09\%$
test_unflatten_speed 0.6578ms 0.5455ms 1.8331 KOps/s 1.7985 KOps/s $\color{#35bf28}+1.92\%$
test_common_ops 0.8925ms 0.6988ms 1.4310 KOps/s 1.4539 KOps/s $\color{#d91a1a}-1.58\%$
test_creation 0.1204ms 3.0930μs 323.3110 KOps/s 322.0560 KOps/s $\color{#35bf28}+0.39\%$
test_creation_empty 46.1210μs 6.8491μs 146.0041 KOps/s 146.8671 KOps/s $\color{#d91a1a}-0.59\%$
test_creation_nested_1 53.1910μs 11.3004μs 88.4925 KOps/s 88.6718 KOps/s $\color{#d91a1a}-0.20\%$
test_creation_nested_2 71.5610μs 12.7899μs 78.1868 KOps/s 75.6392 KOps/s $\color{#35bf28}+3.37\%$
test_creation_many_keys[10] 91.6820μs 20.4337μs 48.9388 KOps/s 50.3539 KOps/s $\color{#d91a1a}-2.81\%$
test_creation_many_keys[50] 0.1326ms 88.3561μs 11.3178 KOps/s 11.4337 KOps/s $\color{#d91a1a}-1.01\%$
test_creation_many_keys[100] 0.2559ms 0.1748ms 5.7192 KOps/s 5.8273 KOps/s $\color{#d91a1a}-1.85\%$
test_creation_nested_many_keys[10] 80.1920μs 44.4036μs 22.5207 KOps/s 22.5230 KOps/s $\color{#d91a1a}-0.01\%$
test_creation_nested_many_keys[50] 0.2253ms 0.1814ms 5.5130 KOps/s 5.5528 KOps/s $\color{#d91a1a}-0.72\%$
test_clone 41.8810μs 13.0625μs 76.5552 KOps/s 78.6055 KOps/s $\color{#d91a1a}-2.61\%$
test_getitem[int] 1.5257ms 15.3507μs 65.1438 KOps/s 66.9493 KOps/s $\color{#d91a1a}-2.70\%$
test_getitem[slice_int] 0.1412ms 25.1074μs 39.8289 KOps/s 41.0657 KOps/s $\color{#d91a1a}-3.01\%$
test_getitem[range] 0.1709ms 62.0810μs 16.1080 KOps/s 16.3179 KOps/s $\color{#d91a1a}-1.29\%$
test_getitem[tuple] 0.1459ms 24.1675μs 41.3780 KOps/s 42.7686 KOps/s $\color{#d91a1a}-3.25\%$
test_getitem[list] 0.1869ms 58.4215μs 17.1170 KOps/s 17.8020 KOps/s $\color{#d91a1a}-3.85\%$
test_setitem_dim[int] 56.1120μs 25.3686μs 39.4189 KOps/s 40.7370 KOps/s $\color{#d91a1a}-3.24\%$
test_setitem_dim[slice_int] 81.4520μs 42.1697μs 23.7137 KOps/s 24.5634 KOps/s $\color{#d91a1a}-3.46\%$
test_setitem_dim[range] 0.1506ms 96.6282μs 10.3489 KOps/s 10.8133 KOps/s $\color{#d91a1a}-4.29\%$
test_setitem_dim[tuple] 63.6010μs 40.0923μs 24.9424 KOps/s 25.7850 KOps/s $\color{#d91a1a}-3.27\%$
test_setitem 73.8420μs 17.3762μs 57.5499 KOps/s 59.3279 KOps/s $\color{#d91a1a}-3.00\%$
test_set 59.8310μs 16.6005μs 60.2390 KOps/s 61.4711 KOps/s $\color{#d91a1a}-2.00\%$
test_set_shared 0.5311ms 0.2116ms 4.7262 KOps/s 4.9224 KOps/s $\color{#d91a1a}-3.99\%$
test_update 0.3878ms 21.6406μs 46.2095 KOps/s 48.3696 KOps/s $\color{#d91a1a}-4.47\%$
test_update_nested 78.8420μs 32.6593μs 30.6191 KOps/s 30.2602 KOps/s $\color{#35bf28}+1.19\%$
test_update__nested 0.4678ms 32.8317μs 30.4584 KOps/s 31.0455 KOps/s $\color{#d91a1a}-1.89\%$
test_set_nested 51.3310μs 18.2070μs 54.9239 KOps/s 57.8045 KOps/s $\color{#d91a1a}-4.98\%$
test_set_nested_new 56.6720μs 22.9287μs 43.6134 KOps/s 44.3979 KOps/s $\color{#d91a1a}-1.77\%$
test_select 88.7920μs 39.3510μs 25.4123 KOps/s 25.8747 KOps/s $\color{#d91a1a}-1.79\%$
test_select_nested 0.1406ms 72.1534μs 13.8594 KOps/s 13.8248 KOps/s $\color{#35bf28}+0.25\%$
test_exclude_nested 0.1410ms 88.1964μs 11.3383 KOps/s 11.6543 KOps/s $\color{#d91a1a}-2.71\%$
test_empty[True] 0.4239ms 0.3844ms 2.6014 KOps/s 2.5981 KOps/s $\color{#35bf28}+0.13\%$
test_empty[False] 20.5555μs 1.3723μs 728.7213 KOps/s 737.5943 KOps/s $\color{#d91a1a}-1.20\%$
test_to 0.1674ms 72.9308μs 13.7116 KOps/s 14.6507 KOps/s $\textbf{\color{#d91a1a}-6.41\%}$
test_to_nonblocking 0.1308ms 69.7228μs 14.3425 KOps/s 14.7455 KOps/s $\color{#d91a1a}-2.73\%$
test_unbind_speed 0.3934ms 0.3315ms 3.0163 KOps/s 3.0346 KOps/s $\color{#d91a1a}-0.60\%$
test_unbind_speed_stack0 0.4947ms 0.3307ms 3.0243 KOps/s 3.0624 KOps/s $\color{#d91a1a}-1.24\%$
test_unbind_speed_stack1 0.1157s 0.8382ms 1.1930 KOps/s 915.4151 Ops/s $\textbf{\color{#35bf28}+30.32\%}$
test_split 0.1160s 1.4198ms 704.3299 Ops/s 789.0804 Ops/s $\textbf{\color{#d91a1a}-10.74\%}$
test_chunk 1.3326ms 1.1989ms 834.0662 Ops/s 822.4250 Ops/s $\color{#35bf28}+1.42\%$
test_to_cpu_blocking 20.7702ms 19.8220ms 50.4491 Ops/s 52.8636 Ops/s $\color{#d91a1a}-4.57\%$
test_to_cpu_global_sync 0.1282s 13.2556ms 75.4398 Ops/s 88.0967 Ops/s $\textbf{\color{#d91a1a}-14.37\%}$
test_to_cpu_event_sync 13.4640ms 12.8024ms 78.1105 Ops/s 81.8809 Ops/s $\color{#d91a1a}-4.60\%$
test_to_cpu_default 13.2996ms 12.7860ms 78.2103 Ops/s 82.0535 Ops/s $\color{#d91a1a}-4.68\%$
test_consolidate[False-None] 4.3398ms 4.1315ms 242.0403 Ops/s 246.9013 Ops/s $\color{#d91a1a}-1.97\%$
test_consolidate[default-None] 3.3117ms 2.3270ms 429.7419 Ops/s 447.6800 Ops/s $\color{#d91a1a}-4.01\%$
test_consolidate[reduce-overhead-None] 3.3071ms 2.2386ms 446.7072 Ops/s 464.5347 Ops/s $\color{#d91a1a}-3.84\%$
test_consolidate_njt[False-None] 5.6512ms 5.1009ms 196.0439 Ops/s 196.6273 Ops/s $\color{#d91a1a}-0.30\%$
test_to[False-False-None] 2.3827ms 2.1830ms 458.0756 Ops/s 464.3660 Ops/s $\color{#d91a1a}-1.35\%$
test_to[True-False-None] 2.1402ms 1.9799ms 505.0662 Ops/s 529.5242 Ops/s $\color{#d91a1a}-4.62\%$
test_to[within-False-None] 6.5624ms 6.2583ms 159.7887 Ops/s 165.2191 Ops/s $\color{#d91a1a}-3.29\%$
test_to[True-default-None] 10.6021ms 10.4065ms 96.0940 Ops/s 97.0803 Ops/s $\color{#d91a1a}-1.02\%$
test_to_njt[False-False-None] 7.0723ms 6.9607ms 143.6643 Ops/s 143.8565 Ops/s $\color{#d91a1a}-0.13\%$
test_to_njt[True-False-None] 4.0616ms 3.8585ms 259.1648 Ops/s 258.8952 Ops/s $\color{#35bf28}+0.10\%$
test_to_njt[within-False-None] 9.2276ms 9.0582ms 110.3974 Ops/s 110.5148 Ops/s $\color{#d91a1a}-0.11\%$
test_creation[device0] 0.4127ms 0.1152ms 8.6835 KOps/s 8.7903 KOps/s $\color{#d91a1a}-1.22\%$
test_creation_from_tensor 0.4333ms 0.1133ms 8.8238 KOps/s 8.9207 KOps/s $\color{#d91a1a}-1.09\%$
test_add_one[memmap_tensor0] 0.3554ms 6.4541μs 154.9391 KOps/s 165.3632 KOps/s $\textbf{\color{#d91a1a}-6.30\%}$
test_contiguous[memmap_tensor0] 24.3110μs 0.6641μs 1.5057 MOps/s 2.1350 MOps/s $\textbf{\color{#d91a1a}-29.47\%}$
test_stack[memmap_tensor0] 81.9920μs 4.2378μs 235.9732 KOps/s 250.4462 KOps/s $\textbf{\color{#d91a1a}-5.78\%}$
test_memmaptd_index 1.2574ms 0.2773ms 3.6065 KOps/s 3.7257 KOps/s $\color{#d91a1a}-3.20\%$
test_memmaptd_index_astensor 0.5311ms 0.3778ms 2.6469 KOps/s 2.7131 KOps/s $\color{#d91a1a}-2.44\%$
test_memmaptd_index_op 0.9975ms 0.6351ms 1.5745 KOps/s 1.6381 KOps/s $\color{#d91a1a}-3.88\%$
test_serialize_model 0.1394s 0.1356s 7.3769 Ops/s 7.4017 Ops/s $\color{#d91a1a}-0.33\%$
test_serialize_model_pickle 1.3473s 1.2099s 0.8265 Ops/s 0.8254 Ops/s $\color{#35bf28}+0.13\%$
test_serialize_weights 0.3240s 0.1626s 6.1494 Ops/s 7.4762 Ops/s $\textbf{\color{#d91a1a}-17.75\%}$
test_serialize_weights_returnearly 71.0103ms 62.6889ms 15.9518 Ops/s 15.0964 Ops/s $\textbf{\color{#35bf28}+5.67\%}$
test_serialize_weights_pickle 1.3495s 1.2132s 0.8243 Ops/s 0.8228 Ops/s $\color{#35bf28}+0.18\%$
test_reshape_pytree 0.2150ms 32.1303μs 31.1233 KOps/s 31.6521 KOps/s $\color{#d91a1a}-1.67\%$
test_reshape_td 84.7920μs 45.0354μs 22.2048 KOps/s 22.3129 KOps/s $\color{#d91a1a}-0.48\%$
test_view_pytree 0.2190ms 31.7097μs 31.5361 KOps/s 32.1295 KOps/s $\color{#d91a1a}-1.85\%$
test_view_td 0.1006ms 54.1151μs 18.4791 KOps/s 18.3671 KOps/s $\color{#35bf28}+0.61\%$
test_unbind_pytree 0.2380ms 35.5483μs 28.1308 KOps/s 28.3260 KOps/s $\color{#d91a1a}-0.69\%$
test_unbind_td 0.1016ms 50.3400μs 19.8649 KOps/s 20.3935 KOps/s $\color{#d91a1a}-2.59\%$
test_split_pytree 0.2604ms 41.2716μs 24.2297 KOps/s 24.5160 KOps/s $\color{#d91a1a}-1.17\%$
test_split_td 0.1284ms 71.5755μs 13.9713 KOps/s 14.0060 KOps/s $\color{#d91a1a}-0.25\%$
test_add_pytree 0.2236ms 40.2217μs 24.8622 KOps/s 25.5486 KOps/s $\color{#d91a1a}-2.69\%$
test_add_td 91.0720μs 53.7932μs 18.5897 KOps/s 18.6748 KOps/s $\color{#d91a1a}-0.46\%$
test_compile_add_one_nested[tensordict-compile] 0.3744ms 0.2726ms 3.6682 KOps/s 3.8289 KOps/s $\color{#d91a1a}-4.20\%$
test_compile_add_one_nested[tensordict-eager] 0.4366ms 0.2011ms 4.9737 KOps/s 4.9136 KOps/s $\color{#35bf28}+1.22\%$
test_compile_add_one_nested[pytree-compile] 0.4298ms 0.1572ms 6.3593 KOps/s 6.3773 KOps/s $\color{#d91a1a}-0.28\%$
test_compile_add_one_nested[pytree-eager] 0.4880ms 0.1771ms 5.6460 KOps/s 5.6483 KOps/s $\color{#d91a1a}-0.04\%$
test_compile_copy_nested[tensordict-compile] 0.1195ms 23.2760μs 42.9626 KOps/s 48.1883 KOps/s $\textbf{\color{#d91a1a}-10.84\%}$
test_compile_copy_nested[tensordict-eager] 0.1254ms 52.0422μs 19.2152 KOps/s 18.6033 KOps/s $\color{#35bf28}+3.29\%$
test_compile_copy_nested[pytree-compile] 0.1837ms 20.1863μs 49.5385 KOps/s 52.9490 KOps/s $\textbf{\color{#d91a1a}-6.44\%}$
test_compile_copy_nested[pytree-eager] 0.3528ms 65.3686μs 15.2979 KOps/s 15.1374 KOps/s $\color{#35bf28}+1.06\%$
test_compile_add_one_flat[tensordict-compile] 0.3725ms 0.3033ms 3.2971 KOps/s 3.1675 KOps/s $\color{#35bf28}+4.09\%$
test_compile_add_one_flat[tensordict-eager] 0.3393ms 0.2867ms 3.4876 KOps/s 3.3424 KOps/s $\color{#35bf28}+4.35\%$
test_compile_add_one_flat[tensorclass-compile] 0.3030ms 0.2082ms 4.8028 KOps/s 4.5452 KOps/s $\textbf{\color{#35bf28}+5.67\%}$
test_compile_add_one_flat[tensorclass-eager] 0.1696ms 75.3950μs 13.2635 KOps/s 13.0332 KOps/s $\color{#35bf28}+1.77\%$
test_compile_add_one_flat[pytree-compile] 0.5495ms 0.2398ms 4.1693 KOps/s 4.0580 KOps/s $\color{#35bf28}+2.74\%$
test_compile_add_one_flat[pytree-eager] 0.7987ms 0.5299ms 1.8871 KOps/s 1.9148 KOps/s $\color{#d91a1a}-1.45\%$
test_compile_add_self_flat[tensordict-eager] 0.4147ms 0.3393ms 2.9470 KOps/s 2.8986 KOps/s $\color{#35bf28}+1.67\%$
test_compile_add_self_flat[tensordict-compile] 0.5717ms 0.3059ms 3.2687 KOps/s 3.1103 KOps/s $\textbf{\color{#35bf28}+5.09\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1609ms 92.3284μs 10.8309 KOps/s 10.9868 KOps/s $\color{#d91a1a}-1.42\%$
test_compile_add_self_flat[tensorclass-compile] 0.3651ms 0.2132ms 4.6910 KOps/s 4.5653 KOps/s $\color{#35bf28}+2.75\%$
test_compile_add_self_flat[pytree-eager] 0.7212ms 0.4418ms 2.2633 KOps/s 2.3390 KOps/s $\color{#d91a1a}-3.24\%$
test_compile_add_self_flat[pytree-compile] 0.6506ms 0.2414ms 4.1433 KOps/s 4.0793 KOps/s $\color{#35bf28}+1.57\%$
test_compile_copy_flat[tensordict-compile] 90.4420μs 27.5619μs 36.2820 KOps/s 37.2264 KOps/s $\color{#d91a1a}-2.54\%$
test_compile_copy_flat[tensordict-eager] 91.5720μs 39.7265μs 25.1721 KOps/s 24.6628 KOps/s $\color{#35bf28}+2.07\%$
test_compile_copy_flat[pytree-compile] 0.2362ms 26.3230μs 37.9897 KOps/s 38.9448 KOps/s $\color{#d91a1a}-2.45\%$
test_compile_copy_flat[pytree-eager] 0.3335ms 50.7081μs 19.7207 KOps/s 19.6989 KOps/s $\color{#35bf28}+0.11\%$
test_compile_assign_and_add[tensordict-compile] 2.1296ms 0.2511ms 3.9831 KOps/s 3.6057 KOps/s $\textbf{\color{#35bf28}+10.47\%}$
test_compile_assign_and_add[tensordict-eager] 3.7087ms 3.3664ms 297.0569 Ops/s 299.9770 Ops/s $\color{#d91a1a}-0.97\%$
test_compile_assign_and_add[pytree-compile] 1.9753ms 0.1929ms 5.1827 KOps/s 5.0881 KOps/s $\color{#35bf28}+1.86\%$
test_compile_assign_and_add[pytree-eager] 3.0256ms 2.7978ms 357.4276 Ops/s 368.7954 Ops/s $\color{#d91a1a}-3.08\%$
test_compile_indexing[tensor-tensordict-compile] 0.3130ms 0.1571ms 6.3672 KOps/s 6.2064 KOps/s $\color{#35bf28}+2.59\%$
test_compile_indexing[tensor-tensordict-eager] 0.2269s 97.2627μs 10.2814 KOps/s 13.0735 KOps/s $\textbf{\color{#d91a1a}-21.36\%}$
test_compile_indexing[tensor-tensorclass-compile] 0.2962ms 0.1360ms 7.3512 KOps/s 7.4540 KOps/s $\color{#d91a1a}-1.38\%$
test_compile_indexing[tensor-tensorclass-eager] 0.2993ms 44.6857μs 22.3785 KOps/s 22.8628 KOps/s $\color{#d91a1a}-2.12\%$
test_compile_indexing[tensor-pytree-compile] 0.2346ms 0.1386ms 7.2150 KOps/s 7.1308 KOps/s $\color{#35bf28}+1.18\%$
test_compile_indexing[tensor-pytree-eager] 0.2773ms 44.7273μs 22.3577 KOps/s 21.9268 KOps/s $\color{#35bf28}+1.97\%$
test_compile_indexing[slice-tensordict-compile] 0.1879ms 96.4485μs 10.3682 KOps/s 11.4893 KOps/s $\textbf{\color{#d91a1a}-9.76\%}$
test_compile_indexing[slice-tensordict-eager] 0.2238ms 27.7248μs 36.0688 KOps/s 36.4824 KOps/s $\color{#d91a1a}-1.13\%$
test_compile_indexing[slice-tensorclass-compile] 0.1844ms 73.5929μs 13.5883 KOps/s 14.3271 KOps/s $\textbf{\color{#d91a1a}-5.16\%}$
test_compile_indexing[slice-tensorclass-eager] 0.2610ms 22.2830μs 44.8773 KOps/s 46.0702 KOps/s $\color{#d91a1a}-2.59\%$
test_compile_indexing[slice-pytree-compile] 0.1348ms 73.8886μs 13.5339 KOps/s 14.1673 KOps/s $\color{#d91a1a}-4.47\%$
test_compile_indexing[slice-pytree-eager] 0.2571ms 22.3831μs 44.6765 KOps/s 46.4440 KOps/s $\color{#d91a1a}-3.81\%$
test_compile_indexing[int-tensordict-compile] 0.2149ms 90.4303μs 11.0582 KOps/s 11.1287 KOps/s $\color{#d91a1a}-0.63\%$
test_compile_indexing[int-tensordict-eager] 0.2115ms 27.8480μs 35.9093 KOps/s 35.0750 KOps/s $\color{#35bf28}+2.38\%$
test_compile_indexing[int-tensorclass-compile] 0.1175ms 73.1917μs 13.6628 KOps/s 14.1990 KOps/s $\color{#d91a1a}-3.78\%$
test_compile_indexing[int-tensorclass-eager] 0.2757ms 22.3541μs 44.7345 KOps/s 46.6525 KOps/s $\color{#d91a1a}-4.11\%$
test_compile_indexing[int-pytree-compile] 0.1150ms 73.2765μs 13.6469 KOps/s 14.2749 KOps/s $\color{#d91a1a}-4.40\%$
test_compile_indexing[int-pytree-eager] 0.2746ms 22.3781μs 44.6865 KOps/s 46.4519 KOps/s $\color{#d91a1a}-3.80\%$
test_compile_replace[single-eager] 0.1104ms 48.2702μs 20.7167 KOps/s 21.3433 KOps/s $\color{#d91a1a}-2.94\%$
test_compile_replace[single-compile] 0.2540ms 0.1449ms 6.8993 KOps/s 6.8986 KOps/s $\color{#35bf28}+0.01\%$
test_compile_replace[multi-eager] 0.6735ms 0.5656ms 1.7682 KOps/s 1.8015 KOps/s $\color{#d91a1a}-1.85\%$
test_compile_replace[multi-compile] 0.2871ms 0.1559ms 6.4123 KOps/s 6.4544 KOps/s $\color{#d91a1a}-0.65\%$
test_compile_tc_getattr_20[eager] 0.2322ms 0.1752ms 5.7066 KOps/s 5.8294 KOps/s $\color{#d91a1a}-2.11\%$
test_compile_tc_getattr_20[compile] 0.3741ms 0.1749ms 5.7185 KOps/s 5.7204 KOps/s $\color{#d91a1a}-0.03\%$
test_compile_clone_shallow[20-eager] 48.2310μs 18.8706μs 52.9925 KOps/s 52.6748 KOps/s $\color{#35bf28}+0.60\%$
test_compile_clone_shallow[20-compile] 90.6420μs 24.4249μs 40.9418 KOps/s 43.8041 KOps/s $\textbf{\color{#d91a1a}-6.53\%}$
test_compile_clone_shallow[40-eager] 71.1620μs 32.3842μs 30.8792 KOps/s 30.2467 KOps/s $\color{#35bf28}+2.09\%$
test_compile_clone_shallow[40-compile] 89.0420μs 26.1088μs 38.3013 KOps/s 40.7200 KOps/s $\textbf{\color{#d91a1a}-5.94\%}$
test_compile_clone_shallow[80-eager] 98.1320μs 60.1197μs 16.6335 KOps/s 16.3018 KOps/s $\color{#35bf28}+2.03\%$
test_compile_clone_shallow[80-compile] 0.1397ms 30.1314μs 33.1880 KOps/s 35.6658 KOps/s $\textbf{\color{#d91a1a}-6.95\%}$
test_compile_update_inplace[eager] 0.2360ms 61.7103μs 16.2048 KOps/s 16.2893 KOps/s $\color{#d91a1a}-0.52\%$
test_compile_update_inplace[compile] 0.4228ms 0.1194ms 8.3785 KOps/s 7.8479 KOps/s $\textbf{\color{#35bf28}+6.76\%}$
test_mod_add[eager] 0.2943ms 0.2250ms 4.4445 KOps/s 4.4503 KOps/s $\color{#d91a1a}-0.13\%$
test_mod_add[compile] 0.3034ms 0.1206ms 8.2890 KOps/s 8.2827 KOps/s $\color{#35bf28}+0.08\%$
test_mod_add[compile-overhead] 0.3390ms 0.1965ms 5.0891 KOps/s 4.9530 KOps/s $\color{#35bf28}+2.75\%$
test_mod_wrap[eager] 0.5542ms 0.4678ms 2.1378 KOps/s 2.1131 KOps/s $\color{#35bf28}+1.17\%$
test_mod_wrap[compile] 0.4537ms 0.3521ms 2.8399 KOps/s 2.9174 KOps/s $\color{#d91a1a}-2.66\%$
test_mod_wrap[compile-overhead] 7.1107ms 3.9991ms 250.0535 Ops/s 250.3029 Ops/s $\color{#d91a1a}-0.10\%$
test_mod_wrap_and_backward[eager] 1.8452ms 1.6871ms 592.7454 Ops/s 606.9798 Ops/s $\color{#d91a1a}-2.35\%$
test_mod_wrap_and_backward[compile] 2.4221ms 1.3639ms 733.2043 Ops/s 755.8591 Ops/s $\color{#d91a1a}-3.00\%$
test_mod_wrap_and_backward[compile-overhead] 1.5632ms 0.9509ms 1.0516 KOps/s 1.0503 KOps/s $\color{#35bf28}+0.13\%$
test_seq_add[eager] 0.6018ms 0.5313ms 1.8821 KOps/s 1.9923 KOps/s $\textbf{\color{#d91a1a}-5.53\%}$
test_seq_add[compile] 0.2794ms 0.1390ms 7.1957 KOps/s 7.4942 KOps/s $\color{#d91a1a}-3.98\%$
test_seq_add[compile-overhead] 0.3674ms 0.2101ms 4.7586 KOps/s 4.8320 KOps/s $\color{#d91a1a}-1.52\%$
test_seq_wrap[eager] 5.4607ms 1.4614ms 684.2791 Ops/s 715.2053 Ops/s $\color{#d91a1a}-4.32\%$
test_seq_wrap[compile] 0.4772ms 0.3785ms 2.6422 KOps/s 2.6839 KOps/s $\color{#d91a1a}-1.55\%$
test_seq_wrap[compile-overhead] 0.4513ms 0.3206ms 3.1189 KOps/s 3.0523 KOps/s $\color{#35bf28}+2.18\%$
test_func_call_runtime[False-eager] 0.9236ms 0.8241ms 1.2134 KOps/s 1.2546 KOps/s $\color{#d91a1a}-3.28\%$
test_func_call_runtime[False-compile] 0.8861ms 0.7612ms 1.3136 KOps/s 1.3145 KOps/s $\color{#d91a1a}-0.07\%$
test_func_call_runtime[False-compile-overhead] 0.6677ms 0.5044ms 1.9824 KOps/s 1.9652 KOps/s $\color{#35bf28}+0.88\%$
test_func_call_runtime[True-eager] 1.1919ms 1.0582ms 944.9943 Ops/s 953.6883 Ops/s $\color{#d91a1a}-0.91\%$
test_func_call_runtime[True-compile] 0.9189ms 0.7903ms 1.2653 KOps/s 1.3033 KOps/s $\color{#d91a1a}-2.92\%$
test_func_call_runtime[True-compile-overhead] 0.6892ms 0.5205ms 1.9212 KOps/s 1.9126 KOps/s $\color{#35bf28}+0.45\%$
test_func_call_cm_runtime[False-eager] 0.9114ms 0.7992ms 1.2512 KOps/s 1.1893 KOps/s $\textbf{\color{#35bf28}+5.20\%}$
test_func_call_cm_runtime[False-compile] 0.9017ms 0.7439ms 1.3443 KOps/s 1.3447 KOps/s $\color{#d91a1a}-0.03\%$
test_func_call_cm_runtime[False-compile-overhead] 0.6323ms 0.4831ms 2.0701 KOps/s 2.0413 KOps/s $\color{#35bf28}+1.41\%$
test_func_call_cm_runtime[True-eager] 1.3129ms 1.2042ms 830.3980 Ops/s 816.3635 Ops/s $\color{#35bf28}+1.72\%$
test_func_call_cm_runtime[True-compile] 0.9056ms 0.8470ms 1.1806 KOps/s 1.1761 KOps/s $\color{#35bf28}+0.38\%$
test_func_call_cm_runtime[True-compile-overhead] 1.1312ms 0.8670ms 1.1534 KOps/s 1.1460 KOps/s $\color{#35bf28}+0.64\%$
test_vmap_func_call_cm_runtime[eager] 3.1176ms 2.6445ms 378.1499 Ops/s 377.8182 Ops/s $\color{#35bf28}+0.09\%$
test_vmap_func_call_cm_runtime[compile] 4.2015ms 3.6400ms 274.7241 Ops/s 275.0944 Ops/s $\color{#d91a1a}-0.13\%$
test_vmap_func_call_cm_runtime[compile-overhead] 4.1818ms 3.6373ms 274.9259 Ops/s 276.1017 Ops/s $\color{#d91a1a}-0.43\%$
test_distributed 1.7920ms 0.1532ms 6.5290 KOps/s 6.9867 KOps/s $\textbf{\color{#d91a1a}-6.55\%}$
test_tdmodule 0.8701ms 0.1901ms 5.2607 KOps/s 5.1803 KOps/s $\color{#35bf28}+1.55\%$
test_tdmodule_dispatch 0.4082ms 0.3700ms 2.7030 KOps/s 2.6953 KOps/s $\color{#35bf28}+0.29\%$
test_tdseq 45.5810μs 26.1338μs 38.2646 KOps/s 37.3816 KOps/s $\color{#35bf28}+2.36\%$
test_tdseq_dispatch 0.2852ms 0.2189ms 4.5693 KOps/s 4.6324 KOps/s $\color{#d91a1a}-1.36\%$
test_instantiation_functorch 2.1297ms 2.0368ms 490.9736 Ops/s 496.0275 Ops/s $\color{#d91a1a}-1.02\%$
test_exec_functorch 0.2389ms 0.1750ms 5.7140 KOps/s 5.7925 KOps/s $\color{#d91a1a}-1.35\%$
test_exec_functional_call 0.4520ms 0.1542ms 6.4837 KOps/s 6.5261 KOps/s $\color{#d91a1a}-0.65\%$
test_exec_td_decorator 0.5083ms 0.2433ms 4.1101 KOps/s 4.1878 KOps/s $\color{#d91a1a}-1.85\%$
test_vmap_mlp_speed_decorator[True-True] 1.3131ms 1.1216ms 891.6158 Ops/s 901.5051 Ops/s $\color{#d91a1a}-1.10\%$
test_vmap_mlp_speed_decorator[True-False] 1.3279ms 1.1146ms 897.1509 Ops/s 906.4150 Ops/s $\color{#d91a1a}-1.02\%$
test_vmap_mlp_speed_decorator[False-True] 1.0487ms 0.8174ms 1.2234 KOps/s 1.2304 KOps/s $\color{#d91a1a}-0.57\%$
test_vmap_mlp_speed_decorator[False-False] 1.0475ms 0.8157ms 1.2259 KOps/s 1.2298 KOps/s $\color{#d91a1a}-0.31\%$
test_vmap_transformer_speed_decorator[True-True] 26.1534ms 24.8122ms 40.3028 Ops/s 40.6002 Ops/s $\color{#d91a1a}-0.73\%$
test_vmap_transformer_speed_decorator[True-False] 24.9987ms 24.6694ms 40.5361 Ops/s 40.7605 Ops/s $\color{#d91a1a}-0.55\%$
test_vmap_transformer_speed_decorator[False-True] 24.5098ms 24.3435ms 41.0787 Ops/s 41.2391 Ops/s $\color{#d91a1a}-0.39\%$
test_vmap_transformer_speed_decorator[False-False] 24.6431ms 24.2788ms 41.1881 Ops/s 41.3889 Ops/s $\color{#d91a1a}-0.49\%$
test_to_module_speed[True] 1.7723ms 1.6056ms 622.8285 Ops/s 618.1857 Ops/s $\color{#35bf28}+0.75\%$
test_to_module_speed[False] 1.6564ms 1.5683ms 637.6256 Ops/s 627.5912 Ops/s $\color{#35bf28}+1.60\%$
test_to_module_plain_tensor_speed[None] 1.8104ms 1.6805ms 595.0657 Ops/s 579.4612 Ops/s $\color{#35bf28}+2.69\%$
test_to_module_plain_tensor_speed[False] 1.6576ms 1.5468ms 646.4815 Ops/s 637.9864 Ops/s $\color{#35bf28}+1.33\%$
test_to_module_plain_tensor_speed[True] 3.3904ms 3.2994ms 303.0830 Ops/s 297.8372 Ops/s $\color{#35bf28}+1.76\%$
test_tc_init 78.7510μs 43.6831μs 22.8922 KOps/s 22.8067 KOps/s $\color{#35bf28}+0.37\%$
test_tc_init_tensor_only 36.6410μs 9.2338μs 108.2975 KOps/s 107.6091 KOps/s $\color{#35bf28}+0.64\%$
test_tc_init_nested 0.1404ms 87.2005μs 11.4678 KOps/s 11.3823 KOps/s $\color{#35bf28}+0.75\%$
test_tc_init_many_fields 52.0210μs 16.9276μs 59.0750 KOps/s 59.0272 KOps/s $\color{#35bf28}+0.08\%$
test_tc_first_layer_tensor 22.4810μs 1.7709μs 564.6743 KOps/s 574.4616 KOps/s $\color{#d91a1a}-1.70\%$
test_tc_first_layer_tensor_only 3.7316μs 0.3920μs 2.5509 MOps/s 2.5474 MOps/s $\color{#35bf28}+0.14\%$
test_tc_first_layer_tensor_set 55.7520μs 4.1612μs 240.3157 KOps/s 243.7797 KOps/s $\color{#d91a1a}-1.42\%$
test_tc_first_layer_tensor_only_set 25.4410μs 3.4249μs 291.9791 KOps/s 294.5996 KOps/s $\color{#d91a1a}-0.89\%$
test_tc_first_layer_nontensor 37.4300μs 5.9645μs 167.6576 KOps/s 168.0723 KOps/s $\color{#d91a1a}-0.25\%$
test_tc_second_layer_tensor 34.3410μs 4.3005μs 232.5326 KOps/s 237.1293 KOps/s $\color{#d91a1a}-1.94\%$
test_tc_second_layer_nontensor 74.9120μs 8.4503μs 118.3390 KOps/s 119.5837 KOps/s $\color{#d91a1a}-1.04\%$
test_unbind 0.2879s 16.4270ms 60.8752 Ops/s 65.1749 Ops/s $\textbf{\color{#d91a1a}-6.60\%}$
test_full_like 4.5076ms 4.3755ms 228.5455 Ops/s 229.0102 Ops/s $\color{#d91a1a}-0.20\%$
test_zeros_like 4.5205ms 4.3642ms 229.1364 Ops/s 229.0254 Ops/s $\color{#35bf28}+0.05\%$
test_ones_like 5.0132ms 4.2853ms 233.3541 Ops/s 229.2059 Ops/s $\color{#35bf28}+1.81\%$
test_clone 6.9126ms 6.4872ms 154.1486 Ops/s 154.7675 Ops/s $\color{#d91a1a}-0.40\%$
test_squeeze 73.1620μs 13.9448μs 71.7112 KOps/s 71.3813 KOps/s $\color{#35bf28}+0.46\%$
test_unsqueeze 0.1771ms 0.1080ms 9.2590 KOps/s 8.9359 KOps/s $\color{#35bf28}+3.62\%$
test_split 0.3955ms 0.1899ms 5.2657 KOps/s 5.3461 KOps/s $\color{#d91a1a}-1.50\%$
test_permute 0.2540ms 0.1923ms 5.2006 KOps/s 5.2173 KOps/s $\color{#d91a1a}-0.32\%$
test_stack 51.2222ms 50.8740ms 19.6564 Ops/s 37.6447 Ops/s $\textbf{\color{#d91a1a}-47.78\%}$
test_cat 51.4524ms 50.9227ms 19.6376 Ops/s 37.6902 Ops/s $\textbf{\color{#d91a1a}-47.90\%}$
test_sequential_tensordict 0.6323ms 0.5626ms 1.7775 KOps/s 1.7700 KOps/s $\color{#35bf28}+0.42\%$
test_sequential_graph_module 0.1788ms 0.1182ms 8.4634 KOps/s 8.8009 KOps/s $\color{#d91a1a}-3.83\%$
test_nested_tensordict 0.9187ms 0.8159ms 1.2256 KOps/s 1.2426 KOps/s $\color{#d91a1a}-1.37\%$
test_nested_graph_module 0.2096ms 0.1275ms 7.8457 KOps/s 8.3339 KOps/s $\textbf{\color{#d91a1a}-5.86\%}$

@github-actions

github-actions Bot commented Jul 9, 2026

Copy link
Copy Markdown
Contributor

$\color{#D29922}\textsf{\Large&amp;#x26A0;\kern{0.2cm}\normalsize Warning}$ Result of GPU Benchmark Tests

Total Benchmarks: 344. Improved: $\large\color{#35bf28}14$. 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.2026ms 0.1638ms 6.1063 KOps/s 6.1118 KOps/s $\color{#d91a1a}-0.09\%$
test_collector_preallocated_write[stack_out-cpu-32-8] 0.3501ms 0.2918ms 3.4265 KOps/s 3.3826 KOps/s $\color{#35bf28}+1.30\%$
test_collector_preallocated_write[stack_out-cpu-32-32] 0.9522ms 0.7756ms 1.2893 KOps/s 1.2607 KOps/s $\color{#35bf28}+2.27\%$
test_collector_preallocated_write[stack_out-cpu-4096-2] 0.3937ms 0.3393ms 2.9471 KOps/s 2.8880 KOps/s $\color{#35bf28}+2.05\%$
test_collector_preallocated_write[stack_out-cpu-4096-8] 1.2720ms 1.1725ms 852.8733 Ops/s 897.8922 Ops/s $\textbf{\color{#d91a1a}-5.01\%}$
test_collector_preallocated_write[stack_out-cpu-4096-32] 7.5411ms 7.3024ms 136.9422 Ops/s 135.5724 Ops/s $\color{#35bf28}+1.01\%$
test_collector_preallocated_write[stack_out-cuda-32-2] 0.3054ms 0.2437ms 4.1038 KOps/s 4.0297 KOps/s $\color{#35bf28}+1.84\%$
test_collector_preallocated_write[stack_out-cuda-32-8] 0.5273ms 0.3644ms 2.7441 KOps/s 2.6711 KOps/s $\color{#35bf28}+2.73\%$
test_collector_preallocated_write[stack_out-cuda-32-32] 0.9570ms 0.8196ms 1.2202 KOps/s 1.2003 KOps/s $\color{#35bf28}+1.66\%$
test_collector_preallocated_write[stack_out-cuda-4096-2] 0.3701ms 0.2429ms 4.1163 KOps/s 3.8403 KOps/s $\textbf{\color{#35bf28}+7.19\%}$
test_collector_preallocated_write[stack_out-cuda-4096-8] 0.4216ms 0.3648ms 2.7411 KOps/s 2.6870 KOps/s $\color{#35bf28}+2.02\%$
test_collector_preallocated_write[stack_out-cuda-4096-32] 0.9676ms 0.8812ms 1.1348 KOps/s 1.1188 KOps/s $\color{#35bf28}+1.43\%$
test_collector_preallocated_write[set_at_loop-cpu-32-2] 0.2573ms 0.2034ms 4.9169 KOps/s 4.7695 KOps/s $\color{#35bf28}+3.09\%$
test_collector_preallocated_write[set_at_loop-cpu-32-8] 0.8875ms 0.7875ms 1.2699 KOps/s 1.2370 KOps/s $\color{#35bf28}+2.66\%$
test_collector_preallocated_write[set_at_loop-cpu-32-32] 3.2653ms 3.1201ms 320.5010 Ops/s 310.5712 Ops/s $\color{#35bf28}+3.20\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-2] 0.5102ms 0.3563ms 2.8065 KOps/s 2.7687 KOps/s $\color{#35bf28}+1.37\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-8] 1.9098ms 1.6641ms 600.9212 Ops/s 606.3159 Ops/s $\color{#d91a1a}-0.89\%$
test_collector_preallocated_write[set_at_loop-cpu-4096-32] 10.1947ms 9.9962ms 100.0381 Ops/s 99.7095 Ops/s $\color{#35bf28}+0.33\%$
test_collector_preallocated_write[set_at_loop-cuda-32-2] 0.4396ms 0.3934ms 2.5421 KOps/s 2.6217 KOps/s $\color{#d91a1a}-3.04\%$
test_collector_preallocated_write[set_at_loop-cuda-32-8] 1.5263ms 1.4267ms 700.8980 Ops/s 697.3675 Ops/s $\color{#35bf28}+0.51\%$
test_collector_preallocated_write[set_at_loop-cuda-32-32] 5.8513ms 5.5229ms 181.0640 Ops/s 175.7247 Ops/s $\color{#35bf28}+3.04\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-2] 0.4385ms 0.3759ms 2.6603 KOps/s 2.6119 KOps/s $\color{#35bf28}+1.85\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-8] 1.5573ms 1.4072ms 710.6226 Ops/s 697.3773 Ops/s $\color{#35bf28}+1.90\%$
test_collector_preallocated_write[set_at_loop-cuda-4096-32] 6.0467ms 5.6137ms 178.1361 Ops/s 178.2748 Ops/s $\color{#d91a1a}-0.08\%$
test_collector_preallocated_write[copy_at-cpu-32-2] 0.6147ms 0.1676ms 5.9658 KOps/s 5.8865 KOps/s $\color{#35bf28}+1.35\%$
test_collector_preallocated_write[copy_at-cpu-32-8] 0.7429ms 0.6448ms 1.5509 KOps/s 1.5339 KOps/s $\color{#35bf28}+1.11\%$
test_collector_preallocated_write[copy_at-cpu-32-32] 2.6601ms 2.5544ms 391.4885 Ops/s 386.1273 Ops/s $\color{#35bf28}+1.39\%$
test_collector_preallocated_write[copy_at-cpu-4096-2] 0.3898ms 0.3075ms 3.2526 KOps/s 3.2185 KOps/s $\color{#35bf28}+1.06\%$
test_collector_preallocated_write[copy_at-cpu-4096-8] 1.6799ms 1.4566ms 686.5438 Ops/s 687.0702 Ops/s $\color{#d91a1a}-0.08\%$
test_collector_preallocated_write[copy_at-cpu-4096-32] 9.5101ms 9.4042ms 106.3355 Ops/s 106.7669 Ops/s $\color{#d91a1a}-0.40\%$
test_collector_preallocated_write[copy_at-cuda-32-2] 0.3987ms 0.3115ms 3.2099 KOps/s 3.1503 KOps/s $\color{#35bf28}+1.89\%$
test_collector_preallocated_write[copy_at-cuda-32-8] 1.2224ms 1.1624ms 860.3247 Ops/s 849.7260 Ops/s $\color{#35bf28}+1.25\%$
test_collector_preallocated_write[copy_at-cuda-32-32] 4.7415ms 4.5371ms 220.4071 Ops/s 217.8125 Ops/s $\color{#35bf28}+1.19\%$
test_collector_preallocated_write[copy_at-cuda-4096-2] 0.3888ms 0.3146ms 3.1782 KOps/s 3.1601 KOps/s $\color{#35bf28}+0.57\%$
test_collector_preallocated_write[copy_at-cuda-4096-8] 1.2584ms 1.1734ms 852.2173 Ops/s 848.2685 Ops/s $\color{#35bf28}+0.47\%$
test_collector_preallocated_write[copy_at-cuda-4096-32] 4.7653ms 4.6016ms 217.3143 Ops/s 215.8388 Ops/s $\color{#35bf28}+0.68\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-2] 0.1531ms 0.1173ms 8.5277 KOps/s 8.5482 KOps/s $\color{#d91a1a}-0.24\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-8] 0.5654ms 0.4564ms 2.1910 KOps/s 2.1771 KOps/s $\color{#35bf28}+0.64\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-32-32] 1.8777ms 1.8206ms 549.2690 Ops/s 547.1257 Ops/s $\color{#35bf28}+0.39\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-2] 0.3423ms 0.2536ms 3.9437 KOps/s 4.0663 KOps/s $\color{#d91a1a}-3.01\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-8] 1.4120ms 1.2164ms 822.1047 Ops/s 837.6081 Ops/s $\color{#d91a1a}-1.85\%$
test_collector_preallocated_write[manual_leaf_copy-cpu-4096-32] 8.6382ms 8.4393ms 118.4935 Ops/s 119.5684 Ops/s $\color{#d91a1a}-0.90\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-2] 0.3205ms 0.2633ms 3.7976 KOps/s 3.7960 KOps/s $\color{#35bf28}+0.04\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-8] 1.0236ms 0.9738ms 1.0269 KOps/s 1.0238 KOps/s $\color{#35bf28}+0.30\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-32-32] 3.9614ms 3.8221ms 261.6385 Ops/s 261.8401 Ops/s $\color{#d91a1a}-0.08\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-2] 0.3066ms 0.2633ms 3.7984 KOps/s 3.8114 KOps/s $\color{#d91a1a}-0.34\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-8] 1.1128ms 0.9795ms 1.0209 KOps/s 1.0205 KOps/s $\color{#35bf28}+0.03\%$
test_collector_preallocated_write[manual_leaf_copy-cuda-4096-32] 3.9427ms 3.7980ms 263.2982 Ops/s 261.0746 Ops/s $\color{#35bf28}+0.85\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-128] 16.6390ms 16.4546ms 60.7732 Ops/s 60.7384 Ops/s $\color{#35bf28}+0.06\%$
test_wide_nested_preallocated_write[stack_out-cpu-128-256] 33.3115ms 33.1202ms 30.1930 Ops/s 30.1766 Ops/s $\color{#35bf28}+0.05\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-128] 84.0279ms 83.2545ms 12.0114 Ops/s 11.9156 Ops/s $\color{#35bf28}+0.80\%$
test_wide_nested_preallocated_write[stack_out-cpu-4096-256] 0.1963s 0.1959s 5.1047 Ops/s 5.1238 Ops/s $\color{#d91a1a}-0.37\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-128] 13.9510ms 13.8056ms 72.4345 Ops/s 72.1969 Ops/s $\color{#35bf28}+0.33\%$
test_wide_nested_preallocated_write[stack_out-cuda-128-256] 27.3242ms 27.0333ms 36.9914 Ops/s 36.8686 Ops/s $\color{#35bf28}+0.33\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-128] 14.0882ms 13.9891ms 71.4845 Ops/s 71.0874 Ops/s $\color{#35bf28}+0.56\%$
test_wide_nested_preallocated_write[stack_out-cuda-4096-256] 27.4797ms 27.3428ms 36.5727 Ops/s 36.4427 Ops/s $\color{#35bf28}+0.36\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-128] 50.6343ms 50.1633ms 19.9349 Ops/s 19.8307 Ops/s $\color{#35bf28}+0.53\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-128-256] 0.1013s 0.1011s 9.8941 Ops/s 9.8937 Ops/s $+0.00\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-128] 0.1439s 0.1437s 6.9587 Ops/s 6.9236 Ops/s $\color{#35bf28}+0.51\%$
test_wide_nested_preallocated_write[set_at_loop-cpu-4096-256] 0.3158s 0.3148s 3.1764 Ops/s 3.2098 Ops/s $\color{#d91a1a}-1.04\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-128] 74.8074ms 72.3741ms 13.8171 Ops/s 14.0570 Ops/s $\color{#d91a1a}-1.71\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-128-256] 0.1443s 0.1430s 6.9916 Ops/s 6.7600 Ops/s $\color{#35bf28}+3.43\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-128] 72.2666ms 71.9263ms 13.9031 Ops/s 13.7553 Ops/s $\color{#35bf28}+1.07\%$
test_wide_nested_preallocated_write[set_at_loop-cuda-4096-256] 0.1446s 0.1440s 6.9423 Ops/s 6.7557 Ops/s $\color{#35bf28}+2.76\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-128] 30.0569ms 29.8915ms 33.4543 Ops/s 33.4835 Ops/s $\color{#d91a1a}-0.09\%$
test_wide_nested_preallocated_write[copy_at-cpu-128-256] 60.5619ms 60.4515ms 16.5422 Ops/s 16.5979 Ops/s $\color{#d91a1a}-0.34\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-128] 0.1173s 0.1169s 8.5560 Ops/s 8.4324 Ops/s $\color{#35bf28}+1.47\%$
test_wide_nested_preallocated_write[copy_at-cpu-4096-256] 0.2630s 0.2627s 3.8064 Ops/s 3.8354 Ops/s $\color{#d91a1a}-0.76\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-128] 42.7939ms 42.2225ms 23.6840 Ops/s 23.5404 Ops/s $\color{#35bf28}+0.61\%$
test_wide_nested_preallocated_write[copy_at-cuda-128-256] 84.6149ms 84.2400ms 11.8708 Ops/s 11.8095 Ops/s $\color{#35bf28}+0.52\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-128] 44.1592ms 42.6258ms 23.4600 Ops/s 23.3498 Ops/s $\color{#35bf28}+0.47\%$
test_wide_nested_preallocated_write[copy_at-cuda-4096-256] 85.3583ms 84.7683ms 11.7969 Ops/s 11.7009 Ops/s $\color{#35bf28}+0.82\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-128] 20.0846ms 19.9528ms 50.1182 Ops/s 49.6973 Ops/s $\color{#35bf28}+0.85\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-128-256] 41.0197ms 40.8685ms 24.4687 Ops/s 24.5073 Ops/s $\color{#d91a1a}-0.16\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-128] 0.1068s 0.1064s 9.3993 Ops/s 9.5007 Ops/s $\color{#d91a1a}-1.07\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cpu-4096-256] 0.2393s 0.2387s 4.1901 Ops/s 4.2406 Ops/s $\color{#d91a1a}-1.19\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-128] 35.4111ms 34.8868ms 28.6641 Ops/s 28.3757 Ops/s $\color{#35bf28}+1.02\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-128-256] 70.0743ms 69.7352ms 14.3400 Ops/s 14.1587 Ops/s $\color{#35bf28}+1.28\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-128] 35.6324ms 35.0438ms 28.5357 Ops/s 28.2489 Ops/s $\color{#35bf28}+1.02\%$
test_wide_nested_preallocated_write[manual_leaf_copy-cuda-4096-256] 71.0791ms 70.6911ms 14.1460 Ops/s 14.1379 Ops/s $\color{#35bf28}+0.06\%$
test_plain_set_nested 39.3610μs 15.1016μs 66.2180 KOps/s 66.6550 KOps/s $\color{#d91a1a}-0.66\%$
test_plain_set_stack_nested 53.8810μs 14.8257μs 67.4504 KOps/s 67.7436 KOps/s $\color{#d91a1a}-0.43\%$
test_plain_set_nested_inplace 49.8120μs 16.5846μs 60.2969 KOps/s 60.7528 KOps/s $\color{#d91a1a}-0.75\%$
test_plain_set_stack_nested_inplace 95.4220μs 16.0815μs 62.1833 KOps/s 63.1301 KOps/s $\color{#d91a1a}-1.50\%$
test_items 43.0810μs 6.0399μs 165.5644 KOps/s 167.3585 KOps/s $\color{#d91a1a}-1.07\%$
test_items_nested 0.4836ms 0.4573ms 2.1866 KOps/s 2.1771 KOps/s $\color{#35bf28}+0.44\%$
test_items_nested_locked 0.4886ms 0.4638ms 2.1562 KOps/s 2.1682 KOps/s $\color{#d91a1a}-0.56\%$
test_items_nested_leaf 0.1634ms 99.4909μs 10.0512 KOps/s 10.1695 KOps/s $\color{#d91a1a}-1.16\%$
test_items_stack_nested 0.5490ms 0.4617ms 2.1659 KOps/s 2.1745 KOps/s $\color{#d91a1a}-0.39\%$
test_items_stack_nested_leaf 0.1386ms 98.6019μs 10.1418 KOps/s 10.1043 KOps/s $\color{#35bf28}+0.37\%$
test_items_stack_nested_locked 0.4881ms 0.4603ms 2.1723 KOps/s 2.1711 KOps/s $\color{#35bf28}+0.06\%$
test_keys 34.2310μs 4.2116μs 237.4401 KOps/s 239.0159 KOps/s $\color{#d91a1a}-0.66\%$
test_keys_nested 0.2373ms 0.1281ms 7.8077 KOps/s 7.8092 KOps/s $\color{#d91a1a}-0.02\%$
test_keys_nested_locked 0.7072ms 0.1367ms 7.3156 KOps/s 7.2772 KOps/s $\color{#35bf28}+0.53\%$
test_keys_nested_leaf 0.1578ms 0.1196ms 8.3626 KOps/s 8.3493 KOps/s $\color{#35bf28}+0.16\%$
test_keys_stack_nested 0.1588ms 0.1269ms 7.8828 KOps/s 7.8351 KOps/s $\color{#35bf28}+0.61\%$
test_keys_stack_nested_leaf 0.1738ms 0.1175ms 8.5102 KOps/s 8.4176 KOps/s $\color{#35bf28}+1.10\%$
test_keys_stack_nested_locked 0.1851ms 0.1353ms 7.3918 KOps/s 7.3492 KOps/s $\color{#35bf28}+0.58\%$
test_values 6.0762μs 1.0574μs 945.7147 KOps/s 982.8822 KOps/s $\color{#d91a1a}-3.78\%$
test_values_nested 83.4420μs 52.1610μs 19.1714 KOps/s 19.0025 KOps/s $\color{#35bf28}+0.89\%$
test_values_nested_locked 86.1720μs 55.8807μs 17.8953 KOps/s 17.7499 KOps/s $\color{#35bf28}+0.82\%$
test_values_nested_leaf 0.1388ms 59.8198μs 16.7169 KOps/s 16.4772 KOps/s $\color{#35bf28}+1.45\%$
test_values_stack_nested 85.6320μs 52.0213μs 19.2229 KOps/s 18.9660 KOps/s $\color{#35bf28}+1.35\%$
test_values_stack_nested_leaf 93.3630μs 59.7428μs 16.7384 KOps/s 16.5600 KOps/s $\color{#35bf28}+1.08\%$
test_values_stack_nested_locked 89.5320μs 55.9290μs 17.8798 KOps/s 17.9821 KOps/s $\color{#d91a1a}-0.57\%$
test_membership 16.6000μs 0.9859μs 1.0143 MOps/s 1.1963 MOps/s $\textbf{\color{#d91a1a}-15.21\%}$
test_membership_nested 77.8820μs 2.9926μs 334.1612 KOps/s 335.8096 KOps/s $\color{#d91a1a}-0.49\%$
test_membership_nested_leaf 36.8710μs 2.9350μs 340.7136 KOps/s 334.8795 KOps/s $\color{#35bf28}+1.74\%$
test_membership_stacked_nested 32.0810μs 2.8497μs 350.9122 KOps/s 347.8977 KOps/s $\color{#35bf28}+0.87\%$
test_membership_stacked_nested_leaf 34.9110μs 2.7506μs 363.5521 KOps/s 346.9217 KOps/s $\color{#35bf28}+4.79\%$
test_membership_nested_last 29.2310μs 4.4447μs 224.9886 KOps/s 223.8521 KOps/s $\color{#35bf28}+0.51\%$
test_membership_nested_leaf_last 34.9710μs 4.4410μs 225.1739 KOps/s 223.0761 KOps/s $\color{#35bf28}+0.94\%$
test_membership_stacked_nested_last 54.1820μs 4.2669μs 234.3641 KOps/s 233.5273 KOps/s $\color{#35bf28}+0.36\%$
test_membership_stacked_nested_leaf_last 45.4010μs 4.2871μs 233.2597 KOps/s 231.7243 KOps/s $\color{#35bf28}+0.66\%$
test_nested_getleaf 99.0330μs 21.4186μs 46.6884 KOps/s 46.5363 KOps/s $\color{#35bf28}+0.33\%$
test_nested_get 66.5420μs 20.1901μs 49.5293 KOps/s 49.0927 KOps/s $\color{#35bf28}+0.89\%$
test_stacked_getleaf 66.8320μs 20.9179μs 47.8059 KOps/s 47.6324 KOps/s $\color{#35bf28}+0.36\%$
test_stacked_get 45.8410μs 19.8582μs 50.3571 KOps/s 49.2961 KOps/s $\color{#35bf28}+2.15\%$
test_nested_getitemleaf 52.3710μs 22.2547μs 44.9343 KOps/s 45.3155 KOps/s $\color{#d91a1a}-0.84\%$
test_nested_getitem 50.9210μs 21.0484μs 47.5095 KOps/s 48.6207 KOps/s $\color{#d91a1a}-2.29\%$
test_stacked_getitemleaf 90.8420μs 21.4304μs 46.6627 KOps/s 46.3423 KOps/s $\color{#35bf28}+0.69\%$
test_stacked_getitem 56.1820μs 20.9038μs 47.8383 KOps/s 48.3412 KOps/s $\color{#d91a1a}-1.04\%$
test_lock_nested 7.9012ms 0.5075ms 1.9706 KOps/s 1.9883 KOps/s $\color{#d91a1a}-0.89\%$
test_lock_stack_nested 0.5340ms 0.5029ms 1.9886 KOps/s 1.9875 KOps/s $\color{#35bf28}+0.05\%$
test_unlock_nested 0.4716ms 0.3943ms 2.5361 KOps/s 2.5357 KOps/s $\color{#35bf28}+0.02\%$
test_unlock_stack_nested 0.4340ms 0.3961ms 2.5246 KOps/s 2.5118 KOps/s $\color{#35bf28}+0.51\%$
test_flatten_speed 0.1926ms 0.1224ms 8.1719 KOps/s 8.0161 KOps/s $\color{#35bf28}+1.94\%$
test_unflatten_speed 0.6382ms 0.5634ms 1.7751 KOps/s 1.7743 KOps/s $\color{#35bf28}+0.05\%$
test_common_ops 0.8306ms 0.6894ms 1.4505 KOps/s 1.4251 KOps/s $\color{#35bf28}+1.78\%$
test_creation 0.1432ms 3.1781μs 314.6496 KOps/s 313.4424 KOps/s $\color{#35bf28}+0.39\%$
test_creation_empty 29.0510μs 7.0054μs 142.7475 KOps/s 141.9341 KOps/s $\color{#35bf28}+0.57\%$
test_creation_nested_1 43.6110μs 11.4718μs 87.1706 KOps/s 87.0222 KOps/s $\color{#35bf28}+0.17\%$
test_creation_nested_2 69.3020μs 12.7793μs 78.2515 KOps/s 74.1485 KOps/s $\textbf{\color{#35bf28}+5.53\%}$
test_creation_many_keys[10] 90.6720μs 21.1344μs 47.3162 KOps/s 47.2723 KOps/s $\color{#35bf28}+0.09\%$
test_creation_many_keys[50] 0.1300ms 90.8465μs 11.0076 KOps/s 11.0518 KOps/s $\color{#d91a1a}-0.40\%$
test_creation_many_keys[100] 0.2401ms 0.1770ms 5.6496 KOps/s 5.5799 KOps/s $\color{#35bf28}+1.25\%$
test_creation_nested_many_keys[10] 73.5220μs 45.0409μs 22.2020 KOps/s 22.1023 KOps/s $\color{#35bf28}+0.45\%$
test_creation_nested_many_keys[50] 0.2758ms 0.1825ms 5.4780 KOps/s 5.3816 KOps/s $\color{#35bf28}+1.79\%$
test_clone 43.5410μs 12.5726μs 79.5383 KOps/s 78.1346 KOps/s $\color{#35bf28}+1.80\%$
test_getitem[int] 1.4913ms 15.1995μs 65.7918 KOps/s 65.5876 KOps/s $\color{#35bf28}+0.31\%$
test_getitem[slice_int] 0.1380ms 24.2953μs 41.1603 KOps/s 40.8061 KOps/s $\color{#35bf28}+0.87\%$
test_getitem[range] 0.1718ms 61.3422μs 16.3020 KOps/s 16.3790 KOps/s $\color{#d91a1a}-0.47\%$
test_getitem[tuple] 0.1433ms 23.7662μs 42.0765 KOps/s 41.7271 KOps/s $\color{#35bf28}+0.84\%$
test_getitem[list] 0.1833ms 56.5608μs 17.6801 KOps/s 17.7713 KOps/s $\color{#d91a1a}-0.51\%$
test_setitem_dim[int] 46.6720μs 24.4507μs 40.8987 KOps/s 39.8805 KOps/s $\color{#35bf28}+2.55\%$
test_setitem_dim[slice_int] 60.8810μs 40.4498μs 24.7220 KOps/s 24.3067 KOps/s $\color{#35bf28}+1.71\%$
test_setitem_dim[range] 0.1320ms 91.7241μs 10.9023 KOps/s 10.9031 KOps/s $-0.01\%$
test_setitem_dim[tuple] 60.7020μs 38.3148μs 26.0996 KOps/s 25.7576 KOps/s $\color{#35bf28}+1.33\%$
test_setitem 50.1210μs 16.9003μs 59.1706 KOps/s 58.8399 KOps/s $\color{#35bf28}+0.56\%$
test_set 49.6210μs 16.1038μs 62.0973 KOps/s 62.1266 KOps/s $\color{#d91a1a}-0.05\%$
test_set_shared 0.5041ms 0.2074ms 4.8212 KOps/s 4.8981 KOps/s $\color{#d91a1a}-1.57\%$
test_update 0.2233ms 20.9588μs 47.7126 KOps/s 47.7180 KOps/s $\color{#d91a1a}-0.01\%$
test_update_nested 65.2710μs 32.6907μs 30.5898 KOps/s 30.5604 KOps/s $\color{#35bf28}+0.10\%$
test_update__nested 0.4948ms 32.2515μs 31.0063 KOps/s 30.9386 KOps/s $\color{#35bf28}+0.22\%$
test_set_nested 52.5220μs 17.8728μs 55.9508 KOps/s 55.8807 KOps/s $\color{#35bf28}+0.13\%$
test_set_nested_new 54.4410μs 22.4102μs 44.6226 KOps/s 44.5546 KOps/s $\color{#35bf28}+0.15\%$
test_select 74.0820μs 38.9869μs 25.6496 KOps/s 25.5292 KOps/s $\color{#35bf28}+0.47\%$
test_select_nested 0.1499ms 74.0191μs 13.5100 KOps/s 13.2793 KOps/s $\color{#35bf28}+1.74\%$
test_exclude_nested 0.1188ms 91.1906μs 10.9660 KOps/s 10.9142 KOps/s $\color{#35bf28}+0.48\%$
test_empty[True] 0.5641ms 0.3956ms 2.5276 KOps/s 2.5267 KOps/s $\color{#35bf28}+0.03\%$
test_empty[False] 10.8702μs 1.3810μs 724.0879 KOps/s 719.5970 KOps/s $\color{#35bf28}+0.62\%$
test_to 95.4020μs 69.9284μs 14.3004 KOps/s 14.4045 KOps/s $\color{#d91a1a}-0.72\%$
test_to_nonblocking 0.1797ms 68.8434μs 14.5257 KOps/s 14.4744 KOps/s $\color{#35bf28}+0.35\%$
test_unbind_speed 0.3816ms 0.3357ms 2.9788 KOps/s 2.9710 KOps/s $\color{#35bf28}+0.26\%$
test_unbind_speed_stack0 0.4158ms 0.3280ms 3.0490 KOps/s 3.0236 KOps/s $\color{#35bf28}+0.84\%$
test_unbind_speed_stack1 0.1160s 0.8561ms 1.1681 KOps/s 1.1581 KOps/s $\color{#35bf28}+0.87\%$
test_split 0.1162s 1.4510ms 689.1861 Ops/s 628.5945 Ops/s $\textbf{\color{#35bf28}+9.64\%}$
test_chunk 1.2761ms 1.2362ms 808.9089 Ops/s 808.5336 Ops/s $\color{#35bf28}+0.05\%$
test_to_cpu_blocking 19.7183ms 19.0347ms 52.5356 Ops/s 52.3018 Ops/s $\color{#35bf28}+0.45\%$
test_to_cpu_global_sync 0.1274s 12.7549ms 78.4012 Ops/s 87.6981 Ops/s $\textbf{\color{#d91a1a}-10.60\%}$
test_to_cpu_event_sync 12.4563ms 12.2100ms 81.8999 Ops/s 81.4788 Ops/s $\color{#35bf28}+0.52\%$
test_to_cpu_default 12.4927ms 12.2322ms 81.7515 Ops/s 72.4867 Ops/s $\textbf{\color{#35bf28}+12.78\%}$
test_consolidate[False-None] 4.2695ms 4.1606ms 240.3500 Ops/s 240.2222 Ops/s $\color{#35bf28}+0.05\%$
test_consolidate[default-None] 3.2320ms 2.2194ms 450.5668 Ops/s 439.4906 Ops/s $\color{#35bf28}+2.52\%$
test_consolidate[reduce-overhead-None] 3.1663ms 2.1372ms 467.8992 Ops/s 453.8302 Ops/s $\color{#35bf28}+3.10\%$
test_consolidate_njt[False-None] 5.3287ms 5.1154ms 195.4871 Ops/s 193.8404 Ops/s $\color{#35bf28}+0.85\%$
test_to[False-False-None] 2.2946ms 2.1703ms 460.7604 Ops/s 461.1275 Ops/s $\color{#d91a1a}-0.08\%$
test_to[True-False-None] 2.2816ms 1.9449ms 514.1559 Ops/s 520.3926 Ops/s $\color{#d91a1a}-1.20\%$
test_to[within-False-None] 6.3337ms 6.1871ms 161.6278 Ops/s 161.8054 Ops/s $\color{#d91a1a}-0.11\%$
test_to[True-default-None] 10.4471ms 10.1609ms 98.4162 Ops/s 97.3473 Ops/s $\color{#35bf28}+1.10\%$
test_to_njt[False-False-None] 7.0221ms 6.9099ms 144.7200 Ops/s 142.8496 Ops/s $\color{#35bf28}+1.31\%$
test_to_njt[True-False-None] 4.0524ms 3.8264ms 261.3420 Ops/s 259.7634 Ops/s $\color{#35bf28}+0.61\%$
test_to_njt[within-False-None] 9.1428ms 9.0421ms 110.5937 Ops/s 109.9783 Ops/s $\color{#35bf28}+0.56\%$
test_creation[device0] 0.3966ms 0.1145ms 8.7364 KOps/s 8.7832 KOps/s $\color{#d91a1a}-0.53\%$
test_creation_from_tensor 0.4428ms 0.1154ms 8.6692 KOps/s 8.8991 KOps/s $\color{#d91a1a}-2.58\%$
test_add_one[memmap_tensor0] 0.2783ms 5.9797μs 167.2323 KOps/s 165.8843 KOps/s $\color{#35bf28}+0.81\%$
test_contiguous[memmap_tensor0] 46.1710μs 0.6777μs 1.4756 MOps/s 2.0475 MOps/s $\textbf{\color{#d91a1a}-27.93\%}$
test_stack[memmap_tensor0] 87.2820μs 4.0662μs 245.9285 KOps/s 246.8994 KOps/s $\color{#d91a1a}-0.39\%$
test_memmaptd_index 1.2995ms 0.2716ms 3.6815 KOps/s 3.6956 KOps/s $\color{#d91a1a}-0.38\%$
test_memmaptd_index_astensor 0.5845ms 0.3762ms 2.6582 KOps/s 2.6666 KOps/s $\color{#d91a1a}-0.31\%$
test_memmaptd_index_op 1.0510ms 0.6126ms 1.6323 KOps/s 1.6179 KOps/s $\color{#35bf28}+0.90\%$
test_serialize_model 0.1366s 0.1351s 7.4045 Ops/s 7.4016 Ops/s $\color{#35bf28}+0.04\%$
test_serialize_model_pickle 1.3719s 1.2148s 0.8232 Ops/s 0.8257 Ops/s $\color{#d91a1a}-0.31\%$
test_serialize_weights 0.1351s 0.1339s 7.4687 Ops/s 7.4615 Ops/s $\color{#35bf28}+0.10\%$
test_serialize_weights_returnearly 65.3601ms 61.0061ms 16.3918 Ops/s 15.7440 Ops/s $\color{#35bf28}+4.11\%$
test_serialize_weights_pickle 1.3529s 1.2301s 0.8129 Ops/s 0.8058 Ops/s $\color{#35bf28}+0.88\%$
test_reshape_pytree 0.2177ms 32.1132μs 31.1398 KOps/s 31.0988 KOps/s $\color{#35bf28}+0.13\%$
test_reshape_td 83.1820μs 44.7056μs 22.3686 KOps/s 22.2905 KOps/s $\color{#35bf28}+0.35\%$
test_view_pytree 0.2233ms 31.7762μs 31.4701 KOps/s 31.5284 KOps/s $\color{#d91a1a}-0.18\%$
test_view_td 93.0620μs 54.6974μs 18.2824 KOps/s 18.3793 KOps/s $\color{#d91a1a}-0.53\%$
test_unbind_pytree 0.2409ms 35.9694μs 27.8014 KOps/s 27.6387 KOps/s $\color{#35bf28}+0.59\%$
test_unbind_td 0.1047ms 50.3350μs 19.8669 KOps/s 19.7261 KOps/s $\color{#35bf28}+0.71\%$
test_split_pytree 0.2579ms 41.1366μs 24.3092 KOps/s 24.0016 KOps/s $\color{#35bf28}+1.28\%$
test_split_td 0.1869ms 71.6506μs 13.9566 KOps/s 13.9117 KOps/s $\color{#35bf28}+0.32\%$
test_add_pytree 0.1942ms 39.9941μs 25.0037 KOps/s 25.3966 KOps/s $\color{#d91a1a}-1.55\%$
test_add_td 0.1575ms 51.8660μs 19.2804 KOps/s 18.7976 KOps/s $\color{#35bf28}+2.57\%$
test_compile_add_one_nested[tensordict-compile] 0.4160ms 0.2735ms 3.6557 KOps/s 3.6950 KOps/s $\color{#d91a1a}-1.06\%$
test_compile_add_one_nested[tensordict-eager] 0.2828ms 0.1998ms 5.0060 KOps/s 4.9870 KOps/s $\color{#35bf28}+0.38\%$
test_compile_add_one_nested[pytree-compile] 0.2153ms 0.1549ms 6.4538 KOps/s 6.4854 KOps/s $\color{#d91a1a}-0.49\%$
test_compile_add_one_nested[pytree-eager] 0.5038ms 0.1774ms 5.6368 KOps/s 5.5980 KOps/s $\color{#35bf28}+0.69\%$
test_compile_copy_nested[tensordict-compile] 68.3110μs 22.2735μs 44.8963 KOps/s 47.7853 KOps/s $\textbf{\color{#d91a1a}-6.05\%}$
test_compile_copy_nested[tensordict-eager] 0.3480ms 52.7712μs 18.9497 KOps/s 18.3785 KOps/s $\color{#35bf28}+3.11\%$
test_compile_copy_nested[pytree-compile] 0.2111ms 19.0068μs 52.6126 KOps/s 51.6753 KOps/s $\color{#35bf28}+1.81\%$
test_compile_copy_nested[pytree-eager] 0.3645ms 66.8358μs 14.9620 KOps/s 14.8748 KOps/s $\color{#35bf28}+0.59\%$
test_compile_add_one_flat[tensordict-compile] 0.5255ms 0.3077ms 3.2503 KOps/s 3.1128 KOps/s $\color{#35bf28}+4.42\%$
test_compile_add_one_flat[tensordict-eager] 0.3769ms 0.2870ms 3.4845 KOps/s 3.4236 KOps/s $\color{#35bf28}+1.78\%$
test_compile_add_one_flat[tensorclass-compile] 0.3907ms 0.2080ms 4.8079 KOps/s 4.6247 KOps/s $\color{#35bf28}+3.96\%$
test_compile_add_one_flat[tensorclass-eager] 0.1883ms 75.8524μs 13.1835 KOps/s 13.2804 KOps/s $\color{#d91a1a}-0.73\%$
test_compile_add_one_flat[pytree-compile] 0.5000ms 0.2422ms 4.1288 KOps/s 4.0427 KOps/s $\color{#35bf28}+2.13\%$
test_compile_add_one_flat[pytree-eager] 0.8989ms 0.5318ms 1.8804 KOps/s 1.8960 KOps/s $\color{#d91a1a}-0.83\%$
test_compile_add_self_flat[tensordict-eager] 0.5066ms 0.3378ms 2.9607 KOps/s 2.9377 KOps/s $\color{#35bf28}+0.78\%$
test_compile_add_self_flat[tensordict-compile] 0.3529ms 0.3059ms 3.2690 KOps/s 3.0842 KOps/s $\textbf{\color{#35bf28}+5.99\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1376ms 92.1215μs 10.8552 KOps/s 10.7675 KOps/s $\color{#35bf28}+0.81\%$
test_compile_add_self_flat[tensorclass-compile] 0.3702ms 0.2093ms 4.7785 KOps/s 4.3086 KOps/s $\textbf{\color{#35bf28}+10.91\%}$
test_compile_add_self_flat[pytree-eager] 0.6786ms 0.4246ms 2.3553 KOps/s 2.3118 KOps/s $\color{#35bf28}+1.88\%$
test_compile_add_self_flat[pytree-compile] 0.3630ms 0.2415ms 4.1411 KOps/s 4.0476 KOps/s $\color{#35bf28}+2.31\%$
test_compile_copy_flat[tensordict-compile] 81.0720μs 25.9266μs 38.5705 KOps/s 34.9578 KOps/s $\textbf{\color{#35bf28}+10.33\%}$
test_compile_copy_flat[tensordict-eager] 70.6620μs 40.3637μs 24.7747 KOps/s 24.3667 KOps/s $\color{#35bf28}+1.67\%$
test_compile_copy_flat[pytree-compile] 0.1779ms 25.3528μs 39.4434 KOps/s 36.7388 KOps/s $\textbf{\color{#35bf28}+7.36\%}$
test_compile_copy_flat[pytree-eager] 0.3330ms 51.6608μs 19.3570 KOps/s 19.4329 KOps/s $\color{#d91a1a}-0.39\%$
test_compile_assign_and_add[tensordict-compile] 2.0722ms 0.2535ms 3.9444 KOps/s 3.5260 KOps/s $\textbf{\color{#35bf28}+11.87\%}$
test_compile_assign_and_add[tensordict-eager] 3.5631ms 3.3012ms 302.9235 Ops/s 289.8383 Ops/s $\color{#35bf28}+4.51\%$
test_compile_assign_and_add[pytree-compile] 1.9352ms 0.1937ms 5.1619 KOps/s 5.0499 KOps/s $\color{#35bf28}+2.22\%$
test_compile_assign_and_add[pytree-eager] 2.9610ms 2.7598ms 362.3489 Ops/s 353.7553 Ops/s $\color{#35bf28}+2.43\%$
test_compile_indexing[tensor-tensordict-compile] 0.3212ms 0.1534ms 6.5177 KOps/s 6.3881 KOps/s $\color{#35bf28}+2.03\%$
test_compile_indexing[tensor-tensordict-eager] 0.3729ms 73.3334μs 13.6363 KOps/s 13.7407 KOps/s $\color{#d91a1a}-0.76\%$
test_compile_indexing[tensor-tensorclass-compile] 0.2883ms 0.1335ms 7.4923 KOps/s 7.4719 KOps/s $\color{#35bf28}+0.27\%$
test_compile_indexing[tensor-tensorclass-eager] 0.3266ms 43.0397μs 23.2343 KOps/s 21.9554 KOps/s $\textbf{\color{#35bf28}+5.83\%}$
test_compile_indexing[tensor-pytree-compile] 0.1753ms 0.1333ms 7.5021 KOps/s 7.1807 KOps/s $\color{#35bf28}+4.48\%$
test_compile_indexing[tensor-pytree-eager] 0.2726ms 43.0097μs 23.2505 KOps/s 21.8328 KOps/s $\textbf{\color{#35bf28}+6.49\%}$
test_compile_indexing[slice-tensordict-compile] 0.1884ms 0.1023ms 9.7748 KOps/s 10.8065 KOps/s $\textbf{\color{#d91a1a}-9.55\%}$
test_compile_indexing[slice-tensordict-eager] 0.2129ms 27.2708μs 36.6692 KOps/s 36.8382 KOps/s $\color{#d91a1a}-0.46\%$
test_compile_indexing[slice-tensorclass-compile] 0.1815ms 71.2554μs 14.0340 KOps/s 13.7215 KOps/s $\color{#35bf28}+2.28\%$
test_compile_indexing[slice-tensorclass-eager] 0.2557ms 21.9401μs 45.5785 KOps/s 45.8129 KOps/s $\color{#d91a1a}-0.51\%$
test_compile_indexing[slice-pytree-compile] 0.1716ms 71.4898μs 13.9880 KOps/s 13.7819 KOps/s $\color{#35bf28}+1.50\%$
test_compile_indexing[slice-pytree-eager] 0.2700ms 21.8627μs 45.7399 KOps/s 46.0165 KOps/s $\color{#d91a1a}-0.60\%$
test_compile_indexing[int-tensordict-compile] 0.1474ms 88.8700μs 11.2524 KOps/s 10.9454 KOps/s $\color{#35bf28}+2.81\%$
test_compile_indexing[int-tensordict-eager] 0.2480ms 27.2761μs 36.6621 KOps/s 36.8868 KOps/s $\color{#d91a1a}-0.61\%$
test_compile_indexing[int-tensorclass-compile] 0.1167ms 71.4443μs 13.9969 KOps/s 13.8764 KOps/s $\color{#35bf28}+0.87\%$
test_compile_indexing[int-tensorclass-eager] 0.2512ms 21.8649μs 45.7354 KOps/s 45.9765 KOps/s $\color{#d91a1a}-0.52\%$
test_compile_indexing[int-pytree-compile] 0.1083ms 71.7554μs 13.9362 KOps/s 13.7099 KOps/s $\color{#35bf28}+1.65\%$
test_compile_indexing[int-pytree-eager] 0.2639ms 21.8882μs 45.6868 KOps/s 46.0712 KOps/s $\color{#d91a1a}-0.83\%$
test_compile_replace[single-eager] 94.8520μs 46.8338μs 21.3521 KOps/s 21.2723 KOps/s $\color{#35bf28}+0.38\%$
test_compile_replace[single-compile] 0.2494ms 0.1434ms 6.9731 KOps/s 6.8606 KOps/s $\color{#35bf28}+1.64\%$
test_compile_replace[multi-eager] 0.6608ms 0.5619ms 1.7796 KOps/s 1.8090 KOps/s $\color{#d91a1a}-1.62\%$
test_compile_replace[multi-compile] 0.2687ms 0.1548ms 6.4592 KOps/s 6.1855 KOps/s $\color{#35bf28}+4.43\%$
test_compile_tc_getattr_20[eager] 0.2315ms 0.1678ms 5.9582 KOps/s 6.0043 KOps/s $\color{#d91a1a}-0.77\%$
test_compile_tc_getattr_20[compile] 0.2364ms 0.1735ms 5.7646 KOps/s 5.6260 KOps/s $\color{#35bf28}+2.46\%$
test_compile_clone_shallow[20-eager] 51.1010μs 18.7659μs 53.2882 KOps/s 53.1053 KOps/s $\color{#35bf28}+0.34\%$
test_compile_clone_shallow[20-compile] 84.6720μs 22.5501μs 44.3456 KOps/s 43.6115 KOps/s $\color{#35bf28}+1.68\%$
test_compile_clone_shallow[40-eager] 67.4920μs 32.6886μs 30.5917 KOps/s 30.7082 KOps/s $\color{#d91a1a}-0.38\%$
test_compile_clone_shallow[40-compile] 79.1820μs 25.0317μs 39.9493 KOps/s 40.5435 KOps/s $\color{#d91a1a}-1.47\%$
test_compile_clone_shallow[80-eager] 0.1375ms 60.6022μs 16.5010 KOps/s 16.0822 KOps/s $\color{#35bf28}+2.60\%$
test_compile_clone_shallow[80-compile] 82.8220μs 28.8421μs 34.6716 KOps/s 34.9723 KOps/s $\color{#d91a1a}-0.86\%$
test_compile_update_inplace[eager] 0.1067ms 61.2322μs 16.3313 KOps/s 16.3366 KOps/s $\color{#d91a1a}-0.03\%$
test_compile_update_inplace[compile] 0.2044ms 0.1185ms 8.4423 KOps/s 8.1802 KOps/s $\color{#35bf28}+3.20\%$
test_mod_add[eager] 0.2837ms 0.2255ms 4.4349 KOps/s 4.4392 KOps/s $\color{#d91a1a}-0.10\%$
test_mod_add[compile] 0.3431ms 0.1176ms 8.5048 KOps/s 8.0927 KOps/s $\textbf{\color{#35bf28}+5.09\%}$
test_mod_add[compile-overhead] 0.3510ms 0.1930ms 5.1810 KOps/s 4.9533 KOps/s $\color{#35bf28}+4.60\%$
test_mod_wrap[eager] 0.5498ms 0.4626ms 2.1616 KOps/s 2.1152 KOps/s $\color{#35bf28}+2.19\%$
test_mod_wrap[compile] 0.4255ms 0.3493ms 2.8632 KOps/s 2.8008 KOps/s $\color{#35bf28}+2.23\%$
test_mod_wrap[compile-overhead] 7.1568ms 4.0315ms 248.0469 Ops/s 250.0441 Ops/s $\color{#d91a1a}-0.80\%$
test_mod_wrap_and_backward[eager] 1.7794ms 1.6441ms 608.2474 Ops/s 602.9962 Ops/s $\color{#35bf28}+0.87\%$
test_mod_wrap_and_backward[compile] 1.8656ms 1.3220ms 756.4205 Ops/s 757.4690 Ops/s $\color{#d91a1a}-0.14\%$
test_mod_wrap_and_backward[compile-overhead] 1.6044ms 0.9334ms 1.0714 KOps/s 1.0536 KOps/s $\color{#35bf28}+1.68\%$
test_seq_add[eager] 0.6575ms 0.5047ms 1.9813 KOps/s 1.9638 KOps/s $\color{#35bf28}+0.89\%$
test_seq_add[compile] 0.2504ms 0.1309ms 7.6373 KOps/s 7.5091 KOps/s $\color{#35bf28}+1.71\%$
test_seq_add[compile-overhead] 0.2488ms 0.2031ms 4.9238 KOps/s 4.6706 KOps/s $\textbf{\color{#35bf28}+5.42\%}$
test_seq_wrap[eager] 1.5199ms 1.3963ms 716.1579 Ops/s 702.0431 Ops/s $\color{#35bf28}+2.01\%$
test_seq_wrap[compile] 0.4141ms 0.3571ms 2.8002 KOps/s 2.7122 KOps/s $\color{#35bf28}+3.25\%$
test_seq_wrap[compile-overhead] 0.4479ms 0.3166ms 3.1583 KOps/s 3.0936 KOps/s $\color{#35bf28}+2.09\%$
test_func_call_runtime[False-eager] 0.9264ms 0.7902ms 1.2656 KOps/s 1.2574 KOps/s $\color{#35bf28}+0.65\%$
test_func_call_runtime[False-compile] 0.8094ms 0.7502ms 1.3331 KOps/s 1.3234 KOps/s $\color{#35bf28}+0.73\%$
test_func_call_runtime[False-compile-overhead] 0.6820ms 0.5071ms 1.9718 KOps/s 1.9558 KOps/s $\color{#35bf28}+0.82\%$
test_func_call_runtime[True-eager] 1.1366ms 1.0518ms 950.7953 Ops/s 948.2524 Ops/s $\color{#35bf28}+0.27\%$
test_func_call_runtime[True-compile] 0.8246ms 0.7651ms 1.3070 KOps/s 1.2957 KOps/s $\color{#35bf28}+0.87\%$
test_func_call_runtime[True-compile-overhead] 0.6855ms 0.5211ms 1.9190 KOps/s 1.8823 KOps/s $\color{#35bf28}+1.95\%$
test_func_call_cm_runtime[False-eager] 0.8582ms 0.7889ms 1.2677 KOps/s 1.2328 KOps/s $\color{#35bf28}+2.83\%$
test_func_call_cm_runtime[False-compile] 0.8571ms 0.7367ms 1.3575 KOps/s 1.3092 KOps/s $\color{#35bf28}+3.69\%$
test_func_call_cm_runtime[False-compile-overhead] 0.5386ms 0.4875ms 2.0511 KOps/s 1.9992 KOps/s $\color{#35bf28}+2.60\%$
test_func_call_cm_runtime[True-eager] 1.3022ms 1.2011ms 832.5378 Ops/s 827.0374 Ops/s $\color{#35bf28}+0.67\%$
test_func_call_cm_runtime[True-compile] 0.9189ms 0.8450ms 1.1834 KOps/s 1.1735 KOps/s $\color{#35bf28}+0.85\%$
test_func_call_cm_runtime[True-compile-overhead] 1.2172ms 0.8494ms 1.1773 KOps/s 1.1278 KOps/s $\color{#35bf28}+4.39\%$
test_vmap_func_call_cm_runtime[eager] 3.2066ms 2.6568ms 376.3885 Ops/s 370.5090 Ops/s $\color{#35bf28}+1.59\%$
test_vmap_func_call_cm_runtime[compile] 4.1051ms 3.6186ms 276.3496 Ops/s 273.0181 Ops/s $\color{#35bf28}+1.22\%$
test_vmap_func_call_cm_runtime[compile-overhead] 4.1027ms 3.6250ms 275.8594 Ops/s 273.5968 Ops/s $\color{#35bf28}+0.83\%$
test_distributed 2.0271ms 0.1579ms 6.3328 KOps/s 6.8830 KOps/s $\textbf{\color{#d91a1a}-7.99\%}$
test_tdmodule 0.7088ms 0.1944ms 5.1453 KOps/s 5.1257 KOps/s $\color{#35bf28}+0.38\%$
test_tdmodule_dispatch 0.4545ms 0.3716ms 2.6914 KOps/s 2.6592 KOps/s $\color{#35bf28}+1.21\%$
test_tdseq 58.2220μs 26.7656μs 37.3613 KOps/s 37.1285 KOps/s $\color{#35bf28}+0.63\%$
test_tdseq_dispatch 0.2531ms 0.2180ms 4.5875 KOps/s 4.5531 KOps/s $\color{#35bf28}+0.76\%$
test_instantiation_functorch 2.1420ms 2.0164ms 495.9361 Ops/s 488.8597 Ops/s $\color{#35bf28}+1.45\%$
test_exec_functorch 0.2761ms 0.1711ms 5.8435 KOps/s 5.8237 KOps/s $\color{#35bf28}+0.34\%$
test_exec_functional_call 0.4316ms 0.1538ms 6.5016 KOps/s 6.4799 KOps/s $\color{#35bf28}+0.33\%$
test_exec_td_decorator 0.5188ms 0.2413ms 4.1444 KOps/s 4.1152 KOps/s $\color{#35bf28}+0.71\%$
test_vmap_mlp_speed_decorator[True-True] 1.2742ms 1.1082ms 902.3711 Ops/s 894.7788 Ops/s $\color{#35bf28}+0.85\%$
test_vmap_mlp_speed_decorator[True-False] 1.3003ms 1.1045ms 905.3932 Ops/s 898.3223 Ops/s $\color{#35bf28}+0.79\%$
test_vmap_mlp_speed_decorator[False-True] 1.0538ms 0.8127ms 1.2305 KOps/s 1.2248 KOps/s $\color{#35bf28}+0.46\%$
test_vmap_mlp_speed_decorator[False-False] 1.0313ms 0.8095ms 1.2353 KOps/s 1.2209 KOps/s $\color{#35bf28}+1.18\%$
test_vmap_transformer_speed_decorator[True-True] 24.9638ms 24.6350ms 40.5926 Ops/s 40.2484 Ops/s $\color{#35bf28}+0.86\%$
test_vmap_transformer_speed_decorator[True-False] 24.9625ms 24.5415ms 40.7473 Ops/s 40.3679 Ops/s $\color{#35bf28}+0.94\%$
test_vmap_transformer_speed_decorator[False-True] 24.4068ms 24.1729ms 41.3687 Ops/s 40.9923 Ops/s $\color{#35bf28}+0.92\%$
test_vmap_transformer_speed_decorator[False-False] 24.2900ms 24.1452ms 41.4161 Ops/s 40.6880 Ops/s $\color{#35bf28}+1.79\%$
test_to_module_speed[True] 1.7449ms 1.6401ms 609.7365 Ops/s 599.3109 Ops/s $\color{#35bf28}+1.74\%$
test_to_module_speed[False] 1.7340ms 1.6245ms 615.5808 Ops/s 613.0135 Ops/s $\color{#35bf28}+0.42\%$
test_to_module_plain_tensor_speed[None] 2.1849ms 1.7336ms 576.8230 Ops/s 573.6866 Ops/s $\color{#35bf28}+0.55\%$
test_to_module_plain_tensor_speed[False] 3.1062ms 1.5784ms 633.5539 Ops/s 630.5065 Ops/s $\color{#35bf28}+0.48\%$
test_to_module_plain_tensor_speed[True] 3.5746ms 3.3976ms 294.3231 Ops/s 293.6560 Ops/s $\color{#35bf28}+0.23\%$
test_tc_init 76.0410μs 44.0803μs 22.6859 KOps/s 22.7166 KOps/s $\color{#d91a1a}-0.14\%$
test_tc_init_tensor_only 32.6610μs 9.5053μs 105.2049 KOps/s 103.7246 KOps/s $\color{#35bf28}+1.43\%$
test_tc_init_nested 0.1238ms 87.9917μs 11.3647 KOps/s 11.3784 KOps/s $\color{#d91a1a}-0.12\%$
test_tc_init_many_fields 47.7610μs 17.4162μs 57.4177 KOps/s 57.6294 KOps/s $\color{#d91a1a}-0.37\%$
test_tc_first_layer_tensor 18.9900μs 1.8386μs 543.8916 KOps/s 563.3037 KOps/s $\color{#d91a1a}-3.45\%$
test_tc_first_layer_tensor_only 1.5265μs 0.4070μs 2.4571 MOps/s 2.4623 MOps/s $\color{#d91a1a}-0.21\%$
test_tc_first_layer_tensor_set 29.9010μs 4.2212μs 236.8999 KOps/s 238.0836 KOps/s $\color{#d91a1a}-0.50\%$
test_tc_first_layer_tensor_only_set 33.6110μs 3.6060μs 277.3130 KOps/s 286.3240 KOps/s $\color{#d91a1a}-3.15\%$
test_tc_first_layer_nontensor 30.9710μs 6.0662μs 164.8483 KOps/s 161.8546 KOps/s $\color{#35bf28}+1.85\%$
test_tc_second_layer_tensor 26.7800μs 4.3821μs 228.1997 KOps/s 226.4190 KOps/s $\color{#35bf28}+0.79\%$
test_tc_second_layer_nontensor 33.7910μs 8.6706μs 115.3321 KOps/s 115.6475 KOps/s $\color{#d91a1a}-0.27\%$
test_unbind 0.2924s 13.5110ms 74.0138 Ops/s 64.4355 Ops/s $\textbf{\color{#35bf28}+14.86\%}$
test_full_like 4.5646ms 4.3856ms 228.0177 Ops/s 228.3497 Ops/s $\color{#d91a1a}-0.15\%$
test_zeros_like 4.5530ms 4.3596ms 229.3769 Ops/s 229.1156 Ops/s $\color{#35bf28}+0.11\%$
test_ones_like 4.5351ms 4.3659ms 229.0488 Ops/s 228.8923 Ops/s $\color{#35bf28}+0.07\%$
test_clone 6.9707ms 6.4656ms 154.6639 Ops/s 156.3746 Ops/s $\color{#d91a1a}-1.09\%$
test_squeeze 76.1320μs 14.2768μs 70.0436 KOps/s 70.8100 KOps/s $\color{#d91a1a}-1.08\%$
test_unsqueeze 0.1735ms 0.1131ms 8.8396 KOps/s 8.9688 KOps/s $\color{#d91a1a}-1.44\%$
test_split 0.2668ms 0.1898ms 5.2686 KOps/s 5.3607 KOps/s $\color{#d91a1a}-1.72\%$
test_permute 0.3217ms 0.1945ms 5.1425 KOps/s 5.2116 KOps/s $\color{#d91a1a}-1.33\%$
test_stack 27.1452ms 26.6587ms 37.5111 Ops/s 55.4376 Ops/s $\textbf{\color{#d91a1a}-32.34\%}$
test_cat 19.1717ms 18.2143ms 54.9018 Ops/s 55.5535 Ops/s $\color{#d91a1a}-1.17\%$
test_sequential_tensordict 0.7127ms 0.5650ms 1.7699 KOps/s 1.7572 KOps/s $\color{#35bf28}+0.72\%$
test_sequential_graph_module 0.2360ms 0.1164ms 8.5886 KOps/s 9.1089 KOps/s $\textbf{\color{#d91a1a}-5.71\%}$
test_nested_tensordict 0.9300ms 0.8071ms 1.2390 KOps/s 1.2423 KOps/s $\color{#d91a1a}-0.26\%$
test_nested_graph_module 0.1695ms 0.1275ms 7.8454 KOps/s 8.2217 KOps/s $\color{#d91a1a}-4.58\%$

@vmoens
vmoens merged commit 99c712b into main Jul 9, 2026
72 of 73 checks passed
@vmoens
vmoens deleted the fix/torch-compat-squeeze-compile branch July 9, 2026 15:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. tensorclass Test

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant