search

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)
https://ffcv.io
Apache License 2.0
2.84k stars 178 forks source link

Modify Tests to Run on Consumer Hardware #294

Closed warner-benjamin closed 1 year ago

warner-benjamin commented 1 year ago

This PR modifies the FFCV tests to work on consumer hardware by adding cpu_count checks and lowering the batch and size in test_cuda_nonblocking.py.

The test_cuda_nonblocking.py changes could be moved to a pytest config file with a pytest extension like pytest-testconfig, assuming pytest doesn't this built in and I am missing it.

The dataset writer tests currently hang on my machine, I believe I fixed this issue by merging in the fix-198 and FFCV 1.0 branches.

andrewilyas commented 1 year ago

This looks like a great and easy fix, thank you!