| |
| """Run existing CPU tensor tests without initializing the historical T5 CUDA default.""" |
| import os |
| from pathlib import Path |
| import sys |
| from unittest.mock import patch |
|
|
| ROOT = Path(__file__).resolve().parents[1] |
| os.environ['CUDA_VISIBLE_DEVICES'] = '' |
| os.environ.setdefault('OMP_NUM_THREADS', '2') |
| os.environ['PYTHONDONTWRITEBYTECODE'] = '1' |
| sys.dont_write_bytecode = True |
| sys.path.insert(0, str(ROOT/'code')) |
| os.chdir(ROOT/'code') |
| import torch |
| import pytest |
|
|
| |
| |
| with patch('torch.cuda.current_device', return_value=0): |
| from trainer.distillation import Trainer |
| from model.wan_kv_mmd import WanKVMMD |
|
|
| raise SystemExit(pytest.main([ |
| '-q', '-p', 'no:cacheprovider', |
| 'tests/test_triple_partial_geometry.py', |
| 'tests/test_readable_config.py', |
| 'tests/test_evaluated_inference.py', |
| 'tests/test_active_encoder_offload.py', |
| 'tests/test_no_dinov3_filtered_bank.py', |
| 'tests/test_no_dinov3_ablation.py', |
| 'tests/test_release_schedule.py', |
| 'tests/test_triple_video_feature_bank.py', |
| 'tests/test_triple_feature_metadata_contract.py', |
| 'tests/test_triple_pixel_cotangent_accumulation.py', |
| 'tests/test_video_nystrom_partial_mmd.py', |
| 'tests/test_multi_representation_video_nystrom_mmd.py', |
| ] + sys.argv[1:])) |
|
|
