[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:
co-authored by
Ma Mingfei
parent
3ce72252de
commit
e26c73c4e9
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user