mirror of
https://github.com/radixark/miles.git
synced 2026-10-01 23:06:14 +08:00
10 lines
239 B
Python
10 lines
239 B
Python
import pytest
|
|
|
|
|
|
def test_fsdp_import():
|
|
try:
|
|
from torch.distributed.fsdp import FullyShardedDataParallel as FSDP
|
|
except ImportError:
|
|
pytest.skip("FSDP not available in this environment")
|
|
assert FSDP is not None
|