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
21using namespace llvm;
22
23constexpr 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
28constexpr static long getMajor(long Ver) { return (Ver / 10000) % 100; }
29constexpr static long getMinor(long Ver) { return (Ver / 100) % 100; }
30constexpr static long getStep(long Ver) { return Ver % 100; }
31
32int 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&: EC)) {
37 if (EC)
38 return 1;
39
40 long Node = 0;
41 if (sys::path::stem(path: Begin->path()).consumeInteger(Radix: 10, Result&: Node))
42 return 1;
43
44 SmallString<0> Path(Begin->path());
45 sys::path::append(path&: Path, a: "properties");
46
47 ErrorOr<std::unique_ptr<MemoryBuffer>> BufferOrErr =
48 MemoryBuffer::getFileOrSTDIN(Filename: 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(Prefix: "gfx_target_version")) {
57 if (Line.drop_while(F: [](char C) { return std::isspace(C); })
58 .consumeInteger(Radix: 10, Result&: 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(Args&: Node, Args&: GFXVersion);
68 }
69
70 // Sort the devices by their node to make sure it prints in order.
71 llvm::sort(C&: Devices, Comp: [](auto &L, auto &R) { return L.first < R.first; });
72 for (const auto &[Node, GFXVersion] : Devices)
73 std::fprintf(stdout, format: "gfx%ld%ld%lx\n", getMajor(Ver: GFXVersion),
74 getMinor(Ver: GFXVersion), getStep(Ver: GFXVersion));
75
76 return 0;
77}
78