[MUSA][17/N] ci: Add MUSA diffusion, sgl-kernel tests, and CI workflow support (#20672)
Co-authored-by: ximin.chen <ximin.chen@mthreads.com> Co-authored-by: R0CKSTAR <xiaodong.ye@mthreads.com>
This commit is contained in:
co-authored by
ximin.chen
R0CKSTAR
parent
15e6572f21
commit
cdf5771f91
@@ -1,6 +1,11 @@
|
||||
import pytest
|
||||
import torch
|
||||
|
||||
from sglang.srt.utils import is_musa
|
||||
|
||||
if is_musa():
|
||||
import torchada # noqa: F401
|
||||
|
||||
|
||||
# This fixture ensures the torch defaults don't get left in modified states between
|
||||
# tests (e.g., when a test fails before restoring the original value), which
|
||||
|
||||
Reference in New Issue
Block a user