7102592af7
The `llvm-omp-device-info` tool is very handy, but broke due to the lazy evaluation of devices. This repairs the functionality and adds a test. The tool is also renamed into `llvm-offload-device-info` as `-omp-` is going away.
33 lines
1.1 KiB
C++
33 lines
1.1 KiB
C++
//===- llvm-offload-device-info.cpp - Device info as seen by LLVM/Offload -===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This is a command line utility that, by using LLVM/Offload, and the device
|
|
// plugins, list devices information as seen by the runtime.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "omptarget.h"
|
|
#include <cstdio>
|
|
|
|
int main(int argc, char **argv) {
|
|
__tgt_bin_desc EmptyDesc = {0, nullptr, nullptr, nullptr};
|
|
__tgt_register_lib(&EmptyDesc);
|
|
__tgt_init_all_rtls();
|
|
|
|
printf("Found %d devices:\n", omp_get_num_devices());
|
|
for (int Dev = 0; Dev < omp_get_num_devices(); Dev++) {
|
|
printf(" Device %d:\n", Dev);
|
|
if (!__tgt_print_device_info(Dev))
|
|
printf(" print_device_info not implemented\n");
|
|
printf("\n");
|
|
}
|
|
|
|
__tgt_unregister_lib(&EmptyDesc);
|
|
return 0;
|
|
}
|