1ecffdaf27
This PR allows `nvptx-loader` to read the resource usage of `_start`, `_begin`, and `_end` when executing CUDA binaries. Example output: ``` $ nvptx-loader --print-resource-usage libc/benchmarks/gpu/src/ctype/libc.benchmarks.gpu.src.ctype.isalnum_benchmark.__build__ [ RUN ] LlvmLibcIsAlNumGpuBenchmark.IsAlnumWrapper [ OK ] LlvmLibcIsAlNumGpuBenchmark.IsAlnumWrapper: 93 cycles, 76 min, 470 max, 23 iterations, 78000 ns, 80 stddev _begin registers: 25 _start registers: 80 _end registers: 62 ``` --------- Co-authored-by: Joseph Huber <huberjn@outlook.com>
101 lines
3.1 KiB
C++
101 lines
3.1 KiB
C++
//===-- Main entry into the loader interface ------------------------------===//
|
|
//
|
|
// 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 file opens a device image passed on the command line and passes it to
|
|
// one of the loader implementations for launch.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "Loader.h"
|
|
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
int main(int argc, char **argv, char **envp) {
|
|
if (argc < 2) {
|
|
printf("USAGE: ./loader [--threads <n>, --blocks <n>, "
|
|
"--print-resource-usage] <device_image> "
|
|
"<args>, ...\n");
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
int offset = 0;
|
|
FILE *file = nullptr;
|
|
char *ptr;
|
|
LaunchParameters params = {1, 1, 1, 1, 1, 1};
|
|
bool print_resource_usage = false;
|
|
while (!file && ++offset < argc) {
|
|
if (argv[offset] == std::string("--threads") ||
|
|
argv[offset] == std::string("--threads-x")) {
|
|
params.num_threads_x =
|
|
offset + 1 < argc ? strtoul(argv[offset + 1], &ptr, 10) : 1;
|
|
offset++;
|
|
continue;
|
|
} else if (argv[offset] == std::string("--threads-y")) {
|
|
params.num_threads_y =
|
|
offset + 1 < argc ? strtoul(argv[offset + 1], &ptr, 10) : 1;
|
|
offset++;
|
|
continue;
|
|
} else if (argv[offset] == std::string("--threads-z")) {
|
|
params.num_threads_z =
|
|
offset + 1 < argc ? strtoul(argv[offset + 1], &ptr, 10) : 1;
|
|
offset++;
|
|
continue;
|
|
} else if (argv[offset] == std::string("--blocks") ||
|
|
argv[offset] == std::string("--blocks-x")) {
|
|
params.num_blocks_x =
|
|
offset + 1 < argc ? strtoul(argv[offset + 1], &ptr, 10) : 1;
|
|
offset++;
|
|
continue;
|
|
} else if (argv[offset] == std::string("--blocks-y")) {
|
|
params.num_blocks_y =
|
|
offset + 1 < argc ? strtoul(argv[offset + 1], &ptr, 10) : 1;
|
|
offset++;
|
|
continue;
|
|
} else if (argv[offset] == std::string("--blocks-z")) {
|
|
params.num_blocks_z =
|
|
offset + 1 < argc ? strtoul(argv[offset + 1], &ptr, 10) : 1;
|
|
offset++;
|
|
continue;
|
|
} else if (argv[offset] == std::string("--print-resource-usage")) {
|
|
print_resource_usage = true;
|
|
continue;
|
|
} else {
|
|
file = fopen(argv[offset], "r");
|
|
if (!file) {
|
|
fprintf(stderr, "Failed to open image file '%s'\n", argv[offset]);
|
|
return EXIT_FAILURE;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!file) {
|
|
fprintf(stderr, "No image file provided\n");
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
// TODO: We should perform some validation on the file.
|
|
fseek(file, 0, SEEK_END);
|
|
const auto size = ftell(file);
|
|
fseek(file, 0, SEEK_SET);
|
|
|
|
void *image = malloc(size * sizeof(char));
|
|
fread(image, sizeof(char), size, file);
|
|
fclose(file);
|
|
|
|
// Drop the loader from the program arguments.
|
|
int ret = load(argc - offset, &argv[offset], envp, image, size, params,
|
|
print_resource_usage);
|
|
|
|
free(image);
|
|
return ret;
|
|
}
|