[DTensor] Add public API module for transfer plan computation by vmoens · Pull Request #1649 · pytorch/tensordict

vmoens · 2026-03-10T19:19:30Z

Stack from ghstack (oldest at bottom):

Create tensordict/dtensor.py as a stable public module re-exporting
the core types and functions from tensordict._dtensor under
non-underscore names:

ShardSpec, ChunkTransfer, TransferPlan
compute_transfer_plan, compute_local_slices, compute_all_local_slices
ShardingDescriptor, execute_transfer_plan
Slice arithmetic helpers

Made-with: Cursor

[ghstack-poisoned]

github-actions · 2026-03-10T19:19:41Z

PR Title Label Error

Unknown or invalid prefix [DTensor].

Current title: [DTensor] Add public API module for transfer plan computation

Supported Prefixes

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

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

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

github-actions · 2026-03-10T19:19:57Z

PR Title Label Error

Unknown or invalid prefix [DTensor].

Current title: [DTensor] Add public API module for transfer plan computation

Supported Prefixes

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

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

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

github-actions · 2026-03-10T19:39:02Z

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

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

Expand to view detailed results

Name	Max	Mean	Ops	Ops on Repo `HEAD`	Change
test_plain_set_nested	51.6510μs	14.7987μs	67.5735 KOps/s	67.1638 KOps/s	$\color{#35bf28}+0.61\%$
test_plain_set_stack_nested	36.1210μs	15.1966μs	65.8042 KOps/s	65.7858 KOps/s	$\color{#35bf28}+0.03\%$
test_plain_set_nested_inplace	43.6310μs	16.8144μs	59.4730 KOps/s	58.5914 KOps/s	$\color{#35bf28}+1.50\%$
test_plain_set_stack_nested_inplace	46.2010μs	16.5969μs	60.2523 KOps/s	59.5688 KOps/s	$\color{#35bf28}+1.15\%$
test_items	27.3500μs	6.0530μs	165.2080 KOps/s	166.9622 KOps/s	$\color{#d91a1a}-1.05\%$
test_items_nested	0.5217ms	0.4731ms	2.1138 KOps/s	2.1271 KOps/s	$\color{#d91a1a}-0.63\%$
test_items_nested_locked	0.5493ms	0.4773ms	2.0950 KOps/s	2.0998 KOps/s	$\color{#d91a1a}-0.23\%$
test_items_nested_leaf	0.1360ms	97.3323μs	10.2741 KOps/s	10.2644 KOps/s	$\color{#35bf28}+0.09\%$
test_items_stack_nested	0.7179ms	0.4805ms	2.0813 KOps/s	2.1234 KOps/s	$\color{#d91a1a}-1.98\%$
test_items_stack_nested_leaf	0.1345ms	97.7963μs	10.2253 KOps/s	10.1578 KOps/s	$\color{#35bf28}+0.66\%$
test_items_stack_nested_locked	0.5864ms	0.4755ms	2.1032 KOps/s	2.1066 KOps/s	$\color{#d91a1a}-0.16\%$
test_keys	29.5000μs	4.2362μs	236.0632 KOps/s	236.1394 KOps/s	$\color{#d91a1a}-0.03\%$
test_keys_nested	0.1816ms	0.1295ms	7.7250 KOps/s	7.6929 KOps/s	$\color{#35bf28}+0.42\%$
test_keys_nested_locked	0.6899ms	0.1375ms	7.2715 KOps/s	7.1688 KOps/s	$\color{#35bf28}+1.43\%$
test_keys_nested_leaf	0.1729ms	0.1198ms	8.3445 KOps/s	8.3277 KOps/s	$\color{#35bf28}+0.20\%$
test_keys_stack_nested	0.2135ms	0.1293ms	7.7319 KOps/s	7.6637 KOps/s	$\color{#35bf28}+0.89\%$
test_keys_stack_nested_leaf	0.1637ms	0.1200ms	8.3350 KOps/s	8.3453 KOps/s	$\color{#d91a1a}-0.12\%$
test_keys_stack_nested_locked	0.1861ms	0.1368ms	7.3124 KOps/s	7.3228 KOps/s	$\color{#d91a1a}-0.14\%$
test_values	5.3720μs	1.0207μs	979.7043 KOps/s	980.7794 KOps/s	$\color{#d91a1a}-0.11\%$
test_values_nested	82.3310μs	52.1875μs	19.1617 KOps/s	18.9354 KOps/s	$\color{#35bf28}+1.20\%$
test_values_nested_locked	87.0320μs	55.2193μs	18.1096 KOps/s	17.7004 KOps/s	$\color{#35bf28}+2.31\%$
test_values_nested_leaf	89.8820μs	58.9487μs	16.9639 KOps/s	16.6679 KOps/s	$\color{#35bf28}+1.78\%$
test_values_stack_nested	83.3920μs	52.8240μs	18.9308 KOps/s	18.9820 KOps/s	$\color{#d91a1a}-0.27\%$
test_values_stack_nested_leaf	89.4420μs	60.4387μs	16.5457 KOps/s	16.5755 KOps/s	$\color{#d91a1a}-0.18\%$
test_values_stack_nested_locked	87.1410μs	55.1293μs	18.1392 KOps/s	17.7759 KOps/s	$\color{#35bf28}+2.04\%$
test_membership	4.3000μs	0.8365μs	1.1954 MOps/s	1.2036 MOps/s	$\color{#d91a1a}-0.68\%$
test_membership_nested	47.8400μs	2.7980μs	357.3929 KOps/s	346.7709 KOps/s	$\color{#35bf28}+3.06\%$
test_membership_nested_leaf	14.6355μs	2.8222μs	354.3316 KOps/s	347.5512 KOps/s	$\color{#35bf28}+1.95\%$
test_membership_stacked_nested	21.3300μs	2.9207μs	342.3812 KOps/s	346.5117 KOps/s	$\color{#d91a1a}-1.19\%$
test_membership_stacked_nested_leaf	25.3510μs	2.9416μs	339.9527 KOps/s	345.7812 KOps/s	$\color{#d91a1a}-1.69\%$
test_membership_nested_last	23.7000μs	4.3759μs	228.5250 KOps/s	228.5449 KOps/s	$-0.01\%$
test_membership_nested_leaf_last	34.1900μs	4.3522μs	229.7663 KOps/s	228.4527 KOps/s	$\color{#35bf28}+0.58\%$
test_membership_stacked_nested_last	36.7410μs	4.3669μs	228.9933 KOps/s	229.5052 KOps/s	$\color{#d91a1a}-0.22\%$
test_membership_stacked_nested_leaf_last	26.0710μs	4.3646μs	229.1184 KOps/s	231.1687 KOps/s	$\color{#d91a1a}-0.89\%$
test_nested_getleaf	45.7610μs	21.4509μs	46.6182 KOps/s	46.3372 KOps/s	$\color{#35bf28}+0.61\%$
test_nested_get	41.6100μs	20.3432μs	49.1566 KOps/s	48.3264 KOps/s	$\color{#35bf28}+1.72\%$
test_stacked_getleaf	50.7610μs	21.6517μs	46.1857 KOps/s	45.9332 KOps/s	$\color{#35bf28}+0.55\%$
test_stacked_get	45.2310μs	20.4460μs	48.9092 KOps/s	47.7831 KOps/s	$\color{#35bf28}+2.36\%$
test_nested_getitemleaf	48.7000μs	22.0861μs	45.2773 KOps/s	45.3411 KOps/s	$\color{#d91a1a}-0.14\%$
test_nested_getitem	43.8900μs	21.0348μs	47.5403 KOps/s	47.7172 KOps/s	$\color{#d91a1a}-0.37\%$
test_stacked_getitemleaf	46.1910μs	22.1013μs	45.2462 KOps/s	45.4019 KOps/s	$\color{#d91a1a}-0.34\%$
test_stacked_getitem	45.1610μs	21.1750μs	47.2255 KOps/s	47.2232 KOps/s	$+0.00\%$
test_lock_nested	7.8135ms	0.4919ms	2.0330 KOps/s	2.0895 KOps/s	$\color{#d91a1a}-2.70\%$
test_lock_stack_nested	0.5210ms	0.4846ms	2.0637 KOps/s	2.0529 KOps/s	$\color{#35bf28}+0.53\%$
test_unlock_nested	0.4826ms	0.3940ms	2.5380 KOps/s	2.5547 KOps/s	$\color{#d91a1a}-0.65\%$
test_unlock_stack_nested	0.4403ms	0.3907ms	2.5596 KOps/s	2.5317 KOps/s	$\color{#35bf28}+1.10\%$
test_flatten_speed	0.1589ms	0.1224ms	8.1680 KOps/s	8.1484 KOps/s	$\color{#35bf28}+0.24\%$
test_unflatten_speed	0.6993ms	0.5743ms	1.7411 KOps/s	1.7396 KOps/s	$\color{#35bf28}+0.09\%$
test_common_ops	0.8689ms	0.6889ms	1.4516 KOps/s	1.4277 KOps/s	$\color{#35bf28}+1.67\%$
test_creation	0.1126ms	3.1620μs	316.2512 KOps/s	314.6739 KOps/s	$\color{#35bf28}+0.50\%$
test_creation_empty	27.0600μs	6.9548μs	143.7863 KOps/s	141.6498 KOps/s	$\color{#35bf28}+1.51\%$
test_creation_nested_1	37.8310μs	11.5863μs	86.3090 KOps/s	85.3676 KOps/s	$\color{#35bf28}+1.10\%$
test_creation_nested_2	33.6510μs	13.2209μs	75.6380 KOps/s	75.0131 KOps/s	$\color{#35bf28}+0.83\%$
test_creation_many_keys[10]	53.6510μs	21.2145μs	47.1376 KOps/s	47.4371 KOps/s	$\color{#d91a1a}-0.63\%$
test_creation_many_keys[50]	0.1236ms	91.1079μs	10.9760 KOps/s	11.0463 KOps/s	$\color{#d91a1a}-0.64\%$
test_creation_many_keys[100]	0.2459ms	0.1784ms	5.6044 KOps/s	5.5635 KOps/s	$\color{#35bf28}+0.73\%$
test_creation_nested_many_keys[10]	75.4510μs	45.0956μs	22.1751 KOps/s	22.0561 KOps/s	$\color{#35bf28}+0.54\%$
test_creation_nested_many_keys[50]	0.2421ms	0.1854ms	5.3930 KOps/s	5.3227 KOps/s	$\color{#35bf28}+1.32\%$
test_clone	49.7410μs	13.2433μs	75.5098 KOps/s	74.9830 KOps/s	$\color{#35bf28}+0.70\%$
test_getitem[int]	1.5938ms	15.2899μs	65.4027 KOps/s	58.9040 KOps/s	$\textbf{\color{#35bf28}+11.03\%}$
test_getitem[slice_int]	0.1357ms	23.8938μs	41.8519 KOps/s	41.7508 KOps/s	$\color{#35bf28}+0.24\%$
test_getitem[range]	0.1759ms	62.2576μs	16.0623 KOps/s	15.8714 KOps/s	$\color{#35bf28}+1.20\%$
test_getitem[tuple]	0.1400ms	24.2010μs	41.3207 KOps/s	41.6773 KOps/s	$\color{#d91a1a}-0.86\%$
test_getitem[list]	0.1801ms	58.8539μs	16.9912 KOps/s	17.2002 KOps/s	$\color{#d91a1a}-1.22\%$
test_setitem_dim[int]	79.8810μs	25.8995μs	38.6108 KOps/s	38.1486 KOps/s	$\color{#35bf28}+1.21\%$
test_setitem_dim[slice_int]	64.6810μs	43.6782μs	22.8947 KOps/s	23.0392 KOps/s	$\color{#d91a1a}-0.63\%$
test_setitem_dim[range]	0.1213ms	96.1155μs	10.4042 KOps/s	10.5033 KOps/s	$\color{#d91a1a}-0.94\%$
test_setitem_dim[tuple]	62.1910μs	40.6136μs	24.6223 KOps/s	24.8779 KOps/s	$\color{#d91a1a}-1.03\%$
test_setitem	52.8810μs	17.6744μs	56.5789 KOps/s	55.6146 KOps/s	$\color{#35bf28}+1.73\%$
test_set	52.7100μs	16.8931μs	59.1957 KOps/s	58.2868 KOps/s	$\color{#35bf28}+1.56\%$
test_set_shared	0.6326ms	0.2066ms	4.8401 KOps/s	4.8747 KOps/s	$\color{#d91a1a}-0.71\%$
test_update	0.2014ms	21.7339μs	46.0111 KOps/s	46.4552 KOps/s	$\color{#d91a1a}-0.96\%$
test_update_nested	76.0610μs	32.9252μs	30.3719 KOps/s	30.7323 KOps/s	$\color{#d91a1a}-1.17\%$
test_update__nested	0.4827ms	34.1301μs	29.2997 KOps/s	28.7363 KOps/s	$\color{#35bf28}+1.96\%$
test_set_nested	48.5810μs	18.7025μs	53.4688 KOps/s	52.3869 KOps/s	$\color{#35bf28}+2.07\%$
test_set_nested_new	61.4510μs	23.6375μs	42.3057 KOps/s	42.5353 KOps/s	$\color{#d91a1a}-0.54\%$
test_select	76.0620μs	40.3442μs	24.7867 KOps/s	24.6054 KOps/s	$\color{#35bf28}+0.74\%$
test_select_nested	0.1068ms	74.6710μs	13.3921 KOps/s	13.3687 KOps/s	$\color{#35bf28}+0.17\%$
test_exclude_nested	0.1270ms	92.6968μs	10.7879 KOps/s	10.8079 KOps/s	$\color{#d91a1a}-0.19\%$
test_empty[True]	0.4508ms	0.3957ms	2.5270 KOps/s	2.5121 KOps/s	$\color{#35bf28}+0.59\%$
test_empty[False]	7.9700μs	1.3133μs	761.4155 KOps/s	762.9360 KOps/s	$\color{#d91a1a}-0.20\%$
test_to	0.1109ms	73.9911μs	13.5151 KOps/s	13.0476 KOps/s	$\color{#35bf28}+3.58\%$
test_to_nonblocking	0.1016ms	63.7771μs	15.6796 KOps/s	15.3276 KOps/s	$\color{#35bf28}+2.30\%$
test_unbind_speed	0.3636ms	0.3370ms	2.9670 KOps/s	3.0297 KOps/s	$\color{#d91a1a}-2.07\%$
test_unbind_speed_stack0	0.4080ms	0.3336ms	2.9980 KOps/s	3.0191 KOps/s	$\color{#d91a1a}-0.70\%$
test_unbind_speed_stack1	0.1035s	0.8390ms	1.1919 KOps/s	1.1818 KOps/s	$\color{#35bf28}+0.86\%$
test_split	0.1034s	1.2622ms	792.2922 Ops/s	789.1739 Ops/s	$\color{#35bf28}+0.40\%$
test_chunk	0.1038s	1.2169ms	821.7794 Ops/s	923.0643 Ops/s	$\textbf{\color{#d91a1a}-10.97\%}$
test_to_cpu_blocking	28.7506ms	28.5628ms	35.0106 Ops/s	51.8907 Ops/s	$\textbf{\color{#d91a1a}-32.53\%}$
test_to_cpu_global_sync	11.3238ms	11.1389ms	89.7755 Ops/s	80.1891 Ops/s	$\textbf{\color{#35bf28}+11.95\%}$
test_to_cpu_event_sync	12.3426ms	12.0662ms	82.8760 Ops/s	82.2381 Ops/s	$\color{#35bf28}+0.78\%$
test_to_cpu_default	12.3236ms	12.0868ms	82.7349 Ops/s	82.0116 Ops/s	$\color{#35bf28}+0.88\%$
test_consolidate[False-None]	4.2995ms	4.2101ms	237.5246 Ops/s	216.5076 Ops/s	$\textbf{\color{#35bf28}+9.71\%}$
test_consolidate[default-None]	2.1467ms	2.0513ms	487.4891 Ops/s	482.0831 Ops/s	$\color{#35bf28}+1.12\%$
test_consolidate[reduce-overhead-None]	2.0459ms	1.9732ms	506.7881 Ops/s	503.9712 Ops/s	$\color{#35bf28}+0.56\%$
test_consolidate_njt[False-None]	8.7813ms	8.5934ms	116.3687 Ops/s	116.5874 Ops/s	$\color{#d91a1a}-0.19\%$
test_to[False-False-None]	2.2002ms	2.1033ms	475.4423 Ops/s	481.5573 Ops/s	$\color{#d91a1a}-1.27\%$
test_to[True-False-None]	2.2333ms	1.9587ms	510.5463 Ops/s	527.2496 Ops/s	$\color{#d91a1a}-3.17\%$
test_to[within-False-None]	0.1894s	7.3689ms	135.7050 Ops/s	164.1883 Ops/s	$\textbf{\color{#d91a1a}-17.35\%}$
test_to[True-default-None]	8.9360ms	8.7475ms	114.3188 Ops/s	113.8923 Ops/s	$\color{#35bf28}+0.37\%$
test_to_njt[False-False-None]	8.7446ms	8.4661ms	118.1184 Ops/s	116.7904 Ops/s	$\color{#35bf28}+1.14\%$
test_to_njt[True-False-None]	7.0801ms	6.9277ms	144.3489 Ops/s	141.8110 Ops/s	$\color{#35bf28}+1.79\%$
test_to_njt[within-False-None]	15.6987ms	15.5270ms	64.4041 Ops/s	64.3823 Ops/s	$\color{#35bf28}+0.03\%$
test_creation[device0]	0.4715ms	0.1150ms	8.6924 KOps/s	8.7372 KOps/s	$\color{#d91a1a}-0.51\%$
test_creation_from_tensor	0.4291ms	0.1129ms	8.8538 KOps/s	8.8802 KOps/s	$\color{#d91a1a}-0.30\%$
test_add_one[memmap_tensor0]	0.3385ms	6.5709μs	152.1853 KOps/s	152.2634 KOps/s	$\color{#d91a1a}-0.05\%$
test_contiguous[memmap_tensor0]	30.8410μs	0.6768μs	1.4775 MOps/s	2.1376 MOps/s	$\textbf{\color{#d91a1a}-30.88\%}$
test_stack[memmap_tensor0]	43.2810μs	4.7591μs	210.1227 KOps/s	214.5939 KOps/s	$\color{#d91a1a}-2.08\%$
test_memmaptd_index	1.1364ms	0.2773ms	3.6065 KOps/s	3.6829 KOps/s	$\color{#d91a1a}-2.07\%$
test_memmaptd_index_astensor	0.5309ms	0.3781ms	2.6447 KOps/s	2.6424 KOps/s	$\color{#35bf28}+0.08\%$
test_memmaptd_index_op	0.8004ms	0.6398ms	1.5631 KOps/s	1.5834 KOps/s	$\color{#d91a1a}-1.29\%$
test_serialize_model	0.3068s	0.1595s	6.2681 Ops/s	7.3672 Ops/s	$\textbf{\color{#d91a1a}-14.92\%}$
test_serialize_model_pickle	1.3472s	1.1939s	0.8376 Ops/s	0.8260 Ops/s	$\color{#35bf28}+1.40\%$
test_serialize_weights	0.1376s	0.1346s	7.4303 Ops/s	7.4974 Ops/s	$\color{#d91a1a}-0.90\%$
test_serialize_weights_returnearly	0.4323s	89.8198ms	11.1334 Ops/s	15.3767 Ops/s	$\textbf{\color{#d91a1a}-27.60\%}$
test_serialize_weights_pickle	1.3659s	1.2140s	0.8237 Ops/s	0.8224 Ops/s	$\color{#35bf28}+0.16\%$
test_reshape_pytree	0.2195ms	32.9746μs	30.3264 KOps/s	30.2957 KOps/s	$\color{#35bf28}+0.10\%$
test_reshape_td	0.2016ms	46.2273μs	21.6323 KOps/s	21.7874 KOps/s	$\color{#d91a1a}-0.71\%$
test_view_pytree	0.2179ms	32.8203μs	30.4689 KOps/s	30.2840 KOps/s	$\color{#35bf28}+0.61\%$
test_view_td	96.0620μs	54.1455μs	18.4688 KOps/s	18.4608 KOps/s	$\color{#35bf28}+0.04\%$
test_unbind_pytree	0.2223ms	36.9891μs	27.0350 KOps/s	27.6630 KOps/s	$\color{#d91a1a}-2.27\%$
test_unbind_td	0.1708ms	49.4839μs	20.2086 KOps/s	20.0915 KOps/s	$\color{#35bf28}+0.58\%$
test_split_pytree	0.2008ms	42.8757μs	23.3232 KOps/s	23.4844 KOps/s	$\color{#d91a1a}-0.69\%$
test_split_td	0.1767ms	65.0104μs	15.3822 KOps/s	15.6220 KOps/s	$\color{#d91a1a}-1.54\%$
test_add_pytree	0.2042ms	42.7649μs	23.3837 KOps/s	23.2026 KOps/s	$\color{#35bf28}+0.78\%$
test_add_td	0.1061ms	55.8775μs	17.8963 KOps/s	18.0595 KOps/s	$\color{#d91a1a}-0.90\%$
test_compile_add_one_nested[tensordict-compile]	0.2142ms	0.1390ms	7.1946 KOps/s	6.6405 KOps/s	$\textbf{\color{#35bf28}+8.34\%}$
test_compile_add_one_nested[tensordict-eager]	0.6862ms	0.2015ms	4.9633 KOps/s	4.9123 KOps/s	$\color{#35bf28}+1.04\%$
test_compile_add_one_nested[pytree-compile]	0.1424ms	0.1079ms	9.2688 KOps/s	9.0438 KOps/s	$\color{#35bf28}+2.49\%$
test_compile_add_one_nested[pytree-eager]	0.4272ms	0.1779ms	5.6220 KOps/s	5.4883 KOps/s	$\color{#35bf28}+2.44\%$
test_compile_copy_nested[tensordict-compile]	0.2502ms	10.3820μs	96.3203 KOps/s	101.6502 KOps/s	$\textbf{\color{#d91a1a}-5.24\%}$
test_compile_copy_nested[tensordict-eager]	0.1353ms	52.9968μs	18.8691 KOps/s	18.5219 KOps/s	$\color{#35bf28}+1.87\%$
test_compile_copy_nested[pytree-compile]	0.1205ms	9.8336μs	101.6923 KOps/s	103.7595 KOps/s	$\color{#d91a1a}-1.99\%$
test_compile_copy_nested[pytree-eager]	0.5049ms	70.0736μs	14.2707 KOps/s	14.2589 KOps/s	$\color{#35bf28}+0.08\%$
test_compile_add_one_flat[tensordict-compile]	0.2120ms	0.1754ms	5.7002 KOps/s	5.1940 KOps/s	$\textbf{\color{#35bf28}+9.75\%}$
test_compile_add_one_flat[tensordict-eager]	0.3139ms	0.2781ms	3.5957 KOps/s	3.4987 KOps/s	$\color{#35bf28}+2.77\%$
test_compile_add_one_flat[tensorclass-compile]	0.3032ms	0.1153ms	8.6708 KOps/s	8.0023 KOps/s	$\textbf{\color{#35bf28}+8.35\%}$
test_compile_add_one_flat[tensorclass-eager]	0.1361ms	72.3253μs	13.8264 KOps/s	13.6099 KOps/s	$\color{#35bf28}+1.59\%$
test_compile_add_one_flat[pytree-compile]	0.1947ms	0.1573ms	6.3576 KOps/s	6.1470 KOps/s	$\color{#35bf28}+3.43\%$
test_compile_add_one_flat[pytree-eager]	0.8092ms	0.5144ms	1.9441 KOps/s	1.9022 KOps/s	$\color{#35bf28}+2.20\%$
test_compile_add_self_flat[tensordict-eager]	0.4294ms	0.3310ms	3.0209 KOps/s	2.9629 KOps/s	$\color{#35bf28}+1.96\%$
test_compile_add_self_flat[tensordict-compile]	0.2371ms	0.1778ms	5.6246 KOps/s	5.1571 KOps/s	$\textbf{\color{#35bf28}+9.06\%}$
test_compile_add_self_flat[tensorclass-eager]	0.2242ms	89.6344μs	11.1564 KOps/s	11.2960 KOps/s	$\color{#d91a1a}-1.24\%$
test_compile_add_self_flat[tensorclass-compile]	0.3304ms	0.1227ms	8.1527 KOps/s	7.9226 KOps/s	$\color{#35bf28}+2.90\%$
test_compile_add_self_flat[pytree-eager]	0.6682ms	0.4474ms	2.2351 KOps/s	2.2634 KOps/s	$\color{#d91a1a}-1.25\%$
test_compile_add_self_flat[pytree-compile]	0.2546ms	0.1625ms	6.1557 KOps/s	6.1550 KOps/s	$\color{#35bf28}+0.01\%$
test_compile_copy_flat[tensordict-compile]	0.1222ms	13.2867μs	75.2634 KOps/s	75.5952 KOps/s	$\color{#d91a1a}-0.44\%$
test_compile_copy_flat[tensordict-eager]	0.1536ms	42.0058μs	23.8063 KOps/s	24.0129 KOps/s	$\color{#d91a1a}-0.86\%$
test_compile_copy_flat[pytree-compile]	0.1259ms	11.0204μs	90.7404 KOps/s	94.4431 KOps/s	$\color{#d91a1a}-3.92\%$
test_compile_copy_flat[pytree-eager]	0.4004ms	52.1523μs	19.1746 KOps/s	19.0469 KOps/s	$\color{#35bf28}+0.67\%$
test_compile_assign_and_add[tensordict-compile]	2.0574ms	0.1778ms	5.6248 KOps/s	5.0943 KOps/s	$\textbf{\color{#35bf28}+10.41\%}$
test_compile_assign_and_add[tensordict-eager]	3.7687ms	3.3147ms	301.6893 Ops/s	293.3295 Ops/s	$\color{#35bf28}+2.85\%$
test_compile_assign_and_add[pytree-compile]	2.0321ms	0.1677ms	5.9619 KOps/s	6.0353 KOps/s	$\color{#d91a1a}-1.22\%$
test_compile_assign_and_add[pytree-eager]	3.3542ms	2.8703ms	348.3973 Ops/s	350.3388 Ops/s	$\color{#d91a1a}-0.55\%$
test_compile_indexing[tensor-tensordict-compile]	0.2323ms	0.1127ms	8.8742 KOps/s	8.5167 KOps/s	$\color{#35bf28}+4.20\%$
test_compile_indexing[tensor-tensordict-eager]	0.3221ms	75.5226μs	13.2411 KOps/s	12.5877 KOps/s	$\textbf{\color{#35bf28}+5.19\%}$
test_compile_indexing[tensor-tensorclass-compile]	0.2925ms	98.9471μs	10.1064 KOps/s	10.1178 KOps/s	$\color{#d91a1a}-0.11\%$
test_compile_indexing[tensor-tensorclass-eager]	0.2501ms	45.1905μs	22.1285 KOps/s	21.2433 KOps/s	$\color{#35bf28}+4.17\%$
test_compile_indexing[tensor-pytree-compile]	0.2410ms	98.9244μs	10.1087 KOps/s	9.9703 KOps/s	$\color{#35bf28}+1.39\%$
test_compile_indexing[tensor-pytree-eager]	0.2496ms	45.3347μs	22.0582 KOps/s	22.7670 KOps/s	$\color{#d91a1a}-3.11\%$
test_compile_indexing[slice-tensordict-compile]	0.2213ms	56.0589μs	17.8384 KOps/s	17.4683 KOps/s	$\color{#35bf28}+2.12\%$
test_compile_indexing[slice-tensordict-eager]	0.2120ms	27.3389μs	36.5779 KOps/s	36.1171 KOps/s	$\color{#35bf28}+1.28\%$
test_compile_indexing[slice-tensorclass-compile]	0.1663ms	45.0292μs	22.2078 KOps/s	22.3378 KOps/s	$\color{#d91a1a}-0.58\%$
test_compile_indexing[slice-tensorclass-eager]	0.2553ms	22.5630μs	44.3203 KOps/s	44.6574 KOps/s	$\color{#d91a1a}-0.75\%$
test_compile_indexing[slice-pytree-compile]	0.1852ms	45.0630μs	22.1912 KOps/s	22.2430 KOps/s	$\color{#d91a1a}-0.23\%$
test_compile_indexing[slice-pytree-eager]	0.3099ms	22.3787μs	44.6853 KOps/s	44.5437 KOps/s	$\color{#35bf28}+0.32\%$
test_compile_indexing[int-tensordict-compile]	95.6820μs	56.3140μs	17.7576 KOps/s	17.0365 KOps/s	$\color{#35bf28}+4.23\%$
test_compile_indexing[int-tensordict-eager]	0.2532ms	27.1387μs	36.8478 KOps/s	36.5236 KOps/s	$\color{#35bf28}+0.89\%$
test_compile_indexing[int-tensorclass-compile]	81.3310μs	44.4231μs	22.5108 KOps/s	22.0912 KOps/s	$\color{#35bf28}+1.90\%$
test_compile_indexing[int-tensorclass-eager]	0.2593ms	22.3797μs	44.6833 KOps/s	44.7253 KOps/s	$\color{#d91a1a}-0.09\%$
test_compile_indexing[int-pytree-compile]	85.3620μs	44.0473μs	22.7028 KOps/s	22.3129 KOps/s	$\color{#35bf28}+1.75\%$
test_compile_indexing[int-pytree-eager]	0.2576ms	22.5161μs	44.4127 KOps/s	44.6443 KOps/s	$\color{#d91a1a}-0.52\%$
test_compile_replace[single-eager]	83.8110μs	47.2665μs	21.1566 KOps/s	20.5339 KOps/s	$\color{#35bf28}+3.03\%$
test_compile_replace[single-compile]	0.1874ms	0.1043ms	9.5833 KOps/s	8.9074 KOps/s	$\textbf{\color{#35bf28}+7.59\%}$
test_compile_replace[multi-eager]	0.6363ms	0.5635ms	1.7746 KOps/s	1.7808 KOps/s	$\color{#d91a1a}-0.35\%$
test_compile_replace[multi-compile]	0.3017ms	0.1102ms	9.0771 KOps/s	8.1893 KOps/s	$\textbf{\color{#35bf28}+10.84\%}$
test_compile_tc_getattr_20[eager]	0.2173ms	0.1667ms	5.9982 KOps/s	6.0748 KOps/s	$\color{#d91a1a}-1.26\%$
test_compile_tc_getattr_20[compile]	0.4850ms	0.1184ms	8.4429 KOps/s	8.0713 KOps/s	$\color{#35bf28}+4.60\%$
test_compile_clone_shallow[20-eager]	50.2610μs	19.5684μs	51.1028 KOps/s	51.6931 KOps/s	$\color{#d91a1a}-1.14\%$
test_compile_clone_shallow[20-compile]	0.1266ms	11.4848μs	87.0719 KOps/s	84.4788 KOps/s	$\color{#35bf28}+3.07\%$
test_compile_clone_shallow[40-eager]	67.7910μs	34.5153μs	28.9727 KOps/s	29.7331 KOps/s	$\color{#d91a1a}-2.56\%$
test_compile_clone_shallow[40-compile]	63.7710μs	12.4593μs	80.2616 KOps/s	81.2144 KOps/s	$\color{#d91a1a}-1.17\%$
test_compile_clone_shallow[80-eager]	0.1068ms	63.9878μs	15.6280 KOps/s	15.9424 KOps/s	$\color{#d91a1a}-1.97\%$
test_compile_clone_shallow[80-compile]	75.2610μs	14.7582μs	67.7587 KOps/s	67.2976 KOps/s	$\color{#35bf28}+0.69\%$
test_compile_update_inplace[eager]	0.1047ms	59.3966μs	16.8360 KOps/s	17.0464 KOps/s	$\color{#d91a1a}-1.23\%$
test_compile_update_inplace[compile]	0.2569ms	0.1388ms	7.2059 KOps/s	6.8994 KOps/s	$\color{#35bf28}+4.44\%$
test_mod_add[eager]	0.1103ms	49.7769μs	20.0896 KOps/s	20.2705 KOps/s	$\color{#d91a1a}-0.89\%$
test_mod_add[compile]	0.4995ms	0.1040ms	9.6171 KOps/s	9.4942 KOps/s	$\color{#35bf28}+1.29\%$
test_mod_add[compile-overhead]	0.2370ms	0.1487ms	6.7247 KOps/s	6.5704 KOps/s	$\color{#35bf28}+2.35\%$
test_mod_wrap[eager]	0.3892ms	0.2944ms	3.3963 KOps/s	3.4197 KOps/s	$\color{#d91a1a}-0.69\%$
test_mod_wrap[compile]	0.4461ms	0.3482ms	2.8715 KOps/s	2.8110 KOps/s	$\color{#35bf28}+2.15\%$
test_mod_wrap[compile-overhead]	5.6157ms	3.2102ms	311.5023 Ops/s	250.2573 Ops/s	$\textbf{\color{#35bf28}+24.47\%}$
test_mod_wrap_and_backward[eager]	1.7425ms	1.6389ms	610.1770 Ops/s	664.0859 Ops/s	$\textbf{\color{#d91a1a}-8.12\%}$
test_mod_wrap_and_backward[compile]	1.6549ms	1.5760ms	634.5076 Ops/s	683.0269 Ops/s	$\textbf{\color{#d91a1a}-7.10\%}$
test_mod_wrap_and_backward[compile-overhead]	1.2912ms	0.9877ms	1.0125 KOps/s	1.0979 KOps/s	$\textbf{\color{#d91a1a}-7.78\%}$
test_seq_add[eager]	0.2370ms	0.1546ms	6.4663 KOps/s	6.4617 KOps/s	$\color{#35bf28}+0.07\%$
test_seq_add[compile]	0.1971ms	0.1169ms	8.5525 KOps/s	8.4848 KOps/s	$\color{#35bf28}+0.80\%$
test_seq_add[compile-overhead]	0.4248ms	0.1530ms	6.5371 KOps/s	6.3024 KOps/s	$\color{#35bf28}+3.72\%$
test_seq_wrap[eager]	0.6121ms	0.5216ms	1.9174 KOps/s	1.9139 KOps/s	$\color{#35bf28}+0.18\%$
test_seq_wrap[compile]	0.4830ms	0.3740ms	2.6738 KOps/s	2.6727 KOps/s	$\color{#35bf28}+0.04\%$
test_seq_wrap[compile-overhead]	0.3511ms	0.2654ms	3.7681 KOps/s	3.7137 KOps/s	$\color{#35bf28}+1.47\%$
test_func_call_runtime[False-eager]	0.9133ms	0.8387ms	1.1923 KOps/s	1.1716 KOps/s	$\color{#35bf28}+1.76\%$
test_func_call_runtime[False-compile]	0.9839ms	0.9076ms	1.1018 KOps/s	1.0829 KOps/s	$\color{#35bf28}+1.75\%$
test_func_call_runtime[False-compile-overhead]	0.5639ms	0.4624ms	2.1628 KOps/s	2.1459 KOps/s	$\color{#35bf28}+0.79\%$
test_func_call_runtime[True-eager]	1.1320ms	1.0740ms	931.0791 Ops/s	922.4840 Ops/s	$\color{#35bf28}+0.93\%$
test_func_call_runtime[True-compile]	0.9758ms	0.9225ms	1.0840 KOps/s	1.0697 KOps/s	$\color{#35bf28}+1.34\%$
test_func_call_runtime[True-compile-overhead]	0.5996ms	0.4747ms	2.1066 KOps/s	2.0718 KOps/s	$\color{#35bf28}+1.68\%$
test_func_call_cm_runtime[False-eager]	0.8909ms	0.8411ms	1.1889 KOps/s	1.1814 KOps/s	$\color{#35bf28}+0.63\%$
test_func_call_cm_runtime[False-compile]	1.0889ms	0.9136ms	1.0946 KOps/s	1.0789 KOps/s	$\color{#35bf28}+1.45\%$
test_func_call_cm_runtime[False-compile-overhead]	0.5730ms	0.4649ms	2.1509 KOps/s	2.1420 KOps/s	$\color{#35bf28}+0.41\%$
test_func_call_cm_runtime[True-eager]	1.3739ms	1.2310ms	812.3381 Ops/s	809.2265 Ops/s	$\color{#35bf28}+0.38\%$
test_func_call_cm_runtime[True-compile]	1.0100ms	0.9584ms	1.0434 KOps/s	1.0339 KOps/s	$\color{#35bf28}+0.92\%$
test_func_call_cm_runtime[True-compile-overhead]	0.6434ms	0.5126ms	1.9509 KOps/s	1.9378 KOps/s	$\color{#35bf28}+0.67\%$
test_vmap_func_call_cm_runtime[eager]	2.8825ms	2.3706ms	421.8331 Ops/s	415.8255 Ops/s	$\color{#35bf28}+1.44\%$
test_vmap_func_call_cm_runtime[compile]	1.1197ms	0.9872ms	1.0130 KOps/s	1.0105 KOps/s	$\color{#35bf28}+0.24\%$
test_vmap_func_call_cm_runtime[compile-overhead]	0.5727ms	0.5171ms	1.9337 KOps/s	1.9150 KOps/s	$\color{#35bf28}+0.98\%$
test_distributed	2.8567ms	0.1629ms	6.1377 KOps/s	5.7009 KOps/s	$\textbf{\color{#35bf28}+7.66\%}$
test_tdmodule	0.1362ms	27.6975μs	36.1043 KOps/s	36.3872 KOps/s	$\color{#d91a1a}-0.78\%$
test_tdmodule_dispatch	79.1910μs	45.4341μs	22.0099 KOps/s	22.1891 KOps/s	$\color{#d91a1a}-0.81\%$
test_tdseq	46.4600μs	27.0240μs	37.0041 KOps/s	37.2347 KOps/s	$\color{#d91a1a}-0.62\%$
test_tdseq_dispatch	69.8710μs	47.9270μs	20.8651 KOps/s	21.1194 KOps/s	$\color{#d91a1a}-1.20\%$
test_instantiation_functorch	2.1606ms	2.0806ms	480.6289 Ops/s	479.9896 Ops/s	$\color{#35bf28}+0.13\%$
test_exec_functorch	0.2304ms	0.1790ms	5.5875 KOps/s	5.5094 KOps/s	$\color{#35bf28}+1.42\%$
test_exec_functional_call	0.2267ms	0.1587ms	6.2993 KOps/s	6.1839 KOps/s	$\color{#35bf28}+1.87\%$
test_exec_td_decorator	0.4440ms	0.2365ms	4.2287 KOps/s	4.1747 KOps/s	$\color{#35bf28}+1.29\%$
test_vmap_mlp_speed_decorator[True-True]	1.0329ms	0.8257ms	1.2111 KOps/s	1.2001 KOps/s	$\color{#35bf28}+0.91\%$
test_vmap_mlp_speed_decorator[True-False]	1.0290ms	0.8270ms	1.2093 KOps/s	1.1990 KOps/s	$\color{#35bf28}+0.86\%$
test_vmap_mlp_speed_decorator[False-True]	0.9417ms	0.7125ms	1.4034 KOps/s	1.3890 KOps/s	$\color{#35bf28}+1.04\%$
test_vmap_mlp_speed_decorator[False-False]	0.8920ms	0.7148ms	1.3989 KOps/s	1.3854 KOps/s	$\color{#35bf28}+0.98\%$
test_vmap_transformer_speed_decorator[True-True]	21.4727ms	20.5478ms	48.6671 Ops/s	48.3165 Ops/s	$\color{#35bf28}+0.73\%$
test_vmap_transformer_speed_decorator[True-False]	21.2821ms	20.5704ms	48.6137 Ops/s	48.3497 Ops/s	$\color{#35bf28}+0.55\%$
test_vmap_transformer_speed_decorator[False-True]	20.4891ms	20.3610ms	49.1135 Ops/s	48.7338 Ops/s	$\color{#35bf28}+0.78\%$
test_vmap_transformer_speed_decorator[False-False]	20.4641ms	20.3327ms	49.1819 Ops/s	48.7538 Ops/s	$\color{#35bf28}+0.88\%$
test_to_module_speed[True]	1.5929ms	1.4787ms	676.2478 Ops/s	679.0269 Ops/s	$\color{#d91a1a}-0.41\%$
test_to_module_speed[False]	1.5720ms	1.4597ms	685.0646 Ops/s	684.8377 Ops/s	$\color{#35bf28}+0.03\%$
test_tc_init	73.3110μs	45.2847μs	22.0825 KOps/s	22.3998 KOps/s	$\color{#d91a1a}-1.42\%$
test_tc_init_tensor_only	40.1400μs	9.7734μs	102.3190 KOps/s	100.9460 KOps/s	$\color{#35bf28}+1.36\%$
test_tc_init_nested	0.1207ms	87.9406μs	11.3713 KOps/s	11.3991 KOps/s	$\color{#d91a1a}-0.24\%$
test_tc_init_many_fields	39.6010μs	16.2941μs	61.3718 KOps/s	60.8034 KOps/s	$\color{#35bf28}+0.93\%$
test_tc_first_layer_tensor	22.7500μs	1.8462μs	541.6587 KOps/s	548.2739 KOps/s	$\color{#d91a1a}-1.21\%$
test_tc_first_layer_tensor_only	2.9790μs	0.4093μs	2.4430 MOps/s	2.5547 MOps/s	$\color{#d91a1a}-4.37\%$
test_tc_first_layer_tensor_set	25.5100μs	4.0240μs	248.5104 KOps/s	256.3678 KOps/s	$\color{#d91a1a}-3.06\%$
test_tc_first_layer_tensor_only_set	22.2700μs	3.2988μs	303.1382 KOps/s	307.9679 KOps/s	$\color{#d91a1a}-1.57\%$
test_tc_first_layer_nontensor	76.0210μs	6.2222μs	160.7140 KOps/s	161.5485 KOps/s	$\color{#d91a1a}-0.52\%$
test_tc_second_layer_tensor	24.5100μs	4.4238μs	226.0501 KOps/s	228.6801 KOps/s	$\color{#d91a1a}-1.15\%$
test_tc_second_layer_nontensor	45.9410μs	8.7836μs	113.8488 KOps/s	115.4156 KOps/s	$\color{#d91a1a}-1.36\%$
test_unbind	0.2498s	16.1774ms	61.8148 Ops/s	67.3421 Ops/s	$\textbf{\color{#d91a1a}-8.21\%}$
test_full_like	5.9793ms	4.4227ms	226.1066 Ops/s	228.4858 Ops/s	$\color{#d91a1a}-1.04\%$
test_zeros_like	4.5998ms	4.3546ms	229.6406 Ops/s	228.8865 Ops/s	$\color{#35bf28}+0.33\%$
test_ones_like	4.6605ms	4.3679ms	228.9432 Ops/s	228.5284 Ops/s	$\color{#35bf28}+0.18\%$
test_clone	6.5528ms	6.3732ms	156.9063 Ops/s	155.4229 Ops/s	$\color{#35bf28}+0.95\%$
test_squeeze	0.1997ms	14.2784μs	70.0360 KOps/s	69.6617 KOps/s	$\color{#35bf28}+0.54\%$
test_unsqueeze	0.1897ms	0.1162ms	8.6059 KOps/s	9.1083 KOps/s	$\textbf{\color{#d91a1a}-5.52\%}$
test_split	0.2415ms	0.1840ms	5.4359 KOps/s	5.3829 KOps/s	$\color{#35bf28}+0.98\%$
test_permute	0.2679ms	0.2120ms	4.7161 KOps/s	4.8921 KOps/s	$\color{#d91a1a}-3.60\%$
test_stack	44.8741ms	43.0285ms	23.2404 Ops/s	19.7236 Ops/s	$\textbf{\color{#35bf28}+17.83\%}$
test_cat	45.2800ms	42.7895ms	23.3702 Ops/s	19.7657 Ops/s	$\textbf{\color{#35bf28}+18.24\%}$
test_sequential_tensordict	0.3836ms	0.2298ms	4.3509 KOps/s	4.5668 KOps/s	$\color{#d91a1a}-4.73\%$
test_sequential_graph_module	0.5406ms	0.1245ms	8.0299 KOps/s	8.2709 KOps/s	$\color{#d91a1a}-2.91\%$
test_nested_tensordict	0.3658ms	0.2963ms	3.3754 KOps/s	3.5070 KOps/s	$\color{#d91a1a}-3.75\%$
test_nested_graph_module	0.2769ms	0.1297ms	7.7127 KOps/s	7.7400 KOps/s	$\color{#d91a1a}-0.35\%$

github-actions · 2026-03-10T19:39:03Z

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

Total Benchmarks: 261. Improved: $\large\color{#35bf28}11$. Worsened: $\large\color{#d91a1a}9$.

Expand to view detailed results

Name	Max	Mean	Ops	Ops on Repo `HEAD`	Change
test_plain_set_nested	32.7010μs	14.8142μs	67.5029 KOps/s	66.6330 KOps/s	$\color{#35bf28}+1.31\%$
test_plain_set_stack_nested	36.0110μs	15.4884μs	64.5646 KOps/s	65.5387 KOps/s	$\color{#d91a1a}-1.49\%$
test_plain_set_nested_inplace	84.5910μs	16.9391μs	59.0351 KOps/s	59.2134 KOps/s	$\color{#d91a1a}-0.30\%$
test_plain_set_stack_nested_inplace	38.1510μs	16.5652μs	60.3677 KOps/s	59.5511 KOps/s	$\color{#35bf28}+1.37\%$
test_items	32.8000μs	6.0076μs	166.4562 KOps/s	166.4028 KOps/s	$\color{#35bf28}+0.03\%$
test_items_nested	0.5415ms	0.4673ms	2.1398 KOps/s	2.1188 KOps/s	$\color{#35bf28}+0.99\%$
test_items_nested_locked	0.5390ms	0.4699ms	2.1279 KOps/s	2.1051 KOps/s	$\color{#35bf28}+1.08\%$
test_items_nested_leaf	0.1586ms	98.2313μs	10.1801 KOps/s	10.0622 KOps/s	$\color{#35bf28}+1.17\%$
test_items_stack_nested	0.5533ms	0.4661ms	2.1454 KOps/s	2.1089 KOps/s	$\color{#35bf28}+1.73\%$
test_items_stack_nested_leaf	0.1544ms	97.7894μs	10.2261 KOps/s	10.1211 KOps/s	$\color{#35bf28}+1.04\%$
test_items_stack_nested_locked	0.5285ms	0.4695ms	2.1300 KOps/s	2.0980 KOps/s	$\color{#35bf28}+1.52\%$
test_keys	22.0600μs	4.2200μs	236.9645 KOps/s	236.1779 KOps/s	$\color{#35bf28}+0.33\%$
test_keys_nested	0.1818ms	0.1294ms	7.7253 KOps/s	7.5860 KOps/s	$\color{#35bf28}+1.84\%$
test_keys_nested_locked	0.7997ms	0.1378ms	7.2558 KOps/s	7.1247 KOps/s	$\color{#35bf28}+1.84\%$
test_keys_nested_leaf	0.1781ms	0.1200ms	8.3342 KOps/s	8.1928 KOps/s	$\color{#35bf28}+1.73\%$
test_keys_stack_nested	0.4971ms	0.1294ms	7.7256 KOps/s	7.5186 KOps/s	$\color{#35bf28}+2.75\%$
test_keys_stack_nested_leaf	0.1806ms	0.1207ms	8.2828 KOps/s	8.1957 KOps/s	$\color{#35bf28}+1.06\%$
test_keys_stack_nested_locked	0.2168ms	0.1379ms	7.2542 KOps/s	7.1926 KOps/s	$\color{#35bf28}+0.86\%$
test_values	8.9060μs	1.0585μs	944.7031 KOps/s	971.4463 KOps/s	$\color{#d91a1a}-2.75\%$
test_values_nested	87.2210μs	51.8837μs	19.2739 KOps/s	18.8307 KOps/s	$\color{#35bf28}+2.35\%$
test_values_nested_locked	94.1210μs	55.3068μs	18.0810 KOps/s	17.7610 KOps/s	$\color{#35bf28}+1.80\%$
test_values_nested_leaf	92.6110μs	59.5881μs	16.7819 KOps/s	16.3465 KOps/s	$\color{#35bf28}+2.66\%$
test_values_stack_nested	80.0910μs	52.1789μs	19.1648 KOps/s	18.7590 KOps/s	$\color{#35bf28}+2.16\%$
test_values_stack_nested_leaf	88.2810μs	59.3863μs	16.8389 KOps/s	16.3485 KOps/s	$\color{#35bf28}+3.00\%$
test_values_stack_nested_locked	86.0810μs	55.4495μs	18.0344 KOps/s	17.8096 KOps/s	$\color{#35bf28}+1.26\%$
test_membership	6.7418μs	0.8378μs	1.1937 MOps/s	1.1696 MOps/s	$\color{#35bf28}+2.06\%$
test_membership_nested	46.4400μs	2.9036μs	344.4058 KOps/s	350.0573 KOps/s	$\color{#d91a1a}-1.61\%$
test_membership_nested_leaf	0.5131ms	2.8701μs	348.4257 KOps/s	359.1058 KOps/s	$\color{#d91a1a}-2.97\%$
test_membership_stacked_nested	24.3500μs	2.8914μs	345.8489 KOps/s	346.2518 KOps/s	$\color{#d91a1a}-0.12\%$
test_membership_stacked_nested_leaf	30.3300μs	2.8789μs	347.3590 KOps/s	345.6727 KOps/s	$\color{#35bf28}+0.49\%$
test_membership_nested_last	36.7610μs	4.3637μs	229.1624 KOps/s	228.7273 KOps/s	$\color{#35bf28}+0.19\%$
test_membership_nested_leaf_last	31.5410μs	4.3445μs	230.1785 KOps/s	232.1878 KOps/s	$\color{#d91a1a}-0.87\%$
test_membership_stacked_nested_last	31.6910μs	4.3506μs	229.8522 KOps/s	229.7993 KOps/s	$\color{#35bf28}+0.02\%$
test_membership_stacked_nested_leaf_last	31.1910μs	4.3622μs	229.2446 KOps/s	232.0931 KOps/s	$\color{#d91a1a}-1.23\%$
test_nested_getleaf	53.2700μs	21.9492μs	45.5597 KOps/s	46.6160 KOps/s	$\color{#d91a1a}-2.27\%$
test_nested_get	45.9210μs	20.6495μs	48.4274 KOps/s	49.5002 KOps/s	$\color{#d91a1a}-2.17\%$
test_stacked_getleaf	52.2410μs	21.6809μs	46.1236 KOps/s	46.4060 KOps/s	$\color{#d91a1a}-0.61\%$
test_stacked_get	44.6610μs	20.5528μs	48.6553 KOps/s	49.3844 KOps/s	$\color{#d91a1a}-1.48\%$
test_nested_getitemleaf	50.0600μs	22.2222μs	45.0000 KOps/s	45.8163 KOps/s	$\color{#d91a1a}-1.78\%$
test_nested_getitem	50.7810μs	21.1536μs	47.2733 KOps/s	47.6194 KOps/s	$\color{#d91a1a}-0.73\%$
test_stacked_getitemleaf	46.6900μs	21.8901μs	45.6827 KOps/s	46.0272 KOps/s	$\color{#d91a1a}-0.75\%$
test_stacked_getitem	55.0310μs	21.0002μs	47.6185 KOps/s	48.5766 KOps/s	$\color{#d91a1a}-1.97\%$
test_lock_nested	7.9871ms	0.4873ms	2.0523 KOps/s	2.0713 KOps/s	$\color{#d91a1a}-0.92\%$
test_lock_stack_nested	0.5492ms	0.4866ms	2.0549 KOps/s	2.0410 KOps/s	$\color{#35bf28}+0.68\%$
test_unlock_nested	0.4942ms	0.3929ms	2.5453 KOps/s	2.5283 KOps/s	$\color{#35bf28}+0.68\%$
test_unlock_stack_nested	0.4793ms	0.3959ms	2.5257 KOps/s	2.5117 KOps/s	$\color{#35bf28}+0.55\%$
test_flatten_speed	0.1529ms	0.1229ms	8.1356 KOps/s	8.0694 KOps/s	$\color{#35bf28}+0.82\%$
test_unflatten_speed	0.6739ms	0.5729ms	1.7456 KOps/s	1.7445 KOps/s	$\color{#35bf28}+0.06\%$
test_common_ops	0.9029ms	0.7069ms	1.4146 KOps/s	1.4215 KOps/s	$\color{#d91a1a}-0.49\%$
test_creation	0.1017ms	3.0572μs	327.0954 KOps/s	315.9969 KOps/s	$\color{#35bf28}+3.51\%$
test_creation_empty	31.7600μs	6.9883μs	143.0956 KOps/s	143.1358 KOps/s	$\color{#d91a1a}-0.03\%$
test_creation_nested_1	35.2900μs	11.6220μs	86.0434 KOps/s	86.1844 KOps/s	$\color{#d91a1a}-0.16\%$
test_creation_nested_2	44.9310μs	13.2965μs	75.2079 KOps/s	74.1564 KOps/s	$\color{#35bf28}+1.42\%$
test_creation_many_keys[10]	97.3110μs	21.2252μs	47.1138 KOps/s	47.3385 KOps/s	$\color{#d91a1a}-0.47\%$
test_creation_many_keys[50]	0.1326ms	92.7354μs	10.7834 KOps/s	11.0387 KOps/s	$\color{#d91a1a}-2.31\%$
test_creation_many_keys[100]	0.2325ms	0.1795ms	5.5713 KOps/s	5.6121 KOps/s	$\color{#d91a1a}-0.73\%$
test_creation_nested_many_keys[10]	80.7710μs	45.9553μs	21.7603 KOps/s	22.1688 KOps/s	$\color{#d91a1a}-1.84\%$
test_creation_nested_many_keys[50]	0.2357ms	0.1884ms	5.3082 KOps/s	5.3761 KOps/s	$\color{#d91a1a}-1.26\%$
test_clone	39.6100μs	13.4576μs	74.3077 KOps/s	73.8847 KOps/s	$\color{#35bf28}+0.57\%$
test_getitem[int]	1.5531ms	15.3386μs	65.1948 KOps/s	59.2613 KOps/s	$\textbf{\color{#35bf28}+10.01\%}$
test_getitem[slice_int]	0.1466ms	24.3545μs	41.0602 KOps/s	40.7457 KOps/s	$\color{#35bf28}+0.77\%$
test_getitem[range]	0.1772ms	64.2596μs	15.5619 KOps/s	15.5240 KOps/s	$\color{#35bf28}+0.24\%$
test_getitem[tuple]	0.1415ms	24.2036μs	41.3162 KOps/s	41.6271 KOps/s	$\color{#d91a1a}-0.75\%$
test_getitem[list]	0.1912ms	58.5794μs	17.0708 KOps/s	16.8733 KOps/s	$\color{#35bf28}+1.17\%$
test_setitem_dim[int]	43.4510μs	25.8669μs	38.6595 KOps/s	38.3745 KOps/s	$\color{#35bf28}+0.74\%$
test_setitem_dim[slice_int]	63.5310μs	43.4529μs	23.0134 KOps/s	23.4118 KOps/s	$\color{#d91a1a}-1.70\%$
test_setitem_dim[range]	0.1245ms	96.2853μs	10.3858 KOps/s	10.4187 KOps/s	$\color{#d91a1a}-0.32\%$
test_setitem_dim[tuple]	63.3910μs	40.7615μs	24.5330 KOps/s	24.3015 KOps/s	$\color{#35bf28}+0.95\%$
test_setitem	47.3410μs	17.7024μs	56.4895 KOps/s	55.3177 KOps/s	$\color{#35bf28}+2.12\%$
test_set	48.2800μs	17.0787μs	58.5524 KOps/s	58.6404 KOps/s	$\color{#d91a1a}-0.15\%$
test_set_shared	0.5035ms	0.2049ms	4.8816 KOps/s	4.8591 KOps/s	$\color{#35bf28}+0.46\%$
test_update	0.3668ms	22.1823μs	45.0809 KOps/s	45.7603 KOps/s	$\color{#d91a1a}-1.48\%$
test_update_nested	64.3810μs	33.5682μs	29.7901 KOps/s	30.2658 KOps/s	$\color{#d91a1a}-1.57\%$
test_update__nested	0.4673ms	34.2350μs	29.2099 KOps/s	28.8188 KOps/s	$\color{#35bf28}+1.36\%$
test_set_nested	56.8110μs	18.9537μs	52.7602 KOps/s	51.7470 KOps/s	$\color{#35bf28}+1.96\%$
test_set_nested_new	53.4410μs	24.2684μs	41.2059 KOps/s	41.2014 KOps/s	$\color{#35bf28}+0.01\%$
test_select	73.1110μs	40.6672μs	24.5898 KOps/s	24.6523 KOps/s	$\color{#d91a1a}-0.25\%$
test_select_nested	0.1023ms	76.1900μs	13.1251 KOps/s	13.1386 KOps/s	$\color{#d91a1a}-0.10\%$
test_exclude_nested	0.1205ms	92.7786μs	10.7784 KOps/s	10.5727 KOps/s	$\color{#35bf28}+1.95\%$
test_empty[True]	0.4830ms	0.4031ms	2.4807 KOps/s	2.4585 KOps/s	$\color{#35bf28}+0.91\%$
test_empty[False]	8.6550μs	1.3309μs	751.3689 KOps/s	757.7801 KOps/s	$\color{#d91a1a}-0.85\%$
test_to	0.1112ms	74.4071μs	13.4396 KOps/s	13.2419 KOps/s	$\color{#35bf28}+1.49\%$
test_to_nonblocking	0.1030ms	65.7684μs	15.2049 KOps/s	15.3491 KOps/s	$\color{#d91a1a}-0.94\%$
test_unbind_speed	0.4028ms	0.3389ms	2.9511 KOps/s	2.9584 KOps/s	$\color{#d91a1a}-0.25\%$
test_unbind_speed_stack0	0.4138ms	0.3362ms	2.9747 KOps/s	3.0042 KOps/s	$\color{#d91a1a}-0.98\%$
test_unbind_speed_stack1	0.1043s	0.8404ms	1.1899 KOps/s	1.1700 KOps/s	$\color{#35bf28}+1.71\%$
test_split	0.1036s	1.2825ms	779.7179 Ops/s	778.4955 Ops/s	$\color{#35bf28}+0.16\%$
test_chunk	0.1040s	1.2206ms	819.2491 Ops/s	915.5542 Ops/s	$\textbf{\color{#d91a1a}-10.52\%}$
test_to_cpu_blocking	28.9647ms	28.7887ms	34.7358 Ops/s	45.9049 Ops/s	$\textbf{\color{#d91a1a}-24.33\%}$
test_to_cpu_global_sync	11.5473ms	11.3985ms	87.7311 Ops/s	87.5868 Ops/s	$\color{#35bf28}+0.16\%$
test_to_cpu_event_sync	12.7097ms	12.4966ms	80.0219 Ops/s	81.0368 Ops/s	$\color{#d91a1a}-1.25\%$
test_to_cpu_default	0.1160s	13.7743ms	72.5988 Ops/s	80.8340 Ops/s	$\textbf{\color{#d91a1a}-10.19\%}$
test_consolidate[False-None]	4.4174ms	4.2228ms	236.8082 Ops/s	239.1893 Ops/s	$\color{#d91a1a}-1.00\%$
test_consolidate[default-None]	2.1911ms	2.0505ms	487.6803 Ops/s	482.1322 Ops/s	$\color{#35bf28}+1.15\%$
test_consolidate[reduce-overhead-None]	2.0826ms	1.9792ms	505.2665 Ops/s	499.9064 Ops/s	$\color{#35bf28}+1.07\%$
test_consolidate_njt[False-None]	0.1913s	10.2759ms	97.3151 Ops/s	116.4999 Ops/s	$\textbf{\color{#d91a1a}-16.47\%}$
test_to[False-False-None]	2.2149ms	2.1262ms	470.3151 Ops/s	475.2136 Ops/s	$\color{#d91a1a}-1.03\%$
test_to[True-False-None]	2.2246ms	1.9565ms	511.1209 Ops/s	517.6776 Ops/s	$\color{#d91a1a}-1.27\%$
test_to[within-False-None]	6.3746ms	6.2446ms	160.1385 Ops/s	161.2825 Ops/s	$\color{#d91a1a}-0.71\%$
test_to[True-default-None]	9.0855ms	8.8438ms	113.0735 Ops/s	112.4552 Ops/s	$\color{#35bf28}+0.55\%$
test_to_njt[False-False-None]	8.6842ms	8.5342ms	117.1752 Ops/s	115.9979 Ops/s	$\color{#35bf28}+1.01\%$
test_to_njt[True-False-None]	7.1390ms	7.0070ms	142.7142 Ops/s	141.9152 Ops/s	$\color{#35bf28}+0.56\%$
test_to_njt[within-False-None]	15.8344ms	15.7169ms	63.6257 Ops/s	63.2808 Ops/s	$\color{#35bf28}+0.55\%$
test_creation[device0]	0.4064ms	0.1156ms	8.6500 KOps/s	8.6770 KOps/s	$\color{#d91a1a}-0.31\%$
test_creation_from_tensor	0.4041ms	0.1134ms	8.8205 KOps/s	8.7780 KOps/s	$\color{#35bf28}+0.48\%$
test_add_one[memmap_tensor0]	0.2131ms	6.6092μs	151.3034 KOps/s	150.2025 KOps/s	$\color{#35bf28}+0.73\%$
test_contiguous[memmap_tensor0]	16.9700μs	0.6663μs	1.5009 MOps/s	2.1319 MOps/s	$\textbf{\color{#d91a1a}-29.60\%}$
test_stack[memmap_tensor0]	33.7600μs	4.7962μs	208.4977 KOps/s	215.4145 KOps/s	$\color{#d91a1a}-3.21\%$
test_memmaptd_index	0.1637s	0.3533ms	2.8308 KOps/s	3.6718 KOps/s	$\textbf{\color{#d91a1a}-22.90\%}$
test_memmaptd_index_astensor	0.5275ms	0.3779ms	2.6463 KOps/s	2.6571 KOps/s	$\color{#d91a1a}-0.41\%$
test_memmaptd_index_op	0.8509ms	0.6450ms	1.5504 KOps/s	1.5820 KOps/s	$\color{#d91a1a}-2.00\%$
test_serialize_model	0.1391s	0.1359s	7.3570 Ops/s	7.4024 Ops/s	$\color{#d91a1a}-0.61\%$
test_serialize_model_pickle	1.3478s	1.2106s	0.8260 Ops/s	0.7865 Ops/s	$\textbf{\color{#35bf28}+5.03\%}$
test_serialize_weights	0.1369s	0.1348s	7.4176 Ops/s	7.4601 Ops/s	$\color{#d91a1a}-0.57\%$
test_serialize_weights_returnearly	0.4351s	85.3972ms	11.7100 Ops/s	15.3755 Ops/s	$\textbf{\color{#d91a1a}-23.84\%}$
test_serialize_weights_pickle	1.3718s	1.2172s	0.8215 Ops/s	0.8213 Ops/s	$\color{#35bf28}+0.03\%$
test_reshape_pytree	0.2098ms	33.0953μs	30.2158 KOps/s	30.5451 KOps/s	$\color{#d91a1a}-1.08\%$
test_reshape_td	82.8510μs	47.6133μs	21.0025 KOps/s	21.8962 KOps/s	$\color{#d91a1a}-4.08\%$
test_view_pytree	0.2260ms	33.0478μs	30.2592 KOps/s	30.4957 KOps/s	$\color{#d91a1a}-0.78\%$
test_view_td	94.0910μs	55.0323μs	18.1712 KOps/s	18.4953 KOps/s	$\color{#d91a1a}-1.75\%$
test_unbind_pytree	0.2318ms	36.9585μs	27.0574 KOps/s	8.4942 KOps/s	$\textbf{\color{#35bf28}+218.54\%}$
test_unbind_td	0.1151ms	50.2449μs	19.9025 KOps/s	19.7312 KOps/s	$\color{#35bf28}+0.87\%$
test_split_pytree	0.1982ms	42.8367μs	23.3445 KOps/s	23.5795 KOps/s	$\color{#d91a1a}-1.00\%$
test_split_td	0.1482ms	64.9055μs	15.4070 KOps/s	15.3359 KOps/s	$\color{#35bf28}+0.46\%$
test_add_pytree	0.1931ms	42.7704μs	23.3806 KOps/s	23.7163 KOps/s	$\color{#d91a1a}-1.42\%$
test_add_td	86.6710μs	55.6115μs	17.9819 KOps/s	17.8704 KOps/s	$\color{#35bf28}+0.62\%$
test_compile_add_one_nested[tensordict-compile]	0.1923ms	0.1410ms	7.0903 KOps/s	6.6628 KOps/s	$\textbf{\color{#35bf28}+6.42\%}$
test_compile_add_one_nested[tensordict-eager]	0.3562ms	0.2032ms	4.9216 KOps/s	4.9761 KOps/s	$\color{#d91a1a}-1.09\%$
test_compile_add_one_nested[pytree-compile]	0.1645ms	0.1091ms	9.1672 KOps/s	9.0299 KOps/s	$\color{#35bf28}+1.52\%$
test_compile_add_one_nested[pytree-eager]	0.4486ms	0.1806ms	5.5358 KOps/s	5.5613 KOps/s	$\color{#d91a1a}-0.46\%$
test_compile_copy_nested[tensordict-compile]	0.2522ms	10.2357μs	97.6970 KOps/s	98.1676 KOps/s	$\color{#d91a1a}-0.48\%$
test_compile_copy_nested[tensordict-eager]	0.1501ms	54.2505μs	18.4330 KOps/s	18.1384 KOps/s	$\color{#35bf28}+1.62\%$
test_compile_copy_nested[pytree-compile]	0.1389ms	9.8551μs	101.4707 KOps/s	102.4601 KOps/s	$\color{#d91a1a}-0.97\%$
test_compile_copy_nested[pytree-eager]	0.4968ms	68.3794μs	14.6243 KOps/s	14.6714 KOps/s	$\color{#d91a1a}-0.32\%$
test_compile_add_one_flat[tensordict-compile]	0.2487ms	0.1792ms	5.5817 KOps/s	5.3522 KOps/s	$\color{#35bf28}+4.29\%$
test_compile_add_one_flat[tensordict-eager]	0.4697ms	0.2798ms	3.5743 KOps/s	3.5419 KOps/s	$\color{#35bf28}+0.91\%$
test_compile_add_one_flat[tensorclass-compile]	0.2803ms	0.1173ms	8.5231 KOps/s	8.2084 KOps/s	$\color{#35bf28}+3.83\%$
test_compile_add_one_flat[tensorclass-eager]	0.1228ms	74.0179μs	13.5102 KOps/s	13.6943 KOps/s	$\color{#d91a1a}-1.34\%$
test_compile_add_one_flat[pytree-compile]	0.2326ms	0.1593ms	6.2784 KOps/s	6.1727 KOps/s	$\color{#35bf28}+1.71\%$
test_compile_add_one_flat[pytree-eager]	0.8246ms	0.5223ms	1.9146 KOps/s	1.9098 KOps/s	$\color{#35bf28}+0.25\%$
test_compile_add_self_flat[tensordict-eager]	0.4946ms	0.3335ms	2.9987 KOps/s	2.9570 KOps/s	$\color{#35bf28}+1.41\%$
test_compile_add_self_flat[tensordict-compile]	0.2205ms	0.1803ms	5.5467 KOps/s	5.0923 KOps/s	$\textbf{\color{#35bf28}+8.92\%}$
test_compile_add_self_flat[tensorclass-eager]	0.1454ms	90.7514μs	11.0191 KOps/s	11.1113 KOps/s	$\color{#d91a1a}-0.83\%$
test_compile_add_self_flat[tensorclass-compile]	0.2447ms	0.1200ms	8.3338 KOps/s	7.8845 KOps/s	$\textbf{\color{#35bf28}+5.70\%}$
test_compile_add_self_flat[pytree-eager]	0.6466ms	0.4308ms	2.3215 KOps/s	2.2910 KOps/s	$\color{#35bf28}+1.33\%$
test_compile_add_self_flat[pytree-compile]	0.2245ms	0.1597ms	6.2630 KOps/s	6.0443 KOps/s	$\color{#35bf28}+3.62\%$
test_compile_copy_flat[tensordict-compile]	0.1032ms	13.3601μs	74.8498 KOps/s	74.5773 KOps/s	$\color{#35bf28}+0.37\%$
test_compile_copy_flat[tensordict-eager]	84.6110μs	41.9168μs	23.8568 KOps/s	24.1075 KOps/s	$\color{#d91a1a}-1.04\%$
test_compile_copy_flat[pytree-compile]	0.1161ms	10.8882μs	91.8423 KOps/s	92.5026 KOps/s	$\color{#d91a1a}-0.71\%$
test_compile_copy_flat[pytree-eager]	0.4036ms	53.0371μs	18.8547 KOps/s	18.9197 KOps/s	$\color{#d91a1a}-0.34\%$
test_compile_assign_and_add[tensordict-compile]	2.0558ms	0.1772ms	5.6442 KOps/s	5.3621 KOps/s	$\textbf{\color{#35bf28}+5.26\%}$
test_compile_assign_and_add[tensordict-eager]	3.4960ms	3.3562ms	297.9556 Ops/s	299.4743 Ops/s	$\color{#d91a1a}-0.51\%$
test_compile_assign_and_add[pytree-compile]	1.9829ms	0.1619ms	6.1769 KOps/s	6.0612 KOps/s	$\color{#35bf28}+1.91\%$
test_compile_assign_and_add[pytree-eager]	2.9799ms	2.7893ms	358.5116 Ops/s	352.3057 Ops/s	$\color{#35bf28}+1.76\%$
test_compile_indexing[tensor-tensordict-compile]	0.2068ms	0.1100ms	9.0890 KOps/s	8.6628 KOps/s	$\color{#35bf28}+4.92\%$
test_compile_indexing[tensor-tensordict-eager]	0.3422ms	74.9222μs	13.3472 KOps/s	13.4256 KOps/s	$\color{#d91a1a}-0.58\%$
test_compile_indexing[tensor-tensorclass-compile]	0.1974ms	96.8357μs	10.3268 KOps/s	10.1848 KOps/s	$\color{#35bf28}+1.39\%$
test_compile_indexing[tensor-tensorclass-eager]	0.2506ms	44.7472μs	22.3478 KOps/s	22.7341 KOps/s	$\color{#d91a1a}-1.70\%$
test_compile_indexing[tensor-pytree-compile]	0.1635ms	98.0758μs	10.1962 KOps/s	10.1172 KOps/s	$\color{#35bf28}+0.78\%$
test_compile_indexing[tensor-pytree-eager]	0.2677ms	44.6649μs	22.3889 KOps/s	23.0075 KOps/s	$\color{#d91a1a}-2.69\%$
test_compile_indexing[slice-tensordict-compile]	0.2662ms	56.4918μs	17.7017 KOps/s	16.7562 KOps/s	$\textbf{\color{#35bf28}+5.64\%}$
test_compile_indexing[slice-tensordict-eager]	0.2261ms	27.8837μs	35.8632 KOps/s	35.6803 KOps/s	$\color{#35bf28}+0.51\%$
test_compile_indexing[slice-tensorclass-compile]	0.1340ms	44.8785μs	22.2824 KOps/s	22.1551 KOps/s	$\color{#35bf28}+0.57\%$
test_compile_indexing[slice-tensorclass-eager]	0.2609ms	22.9454μs	43.5817 KOps/s	44.3589 KOps/s	$\color{#d91a1a}-1.75\%$
test_compile_indexing[slice-pytree-compile]	87.0900μs	45.5588μs	21.9497 KOps/s	21.2844 KOps/s	$\color{#35bf28}+3.13\%$
test_compile_indexing[slice-pytree-eager]	0.2577ms	22.5382μs	44.3692 KOps/s	44.6794 KOps/s	$\color{#d91a1a}-0.69\%$
test_compile_indexing[int-tensordict-compile]	0.1245ms	60.1718μs	16.6191 KOps/s	16.1516 KOps/s	$\color{#35bf28}+2.89\%$
test_compile_indexing[int-tensordict-eager]	0.2675ms	28.4827μs	35.1090 KOps/s	35.9856 KOps/s	$\color{#d91a1a}-2.44\%$
test_compile_indexing[int-tensorclass-compile]	86.8210μs	45.7181μs	21.8732 KOps/s	21.8578 KOps/s	$\color{#35bf28}+0.07\%$
test_compile_indexing[int-tensorclass-eager]	0.2561ms	22.5482μs	44.3495 KOps/s	44.3854 KOps/s	$\color{#d91a1a}-0.08\%$
test_compile_indexing[int-pytree-compile]	70.3510μs	45.4005μs	22.0262 KOps/s	21.3619 KOps/s	$\color{#35bf28}+3.11\%$
test_compile_indexing[int-pytree-eager]	0.2901ms	22.5612μs	44.3240 KOps/s	44.4819 KOps/s	$\color{#d91a1a}-0.35\%$
test_compile_replace[single-eager]	88.4610μs	50.3286μs	19.8694 KOps/s	20.9260 KOps/s	$\textbf{\color{#d91a1a}-5.05\%}$
test_compile_replace[single-compile]	0.2060ms	0.1051ms	9.5122 KOps/s	9.1228 KOps/s	$\color{#35bf28}+4.27\%$
test_compile_replace[multi-eager]	0.6500ms	0.5646ms	1.7712 KOps/s	1.7685 KOps/s	$\color{#35bf28}+0.15\%$
test_compile_replace[multi-compile]	0.1956ms	0.1118ms	8.9429 KOps/s	8.7710 KOps/s	$\color{#35bf28}+1.96\%$
test_compile_tc_getattr_20[eager]	0.2468ms	0.1680ms	5.9527 KOps/s	6.0384 KOps/s	$\color{#d91a1a}-1.42\%$
test_compile_tc_getattr_20[compile]	0.3170ms	0.1200ms	8.3326 KOps/s	8.1709 KOps/s	$\color{#35bf28}+1.98\%$
test_compile_clone_shallow[20-eager]	48.6100μs	19.3018μs	51.8087 KOps/s	51.5547 KOps/s	$\color{#35bf28}+0.49\%$
test_compile_clone_shallow[20-compile]	52.7610μs	11.4493μs	87.3418 KOps/s	86.7901 KOps/s	$\color{#35bf28}+0.64\%$
test_compile_clone_shallow[40-eager]	95.5510μs	33.9843μs	29.4254 KOps/s	28.8965 KOps/s	$\color{#35bf28}+1.83\%$
test_compile_clone_shallow[40-compile]	61.6710μs	12.4768μs	80.1490 KOps/s	75.7131 KOps/s	$\textbf{\color{#35bf28}+5.86\%}$
test_compile_clone_shallow[80-eager]	98.5510μs	62.9203μs	15.8931 KOps/s	15.7178 KOps/s	$\color{#35bf28}+1.12\%$
test_compile_clone_shallow[80-compile]	53.9600μs	15.0180μs	66.5870 KOps/s	66.7094 KOps/s	$\color{#d91a1a}-0.18\%$
test_compile_update_inplace[eager]	0.1056ms	59.7627μs	16.7328 KOps/s	16.7294 KOps/s	$\color{#35bf28}+0.02\%$
test_compile_update_inplace[compile]	0.2750ms	0.1404ms	7.1212 KOps/s	6.8861 KOps/s	$\color{#35bf28}+3.42\%$
test_mod_add[eager]	96.6910μs	48.9719μs	20.4199 KOps/s	19.7562 KOps/s	$\color{#35bf28}+3.36\%$
test_mod_add[compile]	0.4937ms	0.1039ms	9.6210 KOps/s	9.3791 KOps/s	$\color{#35bf28}+2.58\%$
test_mod_add[compile-overhead]	0.5403ms	0.1511ms	6.6182 KOps/s	6.5134 KOps/s	$\color{#35bf28}+1.61\%$
test_mod_wrap[eager]	0.3675ms	0.2911ms	3.4351 KOps/s	3.3922 KOps/s	$\color{#35bf28}+1.26\%$
test_mod_wrap[compile]	0.5104ms	0.3507ms	2.8517 KOps/s	2.8338 KOps/s	$\color{#35bf28}+0.63\%$
test_mod_wrap[compile-overhead]	7.2378ms	3.9977ms	250.1465 Ops/s	246.4985 Ops/s	$\color{#35bf28}+1.48\%$
test_mod_wrap_and_backward[eager]	1.9201ms	1.5266ms	655.0386 Ops/s	667.0185 Ops/s	$\color{#d91a1a}-1.80\%$
test_mod_wrap_and_backward[compile]	1.9055ms	1.4543ms	687.6032 Ops/s	688.7837 Ops/s	$\color{#d91a1a}-0.17\%$
test_mod_wrap_and_backward[compile-overhead]	1.2619ms	0.8911ms	1.1222 KOps/s	1.1072 KOps/s	$\color{#35bf28}+1.35\%$
test_seq_add[eager]	0.2173ms	0.1544ms	6.4782 KOps/s	6.4276 KOps/s	$\color{#35bf28}+0.79\%$
test_seq_add[compile]	0.2062ms	0.1142ms	8.7543 KOps/s	8.4691 KOps/s	$\color{#35bf28}+3.37\%$
test_seq_add[compile-overhead]	0.4460ms	0.1543ms	6.4805 KOps/s	6.2679 KOps/s	$\color{#35bf28}+3.39\%$
test_seq_wrap[eager]	0.6495ms	0.5265ms	1.8993 KOps/s	1.9144 KOps/s	$\color{#d91a1a}-0.79\%$
test_seq_wrap[compile]	0.4271ms	0.3672ms	2.7230 KOps/s	2.6824 KOps/s	$\color{#35bf28}+1.51\%$
test_seq_wrap[compile-overhead]	0.3325ms	0.2664ms	3.7531 KOps/s	3.6557 KOps/s	$\color{#35bf28}+2.66\%$
test_func_call_runtime[False-eager]	0.9693ms	0.8805ms	1.1357 KOps/s	1.1956 KOps/s	$\textbf{\color{#d91a1a}-5.01\%}$
test_func_call_runtime[False-compile]	1.0872ms	0.9223ms	1.0842 KOps/s	1.0945 KOps/s	$\color{#d91a1a}-0.94\%$
test_func_call_runtime[False-compile-overhead]	0.5625ms	0.4665ms	2.1435 KOps/s	2.1329 KOps/s	$\color{#35bf28}+0.50\%$
test_func_call_runtime[True-eager]	1.2052ms	1.0769ms	928.5987 Ops/s	923.2273 Ops/s	$\color{#35bf28}+0.58\%$
test_func_call_runtime[True-compile]	1.0101ms	0.9247ms	1.0814 KOps/s	1.0774 KOps/s	$\color{#35bf28}+0.38\%$
test_func_call_runtime[True-compile-overhead]	0.5533ms	0.4805ms	2.0812 KOps/s	2.0626 KOps/s	$\color{#35bf28}+0.90\%$
test_func_call_cm_runtime[False-eager]	1.3422ms	0.8449ms	1.1836 KOps/s	1.1460 KOps/s	$\color{#35bf28}+3.28\%$
test_func_call_cm_runtime[False-compile]	1.0589ms	0.9164ms	1.0912 KOps/s	1.0890 KOps/s	$\color{#35bf28}+0.21\%$
test_func_call_cm_runtime[False-compile-overhead]	0.5284ms	0.4667ms	2.1426 KOps/s	2.1252 KOps/s	$\color{#35bf28}+0.82\%$
test_func_call_cm_runtime[True-eager]	1.4260ms	1.2351ms	809.6289 Ops/s	814.5816 Ops/s	$\color{#d91a1a}-0.61\%$
test_func_call_cm_runtime[True-compile]	1.0379ms	0.9593ms	1.0424 KOps/s	1.0335 KOps/s	$\color{#35bf28}+0.86\%$
test_func_call_cm_runtime[True-compile-overhead]	0.5949ms	0.5128ms	1.9502 KOps/s	1.9185 KOps/s	$\color{#35bf28}+1.65\%$
test_vmap_func_call_cm_runtime[eager]	2.8726ms	2.3856ms	419.1822 Ops/s	417.6687 Ops/s	$\color{#35bf28}+0.36\%$
test_vmap_func_call_cm_runtime[compile]	1.0873ms	0.9784ms	1.0220 KOps/s	1.0183 KOps/s	$\color{#35bf28}+0.37\%$
test_vmap_func_call_cm_runtime[compile-overhead]	0.5829ms	0.5186ms	1.9283 KOps/s	1.9063 KOps/s	$\color{#35bf28}+1.15\%$
test_distributed	0.5560ms	0.1528ms	6.5458 KOps/s	6.4787 KOps/s	$\color{#35bf28}+1.03\%$
test_tdmodule	0.5420ms	27.9074μs	35.8328 KOps/s	35.8571 KOps/s	$\color{#d91a1a}-0.07\%$
test_tdmodule_dispatch	76.3610μs	45.4980μs	21.9790 KOps/s	22.2197 KOps/s	$\color{#d91a1a}-1.08\%$
test_tdseq	48.1010μs	27.0242μs	37.0039 KOps/s	37.1403 KOps/s	$\color{#d91a1a}-0.37\%$
test_tdseq_dispatch	69.4710μs	47.8500μs	20.8987 KOps/s	20.8744 KOps/s	$\color{#35bf28}+0.12\%$
test_instantiation_functorch	2.1693ms	2.0765ms	481.5759 Ops/s	480.3818 Ops/s	$\color{#35bf28}+0.25\%$
test_exec_functorch	0.2257ms	0.1821ms	5.4925 KOps/s	5.5204 KOps/s	$\color{#d91a1a}-0.50\%$
test_exec_functional_call	0.2075ms	0.1623ms	6.1627 KOps/s	6.1997 KOps/s	$\color{#d91a1a}-0.60\%$
test_exec_td_decorator	0.4530ms	0.2395ms	4.1758 KOps/s	4.2030 KOps/s	$\color{#d91a1a}-0.65\%$
test_vmap_mlp_speed_decorator[True-True]	0.9930ms	0.8261ms	1.2105 KOps/s	1.2014 KOps/s	$\color{#35bf28}+0.76\%$
test_vmap_mlp_speed_decorator[True-False]	0.9939ms	0.8260ms	1.2107 KOps/s	1.2092 KOps/s	$\color{#35bf28}+0.13\%$
test_vmap_mlp_speed_decorator[False-True]	0.9461ms	0.7138ms	1.4010 KOps/s	1.4023 KOps/s	$\color{#d91a1a}-0.09\%$
test_vmap_mlp_speed_decorator[False-False]	0.9011ms	0.7137ms	1.4012 KOps/s	1.3989 KOps/s	$\color{#35bf28}+0.16\%$
test_vmap_transformer_speed_decorator[True-True]	21.4070ms	20.6491ms	48.4283 Ops/s	48.5931 Ops/s	$\color{#d91a1a}-0.34\%$
test_vmap_transformer_speed_decorator[True-False]	21.2142ms	20.6490ms	48.4286 Ops/s	48.4552 Ops/s	$\color{#d91a1a}-0.05\%$
test_vmap_transformer_speed_decorator[False-True]	20.5472ms	20.4047ms	49.0083 Ops/s	49.0372 Ops/s	$\color{#d91a1a}-0.06\%$
test_vmap_transformer_speed_decorator[False-False]	20.5882ms	20.4353ms	48.9348 Ops/s	49.0289 Ops/s	$\color{#d91a1a}-0.19\%$
test_to_module_speed[True]	1.5667ms	1.4827ms	674.4485 Ops/s	672.5730 Ops/s	$\color{#35bf28}+0.28\%$
test_to_module_speed[False]	1.5614ms	1.4659ms	682.1641 Ops/s	684.3982 Ops/s	$\color{#d91a1a}-0.33\%$
test_tc_init	77.8810μs	44.0925μs	22.6796 KOps/s	22.4598 KOps/s	$\color{#35bf28}+0.98\%$
test_tc_init_tensor_only	45.8510μs	9.8275μs	101.7556 KOps/s	102.6951 KOps/s	$\color{#d91a1a}-0.91\%$
test_tc_init_nested	0.1310ms	87.8189μs	11.3871 KOps/s	11.1023 KOps/s	$\color{#35bf28}+2.56\%$
test_tc_init_many_fields	43.5710μs	16.3019μs	61.3426 KOps/s	60.8278 KOps/s	$\color{#35bf28}+0.85\%$
test_tc_first_layer_tensor	23.8310μs	1.8169μs	550.3823 KOps/s	554.0510 KOps/s	$\color{#d91a1a}-0.66\%$
test_tc_first_layer_tensor_only	1.7320μs	0.3950μs	2.5316 MOps/s	2.4429 MOps/s	$\color{#35bf28}+3.63\%$
test_tc_first_layer_tensor_set	33.0110μs	3.9382μs	253.9222 KOps/s	254.8480 KOps/s	$\color{#d91a1a}-0.36\%$
test_tc_first_layer_tensor_only_set	33.0800μs	3.2878μs	304.1554 KOps/s	307.0407 KOps/s	$\color{#d91a1a}-0.94\%$
test_tc_first_layer_nontensor	15.0594ms	6.2995μs	158.7416 KOps/s	160.6485 KOps/s	$\color{#d91a1a}-1.19\%$
test_tc_second_layer_tensor	21.8100μs	4.4567μs	224.3816 KOps/s	221.6650 KOps/s	$\color{#35bf28}+1.23\%$
test_tc_second_layer_nontensor	40.9810μs	8.7351μs	114.4808 KOps/s	113.8922 KOps/s	$\color{#35bf28}+0.52\%$
test_unbind	0.2514s	14.3543ms	69.6655 Ops/s	53.7827 Ops/s	$\textbf{\color{#35bf28}+29.53\%}$
test_full_like	5.3093ms	4.4076ms	226.8813 Ops/s	226.6461 Ops/s	$\color{#35bf28}+0.10\%$
test_zeros_like	4.4858ms	4.3664ms	229.0190 Ops/s	228.8121 Ops/s	$\color{#35bf28}+0.09\%$
test_ones_like	4.5068ms	4.3650ms	229.0962 Ops/s	228.7275 Ops/s	$\color{#35bf28}+0.16\%$
test_clone	6.8212ms	6.4226ms	155.7014 Ops/s	155.5348 Ops/s	$\color{#35bf28}+0.11\%$
test_squeeze	0.1634ms	14.2725μs	70.0646 KOps/s	69.6681 KOps/s	$\color{#35bf28}+0.57\%$
test_unsqueeze	0.1661ms	0.1125ms	8.8888 KOps/s	8.6305 KOps/s	$\color{#35bf28}+2.99\%$
test_split	0.2591ms	0.1866ms	5.3587 KOps/s	5.4891 KOps/s	$\color{#d91a1a}-2.38\%$
test_permute	0.2752ms	0.2076ms	4.8161 KOps/s	4.8701 KOps/s	$\color{#d91a1a}-1.11\%$
test_stack	51.6331ms	51.2283ms	19.5205 Ops/s	19.6354 Ops/s	$\color{#d91a1a}-0.59\%$
test_cat	54.2431ms	51.2560ms	19.5099 Ops/s	19.7013 Ops/s	$\color{#d91a1a}-0.97\%$
test_sequential_tensordict	0.3105ms	0.2245ms	4.4541 KOps/s	4.3775 KOps/s	$\color{#35bf28}+1.75\%$
test_sequential_graph_module	0.2621ms	0.1181ms	8.4705 KOps/s	8.1051 KOps/s	$\color{#35bf28}+4.51\%$
test_nested_tensordict	0.3576ms	0.2825ms	3.5394 KOps/s	3.3567 KOps/s	$\textbf{\color{#35bf28}+5.44\%}$
test_nested_graph_module	0.1853ms	0.1286ms	7.7758 KOps/s	7.4547 KOps/s	$\color{#35bf28}+4.31\%$

github-actions · 2026-03-11T22:29:40Z

PR Title Label Error

Unknown or invalid prefix [DTensor].

Current title: [DTensor] Add public API module for transfer plan computation

Supported Prefixes

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

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

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

[ghstack-poisoned]

github-actions · 2026-03-17T12:55:45Z

PR Title Label Error

Unknown or invalid prefix [DTensor].

Current title: [DTensor] Add public API module for transfer plan computation

Supported Prefixes

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

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

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

github-actions · 2026-03-17T12:55:55Z

PR Title Label Error

Unknown or invalid prefix [DTensor].

Current title: [DTensor] Add public API module for transfer plan computation

Supported Prefixes

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

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

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

[ghstack-poisoned]

github-actions · 2026-03-17T14:27:47Z

PR Title Label Error

Unknown or invalid prefix [DTensor].

Current title: [DTensor] Add public API module for transfer plan computation

Supported Prefixes

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

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

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

[ghstack-poisoned]

github-actions · 2026-03-17T16:01:19Z

PR Title Label Error

Unknown or invalid prefix [DTensor].

Current title: [DTensor] Add public API module for transfer plan computation

Supported Prefixes

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

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

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

github-actions · 2026-03-17T16:01:30Z

PR Title Label Error

Unknown or invalid prefix [DTensor].

Current title: [DTensor] Add public API module for transfer plan computation

Supported Prefixes

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

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

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

[ghstack-poisoned]

github-actions · 2026-03-17T18:41:31Z

PR Title Label Error

Unknown or invalid prefix [DTensor].

Current title: [DTensor] Add public API module for transfer plan computation

Supported Prefixes

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

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

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

Update

59ba7da

[ghstack-poisoned]

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 Mar 10, 2026

vmoens mentioned this pull request Mar 11, 2026

[DTensor] Fix CI: Python 3.10 compat, lint, and pyi stubs #1652

Open

Update

c4b6531

[ghstack-poisoned]

Update

c86cb9d

[ghstack-poisoned]

Update

9937a6f

[ghstack-poisoned]

Update

451c397

[ghstack-poisoned]

Navigation Menu

Uh oh!

[DTensor] Add public API module for transfer plan computation - #1649

[DTensor] Add public API module for transfer plan computation#1649
vmoens wants to merge 5 commits into
gh/vmoens/89/basefrom
gh/vmoens/89/head

vmoens commented Mar 10, 2026 •

edited

Loading

Uh oh!

github-actions Bot commented Mar 10, 2026

Uh oh!

github-actions Bot commented Mar 10, 2026

Uh oh!

github-actions Bot commented Mar 10, 2026 •

edited

Loading

Uh oh!

github-actions Bot commented Mar 10, 2026 •

edited

Loading

Uh oh!

github-actions Bot commented Mar 11, 2026

Uh oh!

github-actions Bot commented Mar 17, 2026

Uh oh!

github-actions Bot commented Mar 17, 2026

Uh oh!

github-actions Bot commented Mar 17, 2026

Uh oh!

github-actions Bot commented Mar 17, 2026

Uh oh!

github-actions Bot commented Mar 17, 2026

Uh oh!

github-actions Bot commented Mar 17, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

Uh oh!

Conversation

vmoens commented Mar 10, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

github-actions Bot commented Mar 10, 2026

PR Title Label Error

Supported Prefixes

Uh oh!

github-actions Bot commented Mar 10, 2026

PR Title Label Error

Supported Prefixes

Uh oh!

github-actions Bot commented Mar 10, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

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

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

Uh oh!

github-actions Bot commented Mar 10, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

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

Total Benchmarks: 261. Improved: $\large\color{#35bf28}11$. Worsened: $\large\color{#d91a1a}9$.

Uh oh!

github-actions Bot commented Mar 11, 2026

PR Title Label Error

Supported Prefixes

Uh oh!

github-actions Bot commented Mar 17, 2026

PR Title Label Error

Supported Prefixes

Uh oh!

github-actions Bot commented Mar 17, 2026

PR Title Label Error

Supported Prefixes

Uh oh!

github-actions Bot commented Mar 17, 2026

PR Title Label Error

Supported Prefixes

Uh oh!

github-actions Bot commented Mar 17, 2026

PR Title Label Error

Supported Prefixes

Uh oh!

github-actions Bot commented Mar 17, 2026

PR Title Label Error

Supported Prefixes

Uh oh!

github-actions Bot commented Mar 17, 2026

PR Title Label Error

Supported Prefixes

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

vmoens commented Mar 10, 2026 •

edited

Loading

github-actions Bot commented Mar 10, 2026 •

edited

Loading

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

github-actions Bot commented Mar 10, 2026 •

edited

Loading

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