diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2017-10-15 12:17:35 +0200 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2017-10-15 12:17:35 +0200 |
commit | 7408da174c848ffeaa1fe2da52f26a057e65b0f1 (patch) | |
tree | 27045f3a79905450829845dd3725f7152282f505 /src/utilities | |
parent | 55a802c63d79264bf6e5e9d82a1df34bbe85ee64 (diff) |
Various fixes to make the first CUDA examples work
Diffstat (limited to 'src/utilities')
-rw-r--r-- | src/utilities/utilities.cpp | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/src/utilities/utilities.cpp b/src/utilities/utilities.cpp index a5c1d45e..f2574104 100644 --- a/src/utilities/utilities.cpp +++ b/src/utilities/utilities.cpp @@ -413,13 +413,17 @@ std::string GetDeviceVendor(const Device& device) { // Mid-level info std::string GetDeviceArchitecture(const Device& device) { auto device_architecture = std::string{""}; - if (device.HasExtension(kKhronosAttributesNVIDIA)) { + #ifdef CUDA_API device_architecture = device.NVIDIAComputeCapability(); - } - else if (device.HasExtension(kKhronosAttributesAMD)) { - device_architecture = device.Name(); // Name is architecture for AMD APP and AMD ROCm - } - // Note: no else - 'device_architecture' might be the empty string + #else + if (device.HasExtension(kKhronosAttributesNVIDIA)) { + device_architecture = device.NVIDIAComputeCapability(); + } + else if (device.HasExtension(kKhronosAttributesAMD)) { + device_architecture = device.Name(); // Name is architecture for AMD APP and AMD ROCm + } + // Note: no else - 'device_architecture' might be the empty string + #endif for (auto &find_and_replace : device_mapping::kArchitectureNames) { // replacing to common names if (device_architecture == find_and_replace.first) { device_architecture = find_and_replace.second; } |