|
@@ -0,0 +1,246 @@
|
|
|
+/*
|
|
|
+
|
|
|
+Copyright (c) 2017-2018, Feral Interactive
|
|
|
+All rights reserved.
|
|
|
+
|
|
|
+Redistribution and use in source and binary forms, with or without
|
|
|
+modification, are permitted provided that the following conditions are met:
|
|
|
+
|
|
|
+ * Redistributions of source code must retain the above copyright notice,
|
|
|
+ this list of conditions and the following disclaimer.
|
|
|
+ * Redistributions in binary form must reproduce the above copyright
|
|
|
+ notice, this list of conditions and the following disclaimer in the
|
|
|
+ documentation and/or other materials provided with the distribution.
|
|
|
+ * Neither the name of Feral Interactive nor the names of its contributors
|
|
|
+ may be used to endorse or promote products derived from this software
|
|
|
+ without specific prior written permission.
|
|
|
+
|
|
|
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
|
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
|
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
|
+ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
|
+LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
|
+CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
|
+SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
|
+INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
|
+CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
|
+ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
|
+POSSIBILITY OF SUCH DAMAGE.
|
|
|
+
|
|
|
+ */
|
|
|
+
|
|
|
+#define _GNU_SOURCE
|
|
|
+
|
|
|
+#include "logging.h"
|
|
|
+
|
|
|
+#include "external-helper.h"
|
|
|
+#include "gpu-control.h"
|
|
|
+
|
|
|
+/* Plausible extras to add:
|
|
|
+ * Intel support - https://blog.ffwll.ch/2013/03/overclocking-your-intel-gpu-on-linux.html
|
|
|
+ * AMD - Allow setting fan speed as well
|
|
|
+ * Store baseline values with get_gpu_state to apply when leaving gamemode
|
|
|
+ */
|
|
|
+
|
|
|
+/* Helper to quit with usage */
|
|
|
+static const char *usage_text =
|
|
|
+ "usage: gpuclockctl PCI_ID DEVICE [get] [set CORE MEM [PERF_LEVEL]]]";
|
|
|
+static void print_usage_and_exit(void)
|
|
|
+{
|
|
|
+ fprintf(stderr, "%s\n", usage_text);
|
|
|
+ exit(EXIT_FAILURE);
|
|
|
+}
|
|
|
+
|
|
|
+/**
|
|
|
+ * Get the gpu state
|
|
|
+ * Populates the struct with the GPU info on the system
|
|
|
+ */
|
|
|
+int get_gpu_state(struct GameModeGPUInfo *info)
|
|
|
+{
|
|
|
+ fprintf(stderr, "Fetching GPU state is currently unimplemented!\n");
|
|
|
+ return info != NULL;
|
|
|
+}
|
|
|
+
|
|
|
+/**
|
|
|
+ * Set the gpu state based on input parameters on Nvidia
|
|
|
+ */
|
|
|
+int set_gpu_state_nv(struct GameModeGPUInfo *info)
|
|
|
+{
|
|
|
+ if (info->vendor != Vendor_NVIDIA)
|
|
|
+ return -1;
|
|
|
+
|
|
|
+ // These commands don't technically even need root
|
|
|
+
|
|
|
+ /* Set the GPUGraphicsClockOffset parameter */
|
|
|
+ char core_arg[128];
|
|
|
+ snprintf(core_arg,
|
|
|
+ 128,
|
|
|
+ "[gpu:%ld]/GPUGraphicsClockOffset[%ld]=%ld",
|
|
|
+ info->device,
|
|
|
+ info->nv_perf_level,
|
|
|
+ info->core);
|
|
|
+ const char *exec_args_core[] = { "/usr/bin/nvidia-settings", "-a", core_arg, NULL };
|
|
|
+ if (run_external_process(exec_args_core) != 0) {
|
|
|
+ LOG_ERROR("Failed to set %s!\n", core_arg);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ /* Set the GPUMemoryTransferRateOffset parameter */
|
|
|
+ char mem_arg[128];
|
|
|
+ snprintf(mem_arg,
|
|
|
+ 128,
|
|
|
+ "[gpu:%ld]/GPUMemoryTransferRateOffset[%ld]=%ld",
|
|
|
+ info->device,
|
|
|
+ info->nv_perf_level,
|
|
|
+ info->mem);
|
|
|
+ const char *exec_args_mem[] = { "/usr/bin/nvidia-settings", "-a", mem_arg, NULL };
|
|
|
+ if (run_external_process(exec_args_mem) != 0) {
|
|
|
+ LOG_ERROR("Failed to set %s!\n", mem_arg);
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+/*
|
|
|
+ * Sets the value in a file in the AMDGPU driver config
|
|
|
+ * Files are:
|
|
|
+ * /sys/class/drm/card0/device/pp_sclk_od
|
|
|
+ * /sys/class/drm/card0/device/pp_mclk_od
|
|
|
+ */
|
|
|
+static int set_gpu_state_amd_file(const char *filename, long device, long value)
|
|
|
+{
|
|
|
+ const char *drm_path = "/sys/class/drm/card%ld/device/%s";
|
|
|
+ char path[64];
|
|
|
+ snprintf(path, 64, drm_path, device, filename);
|
|
|
+
|
|
|
+ FILE *file = fopen(path, "w");
|
|
|
+ if (!file) {
|
|
|
+ LOG_ERROR("Could not open %s for write (%s)!\n", path, strerror(errno));
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (fprintf(file, "%ld", value) < 0) {
|
|
|
+ LOG_ERROR("Could not write to %s (%s)!\n", path, strerror(errno));
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (fclose(file) != 0) {
|
|
|
+ LOG_ERROR("Could not close %s after writing (%s)!\n", path, strerror(errno));
|
|
|
+ return -1;
|
|
|
+ }
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+/**
|
|
|
+ * Set the gpu state based on input parameters on amd
|
|
|
+ */
|
|
|
+int set_gpu_state_amd(struct GameModeGPUInfo *info)
|
|
|
+{
|
|
|
+ if (info->vendor != Vendor_AMD)
|
|
|
+ return -1;
|
|
|
+
|
|
|
+ // Set the the core and mem clock speeds using the OverDrive files
|
|
|
+ if (set_gpu_state_amd_file("pp_sclk_od", info->device, info->core) != 0)
|
|
|
+ return -1;
|
|
|
+ if (set_gpu_state_amd_file("pp_mclk_od", info->device, info->mem) != 0)
|
|
|
+ return -1;
|
|
|
+
|
|
|
+ return 0;
|
|
|
+}
|
|
|
+
|
|
|
+/* Helper to get and verify vendor value */
|
|
|
+static long get_vendor(const char *val)
|
|
|
+{
|
|
|
+ char *end;
|
|
|
+ long ret = strtol(val, &end, 0);
|
|
|
+ if (!GPUVendorValid(ret) || end == val) {
|
|
|
+ LOG_ERROR("Invalid GPU Vendor passed (0x%04x)!\n", (unsigned short)ret);
|
|
|
+ print_usage_and_exit();
|
|
|
+ }
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+/* Helper to get and verify device value */
|
|
|
+static long get_device(const char *val)
|
|
|
+{
|
|
|
+ char *end;
|
|
|
+ long ret = strtol(val, &end, 10);
|
|
|
+ if (ret < 0 || end == val) {
|
|
|
+ LOG_ERROR("Invalid GPU device passed (%ld)!\n", ret);
|
|
|
+ print_usage_and_exit();
|
|
|
+ }
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+/* Helper to get and verify core and mem value */
|
|
|
+static long get_generic_value(const char *val)
|
|
|
+{
|
|
|
+ char *end;
|
|
|
+ long ret = strtol(val, &end, 10);
|
|
|
+ if (ret < 0 || end == val) {
|
|
|
+ LOG_ERROR("Invalid value passed (%ld)!\n", ret);
|
|
|
+ print_usage_and_exit();
|
|
|
+ }
|
|
|
+ return ret;
|
|
|
+}
|
|
|
+
|
|
|
+/**
|
|
|
+ * Main entry point, dispatch to the appropriate helper
|
|
|
+ */
|
|
|
+int main(int argc, char *argv[])
|
|
|
+{
|
|
|
+ if (argc == 4 && strncmp(argv[3], "get", 3) == 0) {
|
|
|
+ /* Get and verify the vendor and device */
|
|
|
+ struct GameModeGPUInfo info;
|
|
|
+ memset(&info, 0, sizeof(info));
|
|
|
+ info.vendor = get_vendor(argv[1]);
|
|
|
+ info.device = get_device(argv[2]);
|
|
|
+
|
|
|
+ /* Fetch the state and print it out */
|
|
|
+ get_gpu_state(&info);
|
|
|
+ printf("%ld %ld\n", info.core, info.mem);
|
|
|
+
|
|
|
+ } else if (argc >= 6 && argc <= 7 && strncmp(argv[3], "set", 3) == 0) {
|
|
|
+ /* Must be root to set the state */
|
|
|
+ if (geteuid() != 0) {
|
|
|
+ fprintf(stderr, "gpuclockctl must be run as root to set values\n");
|
|
|
+ print_usage_and_exit();
|
|
|
+ }
|
|
|
+
|
|
|
+ /* Get and verify the vendor and device */
|
|
|
+ struct GameModeGPUInfo info;
|
|
|
+ memset(&info, 0, sizeof(info));
|
|
|
+ info.vendor = get_vendor(argv[1]);
|
|
|
+ info.device = get_device(argv[2]);
|
|
|
+ info.core = get_generic_value(argv[4]);
|
|
|
+ info.mem = get_generic_value(argv[5]);
|
|
|
+
|
|
|
+ if (info.vendor == Vendor_NVIDIA)
|
|
|
+ info.nv_perf_level = get_generic_value(argv[6]);
|
|
|
+
|
|
|
+ printf("gpuclockctl setting core:%ld mem:%ld on device:%ld with vendor 0x%04x\n",
|
|
|
+ info.core,
|
|
|
+ info.mem,
|
|
|
+ info.device,
|
|
|
+ (unsigned short)info.vendor);
|
|
|
+
|
|
|
+ if (info.vendor == Vendor_NVIDIA)
|
|
|
+ printf("on Performance Level %ld\n", info.nv_perf_level);
|
|
|
+
|
|
|
+ switch (info.vendor) {
|
|
|
+ case Vendor_NVIDIA:
|
|
|
+ return set_gpu_state_nv(&info);
|
|
|
+ case Vendor_AMD:
|
|
|
+ return set_gpu_state_amd(&info);
|
|
|
+ default:
|
|
|
+ printf("Currently unsupported GPU vendor 0x%04x, doing nothing!\n", (short)info.vendor);
|
|
|
+ break;
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ print_usage_and_exit();
|
|
|
+ }
|
|
|
+
|
|
|
+ return EXIT_SUCCESS;
|
|
|
+}
|