()
| 128 | # Offloader Tests |
| 129 | @pytest.fixture |
| 130 | def offloader(): |
| 131 | device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') |
| 132 | return Offloader( |
| 133 | num_blocks=4, |
| 134 | blocks_to_swap=2, |
| 135 | device=device, |
| 136 | debug=False |
| 137 | ) |
| 138 | |
| 139 | |
| 140 | def test_offloader_init(offloader): |