[diffusion] platform: support Intel XPU (#17920)

Signed-off-by: sushil.dubey <sushil.dubey@intel.com>
Signed-off-by: Sushil Dubey <sushil.dubey@intel.com>
Co-authored-by: Ma Mingfei <mingfei.ma@intel.com>
This commit is contained in:
sushil Dubey
2026-04-11 15:09:02 +08:00
committed by GitHub
co-authored by Ma Mingfei
parent 3ce72252de
commit e26c73c4e9
13 changed files with 437 additions and 12 deletions
@@ -338,7 +338,7 @@ def fuse_scale_shift_kernel(
block_l: int = 128,
block_c: int = 128,
):
assert x.is_cuda and scale.is_cuda
assert (x.is_cuda and scale.is_cuda) or (x.is_xpu and scale.is_xpu)
assert x.is_contiguous()
B, L, C = x.shape