| 1 | #include "common.hpp" |
| 2 | |
| 3 | int32_t get_sm_version_num() { |
| 4 | int32_t major_capability, minor_capability; |
| 5 | cudaDeviceGetAttribute(&major_capability, cudaDevAttrComputeCapabilityMajor, |
| 6 | 0); |
| 7 | cudaDeviceGetAttribute(&minor_capability, cudaDevAttrComputeCapabilityMinor, |
| 8 | 0); |
| 9 | int32_t version_num = major_capability * 10 + minor_capability; |
| 10 | return version_num; |
| 11 | } |