[Bugfix] fix mindiesd laserattention unsupported error (#2673)

Signed-off-by: fan2956 <zhoufan53@huawei.com>
This commit is contained in:
fan2956 authored and GitHub committed 2026-04-10 16:18:27 +08:00
1 parent c1da480bbf
commit c2ae58bb84
1 file changed
+3
+3
View File
@@ -69,6 +69,9 @@ class NPUOmniPlatform(OmniPlatform, NPUPlatform):
# Try FLASH_ATTN if mindiesd is available, otherwise fall back to SDPA
if find_spec("mindiesd"):
# Configure ASCEND_CUSTOM_OPP_PATH for mindiesd custom ops upon import
import mindiesd # noqa: F401
logger.info("Defaulting to diffusion attention backend FLASH_ATTN")
return DiffusionAttentionBackendEnum.FLASH_ATTN.get_path()