cusplibrary/build/print_sm_version.cpp

#include <cuda_runtime_api.h>
#include <stdio.h>
#include <stdlib.h>

void usage(const char *name)
{
  printf("usage: %s [device_id]\n", name);
}

int main(int argc, char **argv)
{
  int num_devices = 0;
  int device_id = 0;
  if(argc == 2)
  {
    device_id = atoi(argv[1]);
  }
  else if(argc > 2)
  {
    usage(argv[0]);
    exit(-1);
  }

  cudaGetDeviceCount(&num_devices);
  if(num_devices > device_id)
  {
    cudaDeviceProp properties;
    cudaGetDeviceProperties(&properties, device_id);
    printf("--gpu-architecture=sm_%d%d", properties.major, properties.minor);
    return 0;
  } // end if
  else
  {
    printf("No available device with id %d\n", device_id);
  }

  return -1;
}

1	caltinay	4955	#include <cuda_runtime_api.h>
2			#include <stdio.h>
3			#include <stdlib.h>
4
5			void usage(const char *name)
6			{
7			printf("usage: %s [device_id]\n", name);
8			}
9
10			int main(int argc, char **argv)
11			{
12			int num_devices = 0;
13			int device_id = 0;
14			if(argc == 2)
15			{
16			device_id = atoi(argv[1]);
17			}
18			else if(argc > 2)
19			{
20			usage(argv[0]);
21			exit(-1);
22			}
23
24			cudaGetDeviceCount(&num_devices);
25			if(num_devices > device_id)
26			{
27			cudaDeviceProp properties;
28			cudaGetDeviceProperties(&properties, device_id);
29			printf("--gpu-architecture=sm_%d%d", properties.major, properties.minor);
30			return 0;
31			} // end if
32			else
33			{
34			printf("No available device with id %d\n", device_id);
35			}
36
37			return -1;
38			}
39