cuda_version.cu 506 Bytes
Newer Older
chenych's avatar
chenych committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#include <cuda_runtime_api.h>

#define CUDART_VERSION  11080

namespace adet {
int get_cudart_version() {
  #if !defined(__HIP_PLATFORM_HCC__)
    return CUDART_VERSION;
  #else
    int version = 0;

  #if HIP_VERSION_MAJOR != 0
    // Create a convention similar to that of CUDA, as assumed by other
    // parts of the code.
    version = HIP_VERSION_MINOR;
    version += (HIP_VERSION_MAJOR * 100);
  #else
    hipRuntimeGetVersion(&version);
  #endif
    return version;
  #endif
}
} // namespace adet