bluecast.tests.test_check_gpu_support¶
Module Contents¶
Functions¶
Test that the function falls back to CPU when no GPU is detected. |
|
Test that the function detects GPU support correctly. |
|
Test that the function falls back to CPU if GPU-related warnings are captured. |
- bluecast.tests.test_check_gpu_support.test_check_gpu_support_cpu_fallback()¶
Test that the function falls back to CPU when no GPU is detected.
- bluecast.tests.test_check_gpu_support.test_check_gpu_support_gpu_available()¶
Test that the function detects GPU support correctly.
- bluecast.tests.test_check_gpu_support.test_check_gpu_support_gpu_warning()¶
Test that the function falls back to CPU if GPU-related warnings are captured.