perseusdg
20 lines
631 B
Plaintext
20 lines
631 B
Plaintext
#include <stdio.h>
|
|
|
|
int main(int argc, char **argv){
|
|
cudaDeviceProp dP;
|
|
float min_cc = 5.0;
|
|
|
|
int rc = cudaGetDeviceProperties(&dP, 0);
|
|
if(rc != cudaSuccess) {
|
|
cudaError_t error = cudaGetLastError();
|
|
printf("CUDA error: %s", cudaGetErrorString(error));
|
|
return rc; /* Failure */
|
|
}
|
|
if((dP.major+(dP.minor/10)) < min_cc) {
|
|
printf("Min Compute Capability of %2.1f required: %d.%d found\n Not Building CUDA Code", min_cc, dP.major, dP.minor);
|
|
return 1; /* Failure */
|
|
} else {
|
|
printf("-arch=sm_%d%d", dP.major, dP.minor);
|
|
return 0; /* Success */
|
|
}
|
|
} |