add DeviceName to ROCm api (#4437)

eeabd46a · Thomas Viehmann · masahi · 2d0010f3 · eeabd46a
Commit eeabd46a authored Nov 27, 2019 by Thomas Viehmann Committed by masahi Nov 28, 2019
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletions

src/runtime/rocm/rocm_device_api.cc
+6 -1

No files found.
--- a/src/runtime/rocm/rocm_device_api.cc
+++ b/src/runtime/rocm/rocm_device_api.cc
@@ -78,8 +78,13 @@ class ROCMDeviceAPI final : public DeviceAPI {
        *rv = os.str();
        return;
      }
-      case kDeviceName:
+      case kDeviceName: {
+        std::string name(256, 0);
+        ROCM_CALL(hipDeviceGetName(&name[0], name.size(), ctx.device_id));
+        name.resize(strlen(name.c_str()));
+        *rv = std::move(name);
        return;
+      }
      case kMaxClockRate: {
        ROCM_CALL(hipDeviceGetAttribute(&value, hipDeviceAttributeClockRate,
                                        ctx.device_id));