diffusers
[Neuron] Enable `torch.compile` compatibility with Neuron device
#13485

Merged

[Neuron] Enable `torch.compile` compatibility with Neuron device #13485

yiyixuxu merged 38 commits into huggingface:main from JingyaHuang:neuron-torch-comppile

draft:add neuron as a legit backend

98f6c8c2

Merge branch 'huggingface:main' into add-neuron-backend

c58b8b8b

Merge branch 'huggingface:main' into add-neuron-backend

33674096

Merge branch 'main' into add-neuron-backend

0c517340

feat: neuron-specific changes in the pipeline

a76953cf

tests: eager tests

2480388f

fix: style

929ab728

Merge branch 'huggingface:main' into add-neuron-backend

52cac769

Merge branch 'add-neuron-backend' of github.com:JingyaHuang/diffusers…

28a50862

Merge branch 'huggingface:main' into add-neuron-backend

68689e5c

Merge branch 'main' into add-neuron-backend

da79308d

fix:apr_02 beta

3bb9c7c3

Merge branch 'add-neuron-backend' of github.com:JingyaHuang/diffusers…

c4facab1

Merge branch 'huggingface:main' into add-neuron-backend

1eb5ff9d

fix:pixart

cbe8f28a

fix: rewrite flux swiglu activation to avoid gather op in neuron IR

16b96069

github-actions added lora

github-actions added models

github-actions added tests

github-actions added utils

github-actions added pipelines

github-actions added examples

github-actions added size/L

test: pixart compile mode on neuron

7f13f685

github-actions removed size/L

github-actions added size/L

JingyaHuang changed the title ~~[Neuron] Enable `torch.comppile` compatibility with Neuron device~~ [Neuron] Enable `torch.compile` compatibility with Neuron device 207 days ago

Merge branch 'main' into neuron-torch-comppile

a46cb190

github-actions removed size/L

github-actions added size/L

cleanup & fix style

a354b883

Merge branch 'main' into neuron-torch-comppile

3d1a259e

Merge branch 'main' into neuron-torch-comppile

17c77caf

test: flux2 klein and sdxl-turbo

c6501d5e

github-actions removed lora

github-actions removed examples

JingyaHuang marked this pull request as ready for review 180 days ago

Merge branch 'main' into neuron-torch-comppile

d0bba163

sayakpaul approved these changes on 2026-05-18

sayakpaul requested a review from

DN6 180 days ago

sayakpaul requested a review from

yiyixuxu 180 days ago

github-actions requested changes on 2026-05-18

Merge branch 'main' into neuron-torch-comppile

d566ce01

Merge branch 'main' into neuron-torch-comppile

4b36ec6d

github-actions removed utils

github-actions removed size/L

github-actions added size/M

Merge branch 'main' into neuron-torch-comppile

b586e14c

review: address some comments

cf20ff11

github-actions removed size/M

github-actions added size/L

revert:put back expected slices

d8d0db3c

github-actions removed size/L

github-actions added size/M

cleanup:move the ts index ahead of denoising + cleanup tests

f536c591

Merge branch 'main' into neuron-torch-comppile

7df50b8c

fix: style

7f6a7304

Merge branch 'neuron-torch-comppile' of github.com:JingyaHuang/diffus…

f07d960f

JingyaHuang requested a review from

sayakpaul 162 days ago

sayakpaul commented on 2026-06-05

fix style

90b94bc0

Merge branch 'main' into neuron-torch-comppile

1e5ab322

Merge branch 'main' into neuron-torch-comppile

956af418

review:remove the hook

7f976668

Merge branch 'neuron-torch-comppile' of github.com:JingyaHuang/diffus…

9d8f2443

JingyaHuang commented on 2026-06-15

Merge branch 'main' into neuron-torch-comppile

f72d3f89

sayakpaul approved these changes on 2026-06-17

yiyixuxu merged 236e5dd9 into main 149 days ago

JingyaHuang deleted the neuron-torch-comppile branch 148 days ago

Reviewers

sayakpaul

github-actions

DN6

yiyixuxu

Assignees

No one assigned

Labels

models tests size/M pipelines

Milestone

No milestone

diffusers [Neuron] Enable `torch.compile` compatibility with Neuron device #13485 Merged

[Neuron] Enable `torch.compile` compatibility with Neuron device #13485

diffusers
[Neuron] Enable `torch.compile` compatibility with Neuron device
#13485

Merged