![]() If we try to load the CUDA library on an old GPU, it panics and crashes the server. This checks the compute capability before we load the library so we can gracefully fall back to CPU mode. |
||
---|---|---|
.. | ||
gpu.go | ||
gpu_darwin.go | ||
gpu_info.h | ||
gpu_info_cpu.c | ||
gpu_info_cuda.c | ||
gpu_info_cuda.h | ||
gpu_info_rocm.c | ||
gpu_info_rocm.h | ||
gpu_test.go | ||
types.go |