clang/tools/amdgpu-arch/AMDGPUArchByKFD.cpp

   1 //===- AMDGPUArchByKFD.cpp - list AMDGPU installed ------*- C++ -*---------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file implements a tool for detecting name of AMD GPUs installed in
  10 // system using the Linux sysfs interface for the AMD KFD driver. This file does
  11 // not respect ROCR_VISIBLE_DEVICES like the ROCm environment would.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/Support/FileSystem.h"
  16 #include "llvm/Support/LineIterator.h"
  17 #include "llvm/Support/MemoryBuffer.h"
  18 #include "llvm/Support/Path.h"
  19 #include <memory>
  20
  21 using namespace llvm;
  22
  23 constexpr static const char *KFD_SYSFS_NODE_PATH =
  24     "/sys/devices/virtual/kfd/kfd/topology/nodes";
  25
  26 // See the ROCm implementation for how this is handled.
  27 // https://github.com/ROCm/ROCT-Thunk-Interface/blob/master/src/libhsakmt.h#L126
  28 constexpr static long getMajor(long Ver) { return (Ver / 10000) % 100; }
  29 constexpr static long getMinor(long Ver) { return (Ver / 100) % 100; }
  30 constexpr static long getStep(long Ver) { return Ver % 100; }
  31
  32 int printGPUsByKFD() {
  33   SmallVector<std::pair<long, long>> Devices;
  34   std::error_code EC;
  35   for (sys::fs::directory_iterator Begin(KFD_SYSFS_NODE_PATH, EC), End;
  36        Begin != End; Begin.increment(EC)) {
  37     if (EC)
  38       return 1;
  39
  40     long Node = 0;
  41     if (sys::path::stem(Begin->path()).consumeInteger(10, Node))
  42       return 1;
  43
  44     SmallString<0> Path(Begin->path());
  45     sys::path::append(Path, "properties");
  46
  47     ErrorOr<std::unique_ptr<MemoryBuffer>> BufferOrErr =
  48         MemoryBuffer::getFileOrSTDIN(Path);
  49     if (std::error_code EC = BufferOrErr.getError())
  50       return 1;
  51
  52     long GFXVersion = 0;
  53     for (line_iterator Lines(**BufferOrErr, false); !Lines.is_at_end();
  54          ++Lines) {
  55       StringRef Line(*Lines);
  56       if (Line.consume_front("gfx_target_version")) {
  57         if (Line.drop_while([](char C) { return std::isspace(C); })
  58                 .consumeInteger(10, GFXVersion))
  59           return 1;
  60         break;
  61       }
  62     }
  63
  64     // If this is zero the node is a CPU.
  65     if (GFXVersion == 0)
  66       continue;
  67     Devices.emplace_back(Node, GFXVersion);
  68   }
  69
  70   // Sort the devices by their node to make sure it prints in order.
  71   llvm::sort(Devices, [](auto &L, auto &R) { return L.first < R.first; });
  72   for (const auto &[Node, GFXVersion] : Devices)
  73     std::fprintf(stdout, "gfx%ld%ld%lx\n", getMajor(GFXVersion),
  74                  getMinor(GFXVersion), getStep(GFXVersion));
  75
  76   return 0;
  77 }