refactor: unify CUDA skip guards in tests

- add skip_no_fp8 (CUDA + fp8_mm kernel + cc 8.9+) to tests/conftest.py
- use skip_no_cuda / skip_no_kernel / skip_no_fp8 directly in test modules
- drop _GPU alias and tests.extension.conftest re-exports
- remove unused imports (Union in hf_adapter, make_grpo_config in data conftest)
This commit is contained in:
2026-08-22 21:08:06 +08:00
parent 16a55bb474
commit 75304d084d
8 changed files with 25 additions and 24 deletions
+2 -4
View File
@@ -1,4 +1,3 @@
import pytest
import torch
import torch.distributed as dist
from torch.distributed.fsdp import fully_shard
@@ -8,11 +7,10 @@ from torch.nn.parallel import DistributedDataParallel as DDP
from astrai.model import AutoRegressiveLM
from astrai.optim import NoraNAdamW
from astrai.parallel.setup import find_free_port
from tests.conftest import skip_no_cuda
from tests.helpers import make_tiny_config
pytestmark = pytest.mark.skipif(
torch.cuda.device_count() < 1, reason="CUDA device required"
)
pytestmark = skip_no_cuda
def _assign_grads_and_step(model):