Skip to content

Conversation

@dependabot
Copy link
Contributor

@dependabot dependabot bot commented on behalf of github Dec 16, 2025

Bumps jinja2 from 3.1.4 to 3.1.6.

Release notes

Sourced from jinja2's releases.

3.1.6

This is the Jinja 3.1.6 security release, which fixes security issues but does not otherwise change behavior and should not result in breaking changes compared to the latest feature release.

PyPI: https://pypi.org/project/Jinja2/3.1.6/ Changes: https://jinja.palletsprojects.com/en/stable/changes/#version-3-1-6

  • The |attr filter does not bypass the environment's attribute lookup, allowing the sandbox to apply its checks. GHSA-cpwx-vrp4-4pq7

3.1.5

This is the Jinja 3.1.5 security fix release, which fixes security issues and bugs but does not otherwise change behavior and should not result in breaking changes compared to the latest feature release.

PyPI: https://pypi.org/project/Jinja2/3.1.5/ Changes: https://jinja.palletsprojects.com/changes/#version-3-1-5 Milestone: https://github.com/pallets/jinja/milestone/16?closed=1

  • The sandboxed environment handles indirect calls to str.format, such as by passing a stored reference to a filter that calls its argument. GHSA-q2x7-8rv6-6q7h
  • Escape template name before formatting it into error messages, to avoid issues with names that contain f-string syntax. #1792, GHSA-gmj6-6f8f-6699
  • Sandbox does not allow clear and pop on known mutable sequence types. #2032
  • Calling sync render for an async template uses asyncio.run. #1952
  • Avoid unclosed auto_aiter warnings. #1960
  • Return an aclose-able AsyncGenerator from Template.generate_async. #1960
  • Avoid leaving root_render_func() unclosed in Template.generate_async. #1960
  • Avoid leaving async generators unclosed in blocks, includes and extends. #1960
  • The runtime uses the correct concat function for the current environment when calling block references. #1701
  • Make |unique async-aware, allowing it to be used after another async-aware filter. #1781
  • |int filter handles OverflowError from scientific notation. #1921
  • Make compiling deterministic for tuple unpacking in a {% set ... %} call. #2021
  • Fix dunder protocol (copy/pickle/etc) interaction with Undefined objects. #2025
  • Fix copy/pickle support for the internal missing object. #2027
  • Environment.overlay(enable_async) is applied correctly. #2061
  • The error message from FileSystemLoader includes the paths that were searched. #1661
  • PackageLoader shows a clearer error message when the package does not contain the templates directory. #1705
  • Improve annotations for methods returning copies. #1880
  • urlize does not add mailto: to values like @a@b. #1870
  • Tests decorated with @pass_context can be used with the |select filter. #1624
  • Using set for multiple assignment (a, b = 1, 2) does not fail when the target is a namespace attribute. #1413
  • Using set in all branches of {% if %}{% elif %}{% else %} blocks does not cause the variable to be considered initially undefined. #1253
Changelog

Sourced from jinja2's changelog.

Version 3.1.6

Released 2025-03-05

  • The |attr filter does not bypass the environment's attribute lookup, allowing the sandbox to apply its checks. :ghsa:cpwx-vrp4-4pq7

Version 3.1.5

Released 2024-12-21

  • The sandboxed environment handles indirect calls to str.format, such as by passing a stored reference to a filter that calls its argument. :ghsa:q2x7-8rv6-6q7h
  • Escape template name before formatting it into error messages, to avoid issues with names that contain f-string syntax. :issue:1792, :ghsa:gmj6-6f8f-6699
  • Sandbox does not allow clear and pop on known mutable sequence types. :issue:2032
  • Calling sync render for an async template uses asyncio.run. :pr:1952
  • Avoid unclosed auto_aiter warnings. :pr:1960
  • Return an aclose-able AsyncGenerator from Template.generate_async. :pr:1960
  • Avoid leaving root_render_func() unclosed in Template.generate_async. :pr:1960
  • Avoid leaving async generators unclosed in blocks, includes and extends. :pr:1960
  • The runtime uses the correct concat function for the current environment when calling block references. :issue:1701
  • Make |unique async-aware, allowing it to be used after another async-aware filter. :issue:1781
  • |int filter handles OverflowError from scientific notation. :issue:1921
  • Make compiling deterministic for tuple unpacking in a {% set ... %} call. :issue:2021
  • Fix dunder protocol (copy/pickle/etc) interaction with Undefined objects. :issue:2025
  • Fix copy/pickle support for the internal missing object. :issue:2027
  • Environment.overlay(enable_async) is applied correctly. :pr:2061
  • The error message from FileSystemLoader includes the paths that were searched. :issue:1661
  • PackageLoader shows a clearer error message when the package does not contain the templates directory. :issue:1705
  • Improve annotations for methods returning copies. :pr:1880
  • urlize does not add mailto: to values like @a@b. :pr:1870

... (truncated)

Commits

Dependabot compatibility score

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.


Dependabot commands and options

You can trigger Dependabot actions by commenting on this PR:

  • @dependabot rebase will rebase this PR
  • @dependabot recreate will recreate this PR, overwriting any edits that have been made to it
  • @dependabot merge will merge this PR after your CI passes on it
  • @dependabot squash and merge will squash and merge this PR after your CI passes on it
  • @dependabot cancel merge will cancel a previously requested merge and block automerging
  • @dependabot reopen will reopen this PR if it is closed
  • @dependabot close will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
  • @dependabot show <dependency name> ignore conditions will show all of the ignore conditions of the specified dependency
  • @dependabot ignore this major version will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
  • @dependabot ignore this minor version will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
  • @dependabot ignore this dependency will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)
    You can disable automated security fix PRs for this repo from the Security Alerts page.

Bumps [jinja2](https://github.com/pallets/jinja) from 3.1.4 to 3.1.6.
- [Release notes](https://github.com/pallets/jinja/releases)
- [Changelog](https://github.com/pallets/jinja/blob/main/CHANGES.rst)
- [Commits](pallets/jinja@3.1.4...3.1.6)

---
updated-dependencies:
- dependency-name: jinja2
  dependency-version: 3.1.6
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <[email protected]>
@dependabot dependabot bot added dependencies Pull requests that update a dependency file python Pull requests that update python code labels Dec 16, 2025
@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 Dec 16, 2025
@github-actions
Copy link

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

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

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_plain_set_nested 29.6610μs 14.7659μs 67.7238 KOps/s 66.6242 KOps/s $\color{#35bf28}+1.65\%$
test_plain_set_stack_nested 46.8520μs 15.2147μs 65.7260 KOps/s 65.9501 KOps/s $\color{#d91a1a}-0.34\%$
test_plain_set_nested_inplace 0.1048ms 16.4594μs 60.7556 KOps/s 59.5130 KOps/s $\color{#35bf28}+2.09\%$
test_plain_set_stack_nested_inplace 47.5920μs 16.6008μs 60.2379 KOps/s 59.9039 KOps/s $\color{#35bf28}+0.56\%$
test_items 26.5510μs 5.7589μs 173.6448 KOps/s 172.9665 KOps/s $\color{#35bf28}+0.39\%$
test_items_nested 0.6241ms 0.5354ms 1.8678 KOps/s 1.8240 KOps/s $\color{#35bf28}+2.40\%$
test_items_nested_locked 0.5919ms 0.5390ms 1.8554 KOps/s 1.8096 KOps/s $\color{#35bf28}+2.53\%$
test_items_nested_leaf 0.1352ms 96.1263μs 10.4030 KOps/s 10.3847 KOps/s $\color{#35bf28}+0.18\%$
test_items_stack_nested 0.5753ms 0.5264ms 1.8996 KOps/s 1.8386 KOps/s $\color{#35bf28}+3.32\%$
test_items_stack_nested_leaf 0.1278ms 96.7055μs 10.3407 KOps/s 10.4783 KOps/s $\color{#d91a1a}-1.31\%$
test_items_stack_nested_locked 0.5857ms 0.5358ms 1.8664 KOps/s 1.8307 KOps/s $\color{#35bf28}+1.95\%$
test_keys 0.4320ms 4.2061μs 237.7502 KOps/s 236.0328 KOps/s $\color{#35bf28}+0.73\%$
test_keys_nested 0.2760ms 0.1187ms 8.4244 KOps/s 8.3391 KOps/s $\color{#35bf28}+1.02\%$
test_keys_nested_locked 2.0530ms 0.1296ms 7.7139 KOps/s 7.7694 KOps/s $\color{#d91a1a}-0.71\%$
test_keys_nested_leaf 0.1527ms 0.1101ms 9.0808 KOps/s 9.0628 KOps/s $\color{#35bf28}+0.20\%$
test_keys_stack_nested 0.1537ms 0.1193ms 8.3825 KOps/s 8.2922 KOps/s $\color{#35bf28}+1.09\%$
test_keys_stack_nested_leaf 0.1482ms 0.1103ms 9.0641 KOps/s 9.0332 KOps/s $\color{#35bf28}+0.34\%$
test_keys_stack_nested_locked 0.1592ms 0.1289ms 7.7562 KOps/s 7.7548 KOps/s $\color{#35bf28}+0.02\%$
test_values 5.9762μs 1.0265μs 974.2246 KOps/s 959.0755 KOps/s $\color{#35bf28}+1.58\%$
test_values_nested 78.0140μs 48.3494μs 20.6828 KOps/s 20.7430 KOps/s $\color{#d91a1a}-0.29\%$
test_values_nested_locked 80.9630μs 51.6067μs 19.3773 KOps/s 19.3054 KOps/s $\color{#35bf28}+0.37\%$
test_values_nested_leaf 87.7840μs 54.8702μs 18.2248 KOps/s 18.1925 KOps/s $\color{#35bf28}+0.18\%$
test_values_stack_nested 76.5540μs 48.4895μs 20.6230 KOps/s 20.6952 KOps/s $\color{#d91a1a}-0.35\%$
test_values_stack_nested_leaf 81.9340μs 55.4913μs 18.0209 KOps/s 18.4881 KOps/s $\color{#d91a1a}-2.53\%$
test_values_stack_nested_locked 76.7530μs 51.3896μs 19.4592 KOps/s 19.5519 KOps/s $\color{#d91a1a}-0.47\%$
test_membership 7.7703μs 0.8832μs 1.1323 MOps/s 1.1702 MOps/s $\color{#d91a1a}-3.24\%$
test_membership_nested 28.7920μs 3.2106μs 311.4695 KOps/s 307.2806 KOps/s $\color{#35bf28}+1.36\%$
test_membership_nested_leaf 24.8910μs 3.1850μs 313.9693 KOps/s 312.4130 KOps/s $\color{#35bf28}+0.50\%$
test_membership_stacked_nested 25.9410μs 3.1520μs 317.2557 KOps/s 307.4920 KOps/s $\color{#35bf28}+3.18\%$
test_membership_stacked_nested_leaf 34.0010μs 3.1602μs 316.4345 KOps/s 310.7187 KOps/s $\color{#35bf28}+1.84\%$
test_membership_nested_last 29.5210μs 4.6344μs 215.7784 KOps/s 211.7434 KOps/s $\color{#35bf28}+1.91\%$
test_membership_nested_leaf_last 33.5920μs 4.6843μs 213.4780 KOps/s 211.4928 KOps/s $\color{#35bf28}+0.94\%$
test_membership_stacked_nested_last 36.7510μs 4.6673μs 214.2545 KOps/s 212.7064 KOps/s $\color{#35bf28}+0.73\%$
test_membership_stacked_nested_leaf_last 32.5320μs 4.6748μs 213.9113 KOps/s 215.3746 KOps/s $\color{#d91a1a}-0.68\%$
test_nested_getleaf 44.9720μs 21.6908μs 46.1026 KOps/s 47.5384 KOps/s $\color{#d91a1a}-3.02\%$
test_nested_get 46.5320μs 20.8315μs 48.0043 KOps/s 49.8112 KOps/s $\color{#d91a1a}-3.63\%$
test_stacked_getleaf 50.6720μs 21.8428μs 45.7816 KOps/s 46.7461 KOps/s $\color{#d91a1a}-2.06\%$
test_stacked_get 44.5720μs 20.8109μs 48.0517 KOps/s 49.5132 KOps/s $\color{#d91a1a}-2.95\%$
test_nested_getitemleaf 74.2230μs 22.1160μs 45.2161 KOps/s 45.4847 KOps/s $\color{#d91a1a}-0.59\%$
test_nested_getitem 45.4820μs 21.4099μs 46.7074 KOps/s 47.9365 KOps/s $\color{#d91a1a}-2.56\%$
test_stacked_getitemleaf 53.1120μs 22.4584μs 44.5267 KOps/s 45.0982 KOps/s $\color{#d91a1a}-1.27\%$
test_stacked_getitem 60.7630μs 21.2617μs 47.0330 KOps/s 47.9026 KOps/s $\color{#d91a1a}-1.82\%$
test_lock_nested 0.5823ms 0.4728ms 2.1152 KOps/s 2.1118 KOps/s $\color{#35bf28}+0.16\%$
test_lock_stack_nested 0.5561ms 0.4767ms 2.0977 KOps/s 2.0857 KOps/s $\color{#35bf28}+0.57\%$
test_unlock_nested 0.4913ms 0.3816ms 2.6205 KOps/s 2.6013 KOps/s $\color{#35bf28}+0.74\%$
test_unlock_stack_nested 0.4299ms 0.3796ms 2.6345 KOps/s 2.5925 KOps/s $\color{#35bf28}+1.62\%$
test_flatten_speed 0.1660ms 0.1229ms 8.1365 KOps/s 8.1699 KOps/s $\color{#d91a1a}-0.41\%$
test_unflatten_speed 0.6285ms 0.6000ms 1.6667 KOps/s 1.6614 KOps/s $\color{#35bf28}+0.32\%$
test_common_ops 0.9146ms 0.7451ms 1.3420 KOps/s 1.3154 KOps/s $\color{#35bf28}+2.02\%$
test_creation 92.2840μs 2.7746μs 360.4136 KOps/s 360.8213 KOps/s $\color{#d91a1a}-0.11\%$
test_creation_empty 26.4520μs 9.1027μs 109.8578 KOps/s 108.2677 KOps/s $\color{#35bf28}+1.47\%$
test_creation_nested_1 37.1610μs 12.2508μs 81.6271 KOps/s 80.9075 KOps/s $\color{#35bf28}+0.89\%$
test_creation_nested_2 38.0310μs 16.2554μs 61.5179 KOps/s 60.8335 KOps/s $\color{#35bf28}+1.12\%$
test_clone 50.5630μs 13.7118μs 72.9296 KOps/s 72.5724 KOps/s $\color{#35bf28}+0.49\%$
test_getitem[int] 1.1612ms 14.2356μs 70.2466 KOps/s 69.9844 KOps/s $\color{#35bf28}+0.37\%$
test_getitem[slice_int] 0.1469ms 25.2451μs 39.6117 KOps/s 40.1179 KOps/s $\color{#d91a1a}-1.26\%$
test_getitem[range] 0.1723ms 60.6411μs 16.4905 KOps/s 16.6430 KOps/s $\color{#d91a1a}-0.92\%$
test_getitem[tuple] 0.1415ms 24.8991μs 40.1621 KOps/s 40.4554 KOps/s $\color{#d91a1a}-0.73\%$
test_getitem[list] 0.1819ms 54.8857μs 18.2197 KOps/s 18.4026 KOps/s $\color{#d91a1a}-0.99\%$
test_setitem_dim[int] 44.7520μs 24.3090μs 41.1371 KOps/s 40.2920 KOps/s $\color{#35bf28}+2.10\%$
test_setitem_dim[slice_int] 67.7930μs 44.9766μs 22.2338 KOps/s 22.1725 KOps/s $\color{#35bf28}+0.28\%$
test_setitem_dim[range] 0.1118ms 87.2681μs 11.4589 KOps/s 11.5226 KOps/s $\color{#d91a1a}-0.55\%$
test_setitem_dim[tuple] 63.7330μs 41.6861μs 23.9888 KOps/s 23.8508 KOps/s $\color{#35bf28}+0.58\%$
test_setitem 54.5720μs 18.6163μs 53.7163 KOps/s 52.2583 KOps/s $\color{#35bf28}+2.79\%$
test_set 49.3020μs 17.7752μs 56.2582 KOps/s 55.2455 KOps/s $\color{#35bf28}+1.83\%$
test_set_shared 0.5135ms 0.2045ms 4.8902 KOps/s 4.8977 KOps/s $\color{#d91a1a}-0.15\%$
test_update 0.3412ms 23.0811μs 43.3254 KOps/s 43.2304 KOps/s $\color{#35bf28}+0.22\%$
test_update_nested 77.1540μs 35.4216μs 28.2314 KOps/s 27.3409 KOps/s $\color{#35bf28}+3.26\%$
test_update__nested 0.6082ms 34.6461μs 28.8633 KOps/s 27.7113 KOps/s $\color{#35bf28}+4.16\%$
test_set_nested 55.7120μs 19.6664μs 50.8482 KOps/s 49.6921 KOps/s $\color{#35bf28}+2.33\%$
test_set_nested_new 58.2830μs 24.7336μs 40.4308 KOps/s 38.9887 KOps/s $\color{#35bf28}+3.70\%$
test_select 72.0330μs 43.5915μs 22.9402 KOps/s 23.5976 KOps/s $\color{#d91a1a}-2.79\%$
test_select_nested 0.1033ms 75.5927μs 13.2288 KOps/s 13.2643 KOps/s $\color{#d91a1a}-0.27\%$
test_exclude_nested 0.1266ms 99.8584μs 10.0142 KOps/s 9.9610 KOps/s $\color{#35bf28}+0.53\%$
test_empty[True] 0.7676ms 0.4412ms 2.2664 KOps/s 2.2304 KOps/s $\color{#35bf28}+1.62\%$
test_empty[False] 8.2955μs 1.3338μs 749.7257 KOps/s 768.5799 KOps/s $\color{#d91a1a}-2.45\%$
test_to 0.1028ms 73.1495μs 13.6706 KOps/s 13.3465 KOps/s $\color{#35bf28}+2.43\%$
test_to_nonblocking 0.1089ms 66.2171μs 15.1018 KOps/s 15.2474 KOps/s $\color{#d91a1a}-0.95\%$
test_unbind_speed 0.3705ms 0.3282ms 3.0469 KOps/s 3.1203 KOps/s $\color{#d91a1a}-2.35\%$
test_unbind_speed_stack0 0.3642ms 0.3275ms 3.0537 KOps/s 3.0881 KOps/s $\color{#d91a1a}-1.12\%$
test_unbind_speed_stack1 98.2935ms 0.8593ms 1.1637 KOps/s 1.1548 KOps/s $\color{#35bf28}+0.77\%$
test_split 97.5778ms 1.2856ms 777.8750 Ops/s 869.8182 Ops/s $\textbf{\color{#d91a1a}-10.57\%}$
test_chunk 99.3022ms 1.2247ms 816.5307 Ops/s 771.2015 Ops/s $\textbf{\color{#35bf28}+5.88\%}$
test_consolidate[False-None] 4.0724ms 3.9110ms 255.6911 Ops/s 257.0434 Ops/s $\color{#d91a1a}-0.53\%$
test_consolidate[default-None] 2.2589ms 2.0982ms 476.6028 Ops/s 465.9289 Ops/s $\color{#35bf28}+2.29\%$
test_consolidate[reduce-overhead-None] 2.1488ms 2.0162ms 495.9750 Ops/s 481.5764 Ops/s $\color{#35bf28}+2.99\%$
test_consolidate_njt[False-None] 0.1818s 10.5247ms 95.0142 Ops/s 111.5519 Ops/s $\textbf{\color{#d91a1a}-14.83\%}$
test_to[False-False-None] 2.1959ms 2.0849ms 479.6331 Ops/s 486.5858 Ops/s $\color{#d91a1a}-1.43\%$
test_to[True-False-None] 2.1918ms 1.8918ms 528.5878 Ops/s 527.5472 Ops/s $\color{#35bf28}+0.20\%$
test_to[within-False-None] 6.2705ms 5.8769ms 170.1575 Ops/s 119.8573 Ops/s $\textbf{\color{#35bf28}+41.97\%}$
test_to[True-default-None] 12.4730ms 12.0125ms 83.2468 Ops/s 82.9054 Ops/s $\color{#35bf28}+0.41\%$
test_to_njt[False-False-None] 8.7377ms 8.6362ms 115.7913 Ops/s 115.0892 Ops/s $\color{#35bf28}+0.61\%$
test_to_njt[True-False-None] 7.5810ms 7.4158ms 134.8468 Ops/s 133.8012 Ops/s $\color{#35bf28}+0.78\%$
test_to_njt[within-False-None] 16.6167ms 16.5263ms 60.5095 Ops/s 60.3250 Ops/s $\color{#35bf28}+0.31\%$
test_creation[device0] 0.4062ms 0.1125ms 8.8925 KOps/s 9.0856 KOps/s $\color{#d91a1a}-2.12\%$
test_creation_from_tensor 0.4356ms 0.1151ms 8.6880 KOps/s 8.9515 KOps/s $\color{#d91a1a}-2.94\%$
test_add_one[memmap_tensor0] 0.2135ms 7.3124μs 136.7545 KOps/s 148.1889 KOps/s $\textbf{\color{#d91a1a}-7.72\%}$
test_contiguous[memmap_tensor0] 15.2000μs 0.7022μs 1.4240 MOps/s 1.9249 MOps/s $\textbf{\color{#d91a1a}-26.02\%}$
test_stack[memmap_tensor0] 30.0510μs 4.7309μs 211.3764 KOps/s 210.5946 KOps/s $\color{#35bf28}+0.37\%$
test_memmaptd_index 1.2438ms 0.2867ms 3.4882 KOps/s 3.5709 KOps/s $\color{#d91a1a}-2.32\%$
test_memmaptd_index_astensor 0.5361ms 0.3789ms 2.6394 KOps/s 2.6854 KOps/s $\color{#d91a1a}-1.71\%$
test_memmaptd_index_op 0.8920ms 0.6510ms 1.5362 KOps/s 1.6037 KOps/s $\color{#d91a1a}-4.21\%$
test_serialize_model 0.1320s 0.1312s 7.6205 Ops/s 7.6066 Ops/s $\color{#35bf28}+0.18\%$
test_serialize_model_pickle 1.3492s 1.1943s 0.8373 Ops/s 0.8206 Ops/s $\color{#35bf28}+2.04\%$
test_serialize_weights 0.1309s 0.1303s 7.6762 Ops/s 6.2338 Ops/s $\textbf{\color{#35bf28}+23.14\%}$
test_serialize_weights_returnearly 0.4174s 69.8003ms 14.3266 Ops/s 19.2953 Ops/s $\textbf{\color{#d91a1a}-25.75\%}$
test_serialize_weights_pickle 1.3644s 1.2207s 0.8192 Ops/s 0.8424 Ops/s $\color{#d91a1a}-2.75\%$
test_reshape_pytree 0.3810ms 33.8798μs 29.5161 KOps/s 29.7573 KOps/s $\color{#d91a1a}-0.81\%$
test_reshape_td 74.7030μs 38.7878μs 25.7813 KOps/s 25.6758 KOps/s $\color{#35bf28}+0.41\%$
test_view_pytree 0.2249ms 33.4811μs 29.8676 KOps/s 30.0543 KOps/s $\color{#d91a1a}-0.62\%$
test_view_td 87.2640μs 46.0578μs 21.7119 KOps/s 21.6985 KOps/s $\color{#35bf28}+0.06\%$
test_unbind_pytree 0.2469ms 38.4460μs 26.0105 KOps/s 18.3384 KOps/s $\textbf{\color{#35bf28}+41.84\%}$
test_unbind_td 0.1730ms 49.4304μs 20.2305 KOps/s 20.3407 KOps/s $\color{#d91a1a}-0.54\%$
test_split_pytree 0.2548ms 44.7906μs 22.3261 KOps/s 22.4532 KOps/s $\color{#d91a1a}-0.57\%$
test_split_td 0.2070ms 65.4306μs 15.2834 KOps/s 15.1825 KOps/s $\color{#35bf28}+0.66\%$
test_add_pytree 0.2371ms 45.6268μs 21.9169 KOps/s 22.5067 KOps/s $\color{#d91a1a}-2.62\%$
test_add_td 99.4750μs 54.7214μs 18.2744 KOps/s 18.2689 KOps/s $\color{#35bf28}+0.03\%$
test_compile_add_one_nested[tensordict-compile] 0.2933ms 0.1834ms 5.4527 KOps/s 5.4551 KOps/s $\color{#d91a1a}-0.04\%$
test_compile_add_one_nested[tensordict-eager] 0.2542ms 0.1966ms 5.0868 KOps/s 5.1043 KOps/s $\color{#d91a1a}-0.34\%$
test_compile_add_one_nested[pytree-compile] 0.2036ms 0.1524ms 6.5619 KOps/s 6.4185 KOps/s $\color{#35bf28}+2.23\%$
test_compile_add_one_nested[pytree-eager] 0.4399ms 0.1881ms 5.3174 KOps/s 5.2905 KOps/s $\color{#35bf28}+0.51\%$
test_compile_copy_nested[tensordict-compile] 0.1191ms 28.0207μs 35.6879 KOps/s 35.5673 KOps/s $\color{#35bf28}+0.34\%$
test_compile_copy_nested[tensordict-eager] 88.7140μs 53.5491μs 18.6745 KOps/s 18.7371 KOps/s $\color{#d91a1a}-0.33\%$
test_compile_copy_nested[pytree-compile] 98.4950μs 14.4984μs 68.9731 KOps/s 66.4347 KOps/s $\color{#35bf28}+3.82\%$
test_compile_copy_nested[pytree-eager] 0.3927ms 77.1524μs 12.9614 KOps/s 13.0190 KOps/s $\color{#d91a1a}-0.44\%$
test_compile_add_one_flat[tensordict-compile] 0.3090ms 0.2092ms 4.7805 KOps/s 4.7327 KOps/s $\color{#35bf28}+1.01\%$
test_compile_add_one_flat[tensordict-eager] 0.3316ms 0.2664ms 3.7541 KOps/s 3.7935 KOps/s $\color{#d91a1a}-1.04\%$
test_compile_add_one_flat[tensorclass-compile] 0.2282ms 0.1559ms 6.4142 KOps/s 6.3620 KOps/s $\color{#35bf28}+0.82\%$
test_compile_add_one_flat[tensorclass-eager] 0.1130ms 71.4636μs 13.9931 KOps/s 13.8590 KOps/s $\color{#35bf28}+0.97\%$
test_compile_add_one_flat[pytree-compile] 0.2491ms 0.2062ms 4.8499 KOps/s 4.8189 KOps/s $\color{#35bf28}+0.64\%$
test_compile_add_one_flat[pytree-eager] 0.8108ms 0.5412ms 1.8477 KOps/s 1.8140 KOps/s $\color{#35bf28}+1.86\%$
test_compile_add_self_flat[tensordict-eager] 0.3706ms 0.3171ms 3.1540 KOps/s 3.1541 KOps/s $-0.01\%$
test_compile_add_self_flat[tensordict-compile] 0.3758ms 0.2128ms 4.7001 KOps/s 4.5016 KOps/s $\color{#35bf28}+4.41\%$
test_compile_add_self_flat[tensorclass-eager] 0.1314ms 88.0038μs 11.3631 KOps/s 11.3898 KOps/s $\color{#d91a1a}-0.23\%$
test_compile_add_self_flat[tensorclass-compile] 0.2081ms 0.1593ms 6.2786 KOps/s 6.1303 KOps/s $\color{#35bf28}+2.42\%$
test_compile_add_self_flat[pytree-eager] 0.7014ms 0.4431ms 2.2568 KOps/s 2.1856 KOps/s $\color{#35bf28}+3.26\%$
test_compile_add_self_flat[pytree-compile] 0.2478ms 0.2066ms 4.8404 KOps/s 4.6924 KOps/s $\color{#35bf28}+3.15\%$
test_compile_copy_flat[tensordict-compile] 69.5830μs 24.2949μs 41.1608 KOps/s 39.7229 KOps/s $\color{#35bf28}+3.62\%$
test_compile_copy_flat[tensordict-eager] 72.1930μs 41.7308μs 23.9631 KOps/s 24.0429 KOps/s $\color{#d91a1a}-0.33\%$
test_compile_copy_flat[pytree-compile] 1.3474ms 21.4391μs 46.6438 KOps/s 47.7253 KOps/s $\color{#d91a1a}-2.27\%$
test_compile_copy_flat[pytree-eager] 0.3610ms 70.0509μs 14.2753 KOps/s 14.1330 KOps/s $\color{#35bf28}+1.01\%$
test_compile_assign_and_add[tensordict-compile] 2.0912ms 0.2141ms 4.6704 KOps/s 4.6015 KOps/s $\color{#35bf28}+1.50\%$
test_compile_assign_and_add[tensordict-eager] 3.6275ms 3.3765ms 296.1643 Ops/s 300.5848 Ops/s $\color{#d91a1a}-1.47\%$
test_compile_assign_and_add[pytree-compile] 2.1141ms 0.2124ms 4.7079 KOps/s 4.7457 KOps/s $\color{#d91a1a}-0.80\%$
test_compile_assign_and_add[pytree-eager] 3.1763ms 2.9200ms 342.4716 Ops/s 341.2968 Ops/s $\color{#35bf28}+0.34\%$
test_compile_indexing[tensor-tensordict-compile] 0.3514ms 0.1489ms 6.7165 KOps/s 6.4902 KOps/s $\color{#35bf28}+3.49\%$
test_compile_indexing[tensor-tensordict-eager] 0.3064ms 69.0635μs 14.4794 KOps/s 14.6417 KOps/s $\color{#d91a1a}-1.11\%$
test_compile_indexing[tensor-tensorclass-compile] 0.2298ms 0.1399ms 7.1486 KOps/s 7.1700 KOps/s $\color{#d91a1a}-0.30\%$
test_compile_indexing[tensor-tensorclass-eager] 0.2669ms 51.0573μs 19.5858 KOps/s 21.3814 KOps/s $\textbf{\color{#d91a1a}-8.40\%}$
test_compile_indexing[tensor-pytree-compile] 0.2139ms 0.1418ms 7.0545 KOps/s 6.9608 KOps/s $\color{#35bf28}+1.35\%$
test_compile_indexing[tensor-pytree-eager] 0.2839ms 50.9175μs 19.6396 KOps/s 20.0511 KOps/s $\color{#d91a1a}-2.05\%$
test_compile_indexing[slice-tensordict-compile] 0.1440ms 86.8069μs 11.5198 KOps/s 11.4122 KOps/s $\color{#35bf28}+0.94\%$
test_compile_indexing[slice-tensordict-eager] 0.2117ms 27.6239μs 36.2005 KOps/s 36.7717 KOps/s $\color{#d91a1a}-1.55\%$
test_compile_indexing[slice-tensorclass-compile] 0.1964ms 82.2652μs 12.1558 KOps/s 12.1801 KOps/s $\color{#d91a1a}-0.20\%$
test_compile_indexing[slice-tensorclass-eager] 0.2267ms 24.3659μs 41.0409 KOps/s 41.5714 KOps/s $\color{#d91a1a}-1.28\%$
test_compile_indexing[slice-pytree-compile] 0.1302ms 82.8373μs 12.0719 KOps/s 12.1040 KOps/s $\color{#d91a1a}-0.27\%$
test_compile_indexing[slice-pytree-eager] 0.2849ms 24.1480μs 41.4113 KOps/s 41.4240 KOps/s $\color{#d91a1a}-0.03\%$
test_compile_indexing[int-tensordict-compile] 0.1496ms 88.0718μs 11.3544 KOps/s 10.9948 KOps/s $\color{#35bf28}+3.27\%$
test_compile_indexing[int-tensordict-eager] 0.2377ms 27.3946μs 36.5036 KOps/s 37.3934 KOps/s $\color{#d91a1a}-2.38\%$
test_compile_indexing[int-tensorclass-compile] 0.3043ms 82.1617μs 12.1711 KOps/s 12.0787 KOps/s $\color{#35bf28}+0.77\%$
test_compile_indexing[int-tensorclass-eager] 0.2708ms 24.2560μs 41.2268 KOps/s 41.4850 KOps/s $\color{#d91a1a}-0.62\%$
test_compile_indexing[int-pytree-compile] 0.1271ms 84.1741μs 11.8801 KOps/s 12.1637 KOps/s $\color{#d91a1a}-2.33\%$
test_compile_indexing[int-pytree-eager] 0.2495ms 24.0109μs 41.6478 KOps/s 41.3434 KOps/s $\color{#35bf28}+0.74\%$
test_mod_add[eager] 0.1054ms 53.7729μs 18.5967 KOps/s 18.4373 KOps/s $\color{#35bf28}+0.86\%$
test_mod_add[compile] 0.2563ms 0.1536ms 6.5094 KOps/s 6.3430 KOps/s $\color{#35bf28}+2.62\%$
test_mod_add[compile-overhead] 0.2865ms 0.2020ms 4.9513 KOps/s 4.9576 KOps/s $\color{#d91a1a}-0.13\%$
test_mod_wrap[eager] 0.4490ms 0.3188ms 3.1363 KOps/s 3.0951 KOps/s $\color{#35bf28}+1.33\%$
test_mod_wrap[compile] 0.4976ms 0.4251ms 2.3526 KOps/s 2.4651 KOps/s $\color{#d91a1a}-4.56\%$
test_mod_wrap[compile-overhead] 6.9523ms 3.9047ms 256.1023 Ops/s 251.6326 Ops/s $\color{#35bf28}+1.78\%$
test_mod_wrap_and_backward[eager] 1.7742ms 1.5749ms 634.9450 Ops/s 635.4155 Ops/s $\color{#d91a1a}-0.07\%$
test_mod_wrap_and_backward[compile] 1.7745ms 1.6516ms 605.4877 Ops/s 610.5630 Ops/s $\color{#d91a1a}-0.83\%$
test_mod_wrap_and_backward[compile-overhead] 1.3525ms 0.9976ms 1.0024 KOps/s 1.0011 KOps/s $\color{#35bf28}+0.13\%$
test_seq_add[eager] 0.2466ms 0.1675ms 5.9711 KOps/s 6.1960 KOps/s $\color{#d91a1a}-3.63\%$
test_seq_add[compile] 0.2974ms 0.1628ms 6.1422 KOps/s 6.0884 KOps/s $\color{#35bf28}+0.88\%$
test_seq_add[compile-overhead] 0.2747ms 0.2042ms 4.8976 KOps/s 4.8801 KOps/s $\color{#35bf28}+0.36\%$
test_seq_wrap[eager] 0.7475ms 0.5563ms 1.7975 KOps/s 1.8037 KOps/s $\color{#d91a1a}-0.35\%$
test_seq_wrap[compile] 0.5154ms 0.4237ms 2.3600 KOps/s 2.3629 KOps/s $\color{#d91a1a}-0.12\%$
test_seq_wrap[compile-overhead] 0.4033ms 0.3188ms 3.1364 KOps/s 3.1530 KOps/s $\color{#d91a1a}-0.53\%$
test_func_call_runtime[False-eager] 0.9975ms 0.9317ms 1.0733 KOps/s 1.0597 KOps/s $\color{#35bf28}+1.28\%$
test_func_call_runtime[False-compile] 1.0770ms 0.9703ms 1.0306 KOps/s 1.0306 KOps/s $+0.01\%$
test_func_call_runtime[False-compile-overhead] 0.5733ms 0.5011ms 1.9956 KOps/s 1.9696 KOps/s $\color{#35bf28}+1.32\%$
test_func_call_runtime[True-eager] 1.2521ms 1.1406ms 876.7662 Ops/s 876.0629 Ops/s $\color{#35bf28}+0.08\%$
test_func_call_runtime[True-compile] 1.1146ms 0.9896ms 1.0105 KOps/s 1.0124 KOps/s $\color{#d91a1a}-0.18\%$
test_func_call_runtime[True-compile-overhead] 0.5732ms 0.5241ms 1.9079 KOps/s 1.9013 KOps/s $\color{#35bf28}+0.34\%$
test_func_call_cm_runtime[False-eager] 0.9870ms 0.8944ms 1.1181 KOps/s 1.0682 KOps/s $\color{#35bf28}+4.68\%$
test_func_call_cm_runtime[False-compile] 1.0304ms 0.9778ms 1.0228 KOps/s 1.0372 KOps/s $\color{#d91a1a}-1.39\%$
test_func_call_cm_runtime[False-compile-overhead] 0.5952ms 0.5050ms 1.9802 KOps/s 1.9810 KOps/s $\color{#d91a1a}-0.04\%$
test_func_call_cm_runtime[True-eager] 1.3938ms 1.2758ms 783.8215 Ops/s 774.3991 Ops/s $\color{#35bf28}+1.22\%$
test_func_call_cm_runtime[True-compile] 1.1046ms 1.0305ms 970.4285 Ops/s 977.2128 Ops/s $\color{#d91a1a}-0.69\%$
test_func_call_cm_runtime[True-compile-overhead] 0.7562ms 0.5616ms 1.7808 KOps/s 1.7828 KOps/s $\color{#d91a1a}-0.11\%$
test_vmap_func_call_cm_runtime[eager] 2.8998ms 2.3936ms 417.7826 Ops/s 416.8345 Ops/s $\color{#35bf28}+0.23\%$
test_vmap_func_call_cm_runtime[compile] 1.1272ms 1.0390ms 962.4666 Ops/s 975.2356 Ops/s $\color{#d91a1a}-1.31\%$
test_vmap_func_call_cm_runtime[compile-overhead] 0.6213ms 0.5517ms 1.8126 KOps/s 1.7967 KOps/s $\color{#35bf28}+0.89\%$
test_distributed 2.8331ms 0.1651ms 6.0553 KOps/s 6.4417 KOps/s $\textbf{\color{#d91a1a}-6.00\%}$
test_tdmodule 0.1144ms 29.3900μs 34.0252 KOps/s 34.6605 KOps/s $\color{#d91a1a}-1.83\%$
test_tdmodule_dispatch 88.5840μs 49.2631μs 20.2992 KOps/s 20.2268 KOps/s $\color{#35bf28}+0.36\%$
test_tdseq 45.2620μs 26.8551μs 37.2369 KOps/s 36.7083 KOps/s $\color{#35bf28}+1.44\%$
test_tdseq_dispatch 72.4030μs 50.4544μs 19.8199 KOps/s 19.5061 KOps/s $\color{#35bf28}+1.61\%$
test_instantiation_functorch 2.2263ms 2.1052ms 475.0170 Ops/s 476.5511 Ops/s $\color{#d91a1a}-0.32\%$
test_exec_functorch 0.2553ms 0.1866ms 5.3595 KOps/s 5.3913 KOps/s $\color{#d91a1a}-0.59\%$
test_exec_functional_call 0.2138ms 0.1657ms 6.0341 KOps/s 6.1152 KOps/s $\color{#d91a1a}-1.33\%$
test_exec_td_decorator 0.4545ms 0.2422ms 4.1280 KOps/s 4.1716 KOps/s $\color{#d91a1a}-1.05\%$
test_vmap_mlp_speed_decorator[True-True] 1.0009ms 0.8186ms 1.2217 KOps/s 1.2278 KOps/s $\color{#d91a1a}-0.50\%$
test_vmap_mlp_speed_decorator[True-False] 1.0421ms 0.8214ms 1.2175 KOps/s 1.2321 KOps/s $\color{#d91a1a}-1.19\%$
test_vmap_mlp_speed_decorator[False-True] 0.8707ms 0.7024ms 1.4238 KOps/s 1.4298 KOps/s $\color{#d91a1a}-0.42\%$
test_vmap_mlp_speed_decorator[False-False] 0.9139ms 0.7070ms 1.4145 KOps/s 1.4339 KOps/s $\color{#d91a1a}-1.35\%$
test_vmap_transformer_speed_decorator[True-True] 21.1508ms 20.8344ms 47.9975 Ops/s 48.0142 Ops/s $\color{#d91a1a}-0.03\%$
test_vmap_transformer_speed_decorator[True-False] 21.3941ms 20.7999ms 48.0771 Ops/s 48.0544 Ops/s $\color{#35bf28}+0.05\%$
test_vmap_transformer_speed_decorator[False-True] 20.8771ms 20.6266ms 48.4812 Ops/s 48.6726 Ops/s $\color{#d91a1a}-0.39\%$
test_vmap_transformer_speed_decorator[False-False] 20.8037ms 20.6036ms 48.5353 Ops/s 48.4639 Ops/s $\color{#35bf28}+0.15\%$
test_to_module_speed[True] 1.5628ms 1.4723ms 679.1913 Ops/s 669.7002 Ops/s $\color{#35bf28}+1.42\%$
test_to_module_speed[False] 1.5524ms 1.4612ms 684.3483 Ops/s 675.5299 Ops/s $\color{#35bf28}+1.31\%$
test_tc_init 78.0340μs 51.5127μs 19.4127 KOps/s 19.0783 KOps/s $\color{#35bf28}+1.75\%$
test_tc_init_tensor_only 93.1650μs 15.0860μs 66.2867 KOps/s 66.6646 KOps/s $\color{#d91a1a}-0.57\%$
test_tc_init_nested 0.1340ms 0.1030ms 9.7098 KOps/s 9.6270 KOps/s $\color{#35bf28}+0.86\%$
test_tc_first_layer_tensor 15.1310μs 1.7909μs 558.3652 KOps/s 558.5902 KOps/s $\color{#d91a1a}-0.04\%$
test_tc_first_layer_tensor_only 5.0446μs 0.6874μs 1.4548 MOps/s 1.4441 MOps/s $\color{#35bf28}+0.75\%$
test_tc_first_layer_tensor_set 26.8210μs 4.2188μs 237.0319 KOps/s 236.1562 KOps/s $\color{#35bf28}+0.37\%$
test_tc_first_layer_tensor_only_set 34.2510μs 3.1288μs 319.6132 KOps/s 319.5427 KOps/s $\color{#35bf28}+0.02\%$
test_tc_first_layer_nontensor 41.4410μs 5.9793μs 167.2449 KOps/s 166.4608 KOps/s $\color{#35bf28}+0.47\%$
test_tc_second_layer_tensor 37.0210μs 4.3193μs 231.5174 KOps/s 231.3202 KOps/s $\color{#35bf28}+0.09\%$
test_tc_second_layer_nontensor 36.8920μs 8.4598μs 118.2066 KOps/s 117.3161 KOps/s $\color{#35bf28}+0.76\%$
test_unbind 0.2569s 18.2050ms 54.9300 Ops/s 62.7890 Ops/s $\textbf{\color{#d91a1a}-12.52\%}$
test_full_like 17.5992ms 16.6435ms 60.0834 Ops/s 226.6291 Ops/s $\textbf{\color{#d91a1a}-73.49\%}$
test_zeros_like 16.8569ms 16.6107ms 60.2023 Ops/s 227.2287 Ops/s $\textbf{\color{#d91a1a}-73.51\%}$
test_ones_like 16.9865ms 16.6721ms 59.9805 Ops/s 226.9043 Ops/s $\textbf{\color{#d91a1a}-73.57\%}$
test_clone 17.9799ms 17.7370ms 56.3793 Ops/s 151.0308 Ops/s $\textbf{\color{#d91a1a}-62.67\%}$
test_squeeze 0.1037ms 14.3321μs 69.7733 KOps/s 68.6838 KOps/s $\color{#35bf28}+1.59\%$
test_unsqueeze 0.2090ms 0.1114ms 8.9774 KOps/s 9.1152 KOps/s $\color{#d91a1a}-1.51\%$
test_split 0.2496ms 0.1839ms 5.4384 KOps/s 5.3876 KOps/s $\color{#35bf28}+0.94\%$
test_permute 0.2879ms 0.2058ms 4.8602 KOps/s 4.8472 KOps/s $\color{#35bf28}+0.27\%$
test_stack 52.3991ms 51.7261ms 19.3326 Ops/s 19.2950 Ops/s $\color{#35bf28}+0.19\%$
test_cat 52.4048ms 51.6992ms 19.3426 Ops/s 19.3503 Ops/s $\color{#d91a1a}-0.04\%$

@github-actions
Copy link

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

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

Expand to view detailed results
Name Max Mean Ops Ops on Repo HEAD Change
test_plain_set_nested 83.9020μs 14.9165μs 67.0397 KOps/s 67.5122 KOps/s $\color{#d91a1a}-0.70\%$
test_plain_set_stack_nested 1.7148ms 15.5530μs 64.2962 KOps/s 66.0612 KOps/s $\color{#d91a1a}-2.67\%$
test_plain_set_nested_inplace 55.8810μs 16.1979μs 61.7365 KOps/s 60.1904 KOps/s $\color{#35bf28}+2.57\%$
test_plain_set_stack_nested_inplace 38.5310μs 16.7157μs 59.8241 KOps/s 60.1242 KOps/s $\color{#d91a1a}-0.50\%$
test_items 24.0010μs 5.7250μs 174.6732 KOps/s 173.3531 KOps/s $\color{#35bf28}+0.76\%$
test_items_nested 0.5979ms 0.5395ms 1.8536 KOps/s 1.8793 KOps/s $\color{#d91a1a}-1.37\%$
test_items_nested_locked 0.5787ms 0.5406ms 1.8498 KOps/s 1.8781 KOps/s $\color{#d91a1a}-1.51\%$
test_items_nested_leaf 0.1370ms 96.3467μs 10.3792 KOps/s 10.3540 KOps/s $\color{#35bf28}+0.24\%$
test_items_stack_nested 0.5991ms 0.5360ms 1.8656 KOps/s 1.8661 KOps/s $\color{#d91a1a}-0.03\%$
test_items_stack_nested_leaf 0.1267ms 97.0916μs 10.2995 KOps/s 10.4156 KOps/s $\color{#d91a1a}-1.11\%$
test_items_stack_nested_locked 0.5655ms 0.5380ms 1.8588 KOps/s 1.8473 KOps/s $\color{#35bf28}+0.62\%$
test_keys 24.1300μs 4.2617μs 234.6459 KOps/s 238.5459 KOps/s $\color{#d91a1a}-1.63\%$
test_keys_nested 0.1795ms 0.1206ms 8.2885 KOps/s 8.4290 KOps/s $\color{#d91a1a}-1.67\%$
test_keys_nested_locked 2.1169ms 0.1302ms 7.6781 KOps/s 7.7759 KOps/s $\color{#d91a1a}-1.26\%$
test_keys_nested_leaf 0.1660ms 0.1108ms 9.0242 KOps/s 9.0750 KOps/s $\color{#d91a1a}-0.56\%$
test_keys_stack_nested 0.1836ms 0.1204ms 8.3033 KOps/s 8.3655 KOps/s $\color{#d91a1a}-0.74\%$
test_keys_stack_nested_leaf 0.1591ms 0.1112ms 8.9912 KOps/s 9.0467 KOps/s $\color{#d91a1a}-0.61\%$
test_keys_stack_nested_locked 0.1823ms 0.1289ms 7.7591 KOps/s 7.7043 KOps/s $\color{#35bf28}+0.71\%$
test_values 4.4220μs 1.0289μs 971.9524 KOps/s 967.9589 KOps/s $\color{#35bf28}+0.41\%$
test_values_nested 81.7410μs 48.4770μs 20.6284 KOps/s 20.9458 KOps/s $\color{#d91a1a}-1.52\%$
test_values_nested_locked 82.4210μs 51.2561μs 19.5099 KOps/s 19.5814 KOps/s $\color{#d91a1a}-0.36\%$
test_values_nested_leaf 87.2920μs 55.0660μs 18.1600 KOps/s 18.3988 KOps/s $\color{#d91a1a}-1.30\%$
test_values_stack_nested 86.6020μs 48.2458μs 20.7272 KOps/s 21.0925 KOps/s $\color{#d91a1a}-1.73\%$
test_values_stack_nested_leaf 82.7610μs 54.8115μs 18.2443 KOps/s 18.3880 KOps/s $\color{#d91a1a}-0.78\%$
test_values_stack_nested_locked 90.4520μs 51.2420μs 19.5152 KOps/s 19.6581 KOps/s $\color{#d91a1a}-0.73\%$
test_membership 4.9467μs 0.8626μs 1.1592 MOps/s 1.1538 MOps/s $\color{#35bf28}+0.47\%$
test_membership_nested 25.8410μs 3.2105μs 311.4755 KOps/s 314.6837 KOps/s $\color{#d91a1a}-1.02\%$
test_membership_nested_leaf 31.7710μs 3.2356μs 309.0633 KOps/s 309.2634 KOps/s $\color{#d91a1a}-0.06\%$
test_membership_stacked_nested 55.5110μs 3.2026μs 312.2480 KOps/s 314.3277 KOps/s $\color{#d91a1a}-0.66\%$
test_membership_stacked_nested_leaf 24.7300μs 3.2223μs 310.3361 KOps/s 314.3639 KOps/s $\color{#d91a1a}-1.28\%$
test_membership_nested_last 37.8600μs 4.6504μs 215.0362 KOps/s 217.2331 KOps/s $\color{#d91a1a}-1.01\%$
test_membership_nested_leaf_last 35.6100μs 4.6658μs 214.3252 KOps/s 216.7118 KOps/s $\color{#d91a1a}-1.10\%$
test_membership_stacked_nested_last 25.7210μs 4.6653μs 214.3484 KOps/s 218.0341 KOps/s $\color{#d91a1a}-1.69\%$
test_membership_stacked_nested_leaf_last 34.8400μs 4.6480μs 215.1466 KOps/s 217.4756 KOps/s $\color{#d91a1a}-1.07\%$
test_nested_getleaf 49.2000μs 21.3627μs 46.8107 KOps/s 46.5854 KOps/s $\color{#35bf28}+0.48\%$
test_nested_get 44.8900μs 19.9566μs 50.1088 KOps/s 48.4319 KOps/s $\color{#35bf28}+3.46\%$
test_stacked_getleaf 86.2920μs 21.2858μs 46.9796 KOps/s 46.1288 KOps/s $\color{#35bf28}+1.84\%$
test_stacked_get 49.8010μs 20.1990μs 49.5075 KOps/s 48.2595 KOps/s $\color{#35bf28}+2.59\%$
test_nested_getitemleaf 48.9010μs 21.8744μs 45.7156 KOps/s 45.0428 KOps/s $\color{#35bf28}+1.49\%$
test_nested_getitem 57.7810μs 20.6063μs 48.5288 KOps/s 47.6524 KOps/s $\color{#35bf28}+1.84\%$
test_stacked_getitemleaf 47.0410μs 21.8047μs 45.8617 KOps/s 44.5770 KOps/s $\color{#35bf28}+2.88\%$
test_stacked_getitem 48.8100μs 20.5999μs 48.5438 KOps/s 47.2148 KOps/s $\color{#35bf28}+2.81\%$
test_lock_nested 0.5814ms 0.4712ms 2.1223 KOps/s 2.1071 KOps/s $\color{#35bf28}+0.72\%$
test_lock_stack_nested 0.5112ms 0.4768ms 2.0972 KOps/s 2.0707 KOps/s $\color{#35bf28}+1.28\%$
test_unlock_nested 0.4834ms 0.3844ms 2.6015 KOps/s 2.6001 KOps/s $\color{#35bf28}+0.05\%$
test_unlock_stack_nested 0.4925ms 0.3803ms 2.6297 KOps/s 2.5845 KOps/s $\color{#35bf28}+1.75\%$
test_flatten_speed 0.1825ms 0.1221ms 8.1867 KOps/s 8.1774 KOps/s $\color{#35bf28}+0.11\%$
test_unflatten_speed 0.6941ms 0.5905ms 1.6935 KOps/s 1.6770 KOps/s $\color{#35bf28}+0.99\%$
test_common_ops 0.8786ms 0.7495ms 1.3342 KOps/s 1.3342 KOps/s $-0.00\%$
test_creation 0.1033ms 2.7593μs 362.4107 KOps/s 362.7118 KOps/s $\color{#d91a1a}-0.08\%$
test_creation_empty 38.7410μs 9.2481μs 108.1303 KOps/s 109.5620 KOps/s $\color{#d91a1a}-1.31\%$
test_creation_nested_1 34.7810μs 12.4128μs 80.5618 KOps/s 81.9615 KOps/s $\color{#d91a1a}-1.71\%$
test_creation_nested_2 52.0410μs 16.2617μs 61.4940 KOps/s 61.7353 KOps/s $\color{#d91a1a}-0.39\%$
test_clone 43.1400μs 13.4148μs 74.5445 KOps/s 72.0019 KOps/s $\color{#35bf28}+3.53\%$
test_getitem[int] 1.1343ms 14.2346μs 70.2513 KOps/s 70.0661 KOps/s $\color{#35bf28}+0.26\%$
test_getitem[slice_int] 0.1432ms 24.8251μs 40.2818 KOps/s 40.0539 KOps/s $\color{#35bf28}+0.57\%$
test_getitem[range] 0.1737ms 59.4780μs 16.8129 KOps/s 15.9227 KOps/s $\textbf{\color{#35bf28}+5.59\%}$
test_getitem[tuple] 0.1471ms 24.6114μs 40.6316 KOps/s 40.3217 KOps/s $\color{#35bf28}+0.77\%$
test_getitem[list] 0.1770ms 53.4385μs 18.7131 KOps/s 18.4911 KOps/s $\color{#35bf28}+1.20\%$
test_setitem_dim[int] 46.6600μs 24.7775μs 40.3592 KOps/s 38.2985 KOps/s $\textbf{\color{#35bf28}+5.38\%}$
test_setitem_dim[slice_int] 74.8610μs 45.1613μs 22.1429 KOps/s 21.2054 KOps/s $\color{#35bf28}+4.42\%$
test_setitem_dim[range] 0.1219ms 86.8064μs 11.5199 KOps/s 10.9838 KOps/s $\color{#35bf28}+4.88\%$
test_setitem_dim[tuple] 68.5510μs 42.1564μs 23.7212 KOps/s 23.0038 KOps/s $\color{#35bf28}+3.12\%$
test_setitem 67.1910μs 18.6599μs 53.5908 KOps/s 53.8626 KOps/s $\color{#d91a1a}-0.50\%$
test_set 59.5910μs 17.6509μs 56.6542 KOps/s 56.1790 KOps/s $\color{#35bf28}+0.85\%$
test_set_shared 0.5453ms 0.2052ms 4.8731 KOps/s 4.7204 KOps/s $\color{#35bf28}+3.24\%$
test_update 0.2003ms 22.9368μs 43.5980 KOps/s 42.9452 KOps/s $\color{#35bf28}+1.52\%$
test_update_nested 75.0720μs 35.2550μs 28.3648 KOps/s 27.6122 KOps/s $\color{#35bf28}+2.73\%$
test_update__nested 0.4888ms 33.9542μs 29.4514 KOps/s 27.9476 KOps/s $\textbf{\color{#35bf28}+5.38\%}$
test_set_nested 57.2110μs 19.7910μs 50.5280 KOps/s 50.7152 KOps/s $\color{#d91a1a}-0.37\%$
test_set_nested_new 71.7110μs 25.0376μs 39.9400 KOps/s 39.8229 KOps/s $\color{#35bf28}+0.29\%$
test_select 86.0120μs 42.2921μs 23.6451 KOps/s 22.5875 KOps/s $\color{#35bf28}+4.68\%$
test_select_nested 0.1140ms 74.5050μs 13.4219 KOps/s 13.6846 KOps/s $\color{#d91a1a}-1.92\%$
test_exclude_nested 0.1807ms 98.1872μs 10.1846 KOps/s 10.1429 KOps/s $\color{#35bf28}+0.41\%$
test_empty[True] 0.4861ms 0.4387ms 2.2797 KOps/s 2.2997 KOps/s $\color{#d91a1a}-0.87\%$
test_empty[False] 9.0825μs 1.3223μs 756.2776 KOps/s 763.9516 KOps/s $\color{#d91a1a}-1.00\%$
test_to 0.1028ms 74.9001μs 13.3511 KOps/s 13.7657 KOps/s $\color{#d91a1a}-3.01\%$
test_to_nonblocking 0.1183ms 65.4513μs 15.2785 KOps/s 15.2901 KOps/s $\color{#d91a1a}-0.08\%$
test_unbind_speed 0.3975ms 0.3262ms 3.0655 KOps/s 3.0518 KOps/s $\color{#35bf28}+0.45\%$
test_unbind_speed_stack0 0.3720ms 0.3248ms 3.0787 KOps/s 3.0512 KOps/s $\color{#35bf28}+0.90\%$
test_unbind_speed_stack1 98.0129ms 0.8680ms 1.1521 KOps/s 1.1525 KOps/s $\color{#d91a1a}-0.03\%$
test_split 98.0894ms 1.2848ms 778.3253 Ops/s 649.6386 Ops/s $\textbf{\color{#35bf28}+19.81\%}$
test_chunk 97.7643ms 1.2158ms 822.5043 Ops/s 918.5298 Ops/s $\textbf{\color{#d91a1a}-10.45\%}$
test_consolidate[False-None] 3.9765ms 3.8960ms 256.6718 Ops/s 253.3480 Ops/s $\color{#35bf28}+1.31\%$
test_consolidate[default-None] 2.2239ms 2.1126ms 473.3407 Ops/s 449.6066 Ops/s $\textbf{\color{#35bf28}+5.28\%}$
test_consolidate[reduce-overhead-None] 2.1283ms 2.0415ms 489.8459 Ops/s 466.8480 Ops/s $\color{#35bf28}+4.93\%$
test_consolidate_njt[False-None] 9.4868ms 9.1752ms 108.9890 Ops/s 108.8029 Ops/s $\color{#35bf28}+0.17\%$
test_to[False-False-None] 2.1484ms 2.0696ms 483.1760 Ops/s 480.4283 Ops/s $\color{#35bf28}+0.57\%$
test_to[True-False-None] 0.1734s 2.2260ms 449.2407 Ops/s 528.7194 Ops/s $\textbf{\color{#d91a1a}-15.03\%}$
test_to[within-False-None] 6.1630ms 5.8763ms 170.1741 Ops/s 169.1978 Ops/s $\color{#35bf28}+0.58\%$
test_to[True-default-None] 12.6243ms 12.2027ms 81.9488 Ops/s 83.2296 Ops/s $\color{#d91a1a}-1.54\%$
test_to_njt[False-False-None] 8.8883ms 8.6006ms 116.2715 Ops/s 114.6257 Ops/s $\color{#35bf28}+1.44\%$
test_to_njt[True-False-None] 7.7388ms 7.4549ms 134.1392 Ops/s 133.5793 Ops/s $\color{#35bf28}+0.42\%$
test_to_njt[within-False-None] 17.0121ms 16.5116ms 60.5634 Ops/s 60.3124 Ops/s $\color{#35bf28}+0.42\%$
test_creation[device0] 0.3462ms 0.1135ms 8.8070 KOps/s 9.1155 KOps/s $\color{#d91a1a}-3.39\%$
test_creation_from_tensor 0.3493ms 0.1151ms 8.6843 KOps/s 8.7128 KOps/s $\color{#d91a1a}-0.33\%$
test_add_one[memmap_tensor0] 0.3000ms 7.1031μs 140.7828 KOps/s 148.3632 KOps/s $\textbf{\color{#d91a1a}-5.11\%}$
test_contiguous[memmap_tensor0] 21.9900μs 0.7129μs 1.4028 MOps/s 1.9645 MOps/s $\textbf{\color{#d91a1a}-28.59\%}$
test_stack[memmap_tensor0] 32.0010μs 4.6897μs 213.2333 KOps/s 211.9972 KOps/s $\color{#35bf28}+0.58\%$
test_memmaptd_index 1.0860ms 0.2892ms 3.4579 KOps/s 3.4308 KOps/s $\color{#35bf28}+0.79\%$
test_memmaptd_index_astensor 0.5381ms 0.3836ms 2.6066 KOps/s 2.5974 KOps/s $\color{#35bf28}+0.35\%$
test_memmaptd_index_op 0.8117ms 0.6380ms 1.5673 KOps/s 1.5848 KOps/s $\color{#d91a1a}-1.10\%$
test_serialize_model 0.1334s 0.1321s 7.5710 Ops/s 7.6963 Ops/s $\color{#d91a1a}-1.63\%$
test_serialize_model_pickle 1.3495s 1.2112s 0.8256 Ops/s 0.8253 Ops/s $\color{#35bf28}+0.03\%$
test_serialize_weights 0.3091s 0.1564s 6.3928 Ops/s 7.7578 Ops/s $\textbf{\color{#d91a1a}-17.59\%}$
test_serialize_weights_returnearly 0.3975s 69.2218ms 14.4463 Ops/s 12.0315 Ops/s $\textbf{\color{#35bf28}+20.07\%}$
test_serialize_weights_pickle 1.3497s 1.2116s 0.8253 Ops/s 0.8380 Ops/s $\color{#d91a1a}-1.52\%$
test_reshape_pytree 0.3673ms 33.5514μs 29.8050 KOps/s 29.6444 KOps/s $\color{#35bf28}+0.54\%$
test_reshape_td 75.1220μs 39.9789μs 25.0132 KOps/s 25.7473 KOps/s $\color{#d91a1a}-2.85\%$
test_view_pytree 0.2528ms 34.4664μs 29.0138 KOps/s 29.7070 KOps/s $\color{#d91a1a}-2.33\%$
test_view_td 89.7320μs 47.4483μs 21.0756 KOps/s 21.4999 KOps/s $\color{#d91a1a}-1.97\%$
test_unbind_pytree 0.2469ms 39.0155μs 25.6309 KOps/s 26.0836 KOps/s $\color{#d91a1a}-1.74\%$
test_unbind_td 0.1080ms 48.6246μs 20.5657 KOps/s 20.2633 KOps/s $\color{#35bf28}+1.49\%$
test_split_pytree 0.2585ms 44.7199μs 22.3614 KOps/s 22.2243 KOps/s $\color{#35bf28}+0.62\%$
test_split_td 0.1133ms 66.2879μs 15.0857 KOps/s 15.0325 KOps/s $\color{#35bf28}+0.35\%$
test_add_pytree 0.2292ms 44.8012μs 22.3208 KOps/s 21.3248 KOps/s $\color{#35bf28}+4.67\%$
test_add_td 89.6110μs 56.3352μs 17.7509 KOps/s 17.1318 KOps/s $\color{#35bf28}+3.61\%$
test_compile_add_one_nested[tensordict-compile] 0.2724ms 0.1851ms 5.4011 KOps/s 5.5090 KOps/s $\color{#d91a1a}-1.96\%$
test_compile_add_one_nested[tensordict-eager] 0.3179ms 0.1926ms 5.1923 KOps/s 5.1835 KOps/s $\color{#35bf28}+0.17\%$
test_compile_add_one_nested[pytree-compile] 0.2418ms 0.1508ms 6.6334 KOps/s 6.4283 KOps/s $\color{#35bf28}+3.19\%$
test_compile_add_one_nested[pytree-eager] 0.4398ms 0.1871ms 5.3449 KOps/s 4.3293 KOps/s $\textbf{\color{#35bf28}+23.46\%}$
test_compile_copy_nested[tensordict-compile] 52.6210μs 27.6167μs 36.2099 KOps/s 31.2094 KOps/s $\textbf{\color{#35bf28}+16.02\%}$
test_compile_copy_nested[tensordict-eager] 75.5810μs 54.2215μs 18.4429 KOps/s 18.7261 KOps/s $\color{#d91a1a}-1.51\%$
test_compile_copy_nested[pytree-compile] 40.0610μs 14.6245μs 68.3785 KOps/s 62.5392 KOps/s $\textbf{\color{#35bf28}+9.34\%}$
test_compile_copy_nested[pytree-eager] 0.3937ms 76.0086μs 13.1564 KOps/s 13.1097 KOps/s $\color{#35bf28}+0.36\%$
test_compile_add_one_flat[tensordict-compile] 0.2669ms 0.2082ms 4.8039 KOps/s 4.5580 KOps/s $\textbf{\color{#35bf28}+5.40\%}$
test_compile_add_one_flat[tensordict-eager] 0.3306ms 0.2626ms 3.8088 KOps/s 3.7145 KOps/s $\color{#35bf28}+2.54\%$
test_compile_add_one_flat[tensorclass-compile] 0.2018ms 0.1539ms 6.4973 KOps/s 5.9217 KOps/s $\textbf{\color{#35bf28}+9.72\%}$
test_compile_add_one_flat[tensorclass-eager] 0.1459ms 70.9966μs 14.0852 KOps/s 13.2088 KOps/s $\textbf{\color{#35bf28}+6.63\%}$
test_compile_add_one_flat[pytree-compile] 0.2429ms 0.2022ms 4.9447 KOps/s 4.5173 KOps/s $\textbf{\color{#35bf28}+9.46\%}$
test_compile_add_one_flat[pytree-eager] 0.7981ms 0.5386ms 1.8567 KOps/s 1.6963 KOps/s $\textbf{\color{#35bf28}+9.45\%}$
test_compile_add_self_flat[tensordict-eager] 0.3548ms 0.3151ms 3.1740 KOps/s 3.0820 KOps/s $\color{#35bf28}+2.98\%$
test_compile_add_self_flat[tensordict-compile] 0.2548ms 0.2086ms 4.7936 KOps/s 4.3262 KOps/s $\textbf{\color{#35bf28}+10.81\%}$
test_compile_add_self_flat[tensorclass-eager] 0.1323ms 86.6993μs 11.5341 KOps/s 10.7634 KOps/s $\textbf{\color{#35bf28}+7.16\%}$
test_compile_add_self_flat[tensorclass-compile] 0.2077ms 0.1561ms 6.4045 KOps/s 5.9280 KOps/s $\textbf{\color{#35bf28}+8.04\%}$
test_compile_add_self_flat[pytree-eager] 0.6594ms 0.4532ms 2.2066 KOps/s 2.0483 KOps/s $\textbf{\color{#35bf28}+7.73\%}$
test_compile_add_self_flat[pytree-compile] 0.2563ms 0.2031ms 4.9236 KOps/s 4.5771 KOps/s $\textbf{\color{#35bf28}+7.57\%}$
test_compile_copy_flat[tensordict-compile] 55.6610μs 23.7362μs 42.1298 KOps/s 36.7905 KOps/s $\textbf{\color{#35bf28}+14.51\%}$
test_compile_copy_flat[tensordict-eager] 71.5310μs 41.8245μs 23.9094 KOps/s 24.1058 KOps/s $\color{#d91a1a}-0.81\%$
test_compile_copy_flat[pytree-compile] 52.2410μs 20.5448μs 48.6740 KOps/s 42.0461 KOps/s $\textbf{\color{#35bf28}+15.76\%}$
test_compile_copy_flat[pytree-eager] 0.3565ms 70.0300μs 14.2796 KOps/s 14.4915 KOps/s $\color{#d91a1a}-1.46\%$
test_compile_assign_and_add[tensordict-compile] 2.0987ms 0.2120ms 4.7173 KOps/s 4.5620 KOps/s $\color{#35bf28}+3.40\%$
test_compile_assign_and_add[tensordict-eager] 3.5187ms 3.3383ms 299.5492 Ops/s 303.8688 Ops/s $\color{#d91a1a}-1.42\%$
test_compile_assign_and_add[pytree-compile] 2.0593ms 0.2098ms 4.7673 KOps/s 4.7530 KOps/s $\color{#35bf28}+0.30\%$
test_compile_assign_and_add[pytree-eager] 3.1055ms 2.8931ms 345.6523 Ops/s 346.0858 Ops/s $\color{#d91a1a}-0.13\%$
test_compile_indexing[tensor-tensordict-compile] 0.1936ms 0.1434ms 6.9753 KOps/s 6.8156 KOps/s $\color{#35bf28}+2.34\%$
test_compile_indexing[tensor-tensordict-eager] 0.2992ms 65.7484μs 15.2095 KOps/s 15.1239 KOps/s $\color{#35bf28}+0.57\%$
test_compile_indexing[tensor-tensorclass-compile] 0.1842ms 0.1373ms 7.2856 KOps/s 6.8164 KOps/s $\textbf{\color{#35bf28}+6.88\%}$
test_compile_indexing[tensor-tensorclass-eager] 0.2530ms 46.0818μs 21.7005 KOps/s 21.3706 KOps/s $\color{#35bf28}+1.54\%$
test_compile_indexing[tensor-pytree-compile] 0.1978ms 0.1411ms 7.0884 KOps/s 6.8230 KOps/s $\color{#35bf28}+3.89\%$
test_compile_indexing[tensor-pytree-eager] 0.2794ms 49.5692μs 20.1738 KOps/s 21.2639 KOps/s $\textbf{\color{#d91a1a}-5.13\%}$
test_compile_indexing[slice-tensordict-compile] 0.1325ms 88.1619μs 11.3428 KOps/s 11.2882 KOps/s $\color{#35bf28}+0.48\%$
test_compile_indexing[slice-tensordict-eager] 0.2035ms 27.5206μs 36.3364 KOps/s 35.6698 KOps/s $\color{#35bf28}+1.87\%$
test_compile_indexing[slice-tensorclass-compile] 0.1472ms 85.6412μs 11.6766 KOps/s 12.1479 KOps/s $\color{#d91a1a}-3.88\%$
test_compile_indexing[slice-tensorclass-eager] 0.2410ms 24.3974μs 40.9880 KOps/s 41.1373 KOps/s $\color{#d91a1a}-0.36\%$
test_compile_indexing[slice-pytree-compile] 0.1617ms 88.3862μs 11.3140 KOps/s 12.0088 KOps/s $\textbf{\color{#d91a1a}-5.79\%}$
test_compile_indexing[slice-pytree-eager] 0.2814ms 24.0138μs 41.6427 KOps/s 41.4407 KOps/s $\color{#35bf28}+0.49\%$
test_compile_indexing[int-tensordict-compile] 0.1288ms 89.7327μs 11.1442 KOps/s 11.2406 KOps/s $\color{#d91a1a}-0.86\%$
test_compile_indexing[int-tensordict-eager] 0.2899ms 27.0742μs 36.9355 KOps/s 36.5760 KOps/s $\color{#35bf28}+0.98\%$
test_compile_indexing[int-tensorclass-compile] 0.1295ms 83.3598μs 11.9962 KOps/s 11.9475 KOps/s $\color{#35bf28}+0.41\%$
test_compile_indexing[int-tensorclass-eager] 0.2264ms 23.5470μs 42.4682 KOps/s 41.3287 KOps/s $\color{#35bf28}+2.76\%$
test_compile_indexing[int-pytree-compile] 0.2628ms 84.6797μs 11.8092 KOps/s 12.0372 KOps/s $\color{#d91a1a}-1.89\%$
test_compile_indexing[int-pytree-eager] 0.2377ms 24.7439μs 40.4140 KOps/s 41.6015 KOps/s $\color{#d91a1a}-2.85\%$
test_mod_add[eager] 0.1018ms 53.7174μs 18.6160 KOps/s 19.3485 KOps/s $\color{#d91a1a}-3.79\%$
test_mod_add[compile] 0.2455ms 0.1576ms 6.3457 KOps/s 6.5063 KOps/s $\color{#d91a1a}-2.47\%$
test_mod_add[compile-overhead] 0.2955ms 0.2015ms 4.9618 KOps/s 4.9924 KOps/s $\color{#d91a1a}-0.61\%$
test_mod_wrap[eager] 0.3944ms 0.3060ms 3.2682 KOps/s 3.2375 KOps/s $\color{#35bf28}+0.95\%$
test_mod_wrap[compile] 0.4989ms 0.4038ms 2.4767 KOps/s 2.3975 KOps/s $\color{#35bf28}+3.30\%$
test_mod_wrap[compile-overhead] 7.5193ms 3.9867ms 250.8337 Ops/s 248.5612 Ops/s $\color{#35bf28}+0.91\%$
test_mod_wrap_and_backward[eager] 1.6745ms 1.5674ms 637.9998 Ops/s 628.2470 Ops/s $\color{#35bf28}+1.55\%$
test_mod_wrap_and_backward[compile] 1.7131ms 1.6182ms 617.9565 Ops/s 613.7109 Ops/s $\color{#35bf28}+0.69\%$
test_mod_wrap_and_backward[compile-overhead] 1.3722ms 0.9881ms 1.0120 KOps/s 994.8535 Ops/s $\color{#35bf28}+1.73\%$
test_seq_add[eager] 0.2048ms 0.1598ms 6.2580 KOps/s 5.9333 KOps/s $\textbf{\color{#35bf28}+5.47\%}$
test_seq_add[compile] 0.2779ms 0.1730ms 5.7816 KOps/s 5.7086 KOps/s $\color{#35bf28}+1.28\%$
test_seq_add[compile-overhead] 0.2861ms 0.2040ms 4.9014 KOps/s 4.5290 KOps/s $\textbf{\color{#35bf28}+8.22\%}$
test_seq_wrap[eager] 0.6124ms 0.5396ms 1.8533 KOps/s 1.6881 KOps/s $\textbf{\color{#35bf28}+9.79\%}$
test_seq_wrap[compile] 0.5487ms 0.4379ms 2.2838 KOps/s 2.2372 KOps/s $\color{#35bf28}+2.08\%$
test_seq_wrap[compile-overhead] 0.3803ms 0.3165ms 3.1600 KOps/s 2.9873 KOps/s $\textbf{\color{#35bf28}+5.78\%}$
test_func_call_runtime[False-eager] 0.9577ms 0.8761ms 1.1414 KOps/s 1.0369 KOps/s $\textbf{\color{#35bf28}+10.08\%}$
test_func_call_runtime[False-compile] 1.0191ms 0.9480ms 1.0549 KOps/s 979.3514 Ops/s $\textbf{\color{#35bf28}+7.71\%}$
test_func_call_runtime[False-compile-overhead] 0.5403ms 0.5003ms 1.9988 KOps/s 1.9680 KOps/s $\color{#35bf28}+1.57\%$
test_func_call_runtime[True-eager] 1.2063ms 1.1145ms 897.2914 Ops/s 886.9150 Ops/s $\color{#35bf28}+1.17\%$
test_func_call_runtime[True-compile] 1.0174ms 0.9665ms 1.0347 KOps/s 1.0301 KOps/s $\color{#35bf28}+0.44\%$
test_func_call_runtime[True-compile-overhead] 0.5885ms 0.5193ms 1.9257 KOps/s 1.8831 KOps/s $\color{#35bf28}+2.26\%$
test_func_call_cm_runtime[False-eager] 0.9419ms 0.8645ms 1.1567 KOps/s 1.0875 KOps/s $\textbf{\color{#35bf28}+6.36\%}$
test_func_call_cm_runtime[False-compile] 1.0071ms 0.9522ms 1.0502 KOps/s 1.0511 KOps/s $\color{#d91a1a}-0.08\%$
test_func_call_cm_runtime[False-compile-overhead] 0.5651ms 0.5024ms 1.9906 KOps/s 1.9604 KOps/s $\color{#35bf28}+1.54\%$
test_func_call_cm_runtime[True-eager] 1.3575ms 1.2641ms 791.0859 Ops/s 781.4797 Ops/s $\color{#35bf28}+1.23\%$
test_func_call_cm_runtime[True-compile] 1.1087ms 1.0055ms 994.5074 Ops/s 996.4279 Ops/s $\color{#d91a1a}-0.19\%$
test_func_call_cm_runtime[True-compile-overhead] 0.6230ms 0.5535ms 1.8068 KOps/s 1.7802 KOps/s $\color{#35bf28}+1.50\%$
test_vmap_func_call_cm_runtime[eager] 2.8797ms 2.3797ms 420.2269 Ops/s 418.7666 Ops/s $\color{#35bf28}+0.35\%$
test_vmap_func_call_cm_runtime[compile] 1.1031ms 1.0157ms 984.5675 Ops/s 987.5293 Ops/s $\color{#d91a1a}-0.30\%$
test_vmap_func_call_cm_runtime[compile-overhead] 0.6049ms 0.5472ms 1.8275 KOps/s 1.7205 KOps/s $\textbf{\color{#35bf28}+6.22\%}$
test_distributed 2.8997ms 0.1736ms 5.7587 KOps/s 6.4403 KOps/s $\textbf{\color{#d91a1a}-10.58\%}$
test_tdmodule 0.3053ms 28.4803μs 35.1120 KOps/s 36.0699 KOps/s $\color{#d91a1a}-2.66\%$
test_tdmodule_dispatch 79.0220μs 48.6635μs 20.5493 KOps/s 20.6857 KOps/s $\color{#d91a1a}-0.66\%$
test_tdseq 58.9110μs 26.7220μs 37.4223 KOps/s 37.1953 KOps/s $\color{#35bf28}+0.61\%$
test_tdseq_dispatch 85.9620μs 50.8398μs 19.6696 KOps/s 19.7674 KOps/s $\color{#d91a1a}-0.49\%$
test_instantiation_functorch 2.2286ms 2.1168ms 472.4126 Ops/s 475.8327 Ops/s $\color{#d91a1a}-0.72\%$
test_exec_functorch 0.2310ms 0.1861ms 5.3730 KOps/s 5.3952 KOps/s $\color{#d91a1a}-0.41\%$
test_exec_functional_call 0.2242ms 0.1652ms 6.0518 KOps/s 6.0224 KOps/s $\color{#35bf28}+0.49\%$
test_exec_td_decorator 0.4467ms 0.2430ms 4.1157 KOps/s 4.1358 KOps/s $\color{#d91a1a}-0.49\%$
test_vmap_mlp_speed_decorator[True-True] 0.9984ms 0.8082ms 1.2373 KOps/s 1.2304 KOps/s $\color{#35bf28}+0.56\%$
test_vmap_mlp_speed_decorator[True-False] 0.9848ms 0.8080ms 1.2377 KOps/s 1.2294 KOps/s $\color{#35bf28}+0.68\%$
test_vmap_mlp_speed_decorator[False-True] 0.8486ms 0.6935ms 1.4419 KOps/s 1.4344 KOps/s $\color{#35bf28}+0.52\%$
test_vmap_mlp_speed_decorator[False-False] 0.8869ms 0.6932ms 1.4425 KOps/s 1.4340 KOps/s $\color{#35bf28}+0.60\%$
test_vmap_transformer_speed_decorator[True-True] 20.6718ms 20.5983ms 48.5477 Ops/s 48.1635 Ops/s $\color{#35bf28}+0.80\%$
test_vmap_transformer_speed_decorator[True-False] 20.7144ms 20.6055ms 48.5308 Ops/s 48.1843 Ops/s $\color{#35bf28}+0.72\%$
test_vmap_transformer_speed_decorator[False-True] 20.5314ms 20.4477ms 48.9054 Ops/s 48.6181 Ops/s $\color{#35bf28}+0.59\%$
test_vmap_transformer_speed_decorator[False-False] 20.5549ms 20.4529ms 48.8929 Ops/s 48.5694 Ops/s $\color{#35bf28}+0.67\%$
test_to_module_speed[True] 1.5806ms 1.4869ms 672.5366 Ops/s 674.7118 Ops/s $\color{#d91a1a}-0.32\%$
test_to_module_speed[False] 1.5995ms 1.4572ms 686.2385 Ops/s 679.5530 Ops/s $\color{#35bf28}+0.98\%$
test_tc_init 73.5020μs 52.0968μs 19.1950 KOps/s 19.0946 KOps/s $\color{#35bf28}+0.53\%$
test_tc_init_tensor_only 43.1300μs 14.9943μs 66.6919 KOps/s 65.6992 KOps/s $\color{#35bf28}+1.51\%$
test_tc_init_nested 0.1847ms 0.1035ms 9.6630 KOps/s 9.6195 KOps/s $\color{#35bf28}+0.45\%$
test_tc_first_layer_tensor 20.9800μs 1.7932μs 557.6699 KOps/s 551.7348 KOps/s $\color{#35bf28}+1.08\%$
test_tc_first_layer_tensor_only 4.1530μs 0.6796μs 1.4715 MOps/s 1.4548 MOps/s $\color{#35bf28}+1.15\%$
test_tc_first_layer_tensor_set 31.0800μs 4.1970μs 238.2664 KOps/s 237.6055 KOps/s $\color{#35bf28}+0.28\%$
test_tc_first_layer_tensor_only_set 15.6850μs 3.0295μs 330.0829 KOps/s 329.2541 KOps/s $\color{#35bf28}+0.25\%$
test_tc_first_layer_nontensor 25.9300μs 6.0089μs 166.4191 KOps/s 165.9318 KOps/s $\color{#35bf28}+0.29\%$
test_tc_second_layer_tensor 22.7600μs 4.3003μs 232.5413 KOps/s 230.0751 KOps/s $\color{#35bf28}+1.07\%$
test_tc_second_layer_nontensor 43.1410μs 8.3916μs 119.1662 KOps/s 117.5073 KOps/s $\color{#35bf28}+1.41\%$
test_unbind 0.2558s 18.1441ms 55.1143 Ops/s 46.6594 Ops/s $\textbf{\color{#35bf28}+18.12\%}$
test_full_like 5.6162ms 4.3150ms 231.7521 Ops/s 59.9419 Ops/s $\textbf{\color{#35bf28}+286.63\%}$
test_zeros_like 4.5451ms 4.3756ms 228.5385 Ops/s 57.0286 Ops/s $\textbf{\color{#35bf28}+300.74\%}$
test_ones_like 5.5218ms 4.3774ms 228.4477 Ops/s 60.0588 Ops/s $\textbf{\color{#35bf28}+280.37\%}$
test_clone 7.0272ms 6.5188ms 153.4022 Ops/s 56.4294 Ops/s $\textbf{\color{#35bf28}+171.85\%}$
test_squeeze 0.1859ms 15.1861μs 65.8497 KOps/s 67.4993 KOps/s $\color{#d91a1a}-2.44\%$
test_unsqueeze 0.1600ms 0.1094ms 9.1432 KOps/s 9.0876 KOps/s $\color{#35bf28}+0.61\%$
test_split 0.2391ms 0.1877ms 5.3288 KOps/s 5.3054 KOps/s $\color{#35bf28}+0.44\%$
test_permute 0.2806ms 0.2080ms 4.8078 KOps/s 4.8875 KOps/s $\color{#d91a1a}-1.63\%$
test_stack 51.5712ms 50.9301ms 19.6348 Ops/s 19.5623 Ops/s $\color{#35bf28}+0.37\%$
test_cat 51.8914ms 51.2922ms 19.4961 Ops/s 19.5197 Ops/s $\color{#d91a1a}-0.12\%$

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. dependencies Pull requests that update a dependency file python Pull requests that update python code

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant