Skip to content

[Distributed] Add tensorclass distributed tests - #1598

Merged
vmoens merged 3 commits into
gh/vmoens/78/basefrom
gh/vmoens/78/head
Feb 19, 2026
Merged

[Distributed] Add tensorclass distributed tests#1598
vmoens merged 3 commits into
gh/vmoens/78/basefrom
gh/vmoens/78/head

Conversation

@vmoens

@vmoens vmoens commented Feb 19, 2026 •
edited
Loading

Copy link
Copy Markdown
Collaborator

Stack from ghstack (oldest at bottom):

Add tests verifying that all distributed operations work when called on
tensorclass instances: broadcast, all_reduce, all_gather, scatter,
init_remote, and consolidated send/recv.

Co-authored-by: Cursor cursoragent@cursor.com

@vmoens
Update
26a138e 
[ghstack-poisoned]
vmoens added a commit that referenced this pull request Feb 19, 2026
@vmoens @cursoragent
[Distributed] Add tensorclass distributed tests
d3a5c72 
Add tests verifying that all distributed operations work when called on
tensorclass instances: broadcast, all_reduce, all_gather, scatter,
init_remote, and consolidated send/recv.

Co-authored-by: Cursor <cursoragent@cursor.com>
ghstack-source-id: 1bc6d71
Pull-Request: #1598
@vmoens vmoens mentioned this pull request Feb 19, 2026
Merged
@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Feb 19, 2026
This was referenced Feb 19, 2026
Merged
Merged
Merged
@github-actions

github-actions Bot commented Feb 19, 2026

Copy link
Copy Markdown
Contributor

PR Title Label Error

Unknown or invalid prefix [Distributed].

Current title: [Distributed] Add tensorclass distributed tests

Supported Prefixes

Your PR title must start with exactly one of these prefixes:

Prefix Label Applied Example
[BugFix] BugFix [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] Perf [Perf] Optimize tensor operations
[Deprecation] Deprecation [Deprecation] Mark old function
[Setup] Setup [Setup] Update build configuration

Note: Common variations like singular/plural are supported (e.g., [Doc] or [Docs]).

This was referenced Feb 19, 2026
Merged
Merged
@github-actions github-actions Bot added the Test label Feb 19, 2026
This was referenced Feb 19, 2026
Merged
Merged
Closed
@github-actions

github-actions Bot commented Feb 19, 2026

Copy link
Copy Markdown
Contributor

PR Title Label Error

Unknown or invalid prefix [Distributed].

Current title: [Distributed] Add tensorclass distributed tests

Supported Prefixes

Your PR title must start with exactly one of these prefixes:

Prefix Label Applied Example
[BugFix] BugFix [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] Perf [Perf] Optimize tensor operations
[Deprecation] Deprecation [Deprecation] Mark old function
[Setup] Setup [Setup] Update build configuration

Note: Common variations like singular/plural are supported (e.g., [Doc] or [Docs]).

@vmoens
Update
632e209 
[ghstack-poisoned]
vmoens added a commit that referenced this pull request Feb 19, 2026
@vmoens @cursoragent
[Distributed] Add tensorclass distributed tests
60f3cb8 
Add tests verifying that all distributed operations work when called on
tensorclass instances: broadcast, all_reduce, all_gather, scatter,
init_remote, and consolidated send/recv.

Co-authored-by: Cursor <cursoragent@cursor.com>
ghstack-source-id: 1a09ce8
Pull-Request: #1598
@github-actions

github-actions Bot commented Feb 19, 2026

Copy link
Copy Markdown
Contributor

PR Title Label Error

Unknown or invalid prefix [Distributed].

Current title: [Distributed] Add tensorclass distributed tests

Supported Prefixes

Your PR title must start with exactly one of these prefixes:

Prefix Label Applied Example
[BugFix] BugFix [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] Perf [Perf] Optimize tensor operations
[Deprecation] Deprecation [Deprecation] Mark old function
[Setup] Setup [Setup] Update build configuration

Note: Common variations like singular/plural are supported (e.g., [Doc] or [Docs]).

vmoens added a commit that referenced this pull request Feb 19, 2026
@vmoens @cursoragent
[Distributed] Add tensorclass distributed tests
cb4ce8c 
Add tests verifying that all distributed operations work when called on
tensorclass instances: broadcast, all_reduce, all_gather, scatter,
init_remote, and consolidated send/recv.

Co-authored-by: Cursor <cursoragent@cursor.com>
ghstack-source-id: 1a09ce8
Pull-Request: #1598
@github-actions

github-actions Bot commented Feb 19, 2026

Copy link
Copy Markdown
Contributor

PR Title Label Error

Unknown or invalid prefix [Distributed].

Current title: [Distributed] Add tensorclass distributed tests

Supported Prefixes

Your PR title must start with exactly one of these prefixes:

Prefix Label Applied Example
[BugFix] BugFix [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] Perf [Perf] Optimize tensor operations
[Deprecation] Deprecation [Deprecation] Mark old function
[Setup] Setup [Setup] Update build configuration

Note: Common variations like singular/plural are supported (e.g., [Doc] or [Docs]).

@github-actions

github-actions Bot commented Feb 19, 2026 •
edited
Loading

Copy link
Copy Markdown
Contributor

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

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

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_plain_set_nested 36.8510μs 14.7527μs 67.7843 KOps/s 67.8948 KOps/s $\color{#d91a1a}-0.16\%$
test_plain_set_stack_nested 47.3610μs 15.0347μs 66.5130 KOps/s 66.8575 KOps/s $\color{#d91a1a}-0.52\%$
test_plain_set_nested_inplace 53.9010μs 16.4773μs 60.6896 KOps/s 60.3751 KOps/s $\color{#35bf28}+0.52\%$
test_plain_set_stack_nested_inplace 0.4445ms 16.3261μs 61.2514 KOps/s 61.0549 KOps/s $\color{#35bf28}+0.32\%$
test_items 0.4215ms 5.8370μs 171.3205 KOps/s 172.3657 KOps/s $\color{#d91a1a}-0.61\%$
test_items_nested 0.8705ms 0.4522ms 2.2116 KOps/s 2.1761 KOps/s $\color{#35bf28}+1.63\%$
test_items_nested_locked 0.8714ms 0.4574ms 2.1863 KOps/s 2.1528 KOps/s $\color{#35bf28}+1.56\%$
test_items_nested_leaf 0.1323ms 96.2233μs 10.3925 KOps/s 10.0973 KOps/s $\color{#35bf28}+2.92\%$
test_items_stack_nested 0.8778ms 0.4558ms 2.1939 KOps/s 2.1680 KOps/s $\color{#35bf28}+1.20\%$
test_items_stack_nested_leaf 0.5174ms 95.4848μs 10.4729 KOps/s 10.3744 KOps/s $\color{#35bf28}+0.95\%$
test_items_stack_nested_locked 0.8772ms 0.4573ms 2.1869 KOps/s 2.1573 KOps/s $\color{#35bf28}+1.37\%$
test_keys 0.4192ms 4.1858μs 238.9018 KOps/s 234.9784 KOps/s $\color{#35bf28}+1.67\%$
test_keys_nested 0.5622ms 0.1203ms 8.3129 KOps/s 8.3026 KOps/s $\color{#35bf28}+0.12\%$
test_keys_nested_locked 0.7668ms 0.1289ms 7.7570 KOps/s 6.9727 KOps/s $\textbf{\color{#35bf28}+11.25\%}$
test_keys_nested_leaf 0.1468ms 0.1104ms 9.0569 KOps/s 8.9405 KOps/s $\color{#35bf28}+1.30\%$
test_keys_stack_nested 0.5442ms 0.1211ms 8.2545 KOps/s 8.2047 KOps/s $\color{#35bf28}+0.61\%$
test_keys_stack_nested_leaf 0.5601ms 0.1113ms 8.9823 KOps/s 8.9372 KOps/s $\color{#35bf28}+0.50\%$
test_keys_stack_nested_locked 0.5509ms 0.1301ms 7.6882 KOps/s 7.6570 KOps/s $\color{#35bf28}+0.41\%$
test_values 6.4962μs 1.0230μs 977.5001 KOps/s 973.0081 KOps/s $\color{#35bf28}+0.46\%$
test_values_nested 0.4624ms 47.4690μs 21.0664 KOps/s 20.7837 KOps/s $\color{#35bf28}+1.36\%$
test_values_nested_locked 0.4715ms 51.1375μs 19.5551 KOps/s 19.5249 KOps/s $\color{#35bf28}+0.15\%$
test_values_nested_leaf 0.4702ms 54.3721μs 18.3918 KOps/s 18.4147 KOps/s $\color{#d91a1a}-0.12\%$
test_values_stack_nested 87.7720μs 47.8859μs 20.8830 KOps/s 20.8534 KOps/s $\color{#35bf28}+0.14\%$
test_values_stack_nested_leaf 0.4724ms 54.1950μs 18.4519 KOps/s 18.4558 KOps/s $\color{#d91a1a}-0.02\%$
test_values_stack_nested_locked 0.4747ms 50.3086μs 19.8773 KOps/s 19.7435 KOps/s $\color{#35bf28}+0.68\%$
test_membership 69.5680μs 0.8495μs 1.1772 MOps/s 1.1725 MOps/s $\color{#35bf28}+0.41\%$
test_membership_nested 0.4287ms 2.8625μs 349.3461 KOps/s 336.8692 KOps/s $\color{#35bf28}+3.70\%$
test_membership_nested_leaf 29.1400μs 2.8560μs 350.1384 KOps/s 336.1196 KOps/s $\color{#35bf28}+4.17\%$
test_membership_stacked_nested 21.8710μs 2.9091μs 343.7470 KOps/s 339.9019 KOps/s $\color{#35bf28}+1.13\%$
test_membership_stacked_nested_leaf 21.5400μs 2.8864μs 346.4544 KOps/s 337.3854 KOps/s $\color{#35bf28}+2.69\%$
test_membership_nested_last 27.6010μs 4.3290μs 230.9979 KOps/s 230.4128 KOps/s $\color{#35bf28}+0.25\%$
test_membership_nested_leaf_last 0.4287ms 4.3892μs 227.8335 KOps/s 226.4200 KOps/s $\color{#35bf28}+0.62\%$
test_membership_stacked_nested_last 21.3010μs 4.3706μs 228.8036 KOps/s 228.1859 KOps/s $\color{#35bf28}+0.27\%$
test_membership_stacked_nested_leaf_last 38.0610μs 4.3522μs 229.7705 KOps/s 227.4669 KOps/s $\color{#35bf28}+1.01\%$
test_nested_getleaf 0.4407ms 21.5422μs 46.4204 KOps/s 46.9384 KOps/s $\color{#d91a1a}-1.10\%$
test_nested_get 0.4356ms 20.4303μs 48.9469 KOps/s 49.2301 KOps/s $\color{#d91a1a}-0.58\%$
test_stacked_getleaf 0.4397ms 21.3716μs 46.7911 KOps/s 46.9144 KOps/s $\color{#d91a1a}-0.26\%$
test_stacked_get 0.4421ms 20.2275μs 49.4376 KOps/s 49.3229 KOps/s $\color{#35bf28}+0.23\%$
test_nested_getitemleaf 53.9610μs 21.7406μs 45.9969 KOps/s 44.9936 KOps/s $\color{#35bf28}+2.23\%$
test_nested_getitem 0.4408ms 20.8899μs 47.8701 KOps/s 47.6626 KOps/s $\color{#35bf28}+0.44\%$
test_stacked_getitemleaf 0.4442ms 22.0381μs 45.3760 KOps/s 44.9514 KOps/s $\color{#35bf28}+0.94\%$
test_stacked_getitem 0.4457ms 20.8990μs 47.8492 KOps/s 47.4739 KOps/s $\color{#35bf28}+0.79\%$
test_lock_nested 0.5423ms 0.4689ms 2.1327 KOps/s 2.0738 KOps/s $\color{#35bf28}+2.84\%$
test_lock_stack_nested 0.5220ms 0.4753ms 2.1040 KOps/s 2.0739 KOps/s $\color{#35bf28}+1.45\%$
test_unlock_nested 0.4617ms 0.3788ms 2.6402 KOps/s 2.6065 KOps/s $\color{#35bf28}+1.29\%$
test_unlock_stack_nested 0.6205ms 0.3824ms 2.6148 KOps/s 2.5691 KOps/s $\color{#35bf28}+1.78\%$
test_flatten_speed 0.1686ms 0.1230ms 8.1290 KOps/s 8.2266 KOps/s $\color{#d91a1a}-1.19\%$
test_unflatten_speed 0.6168ms 0.5713ms 1.7505 KOps/s 1.7426 KOps/s $\color{#35bf28}+0.45\%$
test_common_ops 0.8476ms 0.6803ms 1.4700 KOps/s 1.4608 KOps/s $\color{#35bf28}+0.63\%$
test_creation 0.1084ms 2.9236μs 342.0409 KOps/s 337.7007 KOps/s $\color{#35bf28}+1.29\%$
test_creation_empty 33.6500μs 6.1308μs 163.1114 KOps/s 159.8836 KOps/s $\color{#35bf28}+2.02\%$
test_creation_nested_1 47.3010μs 10.5200μs 95.0570 KOps/s 94.1898 KOps/s $\color{#35bf28}+0.92\%$
test_creation_nested_2 40.2500μs 11.9130μs 83.9421 KOps/s 82.8688 KOps/s $\color{#35bf28}+1.30\%$
test_creation_many_keys[10] 44.6710μs 18.2911μs 54.6713 KOps/s 54.3355 KOps/s $\color{#35bf28}+0.62\%$
test_creation_many_keys[50] 0.1092ms 78.3172μs 12.7686 KOps/s 12.5815 KOps/s $\color{#35bf28}+1.49\%$
test_creation_many_keys[100] 0.2026ms 0.1537ms 6.5047 KOps/s 6.4753 KOps/s $\color{#35bf28}+0.45\%$
test_creation_nested_many_keys[10] 68.8510μs 39.5933μs 25.2568 KOps/s 25.1925 KOps/s $\color{#35bf28}+0.26\%$
test_creation_nested_many_keys[50] 0.1981ms 0.1609ms 6.2133 KOps/s 6.2431 KOps/s $\color{#d91a1a}-0.48\%$
test_clone 40.2210μs 13.5785μs 73.6457 KOps/s 73.8381 KOps/s $\color{#d91a1a}-0.26\%$
test_getitem[int] 1.5734ms 14.6594μs 68.2158 KOps/s 57.3859 KOps/s $\textbf{\color{#35bf28}+18.87\%}$
test_getitem[slice_int] 0.1420ms 24.4627μs 40.8786 KOps/s 40.6923 KOps/s $\color{#35bf28}+0.46\%$
test_getitem[range] 0.1753ms 62.5276μs 15.9929 KOps/s 15.8377 KOps/s $\color{#35bf28}+0.98\%$
test_getitem[tuple] 0.1387ms 23.6694μs 42.2487 KOps/s 41.8942 KOps/s $\color{#35bf28}+0.85\%$
test_getitem[list] 0.1788ms 57.6365μs 17.3501 KOps/s 17.1158 KOps/s $\color{#35bf28}+1.37\%$
test_setitem_dim[int] 48.0710μs 26.0146μs 38.4399 KOps/s 38.3549 KOps/s $\color{#35bf28}+0.22\%$
test_setitem_dim[slice_int] 76.3710μs 43.3242μs 23.0818 KOps/s 22.8115 KOps/s $\color{#35bf28}+1.19\%$
test_setitem_dim[range] 0.1272ms 94.3541μs 10.5984 KOps/s 10.4466 KOps/s $\color{#35bf28}+1.45\%$
test_setitem_dim[tuple] 61.1110μs 39.2840μs 25.4557 KOps/s 24.5111 KOps/s $\color{#35bf28}+3.85\%$
test_setitem 44.0000μs 17.7447μs 56.3548 KOps/s 55.7997 KOps/s $\color{#35bf28}+0.99\%$
test_set 56.0810μs 16.9321μs 59.0594 KOps/s 59.0893 KOps/s $\color{#d91a1a}-0.05\%$
test_set_shared 0.5503ms 0.2055ms 4.8663 KOps/s 4.8787 KOps/s $\color{#d91a1a}-0.25\%$
test_update 0.2039ms 21.1472μs 47.2876 KOps/s 46.2244 KOps/s $\color{#35bf28}+2.30\%$
test_update_nested 66.0520μs 33.4893μs 29.8603 KOps/s 29.8283 KOps/s $\color{#35bf28}+0.11\%$
test_update__nested 0.5600ms 34.3797μs 29.0869 KOps/s 28.7633 KOps/s $\color{#35bf28}+1.12\%$
test_set_nested 58.8710μs 18.8542μs 53.0385 KOps/s 52.6793 KOps/s $\color{#35bf28}+0.68\%$
test_set_nested_new 63.2210μs 23.5528μs 42.4578 KOps/s 41.3985 KOps/s $\color{#35bf28}+2.56\%$
test_select 74.4920μs 40.9382μs 24.4271 KOps/s 23.8387 KOps/s $\color{#35bf28}+2.47\%$
test_select_nested 0.1079ms 75.0356μs 13.3270 KOps/s 13.3955 KOps/s $\color{#d91a1a}-0.51\%$
test_exclude_nested 0.1655ms 92.4949μs 10.8114 KOps/s 10.8035 KOps/s $\color{#35bf28}+0.07\%$
test_empty[True] 0.4504ms 0.3922ms 2.5499 KOps/s 2.5525 KOps/s $\color{#d91a1a}-0.10\%$
test_empty[False] 7.0625μs 1.3357μs 748.6539 KOps/s 749.3939 KOps/s $\color{#d91a1a}-0.10\%$
test_to 0.1032ms 76.1776μs 13.1272 KOps/s 13.9104 KOps/s $\textbf{\color{#d91a1a}-5.63\%}$
test_to_nonblocking 0.1049ms 65.1153μs 15.3574 KOps/s 15.5877 KOps/s $\color{#d91a1a}-1.48\%$
test_unbind_speed 0.4091ms 0.3250ms 3.0766 KOps/s 3.0629 KOps/s $\color{#35bf28}+0.45\%$
test_unbind_speed_stack0 0.3712ms 0.3245ms 3.0818 KOps/s 3.0770 KOps/s $\color{#35bf28}+0.16\%$
test_unbind_speed_stack1 0.1033s 1.0253ms 975.3006 Ops/s 1.1742 KOps/s $\textbf{\color{#d91a1a}-16.94\%}$
test_split 1.1890ms 1.1504ms 869.2823 Ops/s 778.9891 Ops/s $\textbf{\color{#35bf28}+11.59\%}$
test_chunk 0.1029s 1.2055ms 829.5579 Ops/s 913.4669 Ops/s $\textbf{\color{#d91a1a}-9.19\%}$
test_to_cpu_blocking 19.7458ms 19.6162ms 50.9782 Ops/s 51.7244 Ops/s $\color{#d91a1a}-1.44\%$
test_to_cpu_global_sync 11.5461ms 11.3957ms 87.7523 Ops/s 80.9443 Ops/s $\textbf{\color{#35bf28}+8.41\%}$
test_to_cpu_event_sync 0.1151s 13.6747ms 73.1275 Ops/s 82.5084 Ops/s $\textbf{\color{#d91a1a}-11.37\%}$
test_to_cpu_default 12.7063ms 12.4634ms 80.2348 Ops/s 82.3797 Ops/s $\color{#d91a1a}-2.60\%$
test_consolidate[False-None] 4.2857ms 4.0843ms 244.8401 Ops/s 221.0958 Ops/s $\textbf{\color{#35bf28}+10.74\%}$
test_consolidate[default-None] 2.1279ms 2.0031ms 499.2361 Ops/s 493.6285 Ops/s $\color{#35bf28}+1.14\%$
test_consolidate[reduce-overhead-None] 1.9989ms 1.9260ms 519.2049 Ops/s 513.0671 Ops/s $\color{#35bf28}+1.20\%$
test_consolidate_njt[False-None] 8.6623ms 8.4644ms 118.1420 Ops/s 118.3910 Ops/s $\color{#d91a1a}-0.21\%$
test_to[False-False-None] 2.1801ms 2.1083ms 474.3061 Ops/s 485.0802 Ops/s $\color{#d91a1a}-2.22\%$
test_to[True-False-None] 2.1735ms 1.9461ms 513.8597 Ops/s 525.9688 Ops/s $\color{#d91a1a}-2.30\%$
test_to[within-False-None] 6.1883ms 6.1066ms 163.7572 Ops/s 165.0948 Ops/s $\color{#d91a1a}-0.81\%$
test_to[True-default-None] 7.7507ms 7.5550ms 132.3635 Ops/s 130.6747 Ops/s $\color{#35bf28}+1.29\%$
test_to_njt[False-False-None] 8.8797ms 8.5527ms 116.9220 Ops/s 116.7764 Ops/s $\color{#35bf28}+0.12\%$
test_to_njt[True-False-None] 7.1086ms 6.9275ms 144.3525 Ops/s 143.3649 Ops/s $\color{#35bf28}+0.69\%$
test_to_njt[within-False-None] 15.6788ms 15.5857ms 64.1614 Ops/s 64.7811 Ops/s $\color{#d91a1a}-0.96\%$
test_creation[device0] 0.3023ms 0.1147ms 8.7162 KOps/s 8.4072 KOps/s $\color{#35bf28}+3.67\%$
test_creation_from_tensor 0.4013ms 0.1114ms 8.9747 KOps/s 8.5527 KOps/s $\color{#35bf28}+4.93\%$
test_add_one[memmap_tensor0] 0.3075ms 6.5215μs 153.3392 KOps/s 152.0900 KOps/s $\color{#35bf28}+0.82\%$
test_contiguous[memmap_tensor0] 31.8600μs 0.6714μs 1.4895 MOps/s 2.1632 MOps/s $\textbf{\color{#d91a1a}-31.14\%}$
test_stack[memmap_tensor0] 32.9110μs 4.6108μs 216.8843 KOps/s 218.1833 KOps/s $\color{#d91a1a}-0.60\%$
test_memmaptd_index 0.9694ms 0.2589ms 3.8620 KOps/s 3.9148 KOps/s $\color{#d91a1a}-1.35\%$
test_memmaptd_index_astensor 0.5128ms 0.3547ms 2.8194 KOps/s 2.8474 KOps/s $\color{#d91a1a}-0.98\%$
test_memmaptd_index_op 0.8436ms 0.5971ms 1.6748 KOps/s 1.6716 KOps/s $\color{#35bf28}+0.19\%$
test_serialize_model 0.1387s 0.1373s 7.2814 Ops/s 7.2885 Ops/s $\color{#d91a1a}-0.10\%$
test_serialize_model_pickle 1.3496s 1.2110s 0.8258 Ops/s 0.8258 Ops/s $-0.01\%$
test_serialize_weights 0.1363s 0.1351s 7.4034 Ops/s 7.3286 Ops/s $\color{#35bf28}+1.02\%$
test_serialize_weights_returnearly 0.2817s 94.5338ms 10.5782 Ops/s 5.6351 Ops/s $\textbf{\color{#35bf28}+87.72\%}$
test_serialize_weights_pickle 1.3677s 1.2131s 0.8244 Ops/s 0.8148 Ops/s $\color{#35bf28}+1.17\%$
test_reshape_pytree 0.2072ms 33.8264μs 29.5627 KOps/s 29.9354 KOps/s $\color{#d91a1a}-1.24\%$
test_reshape_td 75.0010μs 44.1158μs 22.6676 KOps/s 22.4353 KOps/s $\color{#35bf28}+1.04\%$
test_view_pytree 0.1941ms 33.2422μs 30.0822 KOps/s 30.6159 KOps/s $\color{#d91a1a}-1.74\%$
test_view_td 86.5910μs 51.7344μs 19.3295 KOps/s 19.0884 KOps/s $\color{#35bf28}+1.26\%$
test_unbind_pytree 0.2319ms 37.3533μs 26.7714 KOps/s 27.0617 KOps/s $\color{#d91a1a}-1.07\%$
test_unbind_td 0.1701ms 49.1229μs 20.3571 KOps/s 20.3223 KOps/s $\color{#35bf28}+0.17\%$
test_split_pytree 0.2565ms 43.2060μs 23.1449 KOps/s 23.6464 KOps/s $\color{#d91a1a}-2.12\%$
test_split_td 0.1162ms 64.7811μs 15.4366 KOps/s 15.2451 KOps/s $\color{#35bf28}+1.26\%$
test_add_pytree 0.2342ms 42.4713μs 23.5453 KOps/s 23.8935 KOps/s $\color{#d91a1a}-1.46\%$
test_add_td 97.1120μs 53.5346μs 18.6795 KOps/s 18.7921 KOps/s $\color{#d91a1a}-0.60\%$
test_compile_add_one_nested[tensordict-compile] 0.1958ms 0.1410ms 7.0901 KOps/s 6.8209 KOps/s $\color{#35bf28}+3.95\%$
test_compile_add_one_nested[tensordict-eager] 0.3513ms 0.1935ms 5.1679 KOps/s 5.2463 KOps/s $\color{#d91a1a}-1.49\%$
test_compile_add_one_nested[pytree-compile] 0.2085ms 0.1120ms 8.9305 KOps/s 9.0880 KOps/s $\color{#d91a1a}-1.73\%$
test_compile_add_one_nested[pytree-eager] 0.4358ms 0.1847ms 5.4148 KOps/s 5.6101 KOps/s $\color{#d91a1a}-3.48\%$
test_compile_copy_nested[tensordict-compile] 0.2250ms 30.3296μs 32.9711 KOps/s 30.6202 KOps/s $\textbf{\color{#35bf28}+7.68\%}$
test_compile_copy_nested[tensordict-eager] 91.4420μs 51.9122μs 19.2633 KOps/s 19.0500 KOps/s $\color{#35bf28}+1.12\%$
test_compile_copy_nested[pytree-compile] 0.1135ms 9.7837μs 102.2112 KOps/s 100.5141 KOps/s $\color{#35bf28}+1.69\%$
test_compile_copy_nested[pytree-eager] 0.4354ms 69.8432μs 14.3178 KOps/s 14.5251 KOps/s $\color{#d91a1a}-1.43\%$
test_compile_add_one_flat[tensordict-compile] 0.2891ms 0.1780ms 5.6165 KOps/s 5.3144 KOps/s $\textbf{\color{#35bf28}+5.68\%}$
test_compile_add_one_flat[tensordict-eager] 0.3598ms 0.2530ms 3.9519 KOps/s 3.9080 KOps/s $\color{#35bf28}+1.12\%$
test_compile_add_one_flat[tensorclass-compile] 0.1656ms 0.1198ms 8.3487 KOps/s 8.3173 KOps/s $\color{#35bf28}+0.38\%$
test_compile_add_one_flat[tensorclass-eager] 0.1061ms 69.3219μs 14.4255 KOps/s 14.4400 KOps/s $\color{#d91a1a}-0.10\%$
test_compile_add_one_flat[pytree-compile] 0.2295ms 0.1590ms 6.2875 KOps/s 6.1841 KOps/s $\color{#35bf28}+1.67\%$
test_compile_add_one_flat[pytree-eager] 0.8359ms 0.5416ms 1.8464 KOps/s 1.8945 KOps/s $\color{#d91a1a}-2.54\%$
test_compile_add_self_flat[tensordict-eager] 0.3625ms 0.3081ms 3.2457 KOps/s 3.2320 KOps/s $\color{#35bf28}+0.42\%$
test_compile_add_self_flat[tensordict-compile] 0.2498ms 0.1831ms 5.4604 KOps/s 5.0471 KOps/s $\textbf{\color{#35bf28}+8.19\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1429ms 86.6048μs 11.5467 KOps/s 11.7507 KOps/s $\color{#d91a1a}-1.74\%$
test_compile_add_self_flat[tensorclass-compile] 0.1817ms 0.1204ms 8.3036 KOps/s 7.9285 KOps/s $\color{#35bf28}+4.73\%$
test_compile_add_self_flat[pytree-eager] 0.6662ms 0.4399ms 2.2734 KOps/s 2.2513 KOps/s $\color{#35bf28}+0.98\%$
test_compile_add_self_flat[pytree-compile] 0.2000ms 0.1583ms 6.3166 KOps/s 6.0459 KOps/s $\color{#35bf28}+4.48\%$
test_compile_copy_flat[tensordict-compile] 73.1910μs 23.9661μs 41.7255 KOps/s 37.6080 KOps/s $\textbf{\color{#35bf28}+10.95\%}$
test_compile_copy_flat[tensordict-eager] 64.6320μs 40.7597μs 24.5340 KOps/s 24.1740 KOps/s $\color{#35bf28}+1.49\%$
test_compile_copy_flat[pytree-compile] 0.1253ms 10.8758μs 91.9471 KOps/s 91.6637 KOps/s $\color{#35bf28}+0.31\%$
test_compile_copy_flat[pytree-eager] 0.4282ms 52.7728μs 18.9492 KOps/s 19.2639 KOps/s $\color{#d91a1a}-1.63\%$
test_compile_assign_and_add[tensordict-compile] 2.0071ms 0.1756ms 5.6932 KOps/s 5.5499 KOps/s $\color{#35bf28}+2.58\%$
test_compile_assign_and_add[tensordict-eager] 3.6208ms 3.3585ms 297.7512 Ops/s 306.3910 Ops/s $\color{#d91a1a}-2.82\%$
test_compile_assign_and_add[pytree-compile] 2.0051ms 0.1618ms 6.1786 KOps/s 6.0353 KOps/s $\color{#35bf28}+2.37\%$
test_compile_assign_and_add[pytree-eager] 3.0077ms 2.8668ms 348.8254 Ops/s 360.1317 Ops/s $\color{#d91a1a}-3.14\%$
test_compile_indexing[tensor-tensordict-compile] 0.2292ms 0.1098ms 9.1085 KOps/s 8.7778 KOps/s $\color{#35bf28}+3.77\%$
test_compile_indexing[tensor-tensordict-eager] 0.1997s 89.8206μs 11.1333 KOps/s 13.7851 KOps/s $\textbf{\color{#d91a1a}-19.24\%}$
test_compile_indexing[tensor-tensorclass-compile] 0.1533ms 97.7129μs 10.2341 KOps/s 10.1819 KOps/s $\color{#35bf28}+0.51\%$
test_compile_indexing[tensor-tensorclass-eager] 0.2659ms 47.3454μs 21.1214 KOps/s 21.9376 KOps/s $\color{#d91a1a}-3.72\%$
test_compile_indexing[tensor-pytree-compile] 0.1412ms 98.6675μs 10.1350 KOps/s 10.0844 KOps/s $\color{#35bf28}+0.50\%$
test_compile_indexing[tensor-pytree-eager] 0.2934ms 47.7554μs 20.9400 KOps/s 22.3455 KOps/s $\textbf{\color{#d91a1a}-6.29\%}$
test_compile_indexing[slice-tensordict-compile] 0.5035ms 57.1457μs 17.4991 KOps/s 17.3716 KOps/s $\color{#35bf28}+0.73\%$
test_compile_indexing[slice-tensordict-eager] 0.4454ms 27.9751μs 35.7461 KOps/s 35.5634 KOps/s $\color{#35bf28}+0.51\%$
test_compile_indexing[slice-tensorclass-compile] 0.1342ms 45.3751μs 22.0385 KOps/s 21.6206 KOps/s $\color{#35bf28}+1.93\%$
test_compile_indexing[slice-tensorclass-eager] 0.4394ms 22.8848μs 43.6972 KOps/s 44.1743 KOps/s $\color{#d91a1a}-1.08\%$
test_compile_indexing[slice-pytree-compile] 0.4755ms 46.3697μs 21.5658 KOps/s 21.7714 KOps/s $\color{#d91a1a}-0.94\%$
test_compile_indexing[slice-pytree-eager] 0.4528ms 23.2832μs 42.9494 KOps/s 44.1431 KOps/s $\color{#d91a1a}-2.70\%$
test_compile_indexing[int-tensordict-compile] 0.5022ms 57.1600μs 17.4948 KOps/s 17.1146 KOps/s $\color{#35bf28}+2.22\%$
test_compile_indexing[int-tensordict-eager] 0.4580ms 28.0190μs 35.6900 KOps/s 36.3859 KOps/s $\color{#d91a1a}-1.91\%$
test_compile_indexing[int-tensorclass-compile] 0.4853ms 47.8533μs 20.8972 KOps/s 21.2067 KOps/s $\color{#d91a1a}-1.46\%$
test_compile_indexing[int-tensorclass-eager] 0.2679ms 22.9161μs 43.6374 KOps/s 44.3295 KOps/s $\color{#d91a1a}-1.56\%$
test_compile_indexing[int-pytree-compile] 0.4785ms 45.3985μs 22.0271 KOps/s 21.8139 KOps/s $\color{#35bf28}+0.98\%$
test_compile_indexing[int-pytree-eager] 0.4567ms 23.0695μs 43.3473 KOps/s 44.4995 KOps/s $\color{#d91a1a}-2.59\%$
test_mod_add[eager] 0.4965ms 50.3271μs 19.8700 KOps/s 19.8358 KOps/s $\color{#35bf28}+0.17\%$
test_mod_add[compile] 0.1365ms 0.1053ms 9.4931 KOps/s 9.3778 KOps/s $\color{#35bf28}+1.23\%$
test_mod_add[compile-overhead] 0.2348ms 0.1485ms 6.7320 KOps/s 6.6624 KOps/s $\color{#35bf28}+1.05\%$
test_mod_wrap[eager] 0.3767ms 0.2919ms 3.4259 KOps/s 3.3928 KOps/s $\color{#35bf28}+0.98\%$
test_mod_wrap[compile] 0.4257ms 0.3525ms 2.8370 KOps/s 2.8322 KOps/s $\color{#35bf28}+0.17\%$
test_mod_wrap[compile-overhead] 7.8793ms 3.9852ms 250.9268 Ops/s 244.6194 Ops/s $\color{#35bf28}+2.58\%$
test_mod_wrap_and_backward[eager] 1.6176ms 1.4951ms 668.8719 Ops/s 662.0651 Ops/s $\color{#35bf28}+1.03\%$
test_mod_wrap_and_backward[compile] 1.9648ms 1.4638ms 683.1477 Ops/s 686.4069 Ops/s $\color{#d91a1a}-0.47\%$
test_mod_wrap_and_backward[compile-overhead] 1.2459ms 0.8880ms 1.1262 KOps/s 1.1148 KOps/s $\color{#35bf28}+1.02\%$
test_seq_add[eager] 0.2228ms 0.1572ms 6.3599 KOps/s 6.3907 KOps/s $\color{#d91a1a}-0.48\%$
test_seq_add[compile] 0.3167ms 0.1172ms 8.5354 KOps/s 8.4179 KOps/s $\color{#35bf28}+1.40\%$
test_seq_add[compile-overhead] 0.5259ms 0.1572ms 6.3600 KOps/s 6.2796 KOps/s $\color{#35bf28}+1.28\%$
test_seq_wrap[eager] 0.6048ms 0.5293ms 1.8892 KOps/s 1.8973 KOps/s $\color{#d91a1a}-0.43\%$
test_seq_wrap[compile] 0.4486ms 0.3713ms 2.6932 KOps/s 2.6792 KOps/s $\color{#35bf28}+0.52\%$
test_seq_wrap[compile-overhead] 0.3438ms 0.2683ms 3.7275 KOps/s 3.6928 KOps/s $\color{#35bf28}+0.94\%$
test_func_call_runtime[False-eager] 0.9084ms 0.8489ms 1.1781 KOps/s 1.1510 KOps/s $\color{#35bf28}+2.35\%$
test_func_call_runtime[False-compile] 0.9960ms 0.9219ms 1.0847 KOps/s 1.0840 KOps/s $\color{#35bf28}+0.06\%$
test_func_call_runtime[False-compile-overhead] 0.5371ms 0.4645ms 2.1529 KOps/s 2.1370 KOps/s $\color{#35bf28}+0.74\%$
test_func_call_runtime[True-eager] 1.1899ms 1.0867ms 920.2548 Ops/s 917.7579 Ops/s $\color{#35bf28}+0.27\%$
test_func_call_runtime[True-compile] 1.0942ms 0.9337ms 1.0710 KOps/s 1.0736 KOps/s $\color{#d91a1a}-0.24\%$
test_func_call_runtime[True-compile-overhead] 0.5199ms 0.4780ms 2.0921 KOps/s 2.0759 KOps/s $\color{#35bf28}+0.78\%$
test_func_call_cm_runtime[False-eager] 1.2607ms 0.8869ms 1.1275 KOps/s 1.1958 KOps/s $\textbf{\color{#d91a1a}-5.71\%}$
test_func_call_cm_runtime[False-compile] 1.0079ms 0.9257ms 1.0803 KOps/s 1.0895 KOps/s $\color{#d91a1a}-0.84\%$
test_func_call_cm_runtime[False-compile-overhead] 0.5967ms 0.4668ms 2.1423 KOps/s 2.1266 KOps/s $\color{#35bf28}+0.74\%$
test_func_call_cm_runtime[True-eager] 1.3336ms 1.2416ms 805.3971 Ops/s 801.7823 Ops/s $\color{#35bf28}+0.45\%$
test_func_call_cm_runtime[True-compile] 1.0534ms 0.9694ms 1.0316 KOps/s 1.0353 KOps/s $\color{#d91a1a}-0.36\%$
test_func_call_cm_runtime[True-compile-overhead] 0.6091ms 0.5112ms 1.9563 KOps/s 1.9308 KOps/s $\color{#35bf28}+1.32\%$
test_vmap_func_call_cm_runtime[eager] 2.8602ms 2.3754ms 420.9739 Ops/s 420.0931 Ops/s $\color{#35bf28}+0.21\%$
test_vmap_func_call_cm_runtime[compile] 1.0967ms 0.9810ms 1.0193 KOps/s 1.0183 KOps/s $\color{#35bf28}+0.10\%$
test_vmap_func_call_cm_runtime[compile-overhead] 0.5972ms 0.5148ms 1.9426 KOps/s 1.9257 KOps/s $\color{#35bf28}+0.88\%$
test_distributed 0.6307ms 0.1531ms 6.5317 KOps/s 6.4815 KOps/s $\color{#35bf28}+0.77\%$
test_tdmodule 62.5010μs 27.5682μs 36.2736 KOps/s 35.1247 KOps/s $\color{#35bf28}+3.27\%$
test_tdmodule_dispatch 74.0310μs 45.1567μs 22.1451 KOps/s 21.9493 KOps/s $\color{#35bf28}+0.89\%$
test_tdseq 47.4910μs 26.9831μs 37.0602 KOps/s 36.6804 KOps/s $\color{#35bf28}+1.04\%$
test_tdseq_dispatch 68.6810μs 47.4865μs 21.0586 KOps/s 20.6019 KOps/s $\color{#35bf28}+2.22\%$
test_instantiation_functorch 2.1356ms 2.0892ms 478.6473 Ops/s 483.0376 Ops/s $\color{#d91a1a}-0.91\%$
test_exec_functorch 0.2436ms 0.1795ms 5.5698 KOps/s 5.5366 KOps/s $\color{#35bf28}+0.60\%$
test_exec_functional_call 0.2042ms 0.1613ms 6.2014 KOps/s 6.2604 KOps/s $\color{#d91a1a}-0.94\%$
test_exec_td_decorator 0.4515ms 0.2378ms 4.2047 KOps/s 4.2562 KOps/s $\color{#d91a1a}-1.21\%$
test_vmap_mlp_speed_decorator[True-True] 0.9980ms 0.8268ms 1.2095 KOps/s 1.2074 KOps/s $\color{#35bf28}+0.17\%$
test_vmap_mlp_speed_decorator[True-False] 1.0363ms 0.8256ms 1.2113 KOps/s 1.2091 KOps/s $\color{#35bf28}+0.18\%$
test_vmap_mlp_speed_decorator[False-True] 0.9138ms 0.7152ms 1.3982 KOps/s 1.3961 KOps/s $\color{#35bf28}+0.15\%$
test_vmap_mlp_speed_decorator[False-False] 0.8988ms 0.7160ms 1.3966 KOps/s 1.3977 KOps/s $\color{#d91a1a}-0.08\%$
test_vmap_transformer_speed_decorator[True-True] 21.5533ms 20.6102ms 48.5197 Ops/s 48.6393 Ops/s $\color{#d91a1a}-0.25\%$
test_vmap_transformer_speed_decorator[True-False] 20.7187ms 20.5906ms 48.5658 Ops/s 48.5694 Ops/s $-0.01\%$
test_vmap_transformer_speed_decorator[False-True] 20.5504ms 20.3862ms 49.0528 Ops/s 48.9986 Ops/s $\color{#35bf28}+0.11\%$
test_vmap_transformer_speed_decorator[False-False] 20.5650ms 20.3834ms 49.0595 Ops/s 49.0124 Ops/s $\color{#35bf28}+0.10\%$
test_to_module_speed[True] 1.6140ms 1.4757ms 677.6381 Ops/s 664.9706 Ops/s $\color{#35bf28}+1.90\%$
test_to_module_speed[False] 1.5448ms 1.4464ms 691.3599 Ops/s 686.4177 Ops/s $\color{#35bf28}+0.72\%$
test_tc_init 77.4720μs 46.5527μs 21.4810 KOps/s 21.6262 KOps/s $\color{#d91a1a}-0.67\%$
test_tc_init_tensor_only 45.5610μs 9.8321μs 101.7079 KOps/s 100.0319 KOps/s $\color{#35bf28}+1.68\%$
test_tc_init_nested 0.1329ms 89.7376μs 11.1436 KOps/s 10.9725 KOps/s $\color{#35bf28}+1.56\%$
test_tc_init_many_fields 34.8900μs 16.5702μs 60.3494 KOps/s 59.3421 KOps/s $\color{#35bf28}+1.70\%$
test_tc_first_layer_tensor 39.9810μs 1.8705μs 534.6265 KOps/s 537.4052 KOps/s $\color{#d91a1a}-0.52\%$
test_tc_first_layer_tensor_only 5.4730μs 0.7446μs 1.3430 MOps/s 1.3085 MOps/s $\color{#35bf28}+2.63\%$
test_tc_first_layer_tensor_set 31.0200μs 3.9292μs 254.5064 KOps/s 252.8167 KOps/s $\color{#35bf28}+0.67\%$
test_tc_first_layer_tensor_only_set 19.9700μs 3.1697μs 315.4870 KOps/s 306.9813 KOps/s $\color{#35bf28}+2.77\%$
test_tc_first_layer_nontensor 38.8910μs 6.2148μs 160.9061 KOps/s 160.8009 KOps/s $\color{#35bf28}+0.07\%$
test_tc_second_layer_tensor 48.1810μs 4.3896μs 227.8091 KOps/s 222.4352 KOps/s $\color{#35bf28}+2.42\%$
test_tc_second_layer_nontensor 51.9510μs 8.7519μs 114.2609 KOps/s 114.8331 KOps/s $\color{#d91a1a}-0.50\%$
test_unbind 0.2459s 14.0820ms 71.0124 Ops/s 56.3270 Ops/s $\textbf{\color{#35bf28}+26.07\%}$
test_full_like 4.6481ms 4.3994ms 227.3028 Ops/s 227.1056 Ops/s $\color{#35bf28}+0.09\%$
test_zeros_like 10.7291ms 4.4117ms 226.6678 Ops/s 227.7521 Ops/s $\color{#d91a1a}-0.48\%$
test_ones_like 4.5448ms 4.3995ms 227.2993 Ops/s 227.1489 Ops/s $\color{#35bf28}+0.07\%$
test_clone 7.0497ms 6.6256ms 150.9300 Ops/s 149.7535 Ops/s $\color{#35bf28}+0.79\%$
test_squeeze 0.1914ms 14.4387μs 69.2583 KOps/s 70.9632 KOps/s $\color{#d91a1a}-2.40\%$
test_unsqueeze 0.2703ms 0.1111ms 9.0014 KOps/s 9.0600 KOps/s $\color{#d91a1a}-0.65\%$
test_split 0.2583ms 0.1814ms 5.5140 KOps/s 5.4463 KOps/s $\color{#35bf28}+1.24\%$
test_permute 0.2678ms 0.2050ms 4.8775 KOps/s 4.8499 KOps/s $\color{#35bf28}+0.57\%$
test_stack 52.4050ms 51.6413ms 19.3644 Ops/s 19.2302 Ops/s $\color{#35bf28}+0.70\%$
test_cat 44.1605ms 43.5477ms 22.9633 Ops/s 19.2497 Ops/s $\textbf{\color{#35bf28}+19.29\%}$
test_sequential_tensordict 0.2682ms 0.2136ms 4.6823 KOps/s 4.6090 KOps/s $\color{#35bf28}+1.59\%$
test_sequential_graph_module 0.5375ms 0.1205ms 8.3019 KOps/s 8.4408 KOps/s $\color{#d91a1a}-1.65\%$
test_nested_tensordict 0.5958ms 0.2814ms 3.5536 KOps/s 3.5507 KOps/s $\color{#35bf28}+0.08\%$
test_nested_graph_module 0.1768ms 0.1282ms 7.8011 KOps/s 7.8103 KOps/s $\color{#d91a1a}-0.12\%$

@github-actions

github-actions Bot commented Feb 19, 2026 •
edited
Loading

Copy link
Copy Markdown
Contributor

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

Total Benchmarks: 247. Improved: $\large\color{#35bf28}17$. Worsened: $\large\color{#d91a1a}10$.

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_plain_set_nested 39.2910μs 14.6029μs 68.4796 KOps/s 68.2593 KOps/s $\color{#35bf28}+0.32\%$
test_plain_set_stack_nested 81.0930μs 15.0318μs 66.5258 KOps/s 66.7389 KOps/s $\color{#d91a1a}-0.32\%$
test_plain_set_nested_inplace 60.2220μs 16.2962μs 61.3641 KOps/s 60.4903 KOps/s $\color{#35bf28}+1.44\%$
test_plain_set_stack_nested_inplace 41.0710μs 16.1845μs 61.7873 KOps/s 61.0055 KOps/s $\color{#35bf28}+1.28\%$
test_items 35.9210μs 5.9695μs 167.5172 KOps/s 172.5878 KOps/s $\color{#d91a1a}-2.94\%$
test_items_nested 0.5249ms 0.4537ms 2.2042 KOps/s 2.2064 KOps/s $\color{#d91a1a}-0.10\%$
test_items_nested_locked 0.5166ms 0.4590ms 2.1788 KOps/s 2.1783 KOps/s $\color{#35bf28}+0.02\%$
test_items_nested_leaf 0.1326ms 96.9672μs 10.3128 KOps/s 10.3229 KOps/s $\color{#d91a1a}-0.10\%$
test_items_stack_nested 0.5247ms 0.4644ms 2.1535 KOps/s 2.2206 KOps/s $\color{#d91a1a}-3.02\%$
test_items_stack_nested_leaf 0.1279ms 98.1125μs 10.1924 KOps/s 10.4596 KOps/s $\color{#d91a1a}-2.55\%$
test_items_stack_nested_locked 0.5081ms 0.4622ms 2.1637 KOps/s 2.1860 KOps/s $\color{#d91a1a}-1.02\%$
test_keys 33.9910μs 4.2014μs 238.0163 KOps/s 233.4868 KOps/s $\color{#35bf28}+1.94\%$
test_keys_nested 0.1537ms 0.1201ms 8.3297 KOps/s 8.3222 KOps/s $\color{#35bf28}+0.09\%$
test_keys_nested_locked 0.7313ms 0.1286ms 7.7787 KOps/s 7.0273 KOps/s $\textbf{\color{#35bf28}+10.69\%}$
test_keys_nested_leaf 0.1486ms 0.1108ms 9.0270 KOps/s 9.0233 KOps/s $\color{#35bf28}+0.04\%$
test_keys_stack_nested 0.1979ms 0.1192ms 8.3898 KOps/s 8.2179 KOps/s $\color{#35bf28}+2.09\%$
test_keys_stack_nested_leaf 0.1481ms 0.1110ms 9.0055 KOps/s 8.9772 KOps/s $\color{#35bf28}+0.32\%$
test_keys_stack_nested_locked 0.1674ms 0.1302ms 7.6816 KOps/s 7.7273 KOps/s $\color{#d91a1a}-0.59\%$
test_values 5.5262μs 1.0165μs 983.7668 KOps/s 970.3495 KOps/s $\color{#35bf28}+1.38\%$
test_values_nested 88.2530μs 47.9127μs 20.8713 KOps/s 21.1302 KOps/s $\color{#d91a1a}-1.23\%$
test_values_nested_locked 90.4330μs 50.9686μs 19.6199 KOps/s 19.7566 KOps/s $\color{#d91a1a}-0.69\%$
test_values_nested_leaf 77.2420μs 54.5000μs 18.3486 KOps/s 18.5289 KOps/s $\color{#d91a1a}-0.97\%$
test_values_stack_nested 90.1430μs 47.9607μs 20.8504 KOps/s 20.9570 KOps/s $\color{#d91a1a}-0.51\%$
test_values_stack_nested_leaf 81.1830μs 54.1805μs 18.4568 KOps/s 18.4782 KOps/s $\color{#d91a1a}-0.12\%$
test_values_stack_nested_locked 83.0230μs 50.7084μs 19.7206 KOps/s 19.8423 KOps/s $\color{#d91a1a}-0.61\%$
test_membership 5.1385μs 0.8571μs 1.1667 MOps/s 1.1757 MOps/s $\color{#d91a1a}-0.77\%$
test_membership_nested 23.9610μs 2.9394μs 340.2072 KOps/s 343.0988 KOps/s $\color{#d91a1a}-0.84\%$
test_membership_nested_leaf 19.1755μs 2.8675μs 348.7311 KOps/s 342.7111 KOps/s $\color{#35bf28}+1.76\%$
test_membership_stacked_nested 32.1910μs 2.9265μs 341.7010 KOps/s 339.3416 KOps/s $\color{#35bf28}+0.70\%$
test_membership_stacked_nested_leaf 33.4410μs 2.9479μs 339.2224 KOps/s 341.4705 KOps/s $\color{#d91a1a}-0.66\%$
test_membership_nested_last 30.1310μs 4.4270μs 225.8852 KOps/s 230.1545 KOps/s $\color{#d91a1a}-1.85\%$
test_membership_nested_leaf_last 33.1310μs 4.3839μs 228.1081 KOps/s 229.2595 KOps/s $\color{#d91a1a}-0.50\%$
test_membership_stacked_nested_last 31.6910μs 4.3593μs 229.3963 KOps/s 229.0972 KOps/s $\color{#35bf28}+0.13\%$
test_membership_stacked_nested_leaf_last 34.0810μs 4.3698μs 228.8455 KOps/s 229.7290 KOps/s $\color{#d91a1a}-0.38\%$
test_nested_getleaf 60.7220μs 21.3566μs 46.8239 KOps/s 46.4829 KOps/s $\color{#35bf28}+0.73\%$
test_nested_get 44.9020μs 20.1248μs 49.6899 KOps/s 50.0687 KOps/s $\color{#d91a1a}-0.76\%$
test_stacked_getleaf 58.6320μs 21.3729μs 46.7883 KOps/s 47.2601 KOps/s $\color{#d91a1a}-1.00\%$
test_stacked_get 42.4010μs 20.1537μs 49.6186 KOps/s 50.9385 KOps/s $\color{#d91a1a}-2.59\%$
test_nested_getitemleaf 45.8820μs 21.7173μs 46.0463 KOps/s 45.7849 KOps/s $\color{#35bf28}+0.57\%$
test_nested_getitem 50.5810μs 20.7003μs 48.3086 KOps/s 48.3207 KOps/s $\color{#d91a1a}-0.03\%$
test_stacked_getitemleaf 49.6610μs 21.5106μs 46.4888 KOps/s 46.0244 KOps/s $\color{#35bf28}+1.01\%$
test_stacked_getitem 44.5310μs 20.6766μs 48.3639 KOps/s 48.2309 KOps/s $\color{#35bf28}+0.28\%$
test_lock_nested 0.5384ms 0.4705ms 2.1253 KOps/s 2.0933 KOps/s $\color{#35bf28}+1.53\%$
test_lock_stack_nested 0.5164ms 0.4768ms 2.0971 KOps/s 2.0766 KOps/s $\color{#35bf28}+0.99\%$
test_unlock_nested 0.4669ms 0.3797ms 2.6334 KOps/s 2.6160 KOps/s $\color{#35bf28}+0.66\%$
test_unlock_stack_nested 0.4247ms 0.3826ms 2.6140 KOps/s 2.5886 KOps/s $\color{#35bf28}+0.98\%$
test_flatten_speed 0.1685ms 0.1219ms 8.2038 KOps/s 8.2806 KOps/s $\color{#d91a1a}-0.93\%$
test_unflatten_speed 0.6256ms 0.5754ms 1.7378 KOps/s 1.7407 KOps/s $\color{#d91a1a}-0.17\%$
test_common_ops 0.8184ms 0.6825ms 1.4652 KOps/s 1.4152 KOps/s $\color{#35bf28}+3.53\%$
test_creation 0.1137ms 2.9446μs 339.6039 KOps/s 341.2086 KOps/s $\color{#d91a1a}-0.47\%$
test_creation_empty 38.4010μs 6.2298μs 160.5186 KOps/s 162.5882 KOps/s $\color{#d91a1a}-1.27\%$
test_creation_nested_1 37.3810μs 10.6180μs 94.1797 KOps/s 95.2576 KOps/s $\color{#d91a1a}-1.13\%$
test_creation_nested_2 36.9510μs 12.0210μs 83.1875 KOps/s 84.6043 KOps/s $\color{#d91a1a}-1.67\%$
test_creation_many_keys[10] 43.5810μs 18.3546μs 54.4823 KOps/s 55.3784 KOps/s $\color{#d91a1a}-1.62\%$
test_creation_many_keys[50] 0.1048ms 78.6136μs 12.7204 KOps/s 12.8830 KOps/s $\color{#d91a1a}-1.26\%$
test_creation_many_keys[100] 0.1961ms 0.1536ms 6.5106 KOps/s 6.5641 KOps/s $\color{#d91a1a}-0.81\%$
test_creation_nested_many_keys[10] 60.5120μs 39.5611μs 25.2774 KOps/s 25.3798 KOps/s $\color{#d91a1a}-0.40\%$
test_creation_nested_many_keys[50] 0.2079ms 0.1617ms 6.1835 KOps/s 6.3278 KOps/s $\color{#d91a1a}-2.28\%$
test_clone 46.0420μs 13.4576μs 74.3072 KOps/s 73.9850 KOps/s $\color{#35bf28}+0.44\%$
test_getitem[int] 1.7290ms 14.8237μs 67.4596 KOps/s 56.8032 KOps/s $\textbf{\color{#35bf28}+18.76\%}$
test_getitem[slice_int] 0.1423ms 24.5613μs 40.7144 KOps/s 40.9912 KOps/s $\color{#d91a1a}-0.68\%$
test_getitem[range] 0.1771ms 64.5177μs 15.4996 KOps/s 16.1187 KOps/s $\color{#d91a1a}-3.84\%$
test_getitem[tuple] 0.1509ms 23.9444μs 41.7634 KOps/s 41.7024 KOps/s $\color{#35bf28}+0.15\%$
test_getitem[list] 0.1839ms 59.7842μs 16.7268 KOps/s 17.0160 KOps/s $\color{#d91a1a}-1.70\%$
test_setitem_dim[int] 44.7510μs 25.2067μs 39.6720 KOps/s 36.1250 KOps/s $\textbf{\color{#35bf28}+9.82\%}$
test_setitem_dim[slice_int] 64.4630μs 43.4081μs 23.0372 KOps/s 21.9053 KOps/s $\textbf{\color{#35bf28}+5.17\%}$
test_setitem_dim[range] 0.1377ms 98.0372μs 10.2002 KOps/s 10.2005 KOps/s $-0.00\%$
test_setitem_dim[tuple] 61.8320μs 40.6434μs 24.6042 KOps/s 24.6685 KOps/s $\color{#d91a1a}-0.26\%$
test_setitem 46.5610μs 17.4386μs 57.3440 KOps/s 55.3786 KOps/s $\color{#35bf28}+3.55\%$
test_set 65.5620μs 16.8044μs 59.5081 KOps/s 58.3510 KOps/s $\color{#35bf28}+1.98\%$
test_set_shared 0.4975ms 0.2141ms 4.6710 KOps/s 4.8206 KOps/s $\color{#d91a1a}-3.10\%$
test_update 0.3293ms 21.3187μs 46.9072 KOps/s 46.2576 KOps/s $\color{#35bf28}+1.40\%$
test_update_nested 65.2230μs 33.0820μs 30.2279 KOps/s 29.8894 KOps/s $\color{#35bf28}+1.13\%$
test_update__nested 0.4781ms 34.0454μs 29.3726 KOps/s 28.8082 KOps/s $\color{#35bf28}+1.96\%$
test_set_nested 49.2010μs 18.5748μs 53.8362 KOps/s 52.7970 KOps/s $\color{#35bf28}+1.97\%$
test_set_nested_new 60.2720μs 23.7661μs 42.0768 KOps/s 40.9953 KOps/s $\color{#35bf28}+2.64\%$
test_select 76.6130μs 41.6538μs 24.0074 KOps/s 24.0590 KOps/s $\color{#d91a1a}-0.21\%$
test_select_nested 0.1076ms 75.1456μs 13.3075 KOps/s 13.5135 KOps/s $\color{#d91a1a}-1.52\%$
test_exclude_nested 0.1235ms 92.5667μs 10.8030 KOps/s 11.0016 KOps/s $\color{#d91a1a}-1.80\%$
test_empty[True] 0.4445ms 0.3901ms 2.5634 KOps/s 2.5849 KOps/s $\color{#d91a1a}-0.83\%$
test_empty[False] 9.0605μs 1.3306μs 751.5475 KOps/s 753.8989 KOps/s $\color{#d91a1a}-0.31\%$
test_to 0.1036ms 70.9576μs 14.0929 KOps/s 13.4942 KOps/s $\color{#35bf28}+4.44\%$
test_to_nonblocking 0.1061ms 68.1077μs 14.6826 KOps/s 15.6453 KOps/s $\textbf{\color{#d91a1a}-6.15\%}$
test_unbind_speed 0.3853ms 0.3242ms 3.0842 KOps/s 3.0650 KOps/s $\color{#35bf28}+0.63\%$
test_unbind_speed_stack0 0.3759ms 0.3233ms 3.0927 KOps/s 3.1045 KOps/s $\color{#d91a1a}-0.38\%$
test_unbind_speed_stack1 0.1028s 0.8296ms 1.2054 KOps/s 1.1897 KOps/s $\color{#35bf28}+1.32\%$
test_split 0.1027s 1.2731ms 785.4868 Ops/s 785.6883 Ops/s $\color{#d91a1a}-0.03\%$
test_chunk 0.1025s 1.2118ms 825.2078 Ops/s 919.0747 Ops/s $\textbf{\color{#d91a1a}-10.21\%}$
test_to_cpu_blocking 19.4367ms 19.3190ms 51.7624 Ops/s 31.6392 Ops/s $\textbf{\color{#35bf28}+63.60\%}$
test_to_cpu_global_sync 11.1687ms 11.0400ms 90.5794 Ops/s 89.5594 Ops/s $\color{#35bf28}+1.14\%$
test_to_cpu_event_sync 12.2947ms 12.0076ms 83.2806 Ops/s 82.8089 Ops/s $\color{#35bf28}+0.57\%$
test_to_cpu_default 0.1147s 13.2312ms 75.5788 Ops/s 82.7234 Ops/s $\textbf{\color{#d91a1a}-8.64\%}$
test_consolidate[False-None] 4.1813ms 4.0812ms 245.0243 Ops/s 218.3625 Ops/s $\textbf{\color{#35bf28}+12.21\%}$
test_consolidate[default-None] 2.0684ms 1.9855ms 503.6431 Ops/s 478.7657 Ops/s $\textbf{\color{#35bf28}+5.20\%}$
test_consolidate[reduce-overhead-None] 2.0133ms 1.9181ms 521.3477 Ops/s 498.2789 Ops/s $\color{#35bf28}+4.63\%$
test_consolidate_njt[False-None] 8.6136ms 8.4413ms 118.4651 Ops/s 117.1303 Ops/s $\color{#35bf28}+1.14\%$
test_to[False-False-None] 2.1485ms 2.0619ms 484.9948 Ops/s 481.4562 Ops/s $\color{#35bf28}+0.73\%$
test_to[True-False-None] 2.2098ms 1.9236ms 519.8650 Ops/s 514.1619 Ops/s $\color{#35bf28}+1.11\%$
test_to[within-False-None] 6.3330ms 6.0888ms 164.2361 Ops/s 163.3040 Ops/s $\color{#35bf28}+0.57\%$
test_to[True-default-None] 7.6904ms 7.5999ms 131.5815 Ops/s 129.8353 Ops/s $\color{#35bf28}+1.34\%$
test_to_njt[False-False-None] 8.6806ms 8.5388ms 117.1129 Ops/s 117.4972 Ops/s $\color{#d91a1a}-0.33\%$
test_to_njt[True-False-None] 7.1220ms 6.9720ms 143.4309 Ops/s 133.7826 Ops/s $\textbf{\color{#35bf28}+7.21\%}$
test_to_njt[within-False-None] 15.7887ms 15.6551ms 63.8768 Ops/s 64.3003 Ops/s $\color{#d91a1a}-0.66\%$
test_creation[device0] 0.4546ms 0.1166ms 8.5786 KOps/s 8.5495 KOps/s $\color{#35bf28}+0.34\%$
test_creation_from_tensor 0.3915ms 0.1139ms 8.7766 KOps/s 8.7656 KOps/s $\color{#35bf28}+0.13\%$
test_add_one[memmap_tensor0] 0.2760ms 6.3781μs 156.7861 KOps/s 156.0073 KOps/s $\color{#35bf28}+0.50\%$
test_contiguous[memmap_tensor0] 28.3610μs 0.6638μs 1.5064 MOps/s 1.9733 MOps/s $\textbf{\color{#d91a1a}-23.66\%}$
test_stack[memmap_tensor0] 27.0010μs 4.5388μs 220.3224 KOps/s 215.5127 KOps/s $\color{#35bf28}+2.23\%$
test_memmaptd_index 1.0339ms 0.2626ms 3.8083 KOps/s 3.8105 KOps/s $\color{#d91a1a}-0.06\%$
test_memmaptd_index_astensor 0.5041ms 0.3560ms 2.8089 KOps/s 2.8015 KOps/s $\color{#35bf28}+0.26\%$
test_memmaptd_index_op 0.8703ms 0.5942ms 1.6830 KOps/s 1.6863 KOps/s $\color{#d91a1a}-0.19\%$
test_serialize_model 0.1401s 0.1363s 7.3388 Ops/s 7.3135 Ops/s $\color{#35bf28}+0.35\%$
test_serialize_model_pickle 1.3495s 1.2186s 0.8206 Ops/s 0.8347 Ops/s $\color{#d91a1a}-1.69\%$
test_serialize_weights 0.1363s 0.1353s 7.3892 Ops/s 7.3181 Ops/s $\color{#35bf28}+0.97\%$
test_serialize_weights_returnearly 0.3703s 93.0747ms 10.7441 Ops/s 14.8370 Ops/s $\textbf{\color{#d91a1a}-27.59\%}$
test_serialize_weights_pickle 1.3706s 1.1965s 0.8358 Ops/s 0.8368 Ops/s $\color{#d91a1a}-0.13\%$
test_reshape_pytree 0.2093ms 33.0321μs 30.2735 KOps/s 30.3653 KOps/s $\color{#d91a1a}-0.30\%$
test_reshape_td 76.2930μs 43.9087μs 22.7745 KOps/s 22.7947 KOps/s $\color{#d91a1a}-0.09\%$
test_view_pytree 0.2179ms 32.8556μs 30.4362 KOps/s 30.7003 KOps/s $\color{#d91a1a}-0.86\%$
test_view_td 93.3330μs 51.9723μs 19.2410 KOps/s 19.1875 KOps/s $\color{#35bf28}+0.28\%$
test_unbind_pytree 0.2770ms 36.0137μs 27.7672 KOps/s 27.3556 KOps/s $\color{#35bf28}+1.50\%$
test_unbind_td 0.2015ms 48.8249μs 20.4814 KOps/s 20.2668 KOps/s $\color{#35bf28}+1.06\%$
test_split_pytree 0.1971ms 42.3178μs 23.6307 KOps/s 23.4917 KOps/s $\color{#35bf28}+0.59\%$
test_split_td 0.1392ms 64.5761μs 15.4856 KOps/s 15.0723 KOps/s $\color{#35bf28}+2.74\%$
test_add_pytree 0.1732ms 40.8403μs 24.4856 KOps/s 23.7785 KOps/s $\color{#35bf28}+2.97\%$
test_add_td 89.8430μs 51.6653μs 19.3554 KOps/s 18.4740 KOps/s $\color{#35bf28}+4.77\%$
test_compile_add_one_nested[tensordict-compile] 0.1899ms 0.1379ms 7.2526 KOps/s 6.8785 KOps/s $\textbf{\color{#35bf28}+5.44\%}$
test_compile_add_one_nested[tensordict-eager] 0.3569ms 0.1949ms 5.1317 KOps/s 5.2756 KOps/s $\color{#d91a1a}-2.73\%$
test_compile_add_one_nested[pytree-compile] 0.1564ms 0.1080ms 9.2631 KOps/s 8.9690 KOps/s $\color{#35bf28}+3.28\%$
test_compile_add_one_nested[pytree-eager] 0.4293ms 0.1784ms 5.6057 KOps/s 5.6239 KOps/s $\color{#d91a1a}-0.32\%$
test_compile_copy_nested[tensordict-compile] 0.1845ms 29.2723μs 34.1620 KOps/s 30.0680 KOps/s $\textbf{\color{#35bf28}+13.62\%}$
test_compile_copy_nested[tensordict-eager] 88.2330μs 52.5283μs 19.0373 KOps/s 19.2127 KOps/s $\color{#d91a1a}-0.91\%$
test_compile_copy_nested[pytree-compile] 59.8930μs 9.8191μs 101.8421 KOps/s 101.4199 KOps/s $\color{#35bf28}+0.42\%$
test_compile_copy_nested[pytree-eager] 0.4661ms 69.1028μs 14.4712 KOps/s 14.6166 KOps/s $\color{#d91a1a}-0.99\%$
test_compile_add_one_flat[tensordict-compile] 0.2218ms 0.1768ms 5.6555 KOps/s 5.4688 KOps/s $\color{#35bf28}+3.41\%$
test_compile_add_one_flat[tensordict-eager] 0.3749ms 0.2556ms 3.9131 KOps/s 3.9254 KOps/s $\color{#d91a1a}-0.32\%$
test_compile_add_one_flat[tensorclass-compile] 0.1692ms 0.1184ms 8.4460 KOps/s 8.3961 KOps/s $\color{#35bf28}+0.59\%$
test_compile_add_one_flat[tensorclass-eager] 0.1157ms 71.5895μs 13.9685 KOps/s 14.4952 KOps/s $\color{#d91a1a}-3.63\%$
test_compile_add_one_flat[pytree-compile] 0.2010ms 0.1587ms 6.3029 KOps/s 6.1881 KOps/s $\color{#35bf28}+1.86\%$
test_compile_add_one_flat[pytree-eager] 0.9055ms 0.5302ms 1.8860 KOps/s 1.9061 KOps/s $\color{#d91a1a}-1.05\%$
test_compile_add_self_flat[tensordict-eager] 0.4573ms 0.3102ms 3.2239 KOps/s 3.2383 KOps/s $\color{#d91a1a}-0.45\%$
test_compile_add_self_flat[tensordict-compile] 0.2367ms 0.1788ms 5.5936 KOps/s 3.7956 KOps/s $\textbf{\color{#35bf28}+47.37\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1356ms 85.6457μs 11.6760 KOps/s 11.7071 KOps/s $\color{#d91a1a}-0.27\%$
test_compile_add_self_flat[tensorclass-compile] 0.2016ms 0.1190ms 8.4053 KOps/s 7.9887 KOps/s $\textbf{\color{#35bf28}+5.22\%}$
test_compile_add_self_flat[pytree-eager] 0.6584ms 0.4395ms 2.2754 KOps/s 2.2811 KOps/s $\color{#d91a1a}-0.25\%$
test_compile_add_self_flat[pytree-compile] 0.2813ms 0.1608ms 6.2190 KOps/s 6.1305 KOps/s $\color{#35bf28}+1.44\%$
test_compile_copy_flat[tensordict-compile] 0.1177ms 25.1493μs 39.7626 KOps/s 40.2635 KOps/s $\color{#d91a1a}-1.24\%$
test_compile_copy_flat[tensordict-eager] 0.1258ms 40.7369μs 24.5478 KOps/s 24.4736 KOps/s $\color{#35bf28}+0.30\%$
test_compile_copy_flat[pytree-compile] 0.1036ms 11.1399μs 89.7672 KOps/s 91.6819 KOps/s $\color{#d91a1a}-2.09\%$
test_compile_copy_flat[pytree-eager] 0.4209ms 52.6882μs 18.9796 KOps/s 19.1108 KOps/s $\color{#d91a1a}-0.69\%$
test_compile_assign_and_add[tensordict-compile] 2.0057ms 0.1773ms 5.6386 KOps/s 5.4612 KOps/s $\color{#35bf28}+3.25\%$
test_compile_assign_and_add[tensordict-eager] 3.6628ms 3.3147ms 301.6838 Ops/s 307.5918 Ops/s $\color{#d91a1a}-1.92\%$
test_compile_assign_and_add[pytree-compile] 2.0551ms 0.1650ms 6.0620 KOps/s 6.0629 KOps/s $\color{#d91a1a}-0.01\%$
test_compile_assign_and_add[pytree-eager] 3.0217ms 2.8079ms 356.1378 Ops/s 358.0034 Ops/s $\color{#d91a1a}-0.52\%$
test_compile_indexing[tensor-tensordict-compile] 0.2516ms 0.1120ms 8.9291 KOps/s 8.8744 KOps/s $\color{#35bf28}+0.62\%$
test_compile_indexing[tensor-tensordict-eager] 0.3129ms 75.6639μs 13.2163 KOps/s 14.0551 KOps/s $\textbf{\color{#d91a1a}-5.97\%}$
test_compile_indexing[tensor-tensorclass-compile] 0.2528ms 0.1012ms 9.8779 KOps/s 10.2567 KOps/s $\color{#d91a1a}-3.69\%$
test_compile_indexing[tensor-tensorclass-eager] 0.2719ms 47.5110μs 21.0478 KOps/s 22.6070 KOps/s $\textbf{\color{#d91a1a}-6.90\%}$
test_compile_indexing[tensor-pytree-compile] 0.2673ms 0.1022ms 9.7824 KOps/s 10.2218 KOps/s $\color{#d91a1a}-4.30\%$
test_compile_indexing[tensor-pytree-eager] 0.2961ms 47.8031μs 20.9192 KOps/s 22.5372 KOps/s $\textbf{\color{#d91a1a}-7.18\%}$
test_compile_indexing[slice-tensordict-compile] 0.1694ms 58.0046μs 17.2400 KOps/s 17.5268 KOps/s $\color{#d91a1a}-1.64\%$
test_compile_indexing[slice-tensordict-eager] 0.2270ms 28.2336μs 35.4187 KOps/s 35.6501 KOps/s $\color{#d91a1a}-0.65\%$
test_compile_indexing[slice-tensorclass-compile] 0.1692ms 47.1640μs 21.2026 KOps/s 20.4249 KOps/s $\color{#35bf28}+3.81\%$
test_compile_indexing[slice-tensorclass-eager] 0.2631ms 22.3693μs 44.7040 KOps/s 44.2335 KOps/s $\color{#35bf28}+1.06\%$
test_compile_indexing[slice-pytree-compile] 0.1026ms 45.4442μs 22.0050 KOps/s 20.3831 KOps/s $\textbf{\color{#35bf28}+7.96\%}$
test_compile_indexing[slice-pytree-eager] 0.3012ms 22.4883μs 44.4675 KOps/s 44.4134 KOps/s $\color{#35bf28}+0.12\%$
test_compile_indexing[int-tensordict-compile] 0.1048ms 58.2248μs 17.1748 KOps/s 16.8189 KOps/s $\color{#35bf28}+2.12\%$
test_compile_indexing[int-tensordict-eager] 0.2758ms 27.9757μs 35.7453 KOps/s 35.1530 KOps/s $\color{#35bf28}+1.68\%$
test_compile_indexing[int-tensorclass-compile] 0.1818ms 45.7358μs 21.8647 KOps/s 21.6305 KOps/s $\color{#35bf28}+1.08\%$
test_compile_indexing[int-tensorclass-eager] 0.2563ms 22.6410μs 44.1677 KOps/s 44.6458 KOps/s $\color{#d91a1a}-1.07\%$
test_compile_indexing[int-pytree-compile] 82.0330μs 45.5859μs 21.9366 KOps/s 21.5689 KOps/s $\color{#35bf28}+1.70\%$
test_compile_indexing[int-pytree-eager] 0.2818ms 22.3839μs 44.6750 KOps/s 44.5309 KOps/s $\color{#35bf28}+0.32\%$
test_mod_add[eager] 96.5130μs 49.2982μs 20.2847 KOps/s 19.9711 KOps/s $\color{#35bf28}+1.57\%$
test_mod_add[compile] 0.1853ms 0.1042ms 9.5940 KOps/s 9.4225 KOps/s $\color{#35bf28}+1.82\%$
test_mod_add[compile-overhead] 0.2366ms 0.1476ms 6.7736 KOps/s 6.6012 KOps/s $\color{#35bf28}+2.61\%$
test_mod_wrap[eager] 0.3694ms 0.2949ms 3.3907 KOps/s 3.4620 KOps/s $\color{#d91a1a}-2.06\%$
test_mod_wrap[compile] 0.3976ms 0.3494ms 2.8623 KOps/s 2.7620 KOps/s $\color{#35bf28}+3.63\%$
test_mod_wrap[compile-overhead] 7.3989ms 4.0496ms 246.9377 Ops/s 251.1155 Ops/s $\color{#d91a1a}-1.66\%$
test_mod_wrap_and_backward[eager] 1.6914ms 1.5963ms 626.4661 Ops/s 635.9158 Ops/s $\color{#d91a1a}-1.49\%$
test_mod_wrap_and_backward[compile] 1.5234ms 1.4414ms 693.7542 Ops/s 639.6872 Ops/s $\textbf{\color{#35bf28}+8.45\%}$
test_mod_wrap_and_backward[compile-overhead] 1.2482ms 0.8833ms 1.1321 KOps/s 995.0290 Ops/s $\textbf{\color{#35bf28}+13.78\%}$
test_seq_add[eager] 0.2143ms 0.1558ms 6.4181 KOps/s 6.4072 KOps/s $\color{#35bf28}+0.17\%$
test_seq_add[compile] 0.6023ms 0.1158ms 8.6355 KOps/s 8.5386 KOps/s $\color{#35bf28}+1.14\%$
test_seq_add[compile-overhead] 0.4350ms 0.1551ms 6.4495 KOps/s 6.3032 KOps/s $\color{#35bf28}+2.32\%$
test_seq_wrap[eager] 0.5959ms 0.5321ms 1.8795 KOps/s 1.8209 KOps/s $\color{#35bf28}+3.22\%$
test_seq_wrap[compile] 0.4428ms 0.3709ms 2.6962 KOps/s 2.5897 KOps/s $\color{#35bf28}+4.11\%$
test_seq_wrap[compile-overhead] 0.3415ms 0.2663ms 3.7555 KOps/s 3.5496 KOps/s $\textbf{\color{#35bf28}+5.80\%}$
test_func_call_runtime[False-eager] 0.9316ms 0.8567ms 1.1673 KOps/s 1.1990 KOps/s $\color{#d91a1a}-2.65\%$
test_func_call_runtime[False-compile] 1.0967ms 0.9189ms 1.0883 KOps/s 1.0914 KOps/s $\color{#d91a1a}-0.28\%$
test_func_call_runtime[False-compile-overhead] 0.5430ms 0.4632ms 2.1590 KOps/s 2.1338 KOps/s $\color{#35bf28}+1.18\%$
test_func_call_runtime[True-eager] 1.1790ms 1.0820ms 924.2328 Ops/s 933.4817 Ops/s $\color{#d91a1a}-0.99\%$
test_func_call_runtime[True-compile] 0.9833ms 0.9204ms 1.0864 KOps/s 1.0828 KOps/s $\color{#35bf28}+0.33\%$
test_func_call_runtime[True-compile-overhead] 0.5205ms 0.4753ms 2.1039 KOps/s 2.0833 KOps/s $\color{#35bf28}+0.99\%$
test_func_call_cm_runtime[False-eager] 0.8921ms 0.8356ms 1.1967 KOps/s 1.2001 KOps/s $\color{#d91a1a}-0.28\%$
test_func_call_cm_runtime[False-compile] 0.9632ms 0.9177ms 1.0896 KOps/s 1.0884 KOps/s $\color{#35bf28}+0.12\%$
test_func_call_cm_runtime[False-compile-overhead] 0.5060ms 0.4655ms 2.1484 KOps/s 2.1326 KOps/s $\color{#35bf28}+0.74\%$
test_func_call_cm_runtime[True-eager] 1.3448ms 1.2296ms 813.2967 Ops/s 818.4214 Ops/s $\color{#d91a1a}-0.63\%$
test_func_call_cm_runtime[True-compile] 1.1148ms 0.9981ms 1.0019 KOps/s 1.0318 KOps/s $\color{#d91a1a}-2.90\%$
test_func_call_cm_runtime[True-compile-overhead] 0.6251ms 0.5107ms 1.9583 KOps/s 1.9472 KOps/s $\color{#35bf28}+0.57\%$
test_vmap_func_call_cm_runtime[eager] 2.8610ms 2.3607ms 423.6037 Ops/s 422.8782 Ops/s $\color{#35bf28}+0.17\%$
test_vmap_func_call_cm_runtime[compile] 1.0315ms 0.9725ms 1.0282 KOps/s 1.0274 KOps/s $\color{#35bf28}+0.09\%$
test_vmap_func_call_cm_runtime[compile-overhead] 0.5712ms 0.5120ms 1.9529 KOps/s 1.9214 KOps/s $\color{#35bf28}+1.64\%$
test_distributed 0.5007ms 0.1539ms 6.4996 KOps/s 6.4838 KOps/s $\color{#35bf28}+0.25\%$
test_tdmodule 0.4002ms 28.3506μs 35.2726 KOps/s 35.2923 KOps/s $\color{#d91a1a}-0.06\%$
test_tdmodule_dispatch 84.8540μs 45.2800μs 22.0848 KOps/s 21.7204 KOps/s $\color{#35bf28}+1.68\%$
test_tdseq 45.5320μs 26.9384μs 37.1218 KOps/s 36.0321 KOps/s $\color{#35bf28}+3.02\%$
test_tdseq_dispatch 88.9630μs 47.1927μs 21.1897 KOps/s 20.7124 KOps/s $\color{#35bf28}+2.30\%$
test_instantiation_functorch 2.1556ms 2.0703ms 483.0195 Ops/s 481.2753 Ops/s $\color{#35bf28}+0.36\%$
test_exec_functorch 0.2259ms 0.1783ms 5.6076 KOps/s 5.5198 KOps/s $\color{#35bf28}+1.59\%$
test_exec_functional_call 0.2228ms 0.1596ms 6.2640 KOps/s 6.2662 KOps/s $\color{#d91a1a}-0.04\%$
test_exec_td_decorator 0.4493ms 0.2348ms 4.2590 KOps/s 4.2080 KOps/s $\color{#35bf28}+1.21\%$
test_vmap_mlp_speed_decorator[True-True] 1.0258ms 0.8261ms 1.2105 KOps/s 1.2102 KOps/s $\color{#35bf28}+0.02\%$
test_vmap_mlp_speed_decorator[True-False] 1.0132ms 0.8259ms 1.2109 KOps/s 1.2082 KOps/s $\color{#35bf28}+0.22\%$
test_vmap_mlp_speed_decorator[False-True] 0.8925ms 0.7130ms 1.4026 KOps/s 1.3943 KOps/s $\color{#35bf28}+0.60\%$
test_vmap_mlp_speed_decorator[False-False] 0.9000ms 0.7141ms 1.4004 KOps/s 1.4005 KOps/s $\color{#d91a1a}-0.01\%$
test_vmap_transformer_speed_decorator[True-True] 20.7054ms 20.4712ms 48.8491 Ops/s 48.8033 Ops/s $\color{#35bf28}+0.09\%$
test_vmap_transformer_speed_decorator[True-False] 21.1362ms 20.4767ms 48.8359 Ops/s 48.6902 Ops/s $\color{#35bf28}+0.30\%$
test_vmap_transformer_speed_decorator[False-True] 20.9685ms 20.3455ms 49.1510 Ops/s 49.2099 Ops/s $\color{#d91a1a}-0.12\%$
test_vmap_transformer_speed_decorator[False-False] 20.8449ms 20.2869ms 49.2929 Ops/s 49.2859 Ops/s $\color{#35bf28}+0.01\%$
test_to_module_speed[True] 2.0384ms 1.4741ms 678.3705 Ops/s 671.7368 Ops/s $\color{#35bf28}+0.99\%$
test_to_module_speed[False] 1.9915ms 1.4470ms 691.0649 Ops/s 688.7095 Ops/s $\color{#35bf28}+0.34\%$
test_tc_init 89.6930μs 48.0070μs 20.8303 KOps/s 22.0426 KOps/s $\textbf{\color{#d91a1a}-5.50\%}$
test_tc_init_tensor_only 29.7110μs 9.9215μs 100.7916 KOps/s 101.4075 KOps/s $\color{#d91a1a}-0.61\%$
test_tc_init_nested 0.1377ms 93.9249μs 10.6468 KOps/s 11.0739 KOps/s $\color{#d91a1a}-3.86\%$
test_tc_init_many_fields 44.7020μs 17.6760μs 56.5740 KOps/s 59.9174 KOps/s $\textbf{\color{#d91a1a}-5.58\%}$
test_tc_first_layer_tensor 18.6010μs 1.8605μs 537.5018 KOps/s 546.5103 KOps/s $\color{#d91a1a}-1.65\%$
test_tc_first_layer_tensor_only 4.9230μs 0.7489μs 1.3353 MOps/s 1.3168 MOps/s $\color{#35bf28}+1.41\%$
test_tc_first_layer_tensor_set 39.6310μs 3.9427μs 253.6305 KOps/s 262.1283 KOps/s $\color{#d91a1a}-3.24\%$
test_tc_first_layer_tensor_only_set 26.7710μs 3.1919μs 313.2961 KOps/s 319.2881 KOps/s $\color{#d91a1a}-1.88\%$
test_tc_first_layer_nontensor 31.6910μs 6.1659μs 162.1831 KOps/s 161.7783 KOps/s $\color{#35bf28}+0.25\%$
test_tc_second_layer_tensor 27.6410μs 4.3827μs 228.1680 KOps/s 227.5936 KOps/s $\color{#35bf28}+0.25\%$
test_tc_second_layer_nontensor 35.8420μs 8.6511μs 115.5929 KOps/s 114.3896 KOps/s $\color{#35bf28}+1.05\%$
test_unbind 0.2653s 16.7409ms 59.7340 Ops/s 54.2214 Ops/s $\textbf{\color{#35bf28}+10.17\%}$
test_full_like 4.7807ms 4.3786ms 228.3828 Ops/s 232.1808 Ops/s $\color{#d91a1a}-1.64\%$
test_zeros_like 4.4939ms 4.3622ms 229.2419 Ops/s 227.5045 Ops/s $\color{#35bf28}+0.76\%$
test_ones_like 10.5361ms 4.4089ms 226.8163 Ops/s 228.3629 Ops/s $\color{#d91a1a}-0.68\%$
test_clone 7.1976ms 6.5846ms 151.8686 Ops/s 154.3559 Ops/s $\color{#d91a1a}-1.61\%$
test_squeeze 0.1860ms 14.1138μs 70.8526 KOps/s 70.3462 KOps/s $\color{#35bf28}+0.72\%$
test_unsqueeze 0.1619ms 0.1094ms 9.1404 KOps/s 9.1637 KOps/s $\color{#d91a1a}-0.25\%$
test_split 0.2462ms 0.1829ms 5.4666 KOps/s 5.4254 KOps/s $\color{#35bf28}+0.76\%$
test_permute 0.2594ms 0.2037ms 4.9089 KOps/s 4.8683 KOps/s $\color{#35bf28}+0.83\%$
test_stack 52.7661ms 51.7043ms 19.3407 Ops/s 19.2982 Ops/s $\color{#35bf28}+0.22\%$
test_cat 54.0858ms 52.4334ms 19.0718 Ops/s 19.4430 Ops/s $\color{#d91a1a}-1.91\%$
test_sequential_tensordict 0.5806ms 0.2160ms 4.6287 KOps/s 4.5498 KOps/s $\color{#35bf28}+1.73\%$
test_sequential_graph_module 0.2485ms 0.1192ms 8.3890 KOps/s 8.2236 KOps/s $\color{#35bf28}+2.01\%$
test_nested_tensordict 0.6274ms 0.2826ms 3.5383 KOps/s 3.4473 KOps/s $\color{#35bf28}+2.64\%$
test_nested_graph_module 0.2001ms 0.1275ms 7.8444 KOps/s 7.5283 KOps/s $\color{#35bf28}+4.20\%$

@vmoens vmoens mentioned this pull request Feb 19, 2026
Merged
@github-actions

github-actions Bot commented Feb 19, 2026

Copy link
Copy Markdown
Contributor

PR Title Label Error

Unknown or invalid prefix [Distributed].

Current title: [Distributed] Add tensorclass distributed tests

Supported Prefixes

Your PR title must start with exactly one of these prefixes:

Prefix Label Applied Example
[BugFix] BugFix [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] Perf [Perf] Optimize tensor operations
[Deprecation] Deprecation [Deprecation] Mark old function
[Setup] Setup [Setup] Update build configuration

Note: Common variations like singular/plural are supported (e.g., [Doc] or [Docs]).

@vmoens
Update
63cf8d8 
[ghstack-poisoned]
@github-actions

github-actions Bot commented Feb 19, 2026

Copy link
Copy Markdown
Contributor

PR Title Label Error

Unknown or invalid prefix [Distributed].

Current title: [Distributed] Add tensorclass distributed tests

Supported Prefixes

Your PR title must start with exactly one of these prefixes:

Prefix Label Applied Example
[BugFix] BugFix [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] Perf [Perf] Optimize tensor operations
[Deprecation] Deprecation [Deprecation] Mark old function
[Setup] Setup [Setup] Update build configuration

Note: Common variations like singular/plural are supported (e.g., [Doc] or [Docs]).

vmoens added a commit that referenced this pull request Feb 19, 2026
@vmoens @cursoragent
[Distributed] Add tensorclass distributed tests
585142c 
Add tests verifying that all distributed operations work when called on
tensorclass instances: broadcast, all_reduce, all_gather, scatter,
init_remote, and consolidated send/recv.

Co-authored-by: Cursor <cursoragent@cursor.com>
ghstack-source-id: f309608
Pull-Request: #1598
@vmoens
vmoens merged commit 63cf8d8 into gh/vmoens/78/base Feb 19, 2026
80 of 84 checks passed
@vmoens
vmoens deleted the gh/vmoens/78/head branch February 19, 2026 14:43
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. Distributed Test

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@vmoens