Убрал поддиректорию
This commit is contained in:
8
src/gpu_plugin.cu
Normal file
8
src/gpu_plugin.cu
Normal file
@@ -0,0 +1,8 @@
|
||||
#include <cuda_runtime.h>
|
||||
|
||||
extern "C" int gpu_is_available() {
|
||||
int n = 0;
|
||||
cudaError_t err = cudaGetDeviceCount(&n);
|
||||
if (err != cudaSuccess) return 0;
|
||||
return (n > 0) ? 1 : 0;
|
||||
}
|
||||
Reference in New Issue
Block a user